drizzle-graphql-plus 0.8.22 → 0.8.24

package/index.cjs

@@ -2414,6 +2414,7 @@ var generateTypeDefs = (tables, relations) => {
       }
       fields.push(`  ${columnName}: ${typeStr}`);
     }
+    fields.push(`  _operation: OPERATION!`);
     const tableRelations = relations[tableName];
     if (tableRelations) {
       for (const [relationName, relationInfo] of Object.entries(
@@ -2423,8 +2424,21 @@ var generateTypeDefs = (tables, relations) => {
         const targetTableName = relationInfo.targetTableName;
         const targetTypeName = capitalize(targetTableName);
         if (isOne) {
+          const relation = relationInfo.relation;
+          const config = relation.config;
+          let isNonNullable = false;
+          if (config?.fields && config.fields.length > 0) {
+            isNonNullable = config.fields.every((field) => {
+              const fieldColumnName = field.name;
+              const column = Object.values(tableInfo.columns).find(
+                (col) => col.name === fieldColumnName
+              );
+              return column && column.notNull;
+            });
+          }
+          const nullableModifier = isNonNullable ? "!" : "";
           fields.push(
-            `  ${relationName}(where: ${targetTypeName}Filters): ${targetTypeName}`
+            `  ${relationName}(where: ${targetTypeName}Filters): ${targetTypeName}${nullableModifier}`
           );
         } else {
           fields.push(
@@ -2497,6 +2511,12 @@ ${orderByFields.join("\n")}
 }`);
   }
   const allDefs = [];
+  allDefs.push(`enum OPERATION {
+  READ
+  INSERTED
+  UPDATED
+  DELETED
+}`);
   if (customScalars.size > 0) {
     for (const scalarName of Array.from(customScalars).sort()) {
       allDefs.push(`scalar ${scalarName}`);
@@ -2806,7 +2826,10 @@ var createQueryResolver = (queryBase, tableInfo, tables, relations) => {
           allFields
         )
       });
-      return result;
+      return result.map((row) => ({
+        ...row,
+        _operation: "READ"
+      }));
     } catch (e) {
       if (typeof e === "object" && e !== null && "message" in e) {
         throw new import_graphql8.GraphQLError(String(e.message));
@@ -2879,6 +2902,7 @@ var buildWhereClause2 = (tableInfo, where) => {
 };
 var generateMutations = (db, tables, relations) => {
   const mutations = {};
+  const deleteResultResolvers = {};
   for (const [tableName, tableInfo] of Object.entries(tables)) {
     const capitalizedName = capitalize(tableName);
     const queryBase = db.query[tableName];
@@ -2915,7 +2939,7 @@ var generateMutations = (db, tables, relations) => {
         const insertedIds = insertedRows.map(
           (row) => row[primaryKeyColumn.name]
         );
-        return queryResolver(
+        const result = await queryResolver(
           parent,
           {
             where: {
@@ -2925,6 +2949,10 @@ var generateMutations = (db, tables, relations) => {
           context,
           info
         );
+        return result.map((row) => ({
+          ...row,
+          _operation: "INSERTED"
+        }));
       } catch (e) {
         if (typeof e === "object" && e !== null && "message" in e) {
           throw new import_graphql9.GraphQLError(String(e.message));
@@ -2948,7 +2976,7 @@ var generateMutations = (db, tables, relations) => {
         const updatedIds = updatedRows.map(
           (row) => row[primaryKeyColumn.name]
         );
-        return queryResolver(
+        const result = await queryResolver(
           parent,
           {
             where: {
@@ -2958,6 +2986,10 @@ var generateMutations = (db, tables, relations) => {
           context,
           info
         );
+        return result.map((row) => ({
+          ...row,
+          _operation: "UPDATED"
+        }));
       } catch (e) {
         if (typeof e === "object" && e !== null && "message" in e) {
           throw new import_graphql9.GraphQLError(String(e.message));
@@ -2973,8 +3005,11 @@ var generateMutations = (db, tables, relations) => {
         if (whereClause) {
           deleteQuery = deleteQuery.where(whereClause);
         }
-        const result = await deleteQuery.returning();
-        return result;
+        const deletedRows = await deleteQuery.returning();
+        return deletedRows.map((row) => ({
+          ...row,
+          _operation: "DELETED"
+        }));
       } catch (e) {
         if (typeof e === "object" && e !== null && "message" in e) {
           throw new import_graphql9.GraphQLError(String(e.message));
@@ -2983,7 +3018,7 @@ var generateMutations = (db, tables, relations) => {
       }
     };
   }
-  return mutations;
+  return { mutations, deleteResultResolvers: {} };
 };
 
 // src/buildSchemaSDL/index.ts
@@ -3006,12 +3041,17 @@ 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, relations);
+  const { mutations, deleteResultResolvers } = generateMutations(
+    db,
+    tables,
+    relations
+  );
   return {
     typeDefs,
     resolvers: {
       Query: queries,
-      Mutation: mutations
+      Mutation: mutations,
+      ...deleteResultResolvers
     }
   };
 };