Add parser/generation support for directive definitions

rmosolgo · Sep 27, 2016 · 164b5de · 164b5de
1 parent 74c31bc
commit 164b5de
Show file tree

Hide file tree

Showing 8 changed files with 947 additions and 808 deletions.
diff --git a/lib/graphql/language/generation.rb b/lib/graphql/language/generation.rb
@@ -40,7 +40,6 @@ def generate(node, indent: "")
           out << generate_directives(node.directives)
           out << generate_selections(node.selections, indent: indent)
           out
-
         when Nodes::FragmentSpread
           out = "#{indent}...#{node.name}"
           out << generate_directives(node.directives)
@@ -123,6 +122,10 @@ def generate(node, indent: "")
             out << "  #{generate(field)}\n"
           end
           out << "}"
+        when Nodes::DirectiveDefinition
+          out = "directive @#{node.name}"
+          out << "(#{node.arguments.map { |a| generate(a) }.join(", ")})" if node.arguments.any?
+          out << " on #{node.locations.join(' | ')}"
         when Nodes::AbstractNode
           node.to_query_string(indent: indent)
         when FalseClass, Float, Integer, NilClass, String, TrueClass