added support for operation

Author: ChuckJonas

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

@@ -1,9 +1,11 @@
 public class GraphQLNode {
     public String id;
+    public String operation; //optional query, mutation, subscription
     public Boolean typeFrament; // transforms to "...on {id} {"
     public String alias;
     public GraphQLArgument[] args; //not sure this supports all use cases... might need to be GraphQLArgument | GraphQLArgument[]
     public Object[] children; //object or other nodes
+    
 
     public GraphQLNode(String id){
         this.typeFrament = false;
@@ -18,6 +20,12 @@ public class GraphQLNode {
         return this;
     }
 
+    //returns the current node
+    public GraphQLNode setOperation(String operation){
+        this.operation = operation;
+        return this;
+    }
+
     //returns the current node
     public GraphQLNode setAlias(String alias){
         this.alias = alias;
@@ -52,7 +60,12 @@ public class GraphQLNode {
 
     public string build(){
         String qry = this.id;
-        if(this.typeFrament){
+
+        //we should probably refactor to different classes
+        //which all extend GraphQlNode (OperationNode, FragmentNode, etc)
+        if(operation != null){
+            qry = operation + ' ' +  qry;
+        }else if(this.typeFrament){
             qry = '... on ' + this.id;
         }else if(this.alias != null){
             qry = this.alias + ': ' + qry;

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

@@ -1,26 +1,25 @@
 public class GraphQLQuery {
     public string query;
-    public string variables;
-    public GraphQLQuery(string query, string variables){
+    public Object variables;
+    public GraphQLQuery(string query, Object variables){
         this.query = query;
         this.variables = variables;
     }
 
     public GraphQLQuery(GraphQLNode node, Object variables){
         this.query = buildQuery(node);
-        if(variables != null){
-            this.variables = JSON.serialize(variables);
-        }
+        this.variables = variables;
     }
 
     public GraphQLQuery(GraphQLNode[] nodes, Object variables){
         this.query = buildQuery(nodes);
-        if(variables != null){
-            this.variables = JSON.serialize(variables);
-        }
+        this.variables = variables;
     }
 
     private static string buildQuery(GraphQLNode node){
+        if(node.operation != null){
+            return node.build();
+        }
         return '{\n' + node.build() + '\n}';
     }
 

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

@@ -90,6 +90,38 @@ public class GraphQLQueryTests {
         System.assertEquals('{empireHero: hero(episode:EMPIRE){name}jediHero: hero(episode:JEDI){name}}', qry.query.remove('\n'));
     }
 
+
+    
+    // mutation CreateReviewForEpisode($ep: Episode!, $review: ReviewInput!) {
+    //     createReview(episode: $ep, review: $review) {
+    //         stars
+    //         commentary
+    //     }
+    // }
+    @isTest
+    private static void testOperation(){
+        GraphQLNode node = new GraphQLNode('CreateReviewForEpisode')
+            .setOperation('mutation')
+            .addArguments(new GraphQLArgument('$ep', 'Episode!', true))
+            .addArguments(new GraphQLArgument('$review', 'ReviewInput!', true))
+            .add(
+                new GraphQLNode('createReview')
+                .addArguments(new GraphQLArgument[]{
+                    new GraphQLArgument('episode', '$ep', true),
+                    new GraphQLArgument('review', '$review', true)
+                })
+                .add(new Object[]{'stars', 'commentary'})
+            );
+        
+        TestVariables test = new TestVariables();
+        GraphQLQuery qry = new GraphQLQuery(node, test);
+        System.assertEquals(
+            'mutation CreateReviewForEpisode($ep:Episode!, $review:ReviewInput!){ createReview(episode:$ep, review:$review){ stars commentary } }', 
+            qry.query.replace('\n', ' ').trim()
+        );
+    }
+
+
     private class TestVariables {
         public String foo;
         public Decimal bar;