ads support for query variables

Author: terion-name

src/__tests__/jsonToGraphQLQuery.tests.ts

@@ -1,6 +1,6 @@
 
 import { expect } from 'chai';
-import { jsonToGraphQLQuery, EnumType } from '../';
+import { jsonToGraphQLQuery, EnumType, VariableType } from '../';
 
 describe('jsonToGraphQL()', () => {
 
@@ -305,3 +305,41 @@ describe('jsonToGraphQL()', () => {
       );
     });
 });
+
+it('converts a query variables', () => {
+    const query = {
+        query: {
+            __variables: {
+                variableName: 'String!'
+            },
+            Posts: {
+                __args: {
+                    arg1: 20,
+                    arg2: new VariableType('variableName')
+                },
+                id: true,
+                title: true,
+                comments: {
+                    __args: {
+                        offensiveOnly: true
+                    },
+                    id: true,
+                    comment: true,
+                    user: true
+                }
+            }
+        }
+    };
+    expect(jsonToGraphQLQuery(query, { pretty: true })).to.equal(
+`query ($variableName: String!) {
+    Posts (arg1: 20, arg2: $variableName) {
+        id
+        title
+        comments (offensiveOnly: true) {
+            id
+            comment
+            user
+        }
+    }
+}`);
+});

src/index.ts

@@ -1,2 +1,3 @@
-
 export * from './jsonToGraphQLQuery';
+export {EnumType} from './types/EnumType';
+export {VariableType} from './types/VariableType';

src/jsonToGraphQLQuery.ts

@@ -1,7 +1,14 @@
+import {EnumType} from './types/EnumType';
+import {VariableType} from './types/VariableType';
+
 function stringify(obj_from_json: any): string {
     if (obj_from_json instanceof EnumType) {
         return obj_from_json.value;
     }
+    // variables should be prefixed with dollar sign and not e quoted
+    else if (obj_from_json instanceof VariableType) {
+        return `$${obj_from_json.value}`;
+    }
     // Cheers to Derek: https://stackoverflow.com/questions/11233498/json-stringify-without-quotes-on-properties
     else if (typeof obj_from_json !== 'object' || obj_from_json === null) {
         // not an object, stringify using native function
@@ -28,12 +35,20 @@ function buildArgs(argsObj: any): string {
     return args.join(', ');
 }
 
+function buildVariables(varsObj: any): string {
+    const args = [];
+    for (const varName in varsObj) {
+        args.push(`$${varName}: ${varsObj[varName]}`);
+    }
+    return args.join(', ');
+}
+
 function getIndent(level: number): string {
     return Array((level * 4) + 1).join(' ');
 }
 
 function filterNonConfigFields(fieldName: string) {
-    return fieldName !== '__args' && fieldName !== '__alias';
+    return fieldName !== '__args' && fieldName !== '__alias' && fieldName !== '__variables';
 }
 
 function convertQuery(node: any, level: number, output: Array<[ string, number ]>) {
@@ -48,6 +63,9 @@ function convertQuery(node: any, level: number, output: Array<[ string, number ]
             if (typeof node[key].__args === 'object') {
                 token = `${key} (${buildArgs(node[key].__args)})`;
             }
+            else if (typeof node[key].__variables === 'object') {
+                token = `${key} (${buildVariables(node[key].__variables)})`;
+            }
             else {
                 token = `${key}`;
             }
@@ -96,7 +114,3 @@ export function jsonToGraphQLQuery(query: any, options: IJsonToGraphQLOptions =
     });
     return output;
 }
-
-export class EnumType {
-    constructor(public value: string) {}
-}

src/types/EnumType.ts

@@ -0,0 +1,5 @@
+class EnumType {
+    constructor(public value: string) {}
+}
+
+export {EnumType};

src/types/VariableType.ts

@@ -0,0 +1,5 @@
+class VariableType {
+    constructor(public value: string) {}
+}
+
+export {VariableType};