omni-rest 0.2.0 → 0.2.2

package/dist/cli.mjs

@@ -3,7 +3,6 @@ import * as fs3 from 'fs';
 import fs3__default from 'fs';
 import * as path from 'path';
 import path__default from 'path';
-import { Prisma } from '@prisma/client';
 import * as readline from 'readline';
 
 var __require = /* @__PURE__ */ ((x) => typeof require !== "undefined" ? require : typeof Proxy !== "undefined" ? new Proxy(x, {
@@ -12,6 +11,8 @@ var __require = /* @__PURE__ */ ((x) => typeof require !== "undefined" ? require
   if (typeof require !== "undefined") return require.apply(this, arguments);
   throw Error('Dynamic require of "' + x + '" is not supported');
 });
+
+// src/introspect.ts
 function getModels(prisma) {
   let raw;
   if (prisma?._runtimeDataModel?.models) {
@@ -26,9 +27,13 @@ function getModels(prisma) {
     }));
   }
   if (!raw) {
-    const dmmfModels = Prisma?.dmmf?.datamodel?.models || __require("@prisma/client")?.Prisma?.dmmf?.datamodel?.models;
-    if (dmmfModels) {
-      raw = dmmfModels;
+    try {
+      const prismaModule = __require("@prisma/client");
+      const dmmfModels = prismaModule?.Prisma?.dmmf?.datamodel?.models;
+      if (dmmfModels) {
+        raw = dmmfModels;
+      }
+    } catch {
     }
   }
   if (!raw) {
@@ -1503,9 +1508,11 @@ function write(filePath, content) {
   console.log(COLORS2.green(`  \u2713 Written: ${filePath}`));
 }
 function createPrismaClient() {
+  const customClient = tryLoadFromSchemaOutput();
+  if (customClient) return customClient;
   try {
-    const PrismaClientPath = path__default.resolve(cwd, "node_modules/@prisma/client");
-    const { PrismaClient } = __require(PrismaClientPath);
+    const clientPath = __require.resolve("@prisma/client", { paths: [cwd] });
+    const { PrismaClient } = __require(clientPath);
     return new PrismaClient();
   } catch {
     throw new Error(
@@ -1513,6 +1520,53 @@ function createPrismaClient() {
     );
   }
 }
+function tryLoadFromSchemaOutput() {
+  const schemaPaths = [
+    path__default.resolve(cwd, "prisma/schema.prisma"),
+    path__default.resolve(cwd, "schema.prisma")
+  ];
+  for (const schemaPath of schemaPaths) {
+    if (!fs3__default.existsSync(schemaPath)) continue;
+    try {
+      const schema = fs3__default.readFileSync(schemaPath, "utf-8");
+      const match = schema.match(/output\s*=\s*["']([^"']+)["']/);
+      if (!match) continue;
+      const outputDir = path__default.resolve(path__default.dirname(schemaPath), match[1]);
+      const indexPath = path__default.join(outputDir, "index.js");
+      if (!fs3__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();
+      }
+    } 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";
+  }
+}
 var USAGE = `
   Usage:
     npx omni-rest generate            Generate both Zod schemas and OpenAPI spec