drizzle-graphql-plus 0.8.33 → 0.8.34

package/index.cjs

@@ -36,9 +36,13 @@ __export(src_exports, {
   cleanupDataLoaderContext: () => cleanupDataLoaderContext,
   createDataLoaderContext: () => createDataLoaderContext,
   createExportMiddleware: () => createExportMiddleware,
+  exportDirectiveTypeDefs: () => exportDirectiveTypeDefs,
   makeScalarAcceptExports: () => makeScalarAcceptExports,
   setCustomGraphQL: () => setCustomGraphQL,
-  setCustomGraphQLTypes: () => setCustomGraphQLTypes
+  setCustomGraphQLTypes: () => setCustomGraphQLTypes,
+  useDataLoaderCleanup: () => useDataLoaderCleanup,
+  useDataLoaderCleanupOnly: () => useDataLoaderCleanupOnly,
+  useDataLoaderContext: () => useDataLoaderContext
 });
 module.exports = __toCommonJS(src_exports);
 
@@ -3611,6 +3615,7 @@ ${orderByFields.join("\n")}
 }`);
   }
   const allDefs = [];
+  customScalars2.add("JSON");
   if (customScalars2.size > 0) {
     for (const scalarName of Array.from(customScalars2).sort()) {
       allDefs.push(`scalar ${scalarName}`);
@@ -4073,6 +4078,18 @@ async function loadRelationsWithDataLoader(mainResults, tableInfo, tables, relat
   if (!tableRelations) {
     return mainResults;
   }
+  for (const mainResult of mainResults) {
+    for (const [relName, { relation }] of Object.entries(tableRelations)) {
+      const relationField = fields[relName];
+      if (!relationField)
+        continue;
+      if ((0, import_drizzle_orm18.is)(relation, import_drizzle_orm18.One)) {
+        mainResult[relName] = null;
+      } else {
+        mainResult[relName] = [];
+      }
+    }
+  }
   const primaryKeyColumn = Object.values(tableInfo.columns).find((col) => col.primary);
   if (!primaryKeyColumn) {
     throw new Error(`No primary key found for table ${tableInfo.name}`);
@@ -4448,20 +4465,11 @@ var generateMutations2 = (db, tables, relations) => {
   return { mutations, deleteResultResolvers };
 };
 
-// src/build-schema-sdl-with-dl/generator/utils/context.ts
-function createDataLoaderContext() {
-  return {
-    relationLoaders: /* @__PURE__ */ new Map()
-  };
-}
-function cleanupDataLoaderContext(context) {
-  for (const loader of Array.from(context.relationLoaders.values())) {
-    loader.clearAll();
-  }
-  context.relationLoaders.clear();
-}
+// src/export-tool/directive-definitions.ts
+var exportDirectiveTypeDefs = `directive @export(as: String!) on FIELD`;
 
 // src/build-schema-sdl-with-dl/index.ts
+var import_schema3 = require("@graphql-tools/schema");
 var buildSchemaSDL2 = (db) => {
   const schema = db._.fullSchema;
   if (!schema) {
@@ -4486,16 +4494,70 @@ var buildSchemaSDL2 = (db) => {
     tables,
     relations
   );
+  const resolvers = {
+    Query: queries,
+    Mutation: mutations,
+    ...deleteResultResolvers
+  };
   return {
     typeDefs,
-    resolvers: {
-      Query: queries,
-      Mutation: mutations,
-      ...deleteResultResolvers
-    }
+    resolvers
   };
 };
 
+// src/build-schema-sdl-with-dl/generator/utils/context.ts
+function createDataLoaderContext() {
+  return {
+    relationLoaders: /* @__PURE__ */ new Map()
+  };
+}
+function cleanupDataLoaderContext(context) {
+  for (const loader of Array.from(context.relationLoaders.values())) {
+    loader.clearAll();
+  }
+  context.relationLoaders.clear();
+}
+
+// src/build-schema-sdl-with-dl/generator/utils/envelop-plugin.ts
+var useDataLoaderCleanup = (options) => ({
+  onContextBuilding: ({ context, extendContext }) => {
+    const dataLoaderContext = createDataLoaderContext();
+    const contextExtension = dataLoaderContext;
+    if (options?.db) {
+      contextExtension.db = options.db;
+    }
+    extendContext(contextExtension);
+  },
+  onExecute: ({ args }) => ({
+    onExecuteDone: () => {
+      const context = args.contextValue;
+      if (context && context.relationLoaders) {
+        cleanupDataLoaderContext(context);
+      }
+    }
+  })
+});
+var useDataLoaderContext = (options) => ({
+  onContextBuilding: ({ context, extendContext }) => {
+    const dataLoaderContext = createDataLoaderContext();
+    const contextExtension = dataLoaderContext;
+    if (options?.db) {
+      contextExtension.db = options.db;
+    }
+    extendContext(contextExtension);
+  }
+});
+var useDataLoaderCleanupOnly = () => ({
+  onExecute: ({ args }) => ({
+    onExecuteDone: () => {
+      const context = args.contextValue;
+      if (context && context.relationLoaders) {
+        cleanupDataLoaderContext(context);
+      }
+    }
+  })
+});
+
 // src/export-tool/ExportStore.ts
 var ExportStore = class {
   store = /* @__PURE__ */ new Map();
@@ -4685,8 +4747,12 @@ function setCustomGraphQLTypes(table, columnTypes) {
   cleanupDataLoaderContext,
   createDataLoaderContext,
   createExportMiddleware,
+  exportDirectiveTypeDefs,
   makeScalarAcceptExports,
   setCustomGraphQL,
-  setCustomGraphQLTypes
+  setCustomGraphQLTypes,
+  useDataLoaderCleanup,
+  useDataLoaderCleanupOnly,
+  useDataLoaderContext
 });
 //# sourceMappingURL=index.cjs.map