omni-rest 0.2.2 → 0.2.4

package/dist/cli.js

@@ -1532,9 +1532,15 @@ function write(filePath, content) {
 function createPrismaClient() {
   const customClient = tryLoadFromSchemaOutput();
   if (customClient) return customClient;
+  const standardPrismaClient = tryLoadFromStandardOutput();
+  if (standardPrismaClient) return standardPrismaClient;
   try {
     const clientPath = __require.resolve("@prisma/client", { paths: [cwd] });
-    const { PrismaClient } = __require(clientPath);
+    const textResult = extractRuntimeDataModelFromFile(clientPath);
+    if (textResult) return textResult;
+    const mod = __require(clientPath);
+    const PrismaClient = mod.PrismaClient ?? mod.default?.PrismaClient;
+    if (!PrismaClient) throw new Error("PrismaClient not found in @prisma/client");
     return new PrismaClient();
   } catch {
     throw new Error(
@@ -1555,39 +1561,52 @@ function tryLoadFromSchemaOutput() {
       if (!match) continue;
       const outputDir = path__namespace.default.resolve(path__namespace.default.dirname(schemaPath), match[1]);
       const indexPath = path__namespace.default.join(outputDir, "index.js");
-      if (!fs3__namespace.default.existsSync(indexPath)) continue;
-      const mod = __require(indexPath);
-      const PrismaClient = mod.PrismaClient ?? mod.default?.PrismaClient;
-      if (!PrismaClient) continue;
-      const providerMatch = schema.match(/provider\s*=\s*["']([^"']+)["']/);
-      const provider = providerMatch?.[1] ?? "postgresql";
-      const dummyUrl = getDummyUrl(provider);
-      try {
-        return new PrismaClient({ datasourceUrl: dummyUrl });
-      } catch {
-        return new PrismaClient();
-      }
+      const result = extractRuntimeDataModelFromFile(indexPath);
+      if (result) return result;
     } catch {
       continue;
     }
   }
   return null;
 }
-function getDummyUrl(provider) {
-  switch (provider) {
-    case "mysql":
-      return "mysql://root:root@localhost:3306/dummy";
-    case "sqlite":
-      return "file:./dummy.db";
-    case "sqlserver":
-      return "sqlserver://localhost:1433;database=dummy;user=sa;password=dummy";
-    case "mongodb":
-      return "mongodb://localhost:27017/dummy";
-    case "cockroachdb":
-      return "postgresql://root@localhost:26257/dummy";
-    default:
-      return "postgresql://postgres:postgres@localhost:5432/dummy";
+function tryLoadFromStandardOutput() {
+  const candidates = [
+    path__namespace.default.resolve(cwd, "node_modules/.prisma/client/index.js"),
+    path__namespace.default.resolve(cwd, "node_modules/@prisma/client/index.js")
+  ];
+  for (const clientPath of candidates) {
+    const result = extractRuntimeDataModelFromFile(clientPath);
+    if (result) return result;
+  }
+  return null;
+}
+function extractRuntimeDataModelFromFile(filePath) {
+  if (!fs3__namespace.default.existsSync(filePath)) return null;
+  try {
+    const src = fs3__namespace.default.readFileSync(filePath, "utf-8");
+    const match = src.match(/config\.runtimeDataModel\s*=\s*JSON\.parse\("((?:[^"\\]|\\.)*)"\)/);
+    if (match) {
+      const json = match[1].replace(/\\"/g, '"').replace(/\\\\/g, "\\");
+      const runtimeDataModel = JSON.parse(json);
+      if (runtimeDataModel?.models) {
+        return { _runtimeDataModel: runtimeDataModel, $disconnect: async () => {
+        } };
+      }
+    }
+    const match2 = src.match(/runtimeDataModel\s*=\s*(\{[\s\S]*?"models"\s*:\s*\{[\s\S]*?\}\s*\})/);
+    if (match2) {
+      try {
+        const runtimeDataModel = JSON.parse(match2[1]);
+        if (runtimeDataModel?.models) {
+          return { _runtimeDataModel: runtimeDataModel, $disconnect: async () => {
+          } };
+        }
+      } catch {
+      }
+    }
+  } catch {
   }
+  return null;
 }
 var USAGE = `
   Usage: