@vantreeseba/drizzle-graphql 5.3.0 → 5.4.0

package/dist/index.cjs

@@ -2244,6 +2244,7 @@ var computeResolverFieldNames = (tableName, typeNameMapper, prefixes, suffixes)
   const upsertArrayFieldName = `${upsertPrefix}${mapped ? capitalize(mapped.plural) : capitalize(tableName)}`;
   const upsertSingleFieldName = mapped ? `${upsertPrefix}${capitalize(mapped.singular)}` : `${upsertPrefix}${capitalize(tableName)}${suffixes.single}`;
   const updateFieldName = `${prefixes.update}${mapped ? capitalize(mapped.singular) : capitalize(tableName)}`;
+  const updateManyFieldName = `${prefixes.update}${mapped ? capitalize(mapped.plural) : capitalize(tableName)}Many`;
   const deleteFieldName = `${prefixes.delete}${mapped ? capitalize(mapped.singular) : capitalize(tableName)}`;
   const writeSingleSuffix = suffixes.single === "" ? "Single" : suffixes.single;
   const updateSingleFieldName = `${updateFieldName}${writeSingleSuffix}`;
@@ -2259,11 +2260,28 @@ var computeResolverFieldNames = (tableName, typeNameMapper, prefixes, suffixes)
     upsertArrayFieldName,
     upsertSingleFieldName,
     updateFieldName,
+    updateManyFieldName,
     updateSingleFieldName,
     deleteFieldName,
     deleteSingleFieldName
   };
 };
+var generateUpdateManyInput = (params) => {
+  const { typeName, updatePrefix, updateInput, tableFilters } = params;
+  return new import_graphql4.GraphQLInputObjectType({
+    name: `${capitalize(updatePrefix)}${typeName}ManyInput`,
+    description: `One entry of a batch update of ${typeName}: the rows \`where\` matches get this entry's \`set\` applied.`,
+    fields: {
+      where: {
+        type: tableFilters,
+        description: "Rows this entry updates. An omitted filter updates every row."
+      },
+      set: {
+        type: new import_graphql4.GraphQLNonNull(updateInput)
+      }
+    }
+  });
+};
 var extractRequiredFilters = (table, tableName, where, fieldName, relationCtx) => {
   const filters = where ? extractFilters(table, tableName, where, relationCtx) : void 0;
   if (!filters) {
@@ -3113,6 +3131,48 @@ var generateUpdate = (db, tableName, table, setArgs, filterArgs, fieldName, sing
     args: queryArgs
   };
 };
+var generateUpdateMany = (db, tableName, table, updateManyInput, fieldName, filterCtx) => {
+  const queryArgs = {
+    updates: {
+      type: new import_graphql6.GraphQLNonNull(new import_graphql6.GraphQLList(new import_graphql6.GraphQLNonNull(updateManyInput)))
+    }
+  };
+  return {
+    name: fieldName,
+    resolver: async (_source, args, context, _info) => {
+      try {
+        const { updates } = args;
+        if (!updates.length) {
+          throw new import_graphql6.GraphQLError("No updates were provided!");
+        }
+        const entries = updates.map(({ where, set }) => {
+          const input = remapFromGraphQLSingleInput(set, table);
+          if (!Object.keys(input).length) {
+            throw new import_graphql6.GraphQLError("Unable to update with no values specified!");
+          }
+          return {
+            set: input,
+            filters: where ? extractFilters(table, tableName, where, relationFilterCtx(filterCtx, tableName)) : void 0
+          };
+        });
+        const executor = resolveExecutor(db, context);
+        await executor.transaction(async (tx) => {
+          for (const entry of entries) {
+            let query = tx.update(table).set(entry.set);
+            if (entry.filters) {
+              query = query.where(entry.filters);
+            }
+            await query;
+          }
+        });
+        return { isSuccess: true };
+      } catch (e) {
+        throw toGraphQLError(e);
+      }
+    },
+    args: queryArgs
+  };
+};
 var generateDelete = (db, tableName, table, filterArgs, fieldName, single, requireWhere, filterCtx) => {
   const queryArgs = {
     where: {
@@ -3220,6 +3280,7 @@ var generateSchemaData = (db, schema, relations, options) => {
       upsertArrayFieldName,
       upsertSingleFieldName,
       updateFieldName,
+      updateManyFieldName,
       updateSingleFieldName,
       deleteFieldName,
       deleteSingleFieldName
@@ -3282,6 +3343,15 @@ var generateSchemaData = (db, schema, relations, options) => {
       features.requireWhere,
       filterCtx
     ) : void 0;
+    const updateManyInput = features.update && features.updateMany ? generateUpdateManyInput({ typeName, updatePrefix: prefixes.update, updateInput, tableFilters }) : void 0;
+    const updateManyGenerated = updateManyInput ? generateUpdateMany(
+      db,
+      tableName,
+      schema[tableName],
+      updateManyInput,
+      updateManyFieldName,
+      filterCtx
+    ) : void 0;
     const deleteGenerated = features.delete ? generateDelete(
       db,
       tableName,
@@ -3359,6 +3429,7 @@ var generateSchemaData = (db, schema, relations, options) => {
       upsertArrGenerated,
       upsertSingleGenerated,
       updateGenerated,
+      updateManyGenerated,
       updateSingleGenerated,
       deleteGenerated,
       deleteSingleGenerated
@@ -3376,6 +3447,7 @@ var generateSchemaData = (db, schema, relations, options) => {
       ...features.insert || onConflictInput ? [insertInput] : [],
       ...onConflictInput ? [onConflictInput] : [],
       ...features.update ? [updateInput] : [],
+      ...updateManyInput ? [updateManyInput] : [],
       tableFilters,
       tableOrder
     ];
@@ -3795,6 +3867,78 @@ var generateUpdate2 = (db, tableName, table, tables, relationMap, setArgs, filte
     args: queryArgs
   };
 };
+var generateUpdateMany2 = (db, tableName, table, tables, relationMap, updateManyInput, fieldName, typeName, typeNameMapper, filterCtx) => {
+  const queryArgs = {
+    updates: {
+      type: new import_graphql7.GraphQLNonNull(new import_graphql7.GraphQLList(new import_graphql7.GraphQLNonNull(updateManyInput)))
+    }
+  };
+  const pkNames = pgPrimaryKeyPropNames(table);
+  return {
+    name: fieldName,
+    resolver: async (_source, args, context, info) => {
+      try {
+        const { updates } = args;
+        if (!updates.length) {
+          throw new import_graphql7.GraphQLError("No updates were provided!");
+        }
+        const parsedInfo = (0, import_graphql_parse_resolve_info3.parseResolveInfo)(info, {
+          deep: true
+        });
+        const { columns, hasRelations, withParams } = prepareMutationRelationColumns({
+          relationMap,
+          tables,
+          tableName,
+          typeName,
+          typeNameMapper,
+          table,
+          pkNames,
+          parsedInfo
+        });
+        const entries = updates.map(({ where, set }) => {
+          const input = remapFromGraphQLSingleInput(set, table);
+          if (!Object.keys(input).length) {
+            throw new import_graphql7.GraphQLError("Unable to update with no values specified!");
+          }
+          return {
+            set: input,
+            filters: where ? extractFilters(table, tableName, where, relationFilterCtx(filterCtx, tableName)) : void 0
+          };
+        });
+        const executor = resolveExecutor(db, context);
+        const perEntry = await executor.transaction(async (tx) => {
+          const results = [];
+          for (const entry of entries) {
+            let query = tx.update(table).set(entry.set);
+            if (entry.filters) {
+              query = query.where(entry.filters);
+            }
+            results.push(await query.returning(columns));
+          }
+          return results;
+        });
+        const flatRows = perEntry.flat();
+        const enriched = hasRelations ? await eagerLoadMutationRelations(executor, tableName, flatRows, pkNames, withParams) : flatRows;
+        const output = [];
+        let offset = 0;
+        for (const rows of perEntry) {
+          if (!rows.length) {
+            output.push(null);
+            continue;
+          }
+          for (let i = 0; i < rows.length; i++) {
+            output.push(remapToGraphQLSingleOutput(enriched[offset + i], tableName, table, relationMap));
+          }
+          offset += rows.length;
+        }
+        return output;
+      } catch (e) {
+        throw toGraphQLError(e);
+      }
+    },
+    args: queryArgs
+  };
+};
 var generateDelete2 = (db, tableName, table, filterArgs, fieldName, typeName, single, requireWhere, filterCtx, selectionCtx) => {
   const queryArgs = {
     where: {
@@ -3913,6 +4057,7 @@ function generateSchemaData2(db, schema, relations, options) {
       upsertArrayFieldName,
       upsertSingleFieldName,
       updateFieldName,
+      updateManyFieldName,
       updateSingleFieldName,
       deleteFieldName,
       deleteSingleFieldName
@@ -4034,6 +4179,19 @@ function generateSchemaData2(db, schema, relations, options) {
       typeNameMapper,
       filterCtx
     ) : void 0;
+    const updateManyInput = features.update && features.updateMany ? generateUpdateManyInput({ typeName, updatePrefix: prefixes.update, updateInput, tableFilters }) : void 0;
+    const updateManyGenerated = updateManyInput ? generateUpdateMany2(
+      db,
+      tableName,
+      schema[tableName],
+      tables,
+      eagerRelations,
+      updateManyInput,
+      updateManyFieldName,
+      typeName,
+      typeNameMapper,
+      filterCtx
+    ) : void 0;
     const deleteGenerated = features.delete ? generateDelete2(
       db,
       tableName,
@@ -4144,6 +4302,14 @@ function generateSchemaData2(db, schema, relations, options) {
         resolve: updateGenerated.resolver
       };
     }
+    if (updateManyGenerated) {
+      mutations[updateManyGenerated.name] = {
+        // Nullable items: a no-match entry yields `null` in its slot.
+        type: new import_graphql7.GraphQLNonNull(new import_graphql7.GraphQLList(singleTableItemOutput)),
+        args: updateManyGenerated.args,
+        resolve: updateManyGenerated.resolver
+      };
+    }
     if (updateSingleGenerated) {
       mutations[updateSingleGenerated.name] = {
         type: singleTableItemOutput,
@@ -4170,6 +4336,7 @@ function generateSchemaData2(db, schema, relations, options) {
       ...features.insert || onConflictInput ? [insertInput] : [],
       ...onConflictInput ? [onConflictInput] : [],
       ...features.update ? [updateInput] : [],
+      ...updateManyInput ? [updateManyInput] : [],
       tableFilters,
       tableOrder
     ];
@@ -4492,6 +4659,91 @@ var generateUpdate3 = (db, tableName, table, tables, relationMap, setArgs, filte
     args: queryArgs
   };
 };
+var generateUpdateMany3 = (db, tableName, table, tables, relationMap, updateManyInput, fieldName, typeName, typeNameMapper, filterCtx) => {
+  const queryArgs = {
+    updates: {
+      type: new import_graphql8.GraphQLNonNull(new import_graphql8.GraphQLList(new import_graphql8.GraphQLNonNull(updateManyInput)))
+    }
+  };
+  const pkNames = sqlitePrimaryKeyPropNames(table);
+  return {
+    name: fieldName,
+    resolver: async (_source, args, context, info) => {
+      try {
+        const { updates } = args;
+        if (!updates.length) {
+          throw new import_graphql8.GraphQLError("No updates were provided!");
+        }
+        const parsedInfo = (0, import_graphql_parse_resolve_info4.parseResolveInfo)(info, {
+          deep: true
+        });
+        const { columns, hasRelations, withParams } = prepareMutationRelationColumns({
+          relationMap,
+          tables,
+          tableName,
+          typeName,
+          typeNameMapper,
+          table,
+          pkNames,
+          parsedInfo
+        });
+        const entries = updates.map(({ where, set }) => {
+          const input = remapFromGraphQLSingleInput(set, table);
+          if (!Object.keys(input).length) {
+            throw new import_graphql8.GraphQLError("Unable to update with no values specified!");
+          }
+          return {
+            set: input,
+            filters: where ? extractFilters(table, tableName, where, relationFilterCtx(filterCtx, tableName)) : void 0
+          };
+        });
+        const runEntry = (tx, entry) => {
+          let query = tx.update(table).set(entry.set);
+          if (entry.filters) {
+            query = query.where(entry.filters);
+          }
+          return query.returning(columns).all();
+        };
+        const executor = resolveExecutor(db, context);
+        const perEntry = await executor.transaction((tx) => {
+          const first = runEntry(tx, entries[0]);
+          if (typeof first?.then === "function") {
+            return (async () => {
+              const results2 = [await first];
+              for (let i = 1; i < entries.length; i++) {
+                results2.push(await runEntry(tx, entries[i]));
+              }
+              return results2;
+            })();
+          }
+          const results = [first];
+          for (let i = 1; i < entries.length; i++) {
+            results.push(runEntry(tx, entries[i]));
+          }
+          return results;
+        });
+        const flatRows = perEntry.flat();
+        const enriched = hasRelations ? await eagerLoadMutationRelations(executor, tableName, flatRows, pkNames, withParams) : flatRows;
+        const output = [];
+        let offset = 0;
+        for (const rows of perEntry) {
+          if (!rows.length) {
+            output.push(null);
+            continue;
+          }
+          for (let i = 0; i < rows.length; i++) {
+            output.push(remapToGraphQLSingleOutput(enriched[offset + i], tableName, table, relationMap));
+          }
+          offset += rows.length;
+        }
+        return output;
+      } catch (e) {
+        throw toGraphQLError(e);
+      }
+    },
+    args: queryArgs
+  };
+};
 var generateDelete3 = (db, tableName, table, filterArgs, fieldName, typeName, single, requireWhere, filterCtx, selectionCtx) => {
   const queryArgs = {
     where: {
@@ -4611,6 +4863,7 @@ var generateSchemaData3 = (db, schema, relations, options) => {
       upsertArrayFieldName,
       upsertSingleFieldName,
       updateFieldName,
+      updateManyFieldName,
       updateSingleFieldName,
       deleteFieldName,
       deleteSingleFieldName
@@ -4732,6 +4985,19 @@ var generateSchemaData3 = (db, schema, relations, options) => {
       typeNameMapper,
       filterCtx
     ) : void 0;
+    const updateManyInput = features.update && features.updateMany ? generateUpdateManyInput({ typeName, updatePrefix: prefixes.update, updateInput, tableFilters }) : void 0;
+    const updateManyGenerated = updateManyInput ? generateUpdateMany3(
+      db,
+      tableName,
+      schema[tableName],
+      tables,
+      eagerRelations,
+      updateManyInput,
+      updateManyFieldName,
+      typeName,
+      typeNameMapper,
+      filterCtx
+    ) : void 0;
     const deleteGenerated = features.delete ? generateDelete3(
       db,
       tableName,
@@ -4842,6 +5108,14 @@ var generateSchemaData3 = (db, schema, relations, options) => {
         resolve: updateGenerated.resolver
       };
     }
+    if (updateManyGenerated) {
+      mutations[updateManyGenerated.name] = {
+        // Nullable items: a no-match entry yields `null` in its slot.
+        type: new import_graphql8.GraphQLNonNull(new import_graphql8.GraphQLList(singleTableItemOutput)),
+        args: updateManyGenerated.args,
+        resolve: updateManyGenerated.resolver
+      };
+    }
     if (updateSingleGenerated) {
       mutations[updateSingleGenerated.name] = {
         type: singleTableItemOutput,
@@ -4868,6 +5142,7 @@ var generateSchemaData3 = (db, schema, relations, options) => {
       ...features.insert || onConflictInput ? [insertInput] : [],
       ...onConflictInput ? [onConflictInput] : [],
       ...features.update ? [updateInput] : [],
+      ...updateManyInput ? [updateManyInput] : [],
       tableFilters,
       tableOrder
     ];
@@ -4930,6 +5205,7 @@ var buildSchema = (db, config) => {
     distinct: config?.features?.distinct ?? true,
     insert: config?.features?.insert ?? true,
     update: config?.features?.update ?? true,
+    updateMany: config?.features?.updateMany ?? true,
     delete: config?.features?.delete ?? true,
     upsert: config?.features?.upsert ?? false,
     requireWhere: config?.features?.requireWhere ?? false