omni-rest 0.2.2 → 0.2.3

package/dist/cli.js

@@ -1532,9 +1532,18 @@ 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 mod = __require(clientPath);
+    const PrismaClient = mod.PrismaClient ?? mod.default?.PrismaClient;
+    if (!PrismaClient) throw new Error("PrismaClient not found in @prisma/client");
+    const runtimeDataModel = mod?.Prisma?.runtimeDataModel ?? mod?.Prisma?._runtimeDataModel;
+    if (runtimeDataModel?.models) {
+      return { _runtimeDataModel: runtimeDataModel, $disconnect: async () => {
+      } };
+    }
     return new PrismaClient();
   } catch {
     throw new Error(
@@ -1557,15 +1566,29 @@ function tryLoadFromSchemaOutput() {
       const indexPath = path__namespace.default.join(outputDir, "index.js");
       if (!fs3__namespace.default.existsSync(indexPath)) continue;
       const mod = __require(indexPath);
+      const runtimeDataModel = mod?.Prisma?.runtimeDataModel ?? mod?.Prisma?._runtimeDataModel;
+      if (runtimeDataModel?.models) {
+        return {
+          _runtimeDataModel: runtimeDataModel,
+          $disconnect: async () => {
+          }
+        };
+      }
       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();
+      } catch {
+        for (const opt of [
+          { datasourceUrl: "postgresql://x:x@localhost/x" },
+          { datasources: { db: { url: "postgresql://x:x@localhost/x" } } }
+        ]) {
+          try {
+            return new PrismaClient(opt);
+          } catch {
+            continue;
+          }
+        }
       }
     } catch {
       continue;
@@ -1573,21 +1596,25 @@ function tryLoadFromSchemaOutput() {
   }
   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/default.js")
+  ];
+  for (const clientPath of candidates) {
+    if (!fs3__namespace.default.existsSync(clientPath)) continue;
+    try {
+      const mod = __require(clientPath);
+      const runtimeDataModel = mod?.Prisma?.runtimeDataModel ?? mod?.Prisma?._runtimeDataModel;
+      if (runtimeDataModel?.models) {
+        return { _runtimeDataModel: runtimeDataModel, $disconnect: async () => {
+        } };
+      }
+    } catch {
+      continue;
+    }
   }
+  return null;
 }
 var USAGE = `
   Usage: