tinygql 0.1.3 → 0.2.0

data/lib/tinygql/nodes.rb

@@ -3,45 +3,62 @@ module TinyGQL
     class Node
       include Enumerable
 
-    def document?; false; end
-    def operation_definition?; false; end
-    def variable?; false; end
-    def named_type?; false; end
-    def not_null_type?; false; end
-    def list_type?; false; end
-    def variable_definition?; false; end
-    def value?; false; end
-    def argument?; false; end
-    def field?; false; end
-    def object_field?; false; end
-    def int_value?; false; end
-    def float_value?; false; end
-    def string_value?; false; end
-    def boolean_value?; false; end
-    def null_value?; false; end
-    def enum_value?; false; end
-    def list_value?; false; end
-    def object_value?; false; end
-    def directive?; false; end
-    def type_condition?; false; end
-    def inline_fragment?; false; end
-    def fragment_spread?; false; end
-    def fragment_definition?; false; end
-    def root_operation_type_definition?; false; end
-    def schema_definition?; false; end
-    def field_definition?; false; end
-    def input_value_definition?; false; end
-    def object_type_definition?; false; end
-    def interface_type_definition?; false; end
-    def union_type_definition?; false; end
-    def scalar_type_definition?; false; end
-    def enum_value_definition?; false; end
-    def enum_type_definition?; false; end
-    def input_object_type_definition?; false; end
-    def object_type_extension?; false; end
-    def executable_directive_location?; false; end
-    def type_system_directive_location?; false; end
-    def directive_definition?; false; end
+      # `pos` is the position of the _end_ of a token
+      attr_reader :pos
+
+      def initialize pos
+        @pos = pos
+      end
+
+      # Return the line of this node given `doc`
+      def line doc
+        doc[0, @pos].count("\n") + 1
+      end
+
+      def document?; false; end
+      def operation_definition?; false; end
+      def variable?; false; end
+      def named_type?; false; end
+      def not_null_type?; false; end
+      def list_type?; false; end
+      def variable_definition?; false; end
+      def value?; false; end
+      def argument?; false; end
+      def field?; false; end
+      def object_field?; false; end
+      def int_value?; false; end
+      def float_value?; false; end
+      def string_value?; false; end
+      def boolean_value?; false; end
+      def null_value?; false; end
+      def enum_value?; false; end
+      def list_value?; false; end
+      def object_value?; false; end
+      def directive?; false; end
+      def type_condition?; false; end
+      def inline_fragment?; false; end
+      def fragment_spread?; false; end
+      def fragment_definition?; false; end
+      def root_operation_type_definition?; false; end
+      def schema_definition?; false; end
+      def field_definition?; false; end
+      def input_value_definition?; false; end
+      def object_type_definition?; false; end
+      def interface_type_definition?; false; end
+      def union_type_definition?; false; end
+      def scalar_type_definition?; false; end
+      def enum_value_definition?; false; end
+      def enum_type_definition?; false; end
+      def input_object_type_definition?; false; end
+      def object_type_extension?; false; end
+      def executable_directive_location?; false; end
+      def type_system_directive_location?; false; end
+      def directive_definition?; false; end
+      def scalar_type_extension?; false; end
+      def interface_type_extension?; false; end
+      def union_type_extension?; false; end
+      def enum_type_extension?; false; end
+      def input_object_type_extension?; false; end
       def each(&blk)
         yield self
         children.each { |v| v.each(&blk) }
@@ -51,7 +68,8 @@ module TinyGQL
     class Document < Node
       attr_reader :definitions
 
-      def initialize definitions
+      def initialize pos, definitions
+        super(pos)
         @definitions = definitions
       end
 
@@ -72,7 +90,8 @@ module TinyGQL
     class OperationDefinition < Node
       attr_reader :type, :name, :variable_definitions, :directives, :selection_set
 
-      def initialize type, name, variable_definitions, directives, selection_set
+      def initialize pos, type, name, variable_definitions, directives, selection_set
+        super(pos)
         @type = type
         @name = name
         @variable_definitions = variable_definitions
@@ -97,7 +116,8 @@ module TinyGQL
     class Variable < Node
       attr_reader :name
 
-      def initialize name
+      def initialize pos, name
+        super(pos)
         @name = name
       end
 
@@ -118,7 +138,8 @@ module TinyGQL
     class NamedType < Node
       attr_reader :name
 
-      def initialize name
+      def initialize pos, name
+        super(pos)
         @name = name
       end
 
@@ -139,7 +160,8 @@ module TinyGQL
     class NotNullType < Node
       attr_reader :type
 
-      def initialize type
+      def initialize pos, type
+        super(pos)
         @type = type
       end
 
@@ -160,7 +182,8 @@ module TinyGQL
     class ListType < Node
       attr_reader :type
 
-      def initialize type
+      def initialize pos, type
+        super(pos)
         @type = type
       end
 
@@ -181,7 +204,8 @@ module TinyGQL
     class VariableDefinition < Node
       attr_reader :variable, :type, :default_value
 
-      def initialize variable, type, default_value
+      def initialize pos, variable, type, default_value
+        super(pos)
         @variable = variable
         @type = type
         @default_value = default_value
@@ -204,7 +228,8 @@ module TinyGQL
     class Value < Node
       attr_reader :value
 
-      def initialize value
+      def initialize pos, value
+        super(pos)
         @value = value
       end
 
@@ -225,7 +250,8 @@ module TinyGQL
     class Argument < Node
       attr_reader :name, :value
 
-      def initialize name, value
+      def initialize pos, name, value
+        super(pos)
         @name = name
         @value = value
       end
@@ -247,7 +273,8 @@ module TinyGQL
     class Field < Node
       attr_reader :aliaz, :name, :arguments, :directives, :selection_set
 
-      def initialize aliaz, name, arguments, directives, selection_set
+      def initialize pos, aliaz, name, arguments, directives, selection_set
+        super(pos)
         @aliaz = aliaz
         @name = name
         @arguments = arguments
@@ -272,7 +299,8 @@ module TinyGQL
     class ObjectField < Node
       attr_reader :name, :value
 
-      def initialize name, value
+      def initialize pos, name, value
+        super(pos)
         @name = name
         @value = value
       end
@@ -385,7 +413,8 @@ module TinyGQL
     class ObjectValue < Node
       attr_reader :values
 
-      def initialize values
+      def initialize pos, values
+        super(pos)
         @values = values
       end
 
@@ -406,7 +435,8 @@ module TinyGQL
     class Directive < Node
       attr_reader :name, :arguments
 
-      def initialize name, arguments
+      def initialize pos, name, arguments
+        super(pos)
         @name = name
         @arguments = arguments
       end
@@ -422,13 +452,14 @@ module TinyGQL
       def directive?; true; end
 
       def children
-        ary = []; ary.concat(arguments); ary
+        ary = []; ary.concat(arguments) if arguments; ary
       end
     end
     class TypeCondition < Node
       attr_reader :named_type
 
-      def initialize named_type
+      def initialize pos, named_type
+        super(pos)
         @named_type = named_type
       end
 
@@ -449,7 +480,8 @@ module TinyGQL
     class InlineFragment < Node
       attr_reader :type_condition, :directives, :selection_set
 
-      def initialize type_condition, directives, selection_set
+      def initialize pos, type_condition, directives, selection_set
+        super(pos)
         @type_condition = type_condition
         @directives = directives
         @selection_set = selection_set
@@ -472,7 +504,8 @@ module TinyGQL
     class FragmentSpread < Node
       attr_reader :fragment_name, :directives
 
-      def initialize fragment_name, directives
+      def initialize pos, fragment_name, directives
+        super(pos)
         @fragment_name = fragment_name
         @directives = directives
       end
@@ -488,13 +521,14 @@ module TinyGQL
       def fragment_spread?; true; end
 
       def children
-        ary = []; ary << fragment_name; ary.concat(directives) if directives; ary
+        ary = []; ary.concat(directives) if directives; ary
       end
     end
     class FragmentDefinition < Node
       attr_reader :fragment_name, :type_condition, :directives, :selection_set
 
-      def initialize fragment_name, type_condition, directives, selection_set
+      def initialize pos, fragment_name, type_condition, directives, selection_set
+        super(pos)
         @fragment_name = fragment_name
         @type_condition = type_condition
         @directives = directives
@@ -512,13 +546,14 @@ module TinyGQL
       def fragment_definition?; true; end
 
       def children
-        ary = []; ary << fragment_name; ary << type_condition; ary.concat(directives) if directives; ary.concat(selection_set); ary
+        ary = []; ary << type_condition; ary.concat(directives) if directives; ary.concat(selection_set); ary
       end
     end
     class RootOperationTypeDefinition < Node
       attr_reader :operation_type, :named_type
 
-      def initialize operation_type, named_type
+      def initialize pos, operation_type, named_type
+        super(pos)
         @operation_type = operation_type
         @named_type = named_type
       end
@@ -534,13 +569,14 @@ module TinyGQL
       def root_operation_type_definition?; true; end
 
       def children
-        ary = []; ary << operation_type; ary << named_type; ary
+        ary = []; ary << named_type; ary
       end
     end
     class SchemaDefinition < Node
       attr_reader :description, :directives, :root_operation_definitions
 
-      def initialize description, directives, root_operation_definitions
+      def initialize pos, description, directives, root_operation_definitions
+        super(pos)
         @description = description
         @directives = directives
         @root_operation_definitions = root_operation_definitions
@@ -557,13 +593,14 @@ module TinyGQL
       def schema_definition?; true; end
 
       def children
-        ary = []; ary.concat(directives) if directives; ary.concat(root_operation_definitions); ary
+        ary = []; ary << description if description; ary.concat(directives) if directives; ary.concat(root_operation_definitions); ary
       end
     end
     class FieldDefinition < Node
       attr_reader :description, :name, :arguments_definition, :type, :directives
 
-      def initialize description, name, arguments_definition, type, directives
+      def initialize pos, description, name, arguments_definition, type, directives
+        super(pos)
         @description = description
         @name = name
         @arguments_definition = arguments_definition
@@ -582,13 +619,14 @@ module TinyGQL
       def field_definition?; true; end
 
       def children
-        ary = []; ary.concat(arguments_definition) if arguments_definition; ary << type; ary.concat(directives) if directives; ary
+        ary = []; ary << description if description; ary.concat(arguments_definition) if arguments_definition; ary << type; ary.concat(directives) if directives; ary
       end
     end
     class InputValueDefinition < Node
       attr_reader :description, :name, :type, :default_value, :directives
 
-      def initialize description, name, type, default_value, directives
+      def initialize pos, description, name, type, default_value, directives
+        super(pos)
         @description = description
         @name = name
         @type = type
@@ -607,13 +645,14 @@ module TinyGQL
       def input_value_definition?; true; end
 
       def children
-        ary = []; ary << type; ary << default_value if default_value; ary.concat(directives) if directives; ary
+        ary = []; ary << description if description; ary << type; ary << default_value if default_value; ary.concat(directives) if directives; ary
       end
     end
     class ObjectTypeDefinition < Node
       attr_reader :description, :name, :implements_interfaces, :directives, :fields_definition
 
-      def initialize description, name, implements_interfaces, directives, fields_definition
+      def initialize pos, description, name, implements_interfaces, directives, fields_definition
+        super(pos)
         @description = description
         @name = name
         @implements_interfaces = implements_interfaces
@@ -632,13 +671,14 @@ module TinyGQL
       def object_type_definition?; true; end
 
       def children
-        ary = []; ary.concat(implements_interfaces) if implements_interfaces; ary.concat(directives) if directives; ary.concat(fields_definition) if fields_definition; ary
+        ary = []; ary << description if description; ary.concat(implements_interfaces) if implements_interfaces; ary.concat(directives) if directives; ary.concat(fields_definition) if fields_definition; ary
       end
     end
     class InterfaceTypeDefinition < Node
       attr_reader :description, :name, :directives, :fields_definition
 
-      def initialize description, name, directives, fields_definition
+      def initialize pos, description, name, directives, fields_definition
+        super(pos)
         @description = description
         @name = name
         @directives = directives
@@ -656,13 +696,14 @@ module TinyGQL
       def interface_type_definition?; true; end
 
       def children
-        ary = []; ary.concat(directives) if directives; ary.concat(fields_definition) if fields_definition; ary
+        ary = []; ary << description if description; ary.concat(directives) if directives; ary.concat(fields_definition) if fields_definition; ary
       end
     end
     class UnionTypeDefinition < Node
       attr_reader :description, :name, :directives, :union_member_types
 
-      def initialize description, name, directives, union_member_types
+      def initialize pos, description, name, directives, union_member_types
+        super(pos)
         @description = description
         @name = name
         @directives = directives
@@ -680,13 +721,14 @@ module TinyGQL
       def union_type_definition?; true; end
 
       def children
-        ary = []; ary.concat(directives) if directives; ary.concat(union_member_types) if union_member_types; ary
+        ary = []; ary << description if description; ary.concat(directives) if directives; ary.concat(union_member_types) if union_member_types; ary
       end
     end
     class ScalarTypeDefinition < Node
       attr_reader :description, :name, :directives
 
-      def initialize description, name, directives
+      def initialize pos, description, name, directives
+        super(pos)
         @description = description
         @name = name
         @directives = directives
@@ -703,13 +745,14 @@ module TinyGQL
       def scalar_type_definition?; true; end
 
       def children
-        ary = []; ary.concat(directives) if directives; ary
+        ary = []; ary << description if description; ary.concat(directives) if directives; ary
       end
     end
     class EnumValueDefinition < Node
       attr_reader :description, :enum_value, :directives
 
-      def initialize description, enum_value, directives
+      def initialize pos, description, enum_value, directives
+        super(pos)
         @description = description
         @enum_value = enum_value
         @directives = directives
@@ -726,13 +769,14 @@ module TinyGQL
       def enum_value_definition?; true; end
 
       def children
-        ary = []; ary << enum_value; ary.concat(directives) if directives; ary
+        ary = []; ary << description if description; ary << enum_value; ary.concat(directives) if directives; ary
       end
     end
     class EnumTypeDefinition < Node
       attr_reader :description, :name, :directives, :enum_value_definition
 
-      def initialize description, name, directives, enum_value_definition
+      def initialize pos, description, name, directives, enum_value_definition
+        super(pos)
         @description = description
         @name = name
         @directives = directives
@@ -750,13 +794,14 @@ module TinyGQL
       def enum_type_definition?; true; end
 
       def children
-        ary = []; ary.concat(directives) if directives; ary.concat(enum_value_definition) if enum_value_definition; ary
+        ary = []; ary << description if description; ary.concat(directives) if directives; ary.concat(enum_value_definition) if enum_value_definition; ary
       end
     end
     class InputObjectTypeDefinition < Node
       attr_reader :description, :name, :directives, :input_fields_definition
 
-      def initialize description, name, directives, input_fields_definition
+      def initialize pos, description, name, directives, input_fields_definition
+        super(pos)
         @description = description
         @name = name
         @directives = directives
@@ -774,13 +819,14 @@ module TinyGQL
       def input_object_type_definition?; true; end
 
       def children
-        ary = []; ary.concat(directives) if directives; ary.concat(input_fields_definition) if input_fields_definition; ary
+        ary = []; ary << description if description; ary.concat(directives) if directives; ary.concat(input_fields_definition) if input_fields_definition; ary
       end
     end
     class ObjectTypeExtension < Node
       attr_reader :name, :implements_interfaces, :directives, :fields_definition
 
-      def initialize name, implements_interfaces, directives, fields_definition
+      def initialize pos, name, implements_interfaces, directives, fields_definition
+        super(pos)
         @name = name
         @implements_interfaces = implements_interfaces
         @directives = directives
@@ -804,7 +850,8 @@ module TinyGQL
     class ExecutableDirectiveLocation < Node
       attr_reader :name
 
-      def initialize name
+      def initialize pos, name
+        super(pos)
         @name = name
       end
 
@@ -825,7 +872,8 @@ module TinyGQL
     class TypeSystemDirectiveLocation < Node
       attr_reader :name
 
-      def initialize name
+      def initialize pos, name
+        super(pos)
         @name = name
       end
 
@@ -846,7 +894,8 @@ module TinyGQL
     class DirectiveDefinition < Node
       attr_reader :description, :name, :arguments_definition, :directive_locations
 
-      def initialize description, name, arguments_definition, directive_locations
+      def initialize pos, description, name, arguments_definition, directive_locations
+        super(pos)
         @description = description
         @name = name
         @arguments_definition = arguments_definition
@@ -864,7 +913,127 @@ module TinyGQL
       def directive_definition?; true; end
 
       def children
-        ary = []; ary.concat(arguments_definition) if arguments_definition; ary.concat(directive_locations); ary
+        ary = []; ary << description if description; ary.concat(arguments_definition) if arguments_definition; ary.concat(directive_locations); ary
+      end
+    end
+    class ScalarTypeExtension < Node
+      attr_reader :name, :directives
+
+      def initialize pos, name, directives
+        super(pos)
+        @name = name
+        @directives = directives
+      end
+
+      def accept viz
+        viz.handle_scalar_type_extension self
+      end
+
+      def fold viz, seed
+        viz.handle_scalar_type_extension self, seed
+      end
+
+      def scalar_type_extension?; true; end
+
+      def children
+        ary = []; ary.concat(directives) if directives; ary
+      end
+    end
+    class InterfaceTypeExtension < Node
+      attr_reader :name, :implements_interfaces, :directives, :fields_definition
+
+      def initialize pos, name, implements_interfaces, directives, fields_definition
+        super(pos)
+        @name = name
+        @implements_interfaces = implements_interfaces
+        @directives = directives
+        @fields_definition = fields_definition
+      end
+
+      def accept viz
+        viz.handle_interface_type_extension self
+      end
+
+      def fold viz, seed
+        viz.handle_interface_type_extension self, seed
+      end
+
+      def interface_type_extension?; true; end
+
+      def children
+        ary = []; ary.concat(implements_interfaces) if implements_interfaces; ary.concat(directives) if directives; ary.concat(fields_definition) if fields_definition; ary
+      end
+    end
+    class UnionTypeExtension < Node
+      attr_reader :name, :directives, :union_member_types
+
+      def initialize pos, name, directives, union_member_types
+        super(pos)
+        @name = name
+        @directives = directives
+        @union_member_types = union_member_types
+      end
+
+      def accept viz
+        viz.handle_union_type_extension self
+      end
+
+      def fold viz, seed
+        viz.handle_union_type_extension self, seed
+      end
+
+      def union_type_extension?; true; end
+
+      def children
+        ary = []; ary.concat(directives) if directives; ary.concat(union_member_types) if union_member_types; ary
+      end
+    end
+    class EnumTypeExtension < Node
+      attr_reader :name, :directives, :enum_value_definition
+
+      def initialize pos, name, directives, enum_value_definition
+        super(pos)
+        @name = name
+        @directives = directives
+        @enum_value_definition = enum_value_definition
+      end
+
+      def accept viz
+        viz.handle_enum_type_extension self
+      end
+
+      def fold viz, seed
+        viz.handle_enum_type_extension self, seed
+      end
+
+      def enum_type_extension?; true; end
+
+      def children
+        ary = []; ary.concat(directives) if directives; ary.concat(enum_value_definition) if enum_value_definition; ary
+      end
+    end
+    class InputObjectTypeExtension < Node
+      attr_reader :name, :directives, :input_fields_definition
+
+      def initialize pos, name, directives, input_fields_definition
+        super(pos)
+        @name = name
+        @directives = directives
+        @input_fields_definition = input_fields_definition
+      end
+
+      def accept viz
+        viz.handle_input_object_type_extension self
+      end
+
+      def fold viz, seed
+        viz.handle_input_object_type_extension self, seed
+      end
+
+      def input_object_type_extension?; true; end
+
+      def children
+        ary = []; ary.concat(directives) if directives; ary.concat(input_fields_definition) if input_fields_definition; ary
       end
     end
   end

data/lib/tinygql/nodes.rb.erb

@@ -3,8 +3,20 @@ module TinyGQL
     class Node
       include Enumerable
 
+      # `pos` is the position of the _end_ of a token
+      attr_reader :pos
+
+      def initialize pos
+        @pos = pos
+      end
+
+      # Return the line of this node given `doc`
+      def line doc
+        doc[0, @pos].count("\n") + 1
+      end
+
 <%- nodes.each do |node| -%>
-    def <%= node.human_name %>?; false; end
+      def <%= node.human_name %>?; false; end
 <%- end -%>
       def each(&blk)
         yield self
@@ -17,7 +29,8 @@ module TinyGQL
       <%- if node.fields.any? -%>
       attr_reader <%= node.fields.map { |v| ":" + v.name }.join(", ") %>
 
-      def initialize <%= node.fields.map(&:name).join(", ") %>
+      def initialize <%= (["pos"] + node.fields.map(&:name)).join(", ") %>
+        super(pos)
         <%- node.fields.each do |field| -%>
         @<%= field.name %> = <%= field.name %>
         <%- end -%>