tinygql 0.1.0

data/lib/tinygql/visitors.rb

@@ -0,0 +1,442 @@
+module TinyGQL
+  module Visitors
+    module Visitor
+      def handle_document obj
+            obj.definitions.each { _1.accept self }
+      end
+      def handle_operation_definition obj
+        if obj.variable_definitions
+            obj.variable_definitions.each { _1.accept self }
+        end
+        if obj.directives
+            obj.directives.each { _1.accept self }
+        end
+            obj.selection_set.each { _1.accept self }
+      end
+      def handle_variable obj
+      end
+      def handle_named_type obj
+      end
+      def handle_not_null_type obj
+            obj.type.accept self
+      end
+      def handle_list_type obj
+            obj.type.accept self
+      end
+      def handle_variable_definition obj
+            obj.variable.accept self
+            obj.type.accept self
+        if obj.default_value
+            obj.default_value.accept self
+        end
+      end
+      def handle_value obj
+      end
+      def handle_argument obj
+            obj.value.accept self
+      end
+      def handle_field obj
+        if obj.arguments
+            obj.arguments.each { _1.accept self }
+        end
+        if obj.directives
+            obj.directives.each { _1.accept self }
+        end
+        if obj.selection_set
+            obj.selection_set.each { _1.accept self }
+        end
+      end
+      def handle_object_field obj
+            obj.value.accept self
+      end
+      def handle_int_value obj
+      end
+      def handle_float_value obj
+      end
+      def handle_string_value obj
+      end
+      def handle_boolean_value obj
+      end
+      def handle_null_value obj
+      end
+      def handle_enum_value obj
+      end
+      def handle_list_value obj
+      end
+      def handle_object_value obj
+            obj.values.each { _1.accept self }
+      end
+      def handle_directive obj
+            obj.arguments.each { _1.accept self }
+      end
+      def handle_type_condition obj
+            obj.named_type.accept self
+      end
+      def handle_inline_fragment obj
+        if obj.type_condition
+            obj.type_condition.accept self
+        end
+        if obj.directives
+            obj.directives.each { _1.accept self }
+        end
+            obj.selection_set.each { _1.accept self }
+      end
+      def handle_fragment_spread obj
+            obj.fragment_name.accept self
+        if obj.directives
+            obj.directives.each { _1.accept self }
+        end
+      end
+      def handle_fragment_definition obj
+            obj.fragment_name.accept self
+            obj.type_condition.accept self
+        if obj.directives
+            obj.directives.each { _1.accept self }
+        end
+            obj.selection_set.each { _1.accept self }
+      end
+      def handle_root_operation_type_definition obj
+            obj.operation_type.accept self
+            obj.named_type.accept self
+      end
+      def handle_schema_definition obj
+        if obj.directives
+            obj.directives.each { _1.accept self }
+        end
+            obj.root_operation_definitions.each { _1.accept self }
+      end
+      def handle_field_definition obj
+        if obj.arguments_definition
+            obj.arguments_definition.each { _1.accept self }
+        end
+            obj.type.accept self
+        if obj.directives
+            obj.directives.each { _1.accept self }
+        end
+      end
+      def handle_input_value_definition obj
+            obj.type.accept self
+        if obj.default_value
+            obj.default_value.accept self
+        end
+        if obj.directives
+            obj.directives.each { _1.accept self }
+        end
+      end
+      def handle_object_type_definition obj
+        if obj.implements_interfaces
+            obj.implements_interfaces.each { _1.accept self }
+        end
+        if obj.directives
+            obj.directives.each { _1.accept self }
+        end
+        if obj.fields_definition
+            obj.fields_definition.each { _1.accept self }
+        end
+      end
+      def handle_interface_type_definition obj
+        if obj.directives
+            obj.directives.each { _1.accept self }
+        end
+        if obj.fields_definition
+            obj.fields_definition.each { _1.accept self }
+        end
+      end
+      def handle_union_type_definition obj
+        if obj.directives
+            obj.directives.each { _1.accept self }
+        end
+        if obj.union_member_types
+            obj.union_member_types.each { _1.accept self }
+        end
+      end
+      def handle_scalar_type_definition obj
+        if obj.directives
+            obj.directives.each { _1.accept self }
+        end
+      end
+      def handle_enum_value_definition obj
+            obj.enum_value.accept self
+        if obj.directives
+            obj.directives.each { _1.accept self }
+        end
+      end
+      def handle_enum_type_definition obj
+        if obj.directives
+            obj.directives.each { _1.accept self }
+        end
+        if obj.enum_value_definition
+            obj.enum_value_definition.each { _1.accept self }
+        end
+      end
+      def handle_input_object_type_definition obj
+        if obj.directives
+            obj.directives.each { _1.accept self }
+        end
+        if obj.input_fields_definition
+            obj.input_fields_definition.each { _1.accept self }
+        end
+      end
+      def handle_object_type_extension obj
+        if obj.implements_interfaces
+            obj.implements_interfaces.each { _1.accept self }
+        end
+        if obj.directives
+            obj.directives.each { _1.accept self }
+        end
+        if obj.fields_definition
+            obj.fields_definition.each { _1.accept self }
+        end
+      end
+      def handle_executable_directive_location obj
+      end
+      def handle_type_system_directive_location obj
+      end
+      def handle_directive_definition obj
+        if obj.arguments_definition
+            obj.arguments_definition.each { _1.accept self }
+        end
+            obj.directive_locations.each { _1.accept self }
+      end
+    end
+
+    module Fold
+      def handle_document obj, seed
+            obj.definitions.each { seed = _1.fold(self, seed) }
+        seed
+      end
+      def handle_operation_definition obj, seed
+        if obj.variable_definitions
+            obj.variable_definitions.each { seed = _1.fold(self, seed) }
+        end
+        if obj.directives
+            obj.directives.each { seed = _1.fold(self, seed) }
+        end
+            obj.selection_set.each { seed = _1.fold(self, seed) }
+        seed
+      end
+      def handle_variable obj, seed
+        seed
+      end
+      def handle_named_type obj, seed
+        seed
+      end
+      def handle_not_null_type obj, seed
+            seed = obj.type.fold self, seed
+        seed
+      end
+      def handle_list_type obj, seed
+            seed = obj.type.fold self, seed
+        seed
+      end
+      def handle_variable_definition obj, seed
+            seed = obj.variable.fold self, seed
+            seed = obj.type.fold self, seed
+        if obj.default_value
+            seed = obj.default_value.fold self, seed
+        end
+        seed
+      end
+      def handle_value obj, seed
+        seed
+      end
+      def handle_argument obj, seed
+            seed = obj.value.fold self, seed
+        seed
+      end
+      def handle_field obj, seed
+        if obj.arguments
+            obj.arguments.each { seed = _1.fold(self, seed) }
+        end
+        if obj.directives
+            obj.directives.each { seed = _1.fold(self, seed) }
+        end
+        if obj.selection_set
+            obj.selection_set.each { seed = _1.fold(self, seed) }
+        end
+        seed
+      end
+      def handle_object_field obj, seed
+            seed = obj.value.fold self, seed
+        seed
+      end
+      def handle_int_value obj, seed
+        seed
+      end
+      def handle_float_value obj, seed
+        seed
+      end
+      def handle_string_value obj, seed
+        seed
+      end
+      def handle_boolean_value obj, seed
+        seed
+      end
+      def handle_null_value obj, seed
+        seed
+      end
+      def handle_enum_value obj, seed
+        seed
+      end
+      def handle_list_value obj, seed
+        seed
+      end
+      def handle_object_value obj, seed
+            obj.values.each { seed = _1.fold(self, seed) }
+        seed
+      end
+      def handle_directive obj, seed
+            obj.arguments.each { seed = _1.fold(self, seed) }
+        seed
+      end
+      def handle_type_condition obj, seed
+            seed = obj.named_type.fold self, seed
+        seed
+      end
+      def handle_inline_fragment obj, seed
+        if obj.type_condition
+            seed = obj.type_condition.fold self, seed
+        end
+        if obj.directives
+            obj.directives.each { seed = _1.fold(self, seed) }
+        end
+            obj.selection_set.each { seed = _1.fold(self, seed) }
+        seed
+      end
+      def handle_fragment_spread obj, seed
+            seed = obj.fragment_name.fold self, seed
+        if obj.directives
+            obj.directives.each { seed = _1.fold(self, seed) }
+        end
+        seed
+      end
+      def handle_fragment_definition obj, seed
+            seed = obj.fragment_name.fold self, seed
+            seed = obj.type_condition.fold self, seed
+        if obj.directives
+            obj.directives.each { seed = _1.fold(self, seed) }
+        end
+            obj.selection_set.each { seed = _1.fold(self, seed) }
+        seed
+      end
+      def handle_root_operation_type_definition obj, seed
+            seed = obj.operation_type.fold self, seed
+            seed = obj.named_type.fold self, seed
+        seed
+      end
+      def handle_schema_definition obj, seed
+        if obj.directives
+            obj.directives.each { seed = _1.fold(self, seed) }
+        end
+            obj.root_operation_definitions.each { seed = _1.fold(self, seed) }
+        seed
+      end
+      def handle_field_definition obj, seed
+        if obj.arguments_definition
+            obj.arguments_definition.each { seed = _1.fold(self, seed) }
+        end
+            seed = obj.type.fold self, seed
+        if obj.directives
+            obj.directives.each { seed = _1.fold(self, seed) }
+        end
+        seed
+      end
+      def handle_input_value_definition obj, seed
+            seed = obj.type.fold self, seed
+        if obj.default_value
+            seed = obj.default_value.fold self, seed
+        end
+        if obj.directives
+            obj.directives.each { seed = _1.fold(self, seed) }
+        end
+        seed
+      end
+      def handle_object_type_definition obj, seed
+        if obj.implements_interfaces
+            obj.implements_interfaces.each { seed = _1.fold(self, seed) }
+        end
+        if obj.directives
+            obj.directives.each { seed = _1.fold(self, seed) }
+        end
+        if obj.fields_definition
+            obj.fields_definition.each { seed = _1.fold(self, seed) }
+        end
+        seed
+      end
+      def handle_interface_type_definition obj, seed
+        if obj.directives
+            obj.directives.each { seed = _1.fold(self, seed) }
+        end
+        if obj.fields_definition
+            obj.fields_definition.each { seed = _1.fold(self, seed) }
+        end
+        seed
+      end
+      def handle_union_type_definition obj, seed
+        if obj.directives
+            obj.directives.each { seed = _1.fold(self, seed) }
+        end
+        if obj.union_member_types
+            obj.union_member_types.each { seed = _1.fold(self, seed) }
+        end
+        seed
+      end
+      def handle_scalar_type_definition obj, seed
+        if obj.directives
+            obj.directives.each { seed = _1.fold(self, seed) }
+        end
+        seed
+      end
+      def handle_enum_value_definition obj, seed
+            seed = obj.enum_value.fold self, seed
+        if obj.directives
+            obj.directives.each { seed = _1.fold(self, seed) }
+        end
+        seed
+      end
+      def handle_enum_type_definition obj, seed
+        if obj.directives
+            obj.directives.each { seed = _1.fold(self, seed) }
+        end
+        if obj.enum_value_definition
+            obj.enum_value_definition.each { seed = _1.fold(self, seed) }
+        end
+        seed
+      end
+      def handle_input_object_type_definition obj, seed
+        if obj.directives
+            obj.directives.each { seed = _1.fold(self, seed) }
+        end
+        if obj.input_fields_definition
+            obj.input_fields_definition.each { seed = _1.fold(self, seed) }
+        end
+        seed
+      end
+      def handle_object_type_extension obj, seed
+        if obj.implements_interfaces
+            obj.implements_interfaces.each { seed = _1.fold(self, seed) }
+        end
+        if obj.directives
+            obj.directives.each { seed = _1.fold(self, seed) }
+        end
+        if obj.fields_definition
+            obj.fields_definition.each { seed = _1.fold(self, seed) }
+        end
+        seed
+      end
+      def handle_executable_directive_location obj, seed
+        seed
+      end
+      def handle_type_system_directive_location obj, seed
+        seed
+      end
+      def handle_directive_definition obj, seed
+        if obj.arguments_definition
+            obj.arguments_definition.each { seed = _1.fold(self, seed) }
+        end
+            obj.directive_locations.each { seed = _1.fold(self, seed) }
+        seed
+      end
+    end
+  end
+end

data/lib/tinygql/visitors.rb.erb

@@ -0,0 +1,46 @@
+module TinyGQL
+  module Visitors
+    module Visitor
+<%- nodes.each do |node| -%>
+      def handle_<%= node.human_name %> obj
+      <%- node.fields.find_all(&:visitable?).each do |field| -%>
+        <%- if field.nullable? -%>
+        if obj.<%= field.name %>
+        <%- end -%>
+        <%- if field.list? -%>
+            obj.<%= field.name %>.each { _1.accept self }
+        <%- end -%>
+        <%- if field.node? -%>
+            obj.<%= field.name %>.accept self
+        <%- end -%>
+        <%- if field.nullable? -%>
+        end
+        <%- end -%>
+      <%- end -%>
+      end
+<%- end -%>
+    end
+
+    module Fold
+<%- nodes.each do |node| -%>
+      def handle_<%= node.human_name %> obj, seed
+      <%- node.fields.find_all(&:visitable?).each do |field| -%>
+        <%- if field.nullable? -%>
+        if obj.<%= field.name %>
+        <%- end -%>
+        <%- if field.list? -%>
+            obj.<%= field.name %>.each { seed = _1.fold(self, seed) }
+        <%- end -%>
+        <%- if field.node? -%>
+            seed = obj.<%= field.name %>.fold self, seed
+        <%- end -%>
+        <%- if field.nullable? -%>
+        end
+        <%- end -%>
+      <%- end -%>
+        seed
+      end
+<%- end -%>
+    end
+  end
+end

data/lib/tinygql.rb

@@ -0,0 +1,10 @@
+require "tinygql/parser"
+require "tinygql/version"
+
+module TinyGQL
+  autoload :Visitors, "tinygql/visitors"
+
+  def self.parse doc
+    Parser.new(doc).parse
+  end
+end

data/test/helper.rb

@@ -0,0 +1,14 @@
+ENV["MT_NO_PLUGINS"] = "1"
+
+require "tinygql/parser"
+require "minitest/autorun"
+
+Thread.new do
+  sleep 5
+  Process.kill "QUIT", $$
+end
+
+module TinyGQL
+  class Test < Minitest::Test
+  end
+end

data/test/kitchen-sink.graphql

@@ -0,0 +1,59 @@
+# Copyright (c) 2015-present, Facebook, Inc.
+#
+# This source code is licensed under the MIT license found in the
+# LICENSE file in the root directory of this source tree.
+
+query queryName($foo: ComplexType, $site: Site = MOBILE) {
+  whoever123is: node(id: [123, 456]) {
+    id ,
+    ... on User @defer {
+      field2 {
+        id ,
+        alias: field1(first:10, after:$foo,) @include(if: $foo) {
+          id,
+          ...frag
+        }
+      }
+    }
+    ... @skip(unless: $foo) {
+      id
+    }
+    ... {
+      id
+    }
+  }
+}
+
+mutation likeStory {
+  like(story: 123) @defer {
+    story {
+      id
+    }
+  }
+}
+
+subscription StoryLikeSubscription($input: StoryLikeSubscribeInput) {
+  storyLikeSubscribe(input: $input) {
+    story {
+      likers {
+        count
+      }
+      likeSentence {
+        text
+      }
+    }
+  }
+}
+
+fragment frag on Friend {
+  foo(size: $size, bar: $b, obj: {key: "value", block: """
+
+      block string uses \"""
+
+  """})
+}
+
+{
+  unnamed(truthy: true, falsey: false, nullish: null),
+  query
+}

data/test/lexer_test.rb

@@ -0,0 +1,93 @@
+require "helper"
+
+module TinyGQL
+  class LexerTest < Test
+    PUNC_LUT = {"!"=>[:BANG, "!"],
+                "$"=>[:VAR_SIGN, "$"],
+                "("=>[:LPAREN, "("],
+                ")"=>[:RPAREN, ")"],
+                "..."=>[:ELLIPSIS, "..."],
+                ":"=>[:COLON, ":"],
+                "="=>[:EQUALS, "="],
+                "@"=>[:DIR_SIGN, "@"],
+                "["=>[:LBRACKET, "["],
+                "]"=>[:RBRACKET, "]"],
+                "{"=>[:LCURLY, "{"],
+                "|"=>[:PIPE, "|"],
+                "}"=>[:RCURLY, "}"]}
+
+    def test_punc
+      %w{ ! $ ( ) ... : = @ [ ] { | } }.each do |punc|
+        lexer = Lexer.new punc
+        token = lexer.next_token
+        expected = PUNC_LUT[punc]
+        assert_equal(expected || [:ON, "on"], token)
+      end
+    end
+
+    def test_regular_string
+      str = "hello\n# foo\n\"world\"# lol \nlol"
+      lexer = Lexer.new str
+      assert_equal [:IDENTIFIER, "hello"], lexer.next_token
+      assert_equal [:STRING, "world"], lexer.next_token
+      assert_equal [:IDENTIFIER, "lol"], lexer.next_token
+    end
+
+    def test_multiline_comment
+      str = "hello\n# foo\n# lol \nlol"
+      lexer = Lexer.new str
+      assert_equal [:IDENTIFIER, "hello"], lexer.next_token
+      assert_equal [:IDENTIFIER, "lol"], lexer.next_token
+    end
+
+    def test_block_string
+      doc = <<-eos
+"""
+
+      block string uses \\"""
+
+"""
+      eos
+      lexer = Lexer.new doc
+      assert_equal :STRING, lexer.next_token.first
+    end
+
+    def test_tokenize
+      lexer = Lexer.new "on"
+      token = lexer.next_token
+      assert_equal [:ON, "on"], token
+    end
+
+    def test_multi_tok
+      doc = <<-eod
+mutation {
+  likeStory(storyID: 12345) {
+    story {
+      likeCount
+    }
+  }
+}
+eod
+      lexer = Lexer.new doc
+      toks = []
+      while tok = lexer.next_token
+        toks << tok
+      end
+      assert_equal [[:MUTATION, "mutation"],
+                    [:LCURLY, "{"],
+                    [:IDENTIFIER, "likeStory"],
+                    [:LPAREN, "("],
+                    [:IDENTIFIER, "storyID"],
+                    [:COLON, ":"],
+                    [:INT, "12345"],
+                    [:RPAREN, ")"],
+                    [:LCURLY, "{"],
+                    [:IDENTIFIER, "story"],
+                    [:LCURLY, "{"],
+                    [:IDENTIFIER, "likeCount"],
+                    [:RCURLY, "}"],
+                    [:RCURLY, "}"],
+                    [:RCURLY, "}"]], toks
+    end
+  end
+end