You can now set your own ANTLR converter in the parse - NOT API

bbakerman · bbakerman · commit e28e51cf6f51 · 2020-02-12T17:33:42.000+11:00
diff --git a/src/main/java/graphql/parser/Parser.java b/src/main/java/graphql/parser/Parser.java
@@ -74,7 +74,7 @@ public void syntaxError(Recognizer<?, ?> recognizer, Object offendingSymbol, int
         ExtendedBailStrategy bailStrategy = new ExtendedBailStrategy(multiSourceReader);
         parser.setErrorHandler(bailStrategy);
 
-        GraphqlAntlrToLanguage toLanguage = new GraphqlAntlrToLanguage(tokens, multiSourceReader);
+        GraphqlAntlrToLanguage toLanguage = getAntlrToLanguage(tokens, multiSourceReader);
         GraphqlParser.DocumentContext documentContext = parser.document();
 
         Document doc = toLanguage.createDocument(documentContext);
@@ -96,4 +96,7 @@ public void syntaxError(Recognizer<?, ?> recognizer, Object offendingSymbol, int
         return doc;
     }
 
+    protected GraphqlAntlrToLanguage getAntlrToLanguage(CommonTokenStream tokens, MultiSourceReader multiSourceReader) {
+        return new GraphqlAntlrToLanguage(tokens, multiSourceReader);
+    }
 }
diff --git a/src/test/groovy/graphql/parser/ParserTest.groovy b/src/test/groovy/graphql/parser/ParserTest.groovy
@@ -21,6 +21,7 @@ import graphql.language.IntValue
 import graphql.language.InterfaceTypeDefinition
 import graphql.language.ListType
 import graphql.language.Node
+import graphql.language.NodeBuilder
 import graphql.language.NonNullType
 import graphql.language.NullValue
 import graphql.language.ObjectField
@@ -36,6 +37,8 @@ import graphql.language.TypeName
 import graphql.language.UnionTypeDefinition
 import graphql.language.VariableDefinition
 import graphql.language.VariableReference
+import org.antlr.v4.runtime.CommonTokenStream
+import org.antlr.v4.runtime.ParserRuleContext
 import spock.lang.Specification
 import spock.lang.Unroll
 
@@ -772,5 +775,33 @@ triple3 : """edge cases \\""" "" " \\"" \\" edge cases"""
         operationDefinition.getComments()[0].content == " Represents the 😕 emoji."
     }
 
+    def "can override antlr to ast"() {
 
+        def query = '''
+            query {
+                field
+            }
+        '''
+        when:
+        Parser parser = new Parser() {
+            @Override
+            protected GraphqlAntlrToLanguage getAntlrToLanguage(CommonTokenStream tokens, MultiSourceReader multiSourceReader) {
+                // this pattern is used in Nadel - its backdoor but needed
+                return new GraphqlAntlrToLanguage(tokens,multiSourceReader) {
+                    @Override
+                    protected void addCommonData(NodeBuilder nodeBuilder, ParserRuleContext parserRuleContext) {
+                        super.addCommonData(nodeBuilder, parserRuleContext)
+                        nodeBuilder.additionalData("key", "value")
+                    }
+                }
+            }
+        }
+
+        def document = parser.parseDocument(query)
+
+        then:
+        document.getAdditionalData().get("key") == "value"
+        document.children[0].getAdditionalData().get("key") == "value"
+
+    }
 }

Original file line number	Diff line number	Diff line change
`@@ -74,7 +74,7 @@ public void syntaxError(Recognizer<?, ?> recognizer, Object offendingSymbol, int`
`74`	`74`	`ExtendedBailStrategy bailStrategy = new ExtendedBailStrategy(multiSourceReader);`
`75`	`75`	`parser.setErrorHandler(bailStrategy);`
`76`	`76`
`77`		`- GraphqlAntlrToLanguage toLanguage = new GraphqlAntlrToLanguage(tokens, multiSourceReader);`
	`77`	`+ GraphqlAntlrToLanguage toLanguage = getAntlrToLanguage(tokens, multiSourceReader);`
`78`	`78`	`GraphqlParser.DocumentContext documentContext = parser.document();`
`79`	`79`
`80`	`80`	`Document doc = toLanguage.createDocument(documentContext);`
`@@ -96,4 +96,7 @@ public void syntaxError(Recognizer<?, ?> recognizer, Object offendingSymbol, int`
`96`	`96`	`return doc;`
`97`	`97`	`}`
`98`	`98`
	`99`	`+ protected GraphqlAntlrToLanguage getAntlrToLanguage(CommonTokenStream tokens, MultiSourceReader multiSourceReader) {`
	`100`	`+ return new GraphqlAntlrToLanguage(tokens, multiSourceReader);`
	`101`	`+ }`
`99`	`102`	`}`