@stryke/prisma-trpc-generator 0.3.1 → 0.3.2

package/dist/generator.cjs

@@ -676,8 +676,159 @@ init_cjs_shims();
 
 // src/prisma-generator.ts
 init_cjs_shims();
-var import_node_fs6 = require("node:fs");
-var import_node_path7 = __toESM(require("node:path"), 1);
+
+// ../fs/src/helpers.ts
+init_cjs_shims();
+
+// ../path/src/exists.ts
+init_cjs_shims();
+var import_promises = require("node:fs/promises");
+var exists = /* @__PURE__ */ __name(async (filePath) => {
+  return (0, import_promises.access)(filePath, import_promises.constants.F_OK).then(() => true).catch(() => false);
+}, "exists");
+
+// ../fs/src/helpers.ts
+var import_promises2 = require("node:fs/promises");
+async function createDirectory(path6) {
+  if (await exists(path6)) {
+    return;
+  }
+  return (0, import_promises2.mkdir)(path6, {
+    recursive: true
+  });
+}
+__name(createDirectory, "createDirectory");
+
+// ../path/src/join-paths.ts
+init_cjs_shims();
+var _DRIVE_LETTER_START_RE = /^[A-Z]:\//i;
+function normalizeWindowsPath(input = "") {
+  if (!input) {
+    return input;
+  }
+  return input.replace(/\\/g, "/").replace(_DRIVE_LETTER_START_RE, (r) => r.toUpperCase());
+}
+__name(normalizeWindowsPath, "normalizeWindowsPath");
+var _UNC_REGEX = /^[/\\]{2}/;
+var _IS_ABSOLUTE_RE = /^[/\\](?![/\\])|^[/\\]{2}(?!\.)|^[A-Z]:[/\\]/i;
+var _DRIVE_LETTER_RE = /^[A-Z]:$/i;
+var isAbsolute = /* @__PURE__ */ __name(function(p) {
+  return _IS_ABSOLUTE_RE.test(p);
+}, "isAbsolute");
+var correctPaths = /* @__PURE__ */ __name(function(path6) {
+  if (!path6 || path6.length === 0) {
+    return ".";
+  }
+  path6 = normalizeWindowsPath(path6);
+  const isUNCPath = path6.match(_UNC_REGEX);
+  const isPathAbsolute = isAbsolute(path6);
+  const trailingSeparator = path6[path6.length - 1] === "/";
+  path6 = normalizeString(path6, !isPathAbsolute);
+  if (path6.length === 0) {
+    if (isPathAbsolute) {
+      return "/";
+    }
+    return trailingSeparator ? "./" : ".";
+  }
+  if (trailingSeparator) {
+    path6 += "/";
+  }
+  if (_DRIVE_LETTER_RE.test(path6)) {
+    path6 += "/";
+  }
+  if (isUNCPath) {
+    if (!isPathAbsolute) {
+      return `//./${path6}`;
+    }
+    return `//${path6}`;
+  }
+  return isPathAbsolute && !isAbsolute(path6) ? `/${path6}` : path6;
+}, "correctPaths");
+var joinPaths = /* @__PURE__ */ __name(function(...segments) {
+  let path6 = "";
+  for (const seg of segments) {
+    if (!seg) {
+      continue;
+    }
+    if (path6.length > 0) {
+      const pathTrailing = path6[path6.length - 1] === "/";
+      const segLeading = seg[0] === "/";
+      const both = pathTrailing && segLeading;
+      if (both) {
+        path6 += seg.slice(1);
+      } else {
+        path6 += pathTrailing || segLeading ? seg : `/${seg}`;
+      }
+    } else {
+      path6 += seg;
+    }
+  }
+  return correctPaths(path6);
+}, "joinPaths");
+function normalizeString(path6, allowAboveRoot) {
+  let res = "";
+  let lastSegmentLength = 0;
+  let lastSlash = -1;
+  let dots = 0;
+  let char = null;
+  for (let index = 0; index <= path6.length; ++index) {
+    if (index < path6.length) {
+      char = path6[index];
+    } else if (char === "/") {
+      break;
+    } else {
+      char = "/";
+    }
+    if (char === "/") {
+      if (lastSlash === index - 1 || dots === 1) {
+      } else if (dots === 2) {
+        if (res.length < 2 || lastSegmentLength !== 2 || res[res.length - 1] !== "." || res[res.length - 2] !== ".") {
+          if (res.length > 2) {
+            const lastSlashIndex = res.lastIndexOf("/");
+            if (lastSlashIndex === -1) {
+              res = "";
+              lastSegmentLength = 0;
+            } else {
+              res = res.slice(0, lastSlashIndex);
+              lastSegmentLength = res.length - 1 - res.lastIndexOf("/");
+            }
+            lastSlash = index;
+            dots = 0;
+            continue;
+          } else if (res.length > 0) {
+            res = "";
+            lastSegmentLength = 0;
+            lastSlash = index;
+            dots = 0;
+            continue;
+          }
+        }
+        if (allowAboveRoot) {
+          res += res.length > 0 ? "/.." : "..";
+          lastSegmentLength = 2;
+        }
+      } else {
+        if (res.length > 0) {
+          res += `/${path6.slice(lastSlash + 1, index)}`;
+        } else {
+          res = path6.slice(lastSlash + 1, index);
+        }
+        lastSegmentLength = index - lastSlash - 1;
+      }
+      lastSlash = index;
+      dots = 0;
+    } else if (char === "." && dots !== -1) {
+      ++dots;
+    } else {
+      dots = -1;
+    }
+  }
+  return res;
+}
+__name(normalizeString, "normalizeString");
+
+// src/prisma-generator.ts
+var import_node_path6 = __toESM(require("node:path"), 1);
 var import_pluralize = __toESM(require_pluralize(), 1);
 
 // src/config.ts
@@ -1045,8 +1196,8 @@ function getErrorMap() {
 }
 __name(getErrorMap, "getErrorMap");
 var makeIssue = /* @__PURE__ */ __name((params) => {
-  const { data, path: path7, errorMaps, issueData } = params;
-  const fullPath = [...path7, ...issueData.path || []];
+  const { data, path: path6, errorMaps, issueData } = params;
+  const fullPath = [...path6, ...issueData.path || []];
   const fullIssue = {
     ...issueData,
     path: fullPath
@@ -1180,11 +1331,11 @@ var ParseInputLazyPath = class {
   static {
     __name(this, "ParseInputLazyPath");
   }
-  constructor(parent, value, path7, key) {
+  constructor(parent, value, path6, key) {
     this._cachedPath = [];
     this.parent = parent;
     this.data = value;
-    this._path = path7;
+    this._path = path6;
     this._key = key;
   }
   get path() {
@@ -4956,134 +5107,6 @@ init_cjs_shims();
 // ../env/src/get-env-paths.ts
 init_cjs_shims();
 
-// ../path/src/join-paths.ts
-init_cjs_shims();
-var _DRIVE_LETTER_START_RE = /^[A-Z]:\//i;
-function normalizeWindowsPath(input = "") {
-  if (!input) {
-    return input;
-  }
-  return input.replace(/\\/g, "/").replace(_DRIVE_LETTER_START_RE, (r) => r.toUpperCase());
-}
-__name(normalizeWindowsPath, "normalizeWindowsPath");
-var _UNC_REGEX = /^[/\\]{2}/;
-var _IS_ABSOLUTE_RE = /^[/\\](?![/\\])|^[/\\]{2}(?!\.)|^[A-Z]:[/\\]/i;
-var _DRIVE_LETTER_RE = /^[A-Z]:$/i;
-var isAbsolute = /* @__PURE__ */ __name(function(p) {
-  return _IS_ABSOLUTE_RE.test(p);
-}, "isAbsolute");
-var correctPaths = /* @__PURE__ */ __name(function(path7) {
-  if (!path7 || path7.length === 0) {
-    return ".";
-  }
-  path7 = normalizeWindowsPath(path7);
-  const isUNCPath = path7.match(_UNC_REGEX);
-  const isPathAbsolute = isAbsolute(path7);
-  const trailingSeparator = path7[path7.length - 1] === "/";
-  path7 = normalizeString(path7, !isPathAbsolute);
-  if (path7.length === 0) {
-    if (isPathAbsolute) {
-      return "/";
-    }
-    return trailingSeparator ? "./" : ".";
-  }
-  if (trailingSeparator) {
-    path7 += "/";
-  }
-  if (_DRIVE_LETTER_RE.test(path7)) {
-    path7 += "/";
-  }
-  if (isUNCPath) {
-    if (!isPathAbsolute) {
-      return `//./${path7}`;
-    }
-    return `//${path7}`;
-  }
-  return isPathAbsolute && !isAbsolute(path7) ? `/${path7}` : path7;
-}, "correctPaths");
-var joinPaths = /* @__PURE__ */ __name(function(...segments) {
-  let path7 = "";
-  for (const seg of segments) {
-    if (!seg) {
-      continue;
-    }
-    if (path7.length > 0) {
-      const pathTrailing = path7[path7.length - 1] === "/";
-      const segLeading = seg[0] === "/";
-      const both = pathTrailing && segLeading;
-      if (both) {
-        path7 += seg.slice(1);
-      } else {
-        path7 += pathTrailing || segLeading ? seg : `/${seg}`;
-      }
-    } else {
-      path7 += seg;
-    }
-  }
-  return correctPaths(path7);
-}, "joinPaths");
-function normalizeString(path7, allowAboveRoot) {
-  let res = "";
-  let lastSegmentLength = 0;
-  let lastSlash = -1;
-  let dots = 0;
-  let char = null;
-  for (let index = 0; index <= path7.length; ++index) {
-    if (index < path7.length) {
-      char = path7[index];
-    } else if (char === "/") {
-      break;
-    } else {
-      char = "/";
-    }
-    if (char === "/") {
-      if (lastSlash === index - 1 || dots === 1) {
-      } else if (dots === 2) {
-        if (res.length < 2 || lastSegmentLength !== 2 || res[res.length - 1] !== "." || res[res.length - 2] !== ".") {
-          if (res.length > 2) {
-            const lastSlashIndex = res.lastIndexOf("/");
-            if (lastSlashIndex === -1) {
-              res = "";
-              lastSegmentLength = 0;
-            } else {
-              res = res.slice(0, lastSlashIndex);
-              lastSegmentLength = res.length - 1 - res.lastIndexOf("/");
-            }
-            lastSlash = index;
-            dots = 0;
-            continue;
-          } else if (res.length > 0) {
-            res = "";
-            lastSegmentLength = 0;
-            lastSlash = index;
-            dots = 0;
-            continue;
-          }
-        }
-        if (allowAboveRoot) {
-          res += res.length > 0 ? "/.." : "..";
-          lastSegmentLength = 2;
-        }
-      } else {
-        if (res.length > 0) {
-          res += `/${path7.slice(lastSlash + 1, index)}`;
-        } else {
-          res = path7.slice(lastSlash + 1, index);
-        }
-        lastSegmentLength = index - lastSlash - 1;
-      }
-      lastSlash = index;
-      dots = 0;
-    } else if (char === "." && dots !== -1) {
-      ++dots;
-    } else {
-      dots = -1;
-    }
-  }
-  return res;
-}
-__name(normalizeString, "normalizeString");
-
 // ../string-format/src/title-case.ts
 init_cjs_shims();
 
@@ -5297,34 +5320,34 @@ __name2(normalizeWindowsPath2, "normalizeWindowsPath");
 var _UNC_REGEX2 = /^[/\\]{2}/;
 var _IS_ABSOLUTE_RE2 = /^[/\\](?![/\\])|^[/\\]{2}(?!\.)|^[A-Za-z]:[/\\]/;
 var _DRIVE_LETTER_RE2 = /^[A-Za-z]:$/;
-var correctPaths2 = /* @__PURE__ */ __name2(function(path7) {
-  if (!path7 || path7.length === 0) {
+var correctPaths2 = /* @__PURE__ */ __name2(function(path6) {
+  if (!path6 || path6.length === 0) {
     return ".";
   }
-  path7 = normalizeWindowsPath2(path7);
-  const isUNCPath = path7.match(_UNC_REGEX2);
-  const isPathAbsolute = isAbsolute2(path7);
-  const trailingSeparator = path7[path7.length - 1] === "/";
-  path7 = normalizeString2(path7, !isPathAbsolute);
-  if (path7.length === 0) {
+  path6 = normalizeWindowsPath2(path6);
+  const isUNCPath = path6.match(_UNC_REGEX2);
+  const isPathAbsolute = isAbsolute2(path6);
+  const trailingSeparator = path6[path6.length - 1] === "/";
+  path6 = normalizeString2(path6, !isPathAbsolute);
+  if (path6.length === 0) {
     if (isPathAbsolute) {
       return "/";
     }
     return trailingSeparator ? "./" : ".";
   }
   if (trailingSeparator) {
-    path7 += "/";
+    path6 += "/";
   }
-  if (_DRIVE_LETTER_RE2.test(path7)) {
-    path7 += "/";
+  if (_DRIVE_LETTER_RE2.test(path6)) {
+    path6 += "/";
   }
   if (isUNCPath) {
     if (!isPathAbsolute) {
-      return `//./${path7}`;
+      return `//./${path6}`;
     }
-    return `//${path7}`;
+    return `//${path6}`;
   }
-  return isPathAbsolute && !isAbsolute2(path7) ? `/${path7}` : path7;
+  return isPathAbsolute && !isAbsolute2(path6) ? `/${path6}` : path6;
 }, "correctPaths");
 function cwd() {
   if (typeof process !== "undefined" && typeof process.cwd === "function") {
@@ -5334,15 +5357,15 @@ function cwd() {
 }
 __name(cwd, "cwd");
 __name2(cwd, "cwd");
-function normalizeString2(path7, allowAboveRoot) {
+function normalizeString2(path6, allowAboveRoot) {
   let res = "";
   let lastSegmentLength = 0;
   let lastSlash = -1;
   let dots = 0;
   let char = null;
-  for (let index = 0; index <= path7.length; ++index) {
-    if (index < path7.length) {
-      char = path7[index];
+  for (let index = 0; index <= path6.length; ++index) {
+    if (index < path6.length) {
+      char = path6[index];
     } else if (char === "/") {
       break;
     } else {
@@ -5378,9 +5401,9 @@ function normalizeString2(path7, allowAboveRoot) {
         }
       } else {
         if (res.length > 0) {
-          res += `/${path7.slice(lastSlash + 1, index)}`;
+          res += `/${path6.slice(lastSlash + 1, index)}`;
         } else {
-          res = path7.slice(lastSlash + 1, index);
+          res = path6.slice(lastSlash + 1, index);
         }
         lastSegmentLength = index - lastSlash - 1;
       }
@@ -5479,20 +5502,20 @@ init_cjs_shims();
 // ../path/src/is-file.ts
 init_cjs_shims();
 var import_node_fs2 = require("node:fs");
-function isFile(path7, additionalPath) {
-  return Boolean((0, import_node_fs2.statSync)(additionalPath ? joinPaths(additionalPath, path7) : path7, {
+function isFile(path6, additionalPath) {
+  return Boolean((0, import_node_fs2.statSync)(additionalPath ? joinPaths(additionalPath, path6) : path6, {
     throwIfNoEntry: false
   })?.isFile());
 }
 __name(isFile, "isFile");
-function isDirectory(path7, additionalPath) {
-  return Boolean((0, import_node_fs2.statSync)(additionalPath ? joinPaths(additionalPath, path7) : path7, {
+function isDirectory(path6, additionalPath) {
+  return Boolean((0, import_node_fs2.statSync)(additionalPath ? joinPaths(additionalPath, path6) : path6, {
     throwIfNoEntry: false
   })?.isDirectory());
 }
 __name(isDirectory, "isDirectory");
-function isAbsolutePath(path7) {
-  return !/^[/\\](?![/\\])|^[/\\]{2}(?!\.)|^[A-Z]:[/\\]/i.test(path7);
+function isAbsolutePath(path6) {
+  return !/^[/\\](?![/\\])|^[/\\]{2}(?!\.)|^[A-Z]:[/\\]/i.test(path6);
 }
 __name(isAbsolutePath, "isAbsolutePath");
 
@@ -5505,15 +5528,15 @@ function normalizeWindowsPath3(input = "") {
   return input.replace(/\\/g, "/").replace(_DRIVE_LETTER_START_RE3, (r) => r.toUpperCase());
 }
 __name(normalizeWindowsPath3, "normalizeWindowsPath");
-function normalizeString3(path7, allowAboveRoot) {
+function normalizeString3(path6, allowAboveRoot) {
   let res = "";
   let lastSegmentLength = 0;
   let lastSlash = -1;
   let dots = 0;
   let char = null;
-  for (let index = 0; index <= path7.length; ++index) {
-    if (index < path7.length) {
-      char = path7[index];
+  for (let index = 0; index <= path6.length; ++index) {
+    if (index < path6.length) {
+      char = path6[index];
     } else if (char === "/") {
       break;
     } else {
@@ -5549,9 +5572,9 @@ function normalizeString3(path7, allowAboveRoot) {
         }
       } else {
         if (res.length > 0) {
-          res += `/${path7.slice(lastSlash + 1, index)}`;
+          res += `/${path6.slice(lastSlash + 1, index)}`;
         } else {
-          res = path7.slice(lastSlash + 1, index);
+          res = path6.slice(lastSlash + 1, index);
         }
         lastSegmentLength = index - lastSlash - 1;
       }
@@ -5568,17 +5591,17 @@ function normalizeString3(path7, allowAboveRoot) {
 __name(normalizeString3, "normalizeString");
 
 // ../path/src/file-path-fns.ts
-function resolvePath(path7, cwd2 = getWorkspaceRoot()) {
-  const paths = normalizeWindowsPath3(path7).split("/");
+function resolvePath(path6, cwd2 = getWorkspaceRoot()) {
+  const paths = normalizeWindowsPath3(path6).split("/");
   let resolvedPath = "";
   let resolvedAbsolute = false;
   for (let index = paths.length - 1; index >= -1 && !resolvedAbsolute; index--) {
-    const path8 = index >= 0 ? paths[index] : cwd2;
-    if (!path8 || path8.length === 0) {
+    const path7 = index >= 0 ? paths[index] : cwd2;
+    if (!path7 || path7.length === 0) {
       continue;
     }
-    resolvedPath = joinPaths(path8, resolvedPath);
-    resolvedAbsolute = isAbsolutePath(path8);
+    resolvedPath = joinPaths(path7, resolvedPath);
+    resolvedAbsolute = isAbsolutePath(path7);
   }
   resolvedPath = normalizeString3(resolvedPath, !resolvedAbsolute);
   if (resolvedAbsolute && !isAbsolutePath(resolvedPath)) {
@@ -5588,13 +5611,13 @@ function resolvePath(path7, cwd2 = getWorkspaceRoot()) {
 }
 __name(resolvePath, "resolvePath");
 function resolvePaths(...paths) {
-  return resolvePath(joinPaths(...paths.map((path7) => normalizeWindowsPath3(path7))));
+  return resolvePath(joinPaths(...paths.map((path6) => normalizeWindowsPath3(path6))));
 }
 __name(resolvePaths, "resolvePaths");
 
 // ../path/src/get-parent-path.ts
-var resolveParentPath = /* @__PURE__ */ __name((path7) => {
-  return resolvePaths(path7, "..");
+var resolveParentPath = /* @__PURE__ */ __name((path6) => {
+  return resolvePaths(path6, "..");
 }, "resolveParentPath");
 var getParentPath = /* @__PURE__ */ __name((name, cwd2, options) => {
   const ignoreCase = options?.ignoreCase ?? true;
@@ -6075,10 +6098,21 @@ var constructShield = /* @__PURE__ */ __name(async ({ queries, mutations, subscr
   return shieldText;
 }, "constructShield");
 
-// src/prisma-shield-generator.ts
+// src/project.ts
 init_cjs_shims();
-var import_node_fs5 = require("node:fs");
-var import_node_path6 = __toESM(require("node:path"), 1);
+var import_ts_morph = require("ts-morph");
+var compilerOptions = {
+  target: import_ts_morph.ScriptTarget.ESNext,
+  module: import_ts_morph.ModuleKind.ESNext,
+  emitDecoratorMetadata: true,
+  experimentalDecorators: true,
+  esModuleInterop: true
+};
+var project = new import_ts_morph.Project({
+  compilerOptions: {
+    ...compilerOptions
+  }
+});
 
 // src/utils/remove-dir.ts
 init_cjs_shims();
@@ -6136,80 +6170,9 @@ var writeFileSafely = /* @__PURE__ */ __name(async (writeLocation, content) => {
   import_node_fs4.default.writeFileSync(writeLocation, await formatFile(content));
 }, "writeFileSafely");
 
-// src/prisma-shield-generator.ts
-async function generateShield(options) {
-  const internals = await getPrismaInternals();
-  const outputDir = internals.parseEnvValue(options.generator.output);
-  const results = configSchema.safeParse(options.generator.config);
-  if (!results.success) throw new Error("Invalid options passed");
-  const config = results.data;
-  await import_node_fs5.promises.mkdir(outputDir, {
-    recursive: true
-  });
-  await removeDir(outputDir, true);
-  const prismaClientProvider = options.otherGenerators.find((it) => internals.parseEnvValue(it.provider) === "prisma-client-js");
-  const prismaClientDmmf = await internals.getDMMF({
-    datamodel: options.datamodel,
-    previewFeatures: prismaClientProvider?.previewFeatures
-  });
-  const queries = [];
-  const mutations = [];
-  const subscriptions = [];
-  prismaClientDmmf.mappings.modelOperations.forEach((modelOperation) => {
-    const { model: _model, plural: _plural, ...operations } = modelOperation;
-    for (const [opType, opNameWithModel] of Object.entries(operations)) {
-      if ([
-        "findUnique",
-        "findFirst",
-        "findMany",
-        "aggregate",
-        "groupBy"
-      ].includes(opType)) {
-        queries.push(opNameWithModel);
-      }
-      if ([
-        "createOne",
-        "deleteOne",
-        "updateOne",
-        "deleteMany",
-        "updateMany",
-        "upsertOne"
-      ].includes(opType)) {
-        mutations.push(opNameWithModel);
-      }
-    }
-  });
-  queries.sort();
-  mutations.sort();
-  subscriptions.sort();
-  const shieldText = constructShield({
-    queries,
-    mutations,
-    subscriptions
-  }, config, options);
-  await writeFileSafely(import_node_path6.default.join(outputDir, "shield.ts"), shieldText);
-}
-__name(generateShield, "generateShield");
-
-// src/project.ts
-init_cjs_shims();
-var import_ts_morph = require("ts-morph");
-var compilerOptions = {
-  target: import_ts_morph.ScriptTarget.ESNext,
-  module: import_ts_morph.ModuleKind.ESNext,
-  emitDecoratorMetadata: true,
-  experimentalDecorators: true,
-  esModuleInterop: true
-};
-var project = new import_ts_morph.Project({
-  compilerOptions: {
-    ...compilerOptions
-  }
-});
-
 // src/prisma-generator.ts
 async function generate(options) {
-  console.log("[STORM]: Running the Storm Software - Prisma tRPC generator with options: \n", JSON.stringify(options, null, 2));
+  console.log("[STORM]: Running the Storm Software - Prisma tRPC generator");
   const internals = await getPrismaInternals();
   console.log(`[STORM]: Validating configuration options`);
   const outputDir = internals.parseEnvValue(options.generator.output);
@@ -6220,13 +6183,12 @@ async function generate(options) {
   const config = results.data;
   const consoleLog = /* @__PURE__ */ __name((message) => {
     if (config.debug) {
-      console.log(`[STORM]: ${message}`);
+      console.log(`[STORM]: ${message} 
+`);
     }
   }, "consoleLog");
   consoleLog(`Preparing output directory: ${outputDir}`);
-  await import_node_fs6.promises.mkdir(outputDir, {
-    recursive: true
-  });
+  await createDirectory(outputDir);
   await removeDir(outputDir, true);
   if (config.withZod !== false) {
     consoleLog("Generating Zod schemas");
@@ -6235,17 +6197,68 @@ async function generate(options) {
   } else {
     consoleLog("Skipping Zod schemas generation");
   }
+  consoleLog("Finding Prisma Client generator");
+  const prismaClientProvider = options.otherGenerators.find((it) => internals.parseEnvValue(it.provider) === "prisma-client-js");
+  if (!prismaClientProvider) {
+    throw new Error("No Prisma Client generator found. Please add `prisma-client-js` to your generator list.");
+  }
+  consoleLog("Generating Prisma Client DMMF");
+  const prismaClientDmmf = await internals.getDMMF({
+    datamodel: options.datamodel,
+    previewFeatures: prismaClientProvider?.previewFeatures
+  });
+  const modelOperations = prismaClientDmmf.mappings.modelOperations;
+  const models = prismaClientDmmf.datamodel.models;
+  const hiddenModels = [];
   if (config.withShield !== false) {
     consoleLog("Generating tRPC Shield");
-    const shieldOutputPath = import_node_path7.default.join(outputDir, "./shield");
-    await generateShield({
+    const shieldOutputDir = joinPaths(outputDir, "shield");
+    consoleLog("Preparing tRPC Shield output directory");
+    await createDirectory(shieldOutputDir);
+    await removeDir(shieldOutputDir, true);
+    const queries = [];
+    const mutations = [];
+    const subscriptions = [];
+    prismaClientDmmf.mappings.modelOperations.forEach((modelOperation) => {
+      const { model: _model, plural: _plural, ...operations } = modelOperation;
+      for (const [opType, opNameWithModel] of Object.entries(operations)) {
+        if ([
+          "findUnique",
+          "findFirst",
+          "findMany",
+          "aggregate",
+          "groupBy"
+        ].includes(opType)) {
+          queries.push(opNameWithModel);
+        }
+        if ([
+          "createOne",
+          "deleteOne",
+          "updateOne",
+          "deleteMany",
+          "updateMany",
+          "upsertOne"
+        ].includes(opType)) {
+          mutations.push(opNameWithModel);
+        }
+      }
+    });
+    queries.sort();
+    mutations.sort();
+    subscriptions.sort();
+    consoleLog("Constructing tRPC Shield source file");
+    const shieldText = constructShield({
+      queries,
+      mutations,
+      subscriptions
+    }, config, {
       ...options,
       generator: {
         ...options.generator,
         output: {
           fromEnvVar: null,
           ...options.generator.output,
-          value: shieldOutputPath
+          value: shieldOutputDir
         },
         config: {
           ...options.generator.config,
@@ -6253,25 +6266,14 @@ async function generate(options) {
         }
       }
     });
+    consoleLog("Saving tRPC Shield source file to disk");
+    await writeFileSafely(joinPaths(shieldOutputDir, "shield.ts"), shieldText);
   } else {
     consoleLog("Skipping tRPC Shield generation");
   }
-  consoleLog("Finding Prisma Client generator");
-  const prismaClientProvider = options.otherGenerators.find((it) => internals.parseEnvValue(it.provider) === "prisma-client-js");
-  if (!prismaClientProvider) {
-    throw new Error("No Prisma Client generator found. Please add `prisma-client-js` to your generator list.");
-  }
-  consoleLog("Generating Prisma Client DMMF");
-  const prismaClientDmmf = await internals.getDMMF({
-    datamodel: options.datamodel,
-    previewFeatures: prismaClientProvider?.previewFeatures
-  });
-  const modelOperations = prismaClientDmmf.mappings.modelOperations;
-  const models = prismaClientDmmf.datamodel.models;
-  const hiddenModels = [];
   consoleLog(`Generating tRPC source code for ${models.length} models`);
   resolveModelsComments(models, hiddenModels);
-  const createRouter = project.createSourceFile(import_node_path7.default.resolve(outputDir, "routers", "helpers", "createRouter.ts"), void 0, {
+  const createRouter = project.createSourceFile(import_node_path6.default.resolve(outputDir, "routers", "helpers", "createRouter.ts"), void 0, {
     overwrite: true
   });
   consoleLog("Generating tRPC imports");
@@ -6284,7 +6286,7 @@ async function generate(options) {
   createRouter.formatText({
     indentSize: 2
   });
-  const appRouter = project.createSourceFile(import_node_path7.default.resolve(outputDir, "routers", `index.ts`), void 0, {
+  const appRouter = project.createSourceFile(import_node_path6.default.resolve(outputDir, "routers", `index.ts`), void 0, {
     overwrite: true
   });
   consoleLog("Generating tRPC router imports");
@@ -6306,7 +6308,7 @@ async function generate(options) {
     const plural = (0, import_pluralize.default)(model.toLowerCase());
     consoleLog(`Generating tRPC router for model ${model}`);
     generateRouterImport(appRouter, plural, model);
-    const modelRouter = project.createSourceFile(import_node_path7.default.resolve(outputDir, "routers", `${model}.router.ts`), void 0, {
+    const modelRouter = project.createSourceFile(import_node_path6.default.resolve(outputDir, "routers", `${model}.router.ts`), void 0, {
       overwrite: true
     });
     generateCreateRouterImport({