@stryke/prisma-trpc-generator 0.4.8 → 0.5.0

package/dist/generator.cjs

@@ -2631,6 +2631,12 @@ function normalizeString(path6, allowAboveRoot) {
 }
 __name(normalizeString, "normalizeString");
 
+// ../string-format/src/lower-case-first.ts
+init_cjs_shims();
+var lowerCaseFirst = /* @__PURE__ */ __name((input) => {
+  return input ? input.charAt(0).toLowerCase() + input.slice(1) : input;
+}, "lowerCaseFirst");
+
 // src/prisma-generator.ts
 var import_node_path6 = __toESM(require("node:path"), 1);
 var import_pluralize = __toESM(require_pluralize(), 1);
@@ -6886,8 +6892,9 @@ var configSchema = z.object({
   withShield: configShield.default("true"),
   withZod: configBoolean.default("true"),
   contextPath: z.string().default("../src/trpc/context"),
-  trpcOptionsPath: z.string().optional(),
+  trpcOptions: z.boolean().or(z.string()).optional(),
   showModelNameInProcedure: configBoolean.default("true"),
+  useTRPCNext: configBoolean.default("false"),
   generateModelActions: z.string().default(Object.values(ModelAction).join(",")).transform((arg) => {
     return arg.split(",").map((action) => modelActionEnum.parse(action.trim()));
   })
@@ -6896,12 +6903,6 @@ var configSchema = z.object({
 // src/helpers.ts
 init_cjs_shims();
 
-// ../string-format/src/lower-case-first.ts
-init_cjs_shims();
-var lowerCaseFirst = /* @__PURE__ */ __name((input) => {
-  return input ? input.charAt(0).toLowerCase() + input.slice(1) : input;
-}, "lowerCaseFirst");
-
 // src/utils/get-prisma-internals.ts
 init_cjs_shims();
 
@@ -7646,24 +7647,35 @@ var generateRouterImport = /* @__PURE__ */ __name((sourceFile, modelNamePlural,
 async function generateBaseRouter(sourceFile, config, options) {
   const internals = await getPrismaInternals();
   const outputDir = internals.parseEnvValue(options.generator.output);
+  const relativeContextPath = getRelativePath(outputDir, config.contextPath, true, options.schemaPath);
   sourceFile.addStatements(
     /* ts */
     `
-  import type { Context } from '${getRelativePath(outputDir, config.contextPath, true, options.schemaPath)}';
+  import type { Context } from '${relativeContextPath}';
   `
   );
-  if (config.trpcOptionsPath) {
+  if (config.trpcOptions) {
     sourceFile.addStatements(
       /* ts */
       `
-    import trpcOptions from '${getRelativePath(outputDir, config.trpcOptionsPath, true, options.schemaPath)}';
+    import trpcOptions from '${getRelativePath(outputDir, typeof config.trpcOptions === "boolean" ? joinPaths(outputDir, "options") : config.trpcOptions, true, options.schemaPath)}';
     `
     );
   }
+  if (config.useTRPCNext) {
+    sourceFile.addStatements(
+      /* ts */
+      `
+    import { createContext } from '${relativeContextPath}';
+    import { initTRPC, TRPCError } from '@trpc/server';
+    import { createTRPCServerActionHandler } from '@stryke/trpc-next/action-handler';
+  `
+    );
+  }
   sourceFile.addStatements(
     /* ts */
     `
-  export const t = trpc.initTRPC.context<Context>().create(${config.trpcOptionsPath ? "trpcOptions" : ""});
+  export const t = trpc.initTRPC.context<Context>().create(${config.trpcOptions ? "trpcOptions" : ""});
   `
   );
   const middlewares = [];
@@ -7719,11 +7731,44 @@ async function generateBaseRouter(sourceFile, config, options) {
       )
     });
   }
+  sourceFile.addStatements(
+    /* ts */
+    `
+/**
+ * Create a server-side caller
+ * @see https://trpc.io/docs/server/server-side-calls
+ */
+export const createCallerFactory = t.createCallerFactory;`
+  );
   sourceFile.addStatements(
     /* ts */
     `
     export const publicProcedure = t.procedure; `
   );
+  if (config.useTRPCNext) {
+    sourceFile.addStatements(
+      /* ts */
+      `
+export const protectedProcedure = publicProcedure.use((opts) => {
+  const { session } = opts.ctx;
+
+  if (!session?.user) {
+    throw new TRPCError({
+      code: 'UNAUTHORIZED',
+    });
+  }
+
+  return opts.next({ ctx: { session } });
+});
+`
+    );
+    sourceFile.addStatements(
+      /* ts */
+      `
+  export const createAction = createTRPCServerActionHandler(t, createContext);
+  `
+    );
+  }
   if (middlewares.length > 0) {
     const procName = getProcedureName(config);
     middlewares.forEach((middleware, i) => {
@@ -7945,7 +7990,9 @@ var constructShield = /* @__PURE__ */ __name(async ({ queries, mutations, subscr
     })},`;
     rootItems += subscriptionLinesWrapped;
   }
-  if (rootItems.length === 0) return "";
+  if (rootItems.length === 0) {
+    return "";
+  }
   let shieldText = getImports("trpc-shield");
   const internals = await getPrismaInternals();
   const outputDir = internals.parseEnvValue(options.generator.output);
@@ -11612,6 +11659,27 @@ async function generate(options) {
     consoleLog("Skipping tRPC Shield generation");
   }
   consoleLog(`Generating tRPC source code for ${models.length} models`);
+  if (config.trpcOptions && typeof config.trpcOptions === "boolean") {
+    const trpcOptionsOutputPath = joinPaths(outputDir, "options.ts");
+    consoleLog("Generating tRPC options source file");
+    await writeFileSafely(trpcOptionsOutputPath, `import { ZodError } from 'zod';${config.useTRPCNext ? '\nimport { transformer } from "@stryke/trpc-next/shared";' : ""}
+
+export default {${config.useTRPCNext ? "\n transformer," : ""}
+  errorFormatter({ shape, error }) {
+    return {
+      ...shape,
+      data: {
+        ...shape.data,
+        zodError:
+          error.code === 'BAD_REQUEST' && error.cause instanceof ZodError
+            ? error.cause.flatten()
+            : null,
+      },
+    };
+  },
+};
+`);
+  }
   resolveModelsComments(models, hiddenModels);
   const createRouter = project.createSourceFile(import_node_path6.default.resolve(outputDir, "routers", "helpers", "createRouter.ts"), void 0, {
     overwrite: true
@@ -11640,6 +11708,10 @@ async function generate(options) {
       consoleLog(`Skipping model ${model} as it is hidden`);
       continue;
     }
+    if (!model) {
+      consoleLog(`Skipping model ${model} as it is not defined`);
+      continue;
+    }
     const modelActions = Object.keys(operations).filter((opType) => (
       // eslint-disable-next-line unicorn/prefer-includes
       config.generateModelActions.some((generateModelAction) => generateModelAction === opType.replace("One", ""))
@@ -11648,7 +11720,7 @@ async function generate(options) {
       consoleLog(`Skipping model ${model} as it has no actions to generate`);
       continue;
     }
-    const plural = (0, import_pluralize.default)(model.toLowerCase());
+    const plural = (0, import_pluralize.default)(lowerCaseFirst(model));
     consoleLog(`Generating tRPC router for model ${model}`);
     generateRouterImport(appRouter, plural, model);
     const modelRouter = project.createSourceFile(import_node_path6.default.resolve(outputDir, "routers", `${model}.router.ts`), void 0, {

package/dist/generator.js

@@ -2636,6 +2636,12 @@ function normalizeString(path6, allowAboveRoot) {
 }
 __name(normalizeString, "normalizeString");
 
+// ../string-format/src/lower-case-first.ts
+init_esm_shims();
+var lowerCaseFirst = /* @__PURE__ */ __name((input) => {
+  return input ? input.charAt(0).toLowerCase() + input.slice(1) : input;
+}, "lowerCaseFirst");
+
 // src/prisma-generator.ts
 var import_pluralize = __toESM(require_pluralize(), 1);
 import path5 from "node:path";
@@ -6891,8 +6897,9 @@ var configSchema = z.object({
   withShield: configShield.default("true"),
   withZod: configBoolean.default("true"),
   contextPath: z.string().default("../src/trpc/context"),
-  trpcOptionsPath: z.string().optional(),
+  trpcOptions: z.boolean().or(z.string()).optional(),
   showModelNameInProcedure: configBoolean.default("true"),
+  useTRPCNext: configBoolean.default("false"),
   generateModelActions: z.string().default(Object.values(ModelAction).join(",")).transform((arg) => {
     return arg.split(",").map((action) => modelActionEnum.parse(action.trim()));
   })
@@ -6901,12 +6908,6 @@ var configSchema = z.object({
 // src/helpers.ts
 init_esm_shims();
 
-// ../string-format/src/lower-case-first.ts
-init_esm_shims();
-var lowerCaseFirst = /* @__PURE__ */ __name((input) => {
-  return input ? input.charAt(0).toLowerCase() + input.slice(1) : input;
-}, "lowerCaseFirst");
-
 // src/utils/get-prisma-internals.ts
 init_esm_shims();
 
@@ -7651,24 +7652,35 @@ var generateRouterImport = /* @__PURE__ */ __name((sourceFile, modelNamePlural,
 async function generateBaseRouter(sourceFile, config, options) {
   const internals = await getPrismaInternals();
   const outputDir = internals.parseEnvValue(options.generator.output);
+  const relativeContextPath = getRelativePath(outputDir, config.contextPath, true, options.schemaPath);
   sourceFile.addStatements(
     /* ts */
     `
-  import type { Context } from '${getRelativePath(outputDir, config.contextPath, true, options.schemaPath)}';
+  import type { Context } from '${relativeContextPath}';
   `
   );
-  if (config.trpcOptionsPath) {
+  if (config.trpcOptions) {
     sourceFile.addStatements(
       /* ts */
       `
-    import trpcOptions from '${getRelativePath(outputDir, config.trpcOptionsPath, true, options.schemaPath)}';
+    import trpcOptions from '${getRelativePath(outputDir, typeof config.trpcOptions === "boolean" ? joinPaths(outputDir, "options") : config.trpcOptions, true, options.schemaPath)}';
     `
     );
   }
+  if (config.useTRPCNext) {
+    sourceFile.addStatements(
+      /* ts */
+      `
+    import { createContext } from '${relativeContextPath}';
+    import { initTRPC, TRPCError } from '@trpc/server';
+    import { createTRPCServerActionHandler } from '@stryke/trpc-next/action-handler';
+  `
+    );
+  }
   sourceFile.addStatements(
     /* ts */
     `
-  export const t = trpc.initTRPC.context<Context>().create(${config.trpcOptionsPath ? "trpcOptions" : ""});
+  export const t = trpc.initTRPC.context<Context>().create(${config.trpcOptions ? "trpcOptions" : ""});
   `
   );
   const middlewares = [];
@@ -7724,11 +7736,44 @@ async function generateBaseRouter(sourceFile, config, options) {
       )
     });
   }
+  sourceFile.addStatements(
+    /* ts */
+    `
+/**
+ * Create a server-side caller
+ * @see https://trpc.io/docs/server/server-side-calls
+ */
+export const createCallerFactory = t.createCallerFactory;`
+  );
   sourceFile.addStatements(
     /* ts */
     `
     export const publicProcedure = t.procedure; `
   );
+  if (config.useTRPCNext) {
+    sourceFile.addStatements(
+      /* ts */
+      `
+export const protectedProcedure = publicProcedure.use((opts) => {
+  const { session } = opts.ctx;
+
+  if (!session?.user) {
+    throw new TRPCError({
+      code: 'UNAUTHORIZED',
+    });
+  }
+
+  return opts.next({ ctx: { session } });
+});
+`
+    );
+    sourceFile.addStatements(
+      /* ts */
+      `
+  export const createAction = createTRPCServerActionHandler(t, createContext);
+  `
+    );
+  }
   if (middlewares.length > 0) {
     const procName = getProcedureName(config);
     middlewares.forEach((middleware, i) => {
@@ -7950,7 +7995,9 @@ var constructShield = /* @__PURE__ */ __name(async ({ queries, mutations, subscr
     })},`;
     rootItems += subscriptionLinesWrapped;
   }
-  if (rootItems.length === 0) return "";
+  if (rootItems.length === 0) {
+    return "";
+  }
   let shieldText = getImports("trpc-shield");
   const internals = await getPrismaInternals();
   const outputDir = internals.parseEnvValue(options.generator.output);
@@ -11617,6 +11664,27 @@ async function generate(options) {
     consoleLog("Skipping tRPC Shield generation");
   }
   consoleLog(`Generating tRPC source code for ${models.length} models`);
+  if (config.trpcOptions && typeof config.trpcOptions === "boolean") {
+    const trpcOptionsOutputPath = joinPaths(outputDir, "options.ts");
+    consoleLog("Generating tRPC options source file");
+    await writeFileSafely(trpcOptionsOutputPath, `import { ZodError } from 'zod';${config.useTRPCNext ? '\nimport { transformer } from "@stryke/trpc-next/shared";' : ""}
+
+export default {${config.useTRPCNext ? "\n transformer," : ""}
+  errorFormatter({ shape, error }) {
+    return {
+      ...shape,
+      data: {
+        ...shape.data,
+        zodError:
+          error.code === 'BAD_REQUEST' && error.cause instanceof ZodError
+            ? error.cause.flatten()
+            : null,
+      },
+    };
+  },
+};
+`);
+  }
   resolveModelsComments(models, hiddenModels);
   const createRouter = project.createSourceFile(path5.resolve(outputDir, "routers", "helpers", "createRouter.ts"), void 0, {
     overwrite: true
@@ -11645,6 +11713,10 @@ async function generate(options) {
       consoleLog(`Skipping model ${model} as it is hidden`);
       continue;
     }
+    if (!model) {
+      consoleLog(`Skipping model ${model} as it is not defined`);
+      continue;
+    }
     const modelActions = Object.keys(operations).filter((opType) => (
       // eslint-disable-next-line unicorn/prefer-includes
       config.generateModelActions.some((generateModelAction) => generateModelAction === opType.replace("One", ""))
@@ -11653,7 +11725,7 @@ async function generate(options) {
       consoleLog(`Skipping model ${model} as it has no actions to generate`);
       continue;
     }
-    const plural = (0, import_pluralize.default)(model.toLowerCase());
+    const plural = (0, import_pluralize.default)(lowerCaseFirst(model));
     consoleLog(`Generating tRPC router for model ${model}`);
     generateRouterImport(appRouter, plural, model);
     const modelRouter = project.createSourceFile(path5.resolve(outputDir, "routers", `${model}.router.ts`), void 0, {

package/dist/index.cjs

@@ -2627,6 +2627,12 @@ function normalizeString(path6, allowAboveRoot) {
 }
 __name(normalizeString, "normalizeString");
 
+// ../string-format/src/lower-case-first.ts
+init_cjs_shims();
+var lowerCaseFirst = /* @__PURE__ */ __name((input) => {
+  return input ? input.charAt(0).toLowerCase() + input.slice(1) : input;
+}, "lowerCaseFirst");
+
 // src/prisma-generator.ts
 var import_node_path6 = __toESM(require("node:path"), 1);
 var import_pluralize = __toESM(require_pluralize(), 1);
@@ -6882,8 +6888,9 @@ var configSchema = z.object({
   withShield: configShield.default("true"),
   withZod: configBoolean.default("true"),
   contextPath: z.string().default("../src/trpc/context"),
-  trpcOptionsPath: z.string().optional(),
+  trpcOptions: z.boolean().or(z.string()).optional(),
   showModelNameInProcedure: configBoolean.default("true"),
+  useTRPCNext: configBoolean.default("false"),
   generateModelActions: z.string().default(Object.values(ModelAction).join(",")).transform((arg) => {
     return arg.split(",").map((action) => modelActionEnum.parse(action.trim()));
   })
@@ -6892,12 +6899,6 @@ var configSchema = z.object({
 // src/helpers.ts
 init_cjs_shims();
 
-// ../string-format/src/lower-case-first.ts
-init_cjs_shims();
-var lowerCaseFirst = /* @__PURE__ */ __name((input) => {
-  return input ? input.charAt(0).toLowerCase() + input.slice(1) : input;
-}, "lowerCaseFirst");
-
 // src/utils/get-prisma-internals.ts
 init_cjs_shims();
 
@@ -7642,24 +7643,35 @@ var generateRouterImport = /* @__PURE__ */ __name((sourceFile, modelNamePlural,
 async function generateBaseRouter(sourceFile, config, options) {
   const internals = await getPrismaInternals();
   const outputDir = internals.parseEnvValue(options.generator.output);
+  const relativeContextPath = getRelativePath(outputDir, config.contextPath, true, options.schemaPath);
   sourceFile.addStatements(
     /* ts */
     `
-  import type { Context } from '${getRelativePath(outputDir, config.contextPath, true, options.schemaPath)}';
+  import type { Context } from '${relativeContextPath}';
   `
   );
-  if (config.trpcOptionsPath) {
+  if (config.trpcOptions) {
     sourceFile.addStatements(
       /* ts */
       `
-    import trpcOptions from '${getRelativePath(outputDir, config.trpcOptionsPath, true, options.schemaPath)}';
+    import trpcOptions from '${getRelativePath(outputDir, typeof config.trpcOptions === "boolean" ? joinPaths(outputDir, "options") : config.trpcOptions, true, options.schemaPath)}';
     `
     );
   }
+  if (config.useTRPCNext) {
+    sourceFile.addStatements(
+      /* ts */
+      `
+    import { createContext } from '${relativeContextPath}';
+    import { initTRPC, TRPCError } from '@trpc/server';
+    import { createTRPCServerActionHandler } from '@stryke/trpc-next/action-handler';
+  `
+    );
+  }
   sourceFile.addStatements(
     /* ts */
     `
-  export const t = trpc.initTRPC.context<Context>().create(${config.trpcOptionsPath ? "trpcOptions" : ""});
+  export const t = trpc.initTRPC.context<Context>().create(${config.trpcOptions ? "trpcOptions" : ""});
   `
   );
   const middlewares = [];
@@ -7715,11 +7727,44 @@ async function generateBaseRouter(sourceFile, config, options) {
       )
     });
   }
+  sourceFile.addStatements(
+    /* ts */
+    `
+/**
+ * Create a server-side caller
+ * @see https://trpc.io/docs/server/server-side-calls
+ */
+export const createCallerFactory = t.createCallerFactory;`
+  );
   sourceFile.addStatements(
     /* ts */
     `
     export const publicProcedure = t.procedure; `
   );
+  if (config.useTRPCNext) {
+    sourceFile.addStatements(
+      /* ts */
+      `
+export const protectedProcedure = publicProcedure.use((opts) => {
+  const { session } = opts.ctx;
+
+  if (!session?.user) {
+    throw new TRPCError({
+      code: 'UNAUTHORIZED',
+    });
+  }
+
+  return opts.next({ ctx: { session } });
+});
+`
+    );
+    sourceFile.addStatements(
+      /* ts */
+      `
+  export const createAction = createTRPCServerActionHandler(t, createContext);
+  `
+    );
+  }
   if (middlewares.length > 0) {
     const procName = getProcedureName(config);
     middlewares.forEach((middleware, i) => {
@@ -7941,7 +7986,9 @@ var constructShield = /* @__PURE__ */ __name(async ({ queries, mutations, subscr
     })},`;
     rootItems += subscriptionLinesWrapped;
   }
-  if (rootItems.length === 0) return "";
+  if (rootItems.length === 0) {
+    return "";
+  }
   let shieldText = getImports("trpc-shield");
   const internals = await getPrismaInternals();
   const outputDir = internals.parseEnvValue(options.generator.output);
@@ -11608,6 +11655,27 @@ async function generate(options) {
     consoleLog("Skipping tRPC Shield generation");
   }
   consoleLog(`Generating tRPC source code for ${models.length} models`);
+  if (config.trpcOptions && typeof config.trpcOptions === "boolean") {
+    const trpcOptionsOutputPath = joinPaths(outputDir, "options.ts");
+    consoleLog("Generating tRPC options source file");
+    await writeFileSafely(trpcOptionsOutputPath, `import { ZodError } from 'zod';${config.useTRPCNext ? '\nimport { transformer } from "@stryke/trpc-next/shared";' : ""}
+
+export default {${config.useTRPCNext ? "\n transformer," : ""}
+  errorFormatter({ shape, error }) {
+    return {
+      ...shape,
+      data: {
+        ...shape.data,
+        zodError:
+          error.code === 'BAD_REQUEST' && error.cause instanceof ZodError
+            ? error.cause.flatten()
+            : null,
+      },
+    };
+  },
+};
+`);
+  }
   resolveModelsComments(models, hiddenModels);
   const createRouter = project.createSourceFile(import_node_path6.default.resolve(outputDir, "routers", "helpers", "createRouter.ts"), void 0, {
     overwrite: true
@@ -11636,6 +11704,10 @@ async function generate(options) {
       consoleLog(`Skipping model ${model} as it is hidden`);
       continue;
     }
+    if (!model) {
+      consoleLog(`Skipping model ${model} as it is not defined`);
+      continue;
+    }
     const modelActions = Object.keys(operations).filter((opType) => (
       // eslint-disable-next-line unicorn/prefer-includes
       config.generateModelActions.some((generateModelAction) => generateModelAction === opType.replace("One", ""))
@@ -11644,7 +11716,7 @@ async function generate(options) {
       consoleLog(`Skipping model ${model} as it has no actions to generate`);
       continue;
     }
-    const plural = (0, import_pluralize.default)(model.toLowerCase());
+    const plural = (0, import_pluralize.default)(lowerCaseFirst(model));
     consoleLog(`Generating tRPC router for model ${model}`);
     generateRouterImport(appRouter, plural, model);
     const modelRouter = project.createSourceFile(import_node_path6.default.resolve(outputDir, "routers", `${model}.router.ts`), void 0, {

package/dist/index.js

@@ -2632,6 +2632,12 @@ function normalizeString(path6, allowAboveRoot) {
 }
 __name(normalizeString, "normalizeString");
 
+// ../string-format/src/lower-case-first.ts
+init_esm_shims();
+var lowerCaseFirst = /* @__PURE__ */ __name((input) => {
+  return input ? input.charAt(0).toLowerCase() + input.slice(1) : input;
+}, "lowerCaseFirst");
+
 // src/prisma-generator.ts
 var import_pluralize = __toESM(require_pluralize(), 1);
 import path5 from "node:path";
@@ -6887,8 +6893,9 @@ var configSchema = z.object({
   withShield: configShield.default("true"),
   withZod: configBoolean.default("true"),
   contextPath: z.string().default("../src/trpc/context"),
-  trpcOptionsPath: z.string().optional(),
+  trpcOptions: z.boolean().or(z.string()).optional(),
   showModelNameInProcedure: configBoolean.default("true"),
+  useTRPCNext: configBoolean.default("false"),
   generateModelActions: z.string().default(Object.values(ModelAction).join(",")).transform((arg) => {
     return arg.split(",").map((action) => modelActionEnum.parse(action.trim()));
   })
@@ -6897,12 +6904,6 @@ var configSchema = z.object({
 // src/helpers.ts
 init_esm_shims();
 
-// ../string-format/src/lower-case-first.ts
-init_esm_shims();
-var lowerCaseFirst = /* @__PURE__ */ __name((input) => {
-  return input ? input.charAt(0).toLowerCase() + input.slice(1) : input;
-}, "lowerCaseFirst");
-
 // src/utils/get-prisma-internals.ts
 init_esm_shims();
 
@@ -7647,24 +7648,35 @@ var generateRouterImport = /* @__PURE__ */ __name((sourceFile, modelNamePlural,
 async function generateBaseRouter(sourceFile, config, options) {
   const internals = await getPrismaInternals();
   const outputDir = internals.parseEnvValue(options.generator.output);
+  const relativeContextPath = getRelativePath(outputDir, config.contextPath, true, options.schemaPath);
   sourceFile.addStatements(
     /* ts */
     `
-  import type { Context } from '${getRelativePath(outputDir, config.contextPath, true, options.schemaPath)}';
+  import type { Context } from '${relativeContextPath}';
   `
   );
-  if (config.trpcOptionsPath) {
+  if (config.trpcOptions) {
     sourceFile.addStatements(
       /* ts */
       `
-    import trpcOptions from '${getRelativePath(outputDir, config.trpcOptionsPath, true, options.schemaPath)}';
+    import trpcOptions from '${getRelativePath(outputDir, typeof config.trpcOptions === "boolean" ? joinPaths(outputDir, "options") : config.trpcOptions, true, options.schemaPath)}';
     `
     );
   }
+  if (config.useTRPCNext) {
+    sourceFile.addStatements(
+      /* ts */
+      `
+    import { createContext } from '${relativeContextPath}';
+    import { initTRPC, TRPCError } from '@trpc/server';
+    import { createTRPCServerActionHandler } from '@stryke/trpc-next/action-handler';
+  `
+    );
+  }
   sourceFile.addStatements(
     /* ts */
     `
-  export const t = trpc.initTRPC.context<Context>().create(${config.trpcOptionsPath ? "trpcOptions" : ""});
+  export const t = trpc.initTRPC.context<Context>().create(${config.trpcOptions ? "trpcOptions" : ""});
   `
   );
   const middlewares = [];
@@ -7720,11 +7732,44 @@ async function generateBaseRouter(sourceFile, config, options) {
       )
     });
   }
+  sourceFile.addStatements(
+    /* ts */
+    `
+/**
+ * Create a server-side caller
+ * @see https://trpc.io/docs/server/server-side-calls
+ */
+export const createCallerFactory = t.createCallerFactory;`
+  );
   sourceFile.addStatements(
     /* ts */
     `
     export const publicProcedure = t.procedure; `
   );
+  if (config.useTRPCNext) {
+    sourceFile.addStatements(
+      /* ts */
+      `
+export const protectedProcedure = publicProcedure.use((opts) => {
+  const { session } = opts.ctx;
+
+  if (!session?.user) {
+    throw new TRPCError({
+      code: 'UNAUTHORIZED',
+    });
+  }
+
+  return opts.next({ ctx: { session } });
+});
+`
+    );
+    sourceFile.addStatements(
+      /* ts */
+      `
+  export const createAction = createTRPCServerActionHandler(t, createContext);
+  `
+    );
+  }
   if (middlewares.length > 0) {
     const procName = getProcedureName(config);
     middlewares.forEach((middleware, i) => {
@@ -7946,7 +7991,9 @@ var constructShield = /* @__PURE__ */ __name(async ({ queries, mutations, subscr
     })},`;
     rootItems += subscriptionLinesWrapped;
   }
-  if (rootItems.length === 0) return "";
+  if (rootItems.length === 0) {
+    return "";
+  }
   let shieldText = getImports("trpc-shield");
   const internals = await getPrismaInternals();
   const outputDir = internals.parseEnvValue(options.generator.output);
@@ -11613,6 +11660,27 @@ async function generate(options) {
     consoleLog("Skipping tRPC Shield generation");
   }
   consoleLog(`Generating tRPC source code for ${models.length} models`);
+  if (config.trpcOptions && typeof config.trpcOptions === "boolean") {
+    const trpcOptionsOutputPath = joinPaths(outputDir, "options.ts");
+    consoleLog("Generating tRPC options source file");
+    await writeFileSafely(trpcOptionsOutputPath, `import { ZodError } from 'zod';${config.useTRPCNext ? '\nimport { transformer } from "@stryke/trpc-next/shared";' : ""}
+
+export default {${config.useTRPCNext ? "\n transformer," : ""}
+  errorFormatter({ shape, error }) {
+    return {
+      ...shape,
+      data: {
+        ...shape.data,
+        zodError:
+          error.code === 'BAD_REQUEST' && error.cause instanceof ZodError
+            ? error.cause.flatten()
+            : null,
+      },
+    };
+  },
+};
+`);
+  }
   resolveModelsComments(models, hiddenModels);
   const createRouter = project.createSourceFile(path5.resolve(outputDir, "routers", "helpers", "createRouter.ts"), void 0, {
     overwrite: true
@@ -11641,6 +11709,10 @@ async function generate(options) {
       consoleLog(`Skipping model ${model} as it is hidden`);
       continue;
     }
+    if (!model) {
+      consoleLog(`Skipping model ${model} as it is not defined`);
+      continue;
+    }
     const modelActions = Object.keys(operations).filter((opType) => (
       // eslint-disable-next-line unicorn/prefer-includes
       config.generateModelActions.some((generateModelAction) => generateModelAction === opType.replace("One", ""))
@@ -11649,7 +11721,7 @@ async function generate(options) {
       consoleLog(`Skipping model ${model} as it has no actions to generate`);
       continue;
     }
-    const plural = (0, import_pluralize.default)(model.toLowerCase());
+    const plural = (0, import_pluralize.default)(lowerCaseFirst(model));
     consoleLog(`Generating tRPC router for model ${model}`);
     generateRouterImport(appRouter, plural, model);
     const modelRouter = project.createSourceFile(path5.resolve(outputDir, "routers", `${model}.router.ts`), void 0, {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@stryke/prisma-trpc-generator",
-  "version": "0.4.8",
+  "version": "0.5.0",
   "type": "module",
   "description": "A fork of the prisma-trpc-generator code to work in ESM with Prisma v6.",
   "repository": {