Add support for input object definitions

lib/graphql/language/nodes.rb

@@ -775,7 +775,7 @@ class EnumTypeExtension < AbstractNode
       end
 
       class InputObjectTypeDefinition < AbstractNode
-        attr_reader :description
+        attr_reader :comment, :description
         scalar_methods :name
         children_methods({
           directives: GraphQL::Language::Nodes::Directive,

lib/graphql/language/parser.rb

@@ -351,7 +351,7 @@ def definition
             name = parse_name
             directives = parse_directives
             input_fields_definition = parse_input_object_field_definitions
-            InputObjectTypeDefinition.new(pos: loc, definition_pos: defn_loc, description: desc, name: name, directives: directives, fields: input_fields_definition, filename: @filename, source: self)
+            InputObjectTypeDefinition.new(pos: loc, definition_pos: defn_loc, description: desc, comment: comment, name: name, directives: directives, fields: input_fields_definition, filename: @filename, source: self)
           else
             expect_one_of([:SCHEMA, :SCALAR, :TYPE, :ENUM, :INPUT, :UNION, :INTERFACE])
           end

lib/graphql/language/printer.rb

@@ -380,7 +380,7 @@ def print_enum_value_definition(enum_value)
       end
 
       def print_input_object_type_definition(input_object_type, extension: false)
-        extension ? print_string("extend ") : print_description(input_object_type)
+        extension ? print_string("extend ") : print_description_and_comment(input_object_type)
         print_string("input ")
         print_string(input_object_type.name)
         print_directives(input_object_type.directives)

spec/graphql/language/parser_spec.rb

@@ -196,7 +196,7 @@
   end
 
   describe "string comment" do
-    it "is parsed for fields, unions, interfaces, enums, enum values and arguments" do
+    it "is parsed for fields, unions, interfaces, enums, enum values, inputs and arguments" do
       document = subject.parse <<-GRAPHQL
       # type comment
       type Thing {
@@ -223,6 +223,12 @@
       interface CustomInterface {
         name: String!
       }
+
+      # Input comment
+      input CustomInput {
+        # Custom input name comment
+        name: String!
+      }
       GRAPHQL
 
       thing_defn = document.definitions[0]
@@ -252,6 +258,13 @@
 
       custom_interface_defn = document.definitions[4]
       assert_equal "Interface comment", custom_interface_defn.comment
+
+      custom_input_defn = document.definitions[5]
+      assert_equal "Input comment", custom_input_defn.comment
+
+      input_field_defn = custom_input_defn.fields[0]
+      assert_equal "name", input_field_defn.name
+      assert_equal "Custom input name comment", input_field_defn.comment
     end
   end
 

spec/graphql/language/printer_spec.rb

@@ -189,6 +189,7 @@
           """
           Input description
           """
+          # Input type comment
           input InputType {
             key: String!
             answer: Int = 42

spec/graphql/schema/input_object_spec.rb

@@ -18,6 +18,7 @@
     it "has it" do
       assert_equal "EnsembleInput", input_object.graphql_name
       assert_nil input_object.description
+      assert_equal "Ensemble input comment", input_object.comment
       assert_equal 1, input_object.arguments.size
     end
 

spec/support/jazz.rb

@@ -543,6 +543,8 @@ def nullable_ensemble(ensemble: nil)
   end
 
   class EnsembleInput < GraphQL::Schema::InputObject
+    comment "Ensemble input comment"
+
     argument :name, String
   end