Merge pull request #271 from META-DREAMER/fix-multiple-aliases

Fix multiple aliases when generating query

Author: aexol

src/TreeToTS/functions/new/buildQuery.spec.ts

@@ -266,6 +266,40 @@ describe('Test generated function buildQuery', () => {
         }
     }`);
   });
+  test('Query with multiple aliases', () => {
+    const matchExact = replSpace(
+      builder('query', {
+        __alias: {
+          play: {
+            cards: {
+              name: true,
+              age: true,
+              bio: true,
+            },
+          },
+          shuffle: {
+            cards: {
+              name: true,
+              age: true,
+              bio: true,
+            },
+          },
+        },
+      }),
+    );
+    matchExact(`query{
+        play:cards{
+            name
+            age
+            bio
+        }
+        shuffle:cards{
+            name
+            age
+            bio
+        }
+    }`);
+  });
   test('Simple query with enums', () => {
     const enum Status {
       CREATED = 'CREATED',

src/TreeToTS/functions/new/buildQuery.ts

@@ -33,14 +33,18 @@ export const InternalsBuildQuery = (
       return `${ibb(args ? `${k}(${args})` : k, o[1], p, false)}`;
     }
     if (k === '__alias') {
-      const alias = Object.keys(o)[0];
-      const objectUnderAlias = o[alias];
-      if (typeof objectUnderAlias !== 'object' || Array.isArray(objectUnderAlias)) {
-        throw new Error('Invalid alias it should be __alias:{ YOUR_ALIAS_NAME: { OPERATION_NAME: { ...selectors }}}');
-      }
-      const operationName = Object.keys(objectUnderAlias)[0];
-      const operation = objectUnderAlias[operationName];
-      return ibb(`${alias}:${operationName}`, operation, p, false);
+      return Object.entries(o)
+        .map(([alias, objectUnderAlias]) => {
+          if (typeof objectUnderAlias !== 'object' || Array.isArray(objectUnderAlias)) {
+            throw new Error(
+              'Invalid alias it should be __alias:{ YOUR_ALIAS_NAME: { OPERATION_NAME: { ...selectors }}}',
+            );
+          }
+          const operationName = Object.keys(objectUnderAlias)[0];
+          const operation = objectUnderAlias[operationName];
+          return ibb(`${alias}:${operationName}`, operation, p, false);
+        })
+        .join('\n');
     }
     const hasOperationName = root && options?.operationName ? ' ' + options.operationName : '';
     const hasVariables = root && options?.variables?.$params ? `(${options.variables?.$params})` : '';