drizzle-graphql-plus 0.8.20 → 0.8.21

package/index.js

@@ -2608,7 +2608,7 @@ var generateQueryTypeDefs = (tables) => {
   for (const tableName of Object.keys(tables)) {
     const typeName = capitalize(tableName);
     queryFields.push(
-      `  ${tableName}(where: ${typeName}Filters, orderBy: ${typeName}OrderBy, limit: Int, offset: Int): [${typeName}!]!`
+      `  ${tableName}FindMany(where: ${typeName}Filters, orderBy: ${typeName}OrderBy, limit: Int, offset: Int): [${typeName}!]!`
     );
   }
   return `type Query {
@@ -2620,13 +2620,13 @@ var generateMutationTypeDefs = (tables) => {
   for (const tableName of Object.keys(tables)) {
     const typeName = capitalize(tableName);
     mutationFields.push(
-      `  insert${typeName}(values: [${typeName}InsertInput!]!): [${typeName}!]!`
+      `  ${tableName}InsertMany(values: [${typeName}InsertInput!]!): [${typeName}!]!`
     );
     mutationFields.push(
-      `  update${typeName}(where: ${typeName}Filters, set: ${typeName}UpdateInput!): [${typeName}!]!`
+      `  ${tableName}UpdateMany(where: ${typeName}Filters, set: ${typeName}UpdateInput!): [${typeName}!]!`
     );
     mutationFields.push(
-      `  delete${typeName}(where: ${typeName}Filters): [${typeName}!]!`
+      `  ${tableName}DeleteMany(where: ${typeName}Filters): [${typeName}!]!`
     );
   }
   return `type Mutation {
@@ -2854,6 +2854,41 @@ var extractRelationsParams2 = (relationMap, tables, tableName, fields) => {
   }
   return Object.keys(args).length > 0 ? args : void 0;
 };
+var createQueryResolver = (queryBase, tableInfo, tables, relations) => {
+  return async (parent, args, context, info) => {
+    try {
+      const { where, orderBy, limit, offset } = args;
+      const parsedInfo = parseResolveInfo4(info, {
+        deep: true
+      });
+      const allFields = {};
+      if (parsedInfo.fieldsByTypeName) {
+        for (const fields of Object.values(parsedInfo.fieldsByTypeName)) {
+          Object.assign(allFields, fields);
+        }
+      }
+      const result = await queryBase.findMany({
+        columns: extractSelectedColumns(allFields, tableInfo),
+        offset,
+        limit,
+        orderBy: buildOrderByClause(tableInfo, orderBy),
+        where: buildWhereClause(tableInfo, where),
+        with: extractRelationsParams2(
+          relations,
+          tables,
+          tableInfo.name,
+          allFields
+        )
+      });
+      return result;
+    } catch (e) {
+      if (typeof e === "object" && e !== null && "message" in e) {
+        throw new GraphQLError6(String(e.message));
+      }
+      throw e;
+    }
+  };
+};
 var generateQueries = (db, tables, relations) => {
   const queries = {};
   for (const [tableName, tableInfo] of Object.entries(tables)) {
@@ -2863,34 +2898,12 @@ var generateQueries = (db, tables, relations) => {
         `Drizzle-GraphQL Error: Table ${tableName} not found in drizzle instance. Did you forget to pass schema to drizzle constructor?`
       );
     }
-    queries[tableName] = async (parent, args, context, info) => {
-      try {
-        const { where, orderBy, limit, offset } = args;
-        const parsedInfo = parseResolveInfo4(info, {
-          deep: true
-        });
-        const allFields = {};
-        if (parsedInfo.fieldsByTypeName) {
-          for (const fields of Object.values(parsedInfo.fieldsByTypeName)) {
-            Object.assign(allFields, fields);
-          }
-        }
-        const result = await queryBase.findMany({
-          columns: extractSelectedColumns(allFields, tableInfo),
-          offset,
-          limit,
-          orderBy: buildOrderByClause(tableInfo, orderBy),
-          where: buildWhereClause(tableInfo, where),
-          with: extractRelationsParams2(relations, tables, tableName, allFields)
-        });
-        return result;
-      } catch (e) {
-        if (typeof e === "object" && e !== null && "message" in e) {
-          throw new GraphQLError6(String(e.message));
-        }
-        throw e;
-      }
-    };
+    queries[`${tableName}FindMany`] = createQueryResolver(
+      queryBase,
+      tableInfo,
+      tables,
+      relations
+    );
   }
   return queries;
 };
@@ -2938,11 +2951,31 @@ var buildWhereClause2 = (tableInfo, where) => {
     return conditions[0];
   return and3(...conditions);
 };
-var generateMutations = (db, tables) => {
+var generateMutations = (db, tables, relations) => {
   const mutations = {};
   for (const [tableName, tableInfo] of Object.entries(tables)) {
     const capitalizedName = capitalize(tableName);
-    mutations[`insert${capitalizedName}`] = async (parent, args, context, info) => {
+    const queryBase = db.query[tableName];
+    if (!queryBase) {
+      throw new Error(
+        `Drizzle-GraphQL Error: Table ${tableName} not found in drizzle instance. Did you forget to pass schema to drizzle constructor?`
+      );
+    }
+    const primaryKeyColumn = Object.values(tableInfo.columns).find(
+      (col) => col.primary || col.name === "id"
+    );
+    if (!primaryKeyColumn) {
+      throw new Error(
+        `Drizzle-GraphQL Error: Table ${tableName} does not have a primary key column`
+      );
+    }
+    const queryResolver = createQueryResolver(
+      queryBase,
+      tableInfo,
+      tables,
+      relations
+    );
+    mutations[`${tableName}InsertMany`] = async (parent, args, context, info) => {
       try {
         const { values } = args;
         if (!values || values.length === 0) {
@@ -2952,8 +2985,20 @@ var generateMutations = (db, tables) => {
           values,
           tableInfo.table
         );
-        const result = await db.insert(tableInfo.table).values(remappedValues).returning();
-        return result;
+        const insertedRows = await db.insert(tableInfo.table).values(remappedValues).returning();
+        const insertedIds = insertedRows.map(
+          (row) => row[primaryKeyColumn.name]
+        );
+        return queryResolver(
+          parent,
+          {
+            where: {
+              [primaryKeyColumn.name]: { inArray: insertedIds }
+            }
+          },
+          context,
+          info
+        );
       } catch (e) {
         if (typeof e === "object" && e !== null && "message" in e) {
           throw new GraphQLError7(String(e.message));
@@ -2961,7 +3006,7 @@ var generateMutations = (db, tables) => {
         throw e;
       }
     };
-    mutations[`update${capitalizedName}`] = async (parent, args, context, info) => {
+    mutations[`${tableName}UpdateMany`] = async (parent, args, context, info) => {
       try {
         const { where, set } = args;
         if (!set || Object.keys(set).length === 0) {
@@ -2973,8 +3018,20 @@ var generateMutations = (db, tables) => {
         if (whereClause) {
           query = query.where(whereClause);
         }
-        const result = await query.returning();
-        return result;
+        const updatedRows = await query.returning();
+        const updatedIds = updatedRows.map(
+          (row) => row[primaryKeyColumn.name]
+        );
+        return queryResolver(
+          parent,
+          {
+            where: {
+              [primaryKeyColumn.name]: { inArray: updatedIds }
+            }
+          },
+          context,
+          info
+        );
       } catch (e) {
         if (typeof e === "object" && e !== null && "message" in e) {
           throw new GraphQLError7(String(e.message));
@@ -2982,7 +3039,7 @@ var generateMutations = (db, tables) => {
         throw e;
       }
     };
-    mutations[`delete${capitalizedName}`] = async (parent, args, context, info) => {
+    mutations[`${tableName}DeleteMany`] = async (parent, args, context, info) => {
       try {
         const { where } = args;
         const whereClause = buildWhereClause2(tableInfo, where);
@@ -3023,7 +3080,7 @@ var buildSchemaSDL = (db, config) => {
   typeDefsArray.push(generateMutationTypeDefs(tables));
   const typeDefs = typeDefsArray.join("\n\n");
   const queries = generateQueries(db, tables, relations);
-  const mutations = generateMutations(db, tables);
+  const mutations = generateMutations(db, tables, relations);
   return {
     typeDefs,
     resolvers: {