ARGUMENT_DEFINITION support.

Author: LastDragon-ru

src/Utils/SchemaPrinter.php

@@ -268,37 +268,51 @@ protected static function printDescriptionWithComments(string $description, stri
      */
     protected static function printArgs(array $options, array $args, string $indentation = ''): string
     {
+        // Empty?
         if (count($args) === 0) {
             return '';
         }
 
-        // If every arg does not have a description, print them on one line.
-        if (
-            Utils::every(
-                $args,
-                static function ($arg): bool {
-                    return strlen($arg->description ?? '') === 0;
-                }
-            )
-        ) {
-            return '(' . implode(', ', array_map('static::printInputValue', $args)) . ')';
-        }
-
-        return sprintf(
-            "(\n%s\n%s)",
-            implode(
-                "\n",
-                array_map(
-                    static function (FieldArgument $arg, int $i) use ($indentation, $options): string {
-                        return static::printDescription($options, $arg, '  ' . $indentation, $i === 0) . '  ' . $indentation .
-                            static::printInputValue($arg);
-                    },
-                    $args,
-                    array_keys($args)
-                )
-            ),
-            $indentation
-        );
+        // Print arguments
+        $length = 0;
+        $arguments = [];
+        $description = false;
+
+        foreach ($args as $i => $arg) {
+            $value = static::printArg($arg, $options, '  ' . $indentation, $i === 0);
+            $length = $length + mb_strlen($value);
+            $description = $description || mb_strlen($arg->description ?? '') > 0;
+            $arguments[] = $value;
+        }
+
+        // Multiline?
+        $serialized = '';
+
+        if ($length > static::LINE_LENGTH || $description) {
+            $serialized = "(\n" . implode("\n", $arguments) . "\n{$indentation})";
+        } else {
+            $serialized = '(' . implode(', ', array_map('trim', $arguments)) . ')';
+        }
+
+        return $serialized;
+    }
+
+    /**
+     * @param array<string, bool> $options
+     * @phpstan-param Options $options
+     */
+    protected static function printArg(FieldArgument $type, array $options, string $indentation = '', bool $firstInBlock = true): string
+    {
+        $field = static::printDescription($options, $type, $indentation, $firstInBlock) .
+            $indentation .
+            static::printInputValue($type) .
+            static::printTypeDirectives($type, $options, $indentation);
+
+        if (!$firstInBlock && mb_strlen($field) > static::LINE_LENGTH) {
+            $field = "\n".ltrim($field, "\n");
+        }
+
+        return $field;
     }
 
     /**
@@ -536,7 +550,7 @@ protected static function printBlock(array $items): string
     }
 
     /**
-     * @param Type|EnumValueDefinition|EnumType|InterfaceType|FieldDefinition|UnionType|InputObjectType|InputObjectField $type
+     * @param Type|EnumValueDefinition|EnumType|InterfaceType|FieldDefinition|UnionType|InputObjectType|InputObjectField|FieldArgument $type
      * @param array<string, bool> $options
      * @phpstan-param Options $options
      */

tests/Utils/SchemaPrinterTest.php

@@ -1281,7 +1281,7 @@ public function testPrintDirectives(): void {
         $text = str_pad('a', 80, 'a');
         $schema = /** @lang GraphQL */ <<<GRAPHQL
             directive @test(
-              value: String
+              value: String @test(value: "{$text}")
             ) on SCHEMA |
                 SCALAR |
                 OBJECT |
@@ -1352,9 +1352,26 @@ interface InterfaceB implements InterfaceA @test(value: "{$text}") {
             input InputB @test(value: "{$text}") {
               a: ID
             }
+
+            type Query {
+              a(a: String, b: String, c: String): Boolean @test
+              b("desc" a: String): Boolean @test(value: "{$text}")
+              c(a: String @test(value: "{$text}")): Boolean
+              d(
+                a: String @test
+                b: String @test(value: "{$text}")
+                "{$text}"
+                c: String @test
+                "{$text}"
+                d: String = "123" @test(value: "{$text}")
+              ): Boolean
+            }
             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
+            directive @test(
+              value: String
+              @test(value: "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa")
+            ) on SCHEMA | SCALAR | OBJECT | FIELD_DEFINITION | ARGUMENT_DEFINITION | INTERFACE | UNION | ENUM | ENUM_VALUE | INPUT_OBJECT | INPUT_FIELD_DEFINITION
 
             enum EnumA @test {
               a @test @deprecated
@@ -1427,6 +1444,39 @@ interface InterfaceB implements InterfaceA
               a: ID
             }
 
+            type Query {
+              a(a: String, b: String, c: String): Boolean @test
+
+              b(
+                """desc"""
+                a: String
+              ): Boolean
+              @test(value: "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa")
+
+              c(
+                a: String
+                @test(value: "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa")
+              ): Boolean
+
+              d(
+                a: String @test
+
+                b: String
+                @test(value: "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa")
+
+                """
+                aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa
+                """
+                c: String @test
+
+                """
+                aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa
+                """
+                d: String = "123"
+                @test(value: "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa")
+              ): Boolean
+            }
+
             scalar ScalarA @test
 
             scalar ScalarB