add nested updates support

Author: Eldow

src/PostgraphileNestedMutationsPlugin.js

@@ -66,7 +66,8 @@ module.exports = function PostGraphileNestedMutationPlugin(builder) {
       pgQueryFromResolveData: queryFromResolveData,
       pgNestedPluginForwardInputTypes,
       pgNestedPluginReverseInputTypes,
-      pgNestedResolvers,
+      pgNestedCreateResolvers,
+      pgNestedUpdateResolvers,
       pgNestedTableConnectorFields,
       pgNestedTableConnect,
       pgNestedTableDeleterFields,
@@ -97,7 +98,7 @@ module.exports = function PostGraphileNestedMutationPlugin(builder) {
       !pgNestedPluginForwardInputTypes[table.id] &&
       !pgNestedPluginReverseInputTypes[table.id]
     ) {
-      pgNestedResolvers[table.id] = field.resolve;
+      pgNestedCreateResolvers[table.id] = field.resolve;
       return field;
     }
 
@@ -207,7 +208,7 @@ module.exports = function PostGraphileNestedMutationPlugin(builder) {
 
             if (fieldValue.create) {
               const createData = fieldValue.create;
-              const resolver = pgNestedResolvers[foreignTable.id];
+              const resolver = pgNestedCreateResolvers[foreignTable.id];
               const tableVar = inflection.tableFieldName(foreignTable);
 
               const insertData = Object.assign(
@@ -232,6 +233,35 @@ module.exports = function PostGraphileNestedMutationPlugin(builder) {
                   resolveResult.data[`__pk__${k.name}`];
               });
             }
+
+            await Promise.all(Object.keys(fieldValue).map(async k => {
+              if (k.includes('updateBy')) {
+                const rowData = fieldValue[k]
+                const resolver = pgNestedUpdateResolvers[foreignTable.id];
+
+                const updateData = Object.assign(
+                  {},
+                  rowData,
+                  await recurseForwardNestedMutations(
+                    data,
+                    { input: rowData },
+                    { pgClient },
+                    resolveInfo,
+                  ),
+                );
+
+                const resolveResult = await resolver(
+                  data,
+                  { input: updateData },
+                  { pgClient },
+                  resolveInfo,
+                );
+                foreignKeys.forEach((k, idx) => {
+                  output[inflection.column(keys[idx])] =
+                    resolveResult.data[`__pk__${k.name}`];
+                });
+              }
+            }))
           }),
       );
 
@@ -580,7 +610,7 @@ module.exports = function PostGraphileNestedMutationPlugin(builder) {
             if (fieldValue.create) {
               await Promise.all(
                 fieldValue.create.map(async (rowData) => {
-                  const resolver = pgNestedResolvers[foreignTable.id];
+                  const resolver = pgNestedCreateResolvers[foreignTable.id];
                   const tableVar = inflection.tableFieldName(foreignTable);
 
                   const keyData = {};
@@ -610,6 +640,33 @@ module.exports = function PostGraphileNestedMutationPlugin(builder) {
                 }),
               );
             }
+
+            await Promise.all(Object.keys(fieldValue).map(async k => {
+              if (k.includes('updateBy')) {
+                await Promise.all(
+                  fieldValue[k].map(async (rowData) => {
+                    const resolver = pgNestedUpdateResolvers[foreignTable.id];
+
+                    const { data: reverseRow } = await resolver(
+                      data,
+                      {
+                        input: Object.assign({}, rowData),
+                      },
+                      { pgClient },
+                      resolveInfo,
+                    );
+
+                    const rowKeyValues = {};
+                    if (primaryKeys) {
+                      primaryKeys.forEach((k) => {
+                        rowKeyValues[k.name] = reverseRow[`__pk__${k.name}`];
+                      });
+                    }
+                    modifiedRows.push(rowKeyValues);
+                  }),
+                );
+              }
+            }))
           }),
         );
 
@@ -655,7 +712,11 @@ module.exports = function PostGraphileNestedMutationPlugin(builder) {
     };
 
     if (isPgCreateMutationField) {
-      pgNestedResolvers[table.id] = newResolver;
+      pgNestedCreateResolvers[table.id] = newResolver;
+    }
+
+    if (isPgUpdateMutationField) {
+      pgNestedUpdateResolvers[table.id] = newResolver;
     }
 
     return Object.assign({}, field, { resolve: newResolver });

src/PostgraphileNestedTypesPlugin.js

@@ -41,7 +41,8 @@ module.exports = function PostGraphileNestedTypesPlugin(
     return extend(build, {
       pgNestedPluginForwardInputTypes: {},
       pgNestedPluginReverseInputTypes: {},
-      pgNestedResolvers: {},
+      pgNestedCreateResolvers: {},
+      pgNestedUpdateResolvers: {},
       pgNestedFieldName(options) {
         const {
           constraint: {