added alias and support for unescaped string (variables)

Author: ChuckJonas

force-app/main/default/classes/GraphQLArgument.cls

@@ -1,15 +1,23 @@
 public class GraphQLArgument{
     public String name;
     public Object value; //string | GraphQLArgument | GraphQLArgument[]
+    public Boolean isVariable;
 
     public GraphQLArgument(String name, Object value){
         this.name = name;
         this.value = value;
+        this.isVariable = false;
+    }
+
+    public GraphQLArgument(String name, Object value, Boolean isVariable){
+        this.name = name;
+        this.value = value;
+        this.isVariable = isVariable;
     }
 
     public string build(){
         String qry = this.name + ':';
-        if(this.value instanceOf Integer || this.value instanceOf Decimal){
+        if(isVariable || this.value instanceOf Integer || this.value instanceOf Decimal){
             qry += String.valueOf(this.value);
         }else if(this.value instanceOf DateTime){
             qry += '\"' + ((DateTime) this.value).format('yyyy-MM-dd\'T\'HH:mm:ss.SSS\'Z\'') + '\"';
@@ -21,7 +29,7 @@ public class GraphQLArgument{
                 qry += childParam.build();
             }
             qry += '}';
-        }else{ // default to string
+        }else { // default to string
             qry += '\"' + String.valueOf(this.value) + '\"';
         }
         return qry;

force-app/main/default/classes/GraphQLNode.cls

@@ -1,6 +1,7 @@
 public class GraphQLNode {
     public String id;
     public Boolean typeFrament; // transforms to "...on {id} {"
+    public String alias;
     public GraphQLArgument arg; //not sure this supports all use cases... might need to be GraphQLArgument | GraphQLArgument[]
     public Object[] children; //object or other nodes
 
@@ -16,6 +17,12 @@ public class GraphQLNode {
         return this;
     }
 
+    //returns the current node
+    public GraphQLNode setAlias(String alias){
+        this.alias = alias;
+        return this;
+    }
+
     //returns the current node
     public GraphQLNode setArguments(GraphQLArgument arg){
         this.arg = arg;
@@ -37,10 +44,12 @@ public class GraphQLNode {
         return this;
     }
 
-    private string buildInner(){
+    public string build(){
         String qry = this.id;
         if(this.typeFrament){
             qry = '... on ' + this.id;
+        }else if(this.alias != null){
+            qry = this.alias + ': ' + qry;
         }
 
         if(this.arg != null){
@@ -49,7 +58,7 @@ public class GraphQLNode {
         qry += '{\n';
         for(Object child : this.children){
             if(child instanceOf GraphQLNode){
-                qry += ((GraphQLNode) child).buildInner();
+                qry += ((GraphQLNode) child).build();
             }else{
                  qry += ((String) child) + '\n';
             }
@@ -58,9 +67,4 @@ public class GraphQLNode {
         return qry;
     }
 
-    
-    public string build(){
-        return '{\n' + this.buildInner() + '\n}';
-    }
-
 }

force-app/main/default/classes/GraphQLQuery.cls

@@ -7,9 +7,28 @@ public class GraphQLQuery {
     }
 
     public GraphQLQuery(GraphQLNode node, Object variables){
-        this.query = node.build();
+        this.query = buildQuery(node);
         if(variables != null){
             this.variables = JSON.serialize(variables);
         }
     }
+
+    public GraphQLQuery(GraphQLNode[] nodes, Object variables){
+        this.query = buildQuery(nodes);
+        if(variables != null){
+            this.variables = JSON.serialize(variables);
+        }
+    }
+
+    private static string buildQuery(GraphQLNode node){
+        return '{\n' + node.build() + '\n}';
+    }
+
+    private static string buildQuery(GraphQLNode[] nodes){
+        String[] nodeStrings = new String[]{};
+        for(GraphQLNode node : nodes){
+            nodeStrings.add(node.build());
+        }
+        return '{\n' + String.join(nodeStrings, '\n') + '\n}';
+    }
 }

force-app/main/default/classes/GraphQLQueryTests.cls

@@ -19,15 +19,15 @@ public class GraphQLQueryTests {
     private static void simpleNode(){
         GraphQLNode n = new GraphQLNode('hello')
         .add('world');
-        System.assertEquals('{hello{world}}', n.build().deleteWhitespace());
+        System.assertEquals('hello{world}', n.build().deleteWhitespace());
     }
 
     @isTest
     private static void testSimpleNodeInput(){
         GraphQLNode n = new GraphQLNode('hello')
         .setArguments(new GraphQLArgument('key', 'value'))
         .add('world');
-        System.assertEquals('{hello(key:\"value\"){world}}', n.build().deleteWhitespace());
+        System.assertEquals('hello(key:\"value\"){world}', n.build().deleteWhitespace());
     }
 
     @isTest
@@ -36,7 +36,7 @@ public class GraphQLQueryTests {
         .add(
             new GraphQLNode('hello').add('world')
         );
-        System.assertEquals('{message{hello{world}}}', n.build().deleteWhitespace());
+        System.assertEquals('message{hello{world}}', n.build().deleteWhitespace());
     }
 
     @isTest
@@ -47,11 +47,28 @@ public class GraphQLQueryTests {
             .setTypeFragment(true)
             .add('x')
         });
-        System.assertEquals('{foo{... on bar{x}}}', n.build().remove('\n'));
+        System.assertEquals('foo{... on bar{x}}', n.build().remove('\n'));
+    }
+
+    @isTest
+    private static void testAlias(){
+        GraphQLNode[] nodes = new GraphQLNode[]{
+             new GraphQLNode('hero')
+            .setAlias('empireHero')
+            .setArguments(new GraphQLArgument('episode', 'EMPIRE', true))
+            .add('name'),
+            new GraphQLNode('hero')
+            .setAlias('jediHero')
+            .setArguments(new GraphQLArgument('episode', 'JEDI', true))
+            .add('name')
+        };
+        
+        GraphQLQuery qry = new GraphQLQuery(nodes, null);
+        System.assertEquals('{empireHero: hero(episode:EMPIRE){name}jediHero: hero(episode:JEDI){name}}', qry.query.remove('\n'));
     }
 
     private class TestVariables {
         public String foo;
         public Decimal bar;
     }
-}
+}