ENUM support.

Author: LastDragon-ru

src/Utils/SchemaPrinter.php

@@ -7,6 +7,7 @@
 use Closure;
 use GraphQL\Error\Error;
 use GraphQL\Language\AST\DirectiveNode;
+use GraphQL\Language\AST\EnumTypeDefinitionNode;
 use GraphQL\Language\AST\EnumValueDefinitionNode;
 use GraphQL\Language\AST\ScalarTypeDefinitionNode;
 use GraphQL\Language\BlockString;
@@ -437,6 +438,7 @@ static function (EnumValueDefinition $value, int $i) use ($options): string {
 
         return static::printDescription($options, $type) .
             sprintf('enum %s', $type->name) .
+            static::printTypeDirectives($type, $options) .
             static::printBlock($values);
     }
 
@@ -489,7 +491,7 @@ protected static function printBlock(array $items): string
     }
 
     /**
-     * @param Type|EnumValueDefinition $type
+     * @param Type|EnumValueDefinition|EnumType $type
      * @param array<string, bool> $options
      * @phpstan-param Options $options
      */
@@ -508,7 +510,7 @@ protected static function printTypeDirectives($type, array $options, string $ind
         // AST Node available and has directives?
         $node = $type->astNode;
 
-        if (!($node instanceof ScalarTypeDefinitionNode || $node instanceof EnumValueDefinitionNode)) {
+        if (!($node instanceof ScalarTypeDefinitionNode || $node instanceof EnumValueDefinitionNode || $node instanceof EnumTypeDefinitionNode)) {
             return '';
         }
 
@@ -535,8 +537,12 @@ protected static function printTypeDirectives($type, array $options, string $ind
         }
 
         // Multiline?
-        $delimiter = $length > static::LINE_LENGTH ? "\n{$indentation}" : ' ';
-        $serialized = $delimiter.implode($delimiter, $directives);
+        $serialized = '';
+
+        if ($directives) {
+            $delimiter  = $length > static::LINE_LENGTH ? "\n{$indentation}" : ' ';
+            $serialized = $delimiter.implode($delimiter, $directives);
+        }
 
         // Return
         return $serialized;

tests/Utils/SchemaPrinterTest.php

@@ -1278,7 +1278,7 @@ enum __TypeKind {
     }
 
     public function testPrintDirectives(): void {
-        $text   = str_pad('a', 80, 'a');
+        $text = str_pad('a', 80, 'a');
         $schema = /** @lang GraphQL */ <<<GRAPHQL
         directive @test(
           value: String
@@ -1297,19 +1297,23 @@ public function testPrintDirectives(): void {
         scalar ScalarA @test
         scalar ScalarB @test(value: "{$text}")
 
-        enum EnumA {
+        enum EnumA @test {
           a @test @deprecated
           b @test(value: "{$text}")
           "{$text}"
           c @test
           "{$text}"
-          d @test(value: "{$text}")
+          d @test(value: "{$text}") @deprecated
+        }
+
+        enum EnumB @test(value: "{$text}") {
+          a
         }
         GRAPHQL;
-        $expected = <<<'GRAPHQL'
+        $expected = /** @lang GraphQL */ <<<'GRAPHQL'
         directive @test(value: String) on SCHEMA | SCALAR | OBJECT | FIELD_DEFINITION | ARGUMENT_DEFINITION | INTERFACE | UNION | ENUM | ENUM_VALUE | INPUT_OBJECT | INPUT_FIELD_DEFINITION
 
-        enum EnumA {
+        enum EnumA @test {
           a @test @deprecated
 
           b
@@ -1325,6 +1329,12 @@ enum EnumA {
           """
           d
           @test(value: "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa")
+          @deprecated
+        }
+
+        enum EnumB
+        @test(value: "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa") {
+          a
         }
 
         scalar ScalarA @test