@devticon-os/graphql-codegen-axios 0.3.13 → 0.3.14

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@devticon-os/graphql-codegen-axios",
-  "version": "0.3.13",
+  "version": "0.3.14",
   "license": "MIT",
   "main": "lib/index.js",
   "repository": "https://github.com/devticon/graphql-codegen-axios",

package/templates/helpers.ts

@@ -10,14 +10,45 @@ type GraphqlRequestParams = {
   query: string;
   variables?: any;
 };
-const first = (key: string) => (data: any) => {
-  data[key] = data[key][0];
+
+const get = (key: string, data: any) => {
+  const p = key.split('.');
+  let d = data;
+  while (p.length) {
+    const k = p.shift();
+    d = d[k];
+    if (p.length === 0) {
+      return d;
+    }
+    if (d === undefined) {
+      throw new Error(`Invalid path ${key} missing ${k}`);
+    }
+  }
+};
+
+const set = (key: string, value: any, data: any) => {
+  const p = key.split('.');
+  let d = data;
+  while (p.length) {
+    const k = p.shift();
+    if (p.length === 0) {
+      d[k] = value;
+      return;
+    }
+    d = d[k];
+    if (d === undefined) {
+      throw new Error(`Invalid path ${key} missing ${k}`);
+    }
+  }
+};
+export const first = (key: string) => (data: any) => {
+  set(key, get(key + '.0', data), data);
   return data;
 };
 
-const firstOrFail = (key: string) => (data: any) => {
-  data[key] = (data as any)[key][0];
-  if (!data[key]) {
+export const firstOrFail = (key: string) => (data: any) => {
+  first(key)(data);
+  if (!get(key, data)) {
     throw {
       message: `Empty list for ${key} `,
       code: 'EMPTY_LIST',
@@ -75,7 +106,7 @@ const execute = (
     return d;
   });
 
-export const singleResult = (key: string) => (data: any) => data[key];
+export const singleResult = (key: string) => (data: any) => get(key, data);
 
 export class GraphqlError extends Error {
   constructor(message: string, public gqlErrors: GraphQLError[]) {