@suss/contract-prisma 0.34.0 → 0.34.1

package/dist/index.cjs

@@ -31,7 +31,8 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 var index_exports = {};
 __export(index_exports, {
   prismaSchemaFileToSummaries: () => prismaSchemaFileToSummaries,
-  prismaSchemaToSummaries: () => prismaSchemaToSummaries
+  prismaSchemaToSummaries: () => prismaSchemaToSummaries,
+  prismaStorageSystem: () => prismaStorageSystem
 });
 module.exports = __toCommonJS(index_exports);
 var import_node_fs = __toESM(require("fs"), 1);
@@ -60,7 +61,7 @@ function prismaSchemaToSummaries(source, options = {}) {
   const list = ast.list;
   const modelNames = /* @__PURE__ */ new Set();
   const enumNames = /* @__PURE__ */ new Set();
-  let storageSystem = null;
+  const storageSystem = storageSystemIn(list);
   for (const node of list) {
     const n = node;
     if (n.type === "model" || n.type === "view") {
@@ -71,12 +72,6 @@ function prismaSchemaToSummaries(source, options = {}) {
       if (typeof n.name === "string") {
         enumNames.add(n.name);
       }
-    } else if (n.type === "datasource") {
-      const ds = node;
-      const provider = readProviderString(ds);
-      if (provider !== null && provider in PROVIDER_TO_SYSTEM) {
-        storageSystem = PROVIDER_TO_SYSTEM[provider];
-      }
     }
   }
   if (storageSystem === null) {
@@ -441,6 +436,25 @@ function arrayNames(value) {
   }
   return out;
 }
+function prismaStorageSystem(source) {
+  try {
+    return storageSystemIn((0, import_prisma_ast.getSchema)(source).list);
+  } catch {
+    return null;
+  }
+}
+function storageSystemIn(list) {
+  for (const node of list) {
+    if (node.type !== "datasource") {
+      continue;
+    }
+    const provider = readProviderString(node);
+    if (provider !== null && provider in PROVIDER_TO_SYSTEM) {
+      return PROVIDER_TO_SYSTEM[provider];
+    }
+  }
+  return null;
+}
 function readProviderString(ds) {
   for (const a of ds.assignments) {
     if (a.key !== "provider") {
@@ -455,5 +469,6 @@ function readProviderString(ds) {
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   prismaSchemaFileToSummaries,
-  prismaSchemaToSummaries
+  prismaSchemaToSummaries,
+  prismaStorageSystem
 });

package/dist/index.d.cts

@@ -29,5 +29,11 @@ declare function prismaSchemaToSummaries(source: string, options?: PrismaSchemaT
  * not exist.
  */
 declare function prismaSchemaFileToSummaries(schemaPath: string, options?: PrismaSchemaToSummariesOptions): BehavioralSummary[];
+type PrismaStorageSystem = "postgresql" | "mysql" | "sqlite";
+/**
+ * The storage system the datasource provider says, or null for MongoDB or
+ * a file with no datasource. The client pack binds its calls with this.
+ */
+declare function prismaStorageSystem(source: string): PrismaStorageSystem | null;
 
-export { type PrismaSchemaToSummariesOptions, prismaSchemaFileToSummaries, prismaSchemaToSummaries };
+export { type PrismaSchemaToSummariesOptions, type PrismaStorageSystem, prismaSchemaFileToSummaries, prismaSchemaToSummaries, prismaStorageSystem };

package/dist/index.d.ts

@@ -29,5 +29,11 @@ declare function prismaSchemaToSummaries(source: string, options?: PrismaSchemaT
  * not exist.
  */
 declare function prismaSchemaFileToSummaries(schemaPath: string, options?: PrismaSchemaToSummariesOptions): BehavioralSummary[];
+type PrismaStorageSystem = "postgresql" | "mysql" | "sqlite";
+/**
+ * The storage system the datasource provider says, or null for MongoDB or
+ * a file with no datasource. The client pack binds its calls with this.
+ */
+declare function prismaStorageSystem(source: string): PrismaStorageSystem | null;
 
-export { type PrismaSchemaToSummariesOptions, prismaSchemaFileToSummaries, prismaSchemaToSummaries };
+export { type PrismaSchemaToSummariesOptions, type PrismaStorageSystem, prismaSchemaFileToSummaries, prismaSchemaToSummaries, prismaStorageSystem };

package/dist/index.js

@@ -25,7 +25,7 @@ function prismaSchemaToSummaries(source, options = {}) {
   const list = ast.list;
   const modelNames = /* @__PURE__ */ new Set();
   const enumNames = /* @__PURE__ */ new Set();
-  let storageSystem = null;
+  const storageSystem = storageSystemIn(list);
   for (const node of list) {
     const n = node;
     if (n.type === "model" || n.type === "view") {
@@ -36,12 +36,6 @@ function prismaSchemaToSummaries(source, options = {}) {
       if (typeof n.name === "string") {
         enumNames.add(n.name);
       }
-    } else if (n.type === "datasource") {
-      const ds = node;
-      const provider = readProviderString(ds);
-      if (provider !== null && provider in PROVIDER_TO_SYSTEM) {
-        storageSystem = PROVIDER_TO_SYSTEM[provider];
-      }
     }
   }
   if (storageSystem === null) {
@@ -406,6 +400,25 @@ function arrayNames(value) {
   }
   return out;
 }
+function prismaStorageSystem(source) {
+  try {
+    return storageSystemIn(getSchema(source).list);
+  } catch {
+    return null;
+  }
+}
+function storageSystemIn(list) {
+  for (const node of list) {
+    if (node.type !== "datasource") {
+      continue;
+    }
+    const provider = readProviderString(node);
+    if (provider !== null && provider in PROVIDER_TO_SYSTEM) {
+      return PROVIDER_TO_SYSTEM[provider];
+    }
+  }
+  return null;
+}
 function readProviderString(ds) {
   for (const a of ds.assignments) {
     if (a.key !== "provider") {
@@ -419,5 +432,6 @@ function readProviderString(ds) {
 }
 export {
   prismaSchemaFileToSummaries,
-  prismaSchemaToSummaries
+  prismaSchemaToSummaries,
+  prismaStorageSystem
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@suss/contract-prisma",
-  "version": "0.34.0",
+  "version": "0.34.1",
   "type": "module",
   "description": "Turn a Prisma schema into a suss contract and check it against the queries your code runs on those tables.",
   "keywords": [
@@ -48,7 +48,7 @@
   },
   "dependencies": {
     "@mrleebo/prisma-ast": "^0.15.0",
-    "@suss/behavioral-ir": "0.34.0"
+    "@suss/behavioral-ir": "0.34.1"
   },
   "devDependencies": {
     "tsup": "*",