@stryke/prisma-trpc-generator 0.9.4 → 0.10.1

package/dist/generator.cjs

@@ -6872,15 +6872,15 @@ var ModelAction = /* @__PURE__ */ function(ModelAction2) {
   ModelAction2["findRaw"] = "findRaw";
   ModelAction2["aggregateRaw"] = "aggregateRaw";
   return ModelAction2;
-}(ModelAction || {});
+}({});
 var modelActionEnum = z.nativeEnum(ModelAction);
 var configBoolean = z.string().trim().transform((value) => !value || value.toLowerCase() === "false" || value.toLowerCase() === "n" || value.toLowerCase() === "no" || value === "0" ? false : value.toLowerCase() === "true" || value.toLowerCase() === "y" || value.toLowerCase() === "yes" || value === "1" ? true : value);
 var configSchema = z.object({
   debug: configBoolean.default("false"),
-  withMiddleware: configBoolean.default("false"),
-  withShield: configBoolean.default("true"),
   withZod: configBoolean.default("true"),
   withNext: configBoolean.default("true"),
+  withMiddleware: configBoolean.default("false"),
+  withShield: configBoolean.default("false"),
   contextPath: z.string().trim().default("../context"),
   trpcOptions: configBoolean.default("true"),
   showModelNameInProcedure: configBoolean.default("true"),
@@ -7625,9 +7625,6 @@ async function generateTRPCExports(sourceFile, config, options, outputDir) {
         "permissions"
       ]
     });
-    sourceFile.formatText({
-      indentSize: 2
-    });
   }
   sourceFile.addStatements(
     /* ts */
@@ -7745,6 +7742,9 @@ export const createCallerFactory = t.createCallerFactory;`
   `
     );
   }
+  sourceFile.formatText({
+    indentSize: 2
+  });
 }
 __name(generateTRPCExports, "generateTRPCExports");
 function generateProcedure(sourceFile, name, typeName, modelName, opType, baseOpType, config) {
@@ -10355,7 +10355,7 @@ var writeIndexFile = /* @__PURE__ */ __name(async (indexPath) => {
   await writeFileSafely(indexPath, rows.join("\n"), false);
 }, "writeIndexFile");
 
-// src/zod-helpers/aggregate-helpers.ts
+// src/zod/aggregate-helpers.ts
 init_cjs_shims();
 var isAggregateOutputType = /* @__PURE__ */ __name((name) => /(?:Count|Avg|Sum|Min|Max)AggregateOutputType$/.test(name), "isAggregateOutputType");
 var isAggregateInputType = /* @__PURE__ */ __name((name) => name.endsWith("CountAggregateInput") || name.endsWith("SumAggregateInput") || name.endsWith("AvgAggregateInput") || name.endsWith("MinAggregateInput") || name.endsWith("MaxAggregateInput"), "isAggregateInputType");
@@ -10427,7 +10427,7 @@ function resolveZodAggregateOperationSupport(inputObjectTypes) {
 }
 __name(resolveZodAggregateOperationSupport, "resolveZodAggregateOperationSupport");
 
-// src/zod-helpers/comments-helpers.ts
+// src/zod/comments-helpers.ts
 init_cjs_shims();
 var modelAttributeRegex = /(?:@@Gen\.)+[A-z]+\(.+\)/;
 var attributeNameRegex = /\.+[A-Z]+\(+/i;
@@ -10517,14 +10517,14 @@ function hideZodInputObjectTypesAndRelatedFields(inputObjectTypes, hiddenModels,
 }
 __name(hideZodInputObjectTypesAndRelatedFields, "hideZodInputObjectTypesAndRelatedFields");
 
-// src/zod-helpers/generator-helpers.ts
+// src/zod/generator-helpers.ts
 init_cjs_shims();
 
-// src/zod-helpers/transformer.ts
+// src/zod/transformer.ts
 init_cjs_shims();
 var import_node_path4 = __toESM(require("node:path"), 1);
 
-// src/zod-helpers/model-helpers.ts
+// src/zod/model-helpers.ts
 init_cjs_shims();
 function checkModelHasModelRelation(model) {
   const { fields: modelFields } = model;
@@ -10562,7 +10562,7 @@ function findModelByName(models, modelName) {
 }
 __name(findModelByName, "findModelByName");
 
-// src/zod-helpers/mongodb-helpers.ts
+// src/zod/mongodb-helpers.ts
 init_cjs_shims();
 function addMissingInputObjectTypesForMongoDbRawOpsAndQueries(modelOperations, outputObjectTypes, inputObjectTypes) {
   const rawOpsMap = resolveMongoDbRawOperations(modelOperations);
@@ -10623,7 +10623,7 @@ function getMongoDbRawQueries(outputObjectTypes) {
 __name(getMongoDbRawQueries, "getMongoDbRawQueries");
 var isMongodbRawOp = /* @__PURE__ */ __name((name) => /find[\s\S]*?Raw/.test(name) || /aggregate[\s\S]*?Raw/.test(name), "isMongodbRawOp");
 
-// src/zod-helpers/transformer.ts
+// src/zod/transformer.ts
 var Transformer = class _Transformer {
   static {
     __name(this, "Transformer");
@@ -10705,7 +10705,7 @@ ${this.generateExportSchemaStatement(`${name}`, `z.enum(${JSON.stringify(values)
     if (lines.length === 0) {
       return [];
     }
-    let alternatives = lines.reduce((result, inputType) => {
+    let alternatives = lines.slice(0, 1).reduce((result, inputType) => {
       if (inputType.type === "String") {
         result.push(this.wrapWithZodValidators("z.string()", field));
       } else if (inputType.type === "Int" || inputType.type === "Float" || inputType.type === "Decimal") {
@@ -10729,7 +10729,7 @@ ${this.generateExportSchemaStatement(`${name}`, `z.enum(${JSON.stringify(values)
           if (inputType.type !== this.name && typeof inputType.type === "string") {
             this.addSchemaImport(inputType.type);
           }
-          result.push(this.generatePrismaStringLine(field, lines.length));
+          result.push(this.generatePrismaStringLine(field, inputType, lines.length));
         }
       }
       return result;
@@ -10768,11 +10768,7 @@ ${this.generateExportSchemaStatement(`${name}`, `z.enum(${JSON.stringify(values)
   addSchemaImport(name) {
     this.schemaImports.add(name);
   }
-  generatePrismaStringLine(field, inputsLength) {
-    if (field.inputTypes.length === 0 || !field.inputTypes[0]) {
-      return "";
-    }
-    const inputType = field.inputTypes[0];
+  generatePrismaStringLine(field, inputType, inputsLength) {
     const isEnum = inputType.location === "enumTypes";
     const inputTypeString = inputType.type;
     const { isModelQueryType, modelName, queryName } = this.checkIsModelQueryType(inputTypeString);
@@ -10898,8 +10894,8 @@ ${this.generateExportSchemaStatement(`${name}`, `z.enum(${JSON.stringify(values)
     };
   }
   resolveModelQuerySchemaName(modelName, queryName) {
-    const modelNameCapitalized = modelName.charAt(0).toUpperCase() + modelName.slice(1);
-    const queryNameCapitalized = queryName.charAt(0).toUpperCase() + queryName.slice(1);
+    const modelNameCapitalized = upperCaseFirst(modelName);
+    const queryNameCapitalized = upperCaseFirst(queryName);
     return `${modelNameCapitalized}${queryNameCapitalized}Schema`;
   }
   wrapWithZodUnion(zodStringFields) {
@@ -11151,7 +11147,7 @@ ${this.generateExportSchemaStatement(`${name}`, `z.enum(${JSON.stringify(values)
   }
 };
 
-// src/zod-helpers/generator-helpers.ts
+// src/zod/generator-helpers.ts
 async function generateZodEnumSchemas(prismaSchemaEnum, modelSchemaEnum) {
   const enumTypes = [
     ...prismaSchemaEnum,
@@ -11191,10 +11187,10 @@ async function generateZodIndex() {
 }
 __name(generateZodIndex, "generateZodIndex");
 
-// src/zod-helpers/helpers.ts
+// src/zod/helpers.ts
 init_cjs_shims();
 
-// src/zod-helpers/include-helpers.ts
+// src/zod/include-helpers.ts
 init_cjs_shims();
 function addMissingInputObjectTypesForInclude(inputObjectTypes, models, isGenerateSelect) {
   const generatedIncludeInputObjectTypes = generateModelIncludeInputObjectTypes(models, isGenerateSelect);
@@ -11277,7 +11273,7 @@ function generateModelIncludeInputObjectTypes(models, isGenerateSelect) {
 }
 __name(generateModelIncludeInputObjectTypes, "generateModelIncludeInputObjectTypes");
 
-// src/zod-helpers/modelArgs-helpers.ts
+// src/zod/modelArgs-helpers.ts
 init_cjs_shims();
 function addMissingInputObjectTypesForModelArgs(inputObjectTypes, models, isGenerateSelect, isGenerateInclude) {
   const modelArgsInputObjectTypes = generateModelArgsInputObjectTypes(models, isGenerateSelect, isGenerateInclude);
@@ -11338,7 +11334,7 @@ function generateModelArgsInputObjectTypes(models, isGenerateSelect, isGenerateI
 }
 __name(generateModelArgsInputObjectTypes, "generateModelArgsInputObjectTypes");
 
-// src/zod-helpers/select-helpers.ts
+// src/zod/select-helpers.ts
 init_cjs_shims();
 function addMissingInputObjectTypesForSelect(inputObjectTypes, outputObjectTypes, models) {
   const modelCountOutputTypes = getModelCountOutputTypes(outputObjectTypes);
@@ -11486,7 +11482,7 @@ function generateModelSelectInputObjectTypes(models) {
 }
 __name(generateModelSelectInputObjectTypes, "generateModelSelectInputObjectTypes");
 
-// src/zod-helpers/whereUniqueInput-helpers.ts
+// src/zod/whereUniqueInput-helpers.ts
 init_cjs_shims();
 function changeOptionalToRequiredFields(inputObjectTypes) {
   inputObjectTypes.map((item) => {
@@ -11505,7 +11501,7 @@ function changeOptionalToRequiredFields(inputObjectTypes) {
 }
 __name(changeOptionalToRequiredFields, "changeOptionalToRequiredFields");
 
-// src/zod-helpers/helpers.ts
+// src/zod/helpers.ts
 function addMissingZodInputObjectTypes(inputObjectTypes, outputObjectTypes, models, modelOperations, dataSourceProvider, options) {
   if (dataSourceProvider === "mongodb") {
     addMissingInputObjectTypesForMongoDbRawOpsAndQueries(modelOperations, outputObjectTypes, inputObjectTypes);

package/dist/generator.js

@@ -6877,15 +6877,15 @@ var ModelAction = /* @__PURE__ */ function(ModelAction2) {
   ModelAction2["findRaw"] = "findRaw";
   ModelAction2["aggregateRaw"] = "aggregateRaw";
   return ModelAction2;
-}(ModelAction || {});
+}({});
 var modelActionEnum = z.nativeEnum(ModelAction);
 var configBoolean = z.string().trim().transform((value) => !value || value.toLowerCase() === "false" || value.toLowerCase() === "n" || value.toLowerCase() === "no" || value === "0" ? false : value.toLowerCase() === "true" || value.toLowerCase() === "y" || value.toLowerCase() === "yes" || value === "1" ? true : value);
 var configSchema = z.object({
   debug: configBoolean.default("false"),
-  withMiddleware: configBoolean.default("false"),
-  withShield: configBoolean.default("true"),
   withZod: configBoolean.default("true"),
   withNext: configBoolean.default("true"),
+  withMiddleware: configBoolean.default("false"),
+  withShield: configBoolean.default("false"),
   contextPath: z.string().trim().default("../context"),
   trpcOptions: configBoolean.default("true"),
   showModelNameInProcedure: configBoolean.default("true"),
@@ -7630,9 +7630,6 @@ async function generateTRPCExports(sourceFile, config, options, outputDir) {
         "permissions"
       ]
     });
-    sourceFile.formatText({
-      indentSize: 2
-    });
   }
   sourceFile.addStatements(
     /* ts */
@@ -7750,6 +7747,9 @@ export const createCallerFactory = t.createCallerFactory;`
   `
     );
   }
+  sourceFile.formatText({
+    indentSize: 2
+  });
 }
 __name(generateTRPCExports, "generateTRPCExports");
 function generateProcedure(sourceFile, name, typeName, modelName, opType, baseOpType, config) {
@@ -10360,7 +10360,7 @@ var writeIndexFile = /* @__PURE__ */ __name(async (indexPath) => {
   await writeFileSafely(indexPath, rows.join("\n"), false);
 }, "writeIndexFile");
 
-// src/zod-helpers/aggregate-helpers.ts
+// src/zod/aggregate-helpers.ts
 init_esm_shims();
 var isAggregateOutputType = /* @__PURE__ */ __name((name) => /(?:Count|Avg|Sum|Min|Max)AggregateOutputType$/.test(name), "isAggregateOutputType");
 var isAggregateInputType = /* @__PURE__ */ __name((name) => name.endsWith("CountAggregateInput") || name.endsWith("SumAggregateInput") || name.endsWith("AvgAggregateInput") || name.endsWith("MinAggregateInput") || name.endsWith("MaxAggregateInput"), "isAggregateInputType");
@@ -10432,7 +10432,7 @@ function resolveZodAggregateOperationSupport(inputObjectTypes) {
 }
 __name(resolveZodAggregateOperationSupport, "resolveZodAggregateOperationSupport");
 
-// src/zod-helpers/comments-helpers.ts
+// src/zod/comments-helpers.ts
 init_esm_shims();
 var modelAttributeRegex = /(?:@@Gen\.)+[A-z]+\(.+\)/;
 var attributeNameRegex = /\.+[A-Z]+\(+/i;
@@ -10522,14 +10522,14 @@ function hideZodInputObjectTypesAndRelatedFields(inputObjectTypes, hiddenModels,
 }
 __name(hideZodInputObjectTypesAndRelatedFields, "hideZodInputObjectTypesAndRelatedFields");
 
-// src/zod-helpers/generator-helpers.ts
+// src/zod/generator-helpers.ts
 init_esm_shims();
 
-// src/zod-helpers/transformer.ts
+// src/zod/transformer.ts
 init_esm_shims();
 import path3 from "node:path";
 
-// src/zod-helpers/model-helpers.ts
+// src/zod/model-helpers.ts
 init_esm_shims();
 function checkModelHasModelRelation(model) {
   const { fields: modelFields } = model;
@@ -10567,7 +10567,7 @@ function findModelByName(models, modelName) {
 }
 __name(findModelByName, "findModelByName");
 
-// src/zod-helpers/mongodb-helpers.ts
+// src/zod/mongodb-helpers.ts
 init_esm_shims();
 function addMissingInputObjectTypesForMongoDbRawOpsAndQueries(modelOperations, outputObjectTypes, inputObjectTypes) {
   const rawOpsMap = resolveMongoDbRawOperations(modelOperations);
@@ -10628,7 +10628,7 @@ function getMongoDbRawQueries(outputObjectTypes) {
 __name(getMongoDbRawQueries, "getMongoDbRawQueries");
 var isMongodbRawOp = /* @__PURE__ */ __name((name) => /find[\s\S]*?Raw/.test(name) || /aggregate[\s\S]*?Raw/.test(name), "isMongodbRawOp");
 
-// src/zod-helpers/transformer.ts
+// src/zod/transformer.ts
 var Transformer = class _Transformer {
   static {
     __name(this, "Transformer");
@@ -10710,7 +10710,7 @@ ${this.generateExportSchemaStatement(`${name}`, `z.enum(${JSON.stringify(values)
     if (lines.length === 0) {
       return [];
     }
-    let alternatives = lines.reduce((result, inputType) => {
+    let alternatives = lines.slice(0, 1).reduce((result, inputType) => {
       if (inputType.type === "String") {
         result.push(this.wrapWithZodValidators("z.string()", field));
       } else if (inputType.type === "Int" || inputType.type === "Float" || inputType.type === "Decimal") {
@@ -10734,7 +10734,7 @@ ${this.generateExportSchemaStatement(`${name}`, `z.enum(${JSON.stringify(values)
           if (inputType.type !== this.name && typeof inputType.type === "string") {
             this.addSchemaImport(inputType.type);
           }
-          result.push(this.generatePrismaStringLine(field, lines.length));
+          result.push(this.generatePrismaStringLine(field, inputType, lines.length));
         }
       }
       return result;
@@ -10773,11 +10773,7 @@ ${this.generateExportSchemaStatement(`${name}`, `z.enum(${JSON.stringify(values)
   addSchemaImport(name) {
     this.schemaImports.add(name);
   }
-  generatePrismaStringLine(field, inputsLength) {
-    if (field.inputTypes.length === 0 || !field.inputTypes[0]) {
-      return "";
-    }
-    const inputType = field.inputTypes[0];
+  generatePrismaStringLine(field, inputType, inputsLength) {
     const isEnum = inputType.location === "enumTypes";
     const inputTypeString = inputType.type;
     const { isModelQueryType, modelName, queryName } = this.checkIsModelQueryType(inputTypeString);
@@ -10903,8 +10899,8 @@ ${this.generateExportSchemaStatement(`${name}`, `z.enum(${JSON.stringify(values)
     };
   }
   resolveModelQuerySchemaName(modelName, queryName) {
-    const modelNameCapitalized = modelName.charAt(0).toUpperCase() + modelName.slice(1);
-    const queryNameCapitalized = queryName.charAt(0).toUpperCase() + queryName.slice(1);
+    const modelNameCapitalized = upperCaseFirst(modelName);
+    const queryNameCapitalized = upperCaseFirst(queryName);
     return `${modelNameCapitalized}${queryNameCapitalized}Schema`;
   }
   wrapWithZodUnion(zodStringFields) {
@@ -11156,7 +11152,7 @@ ${this.generateExportSchemaStatement(`${name}`, `z.enum(${JSON.stringify(values)
   }
 };
 
-// src/zod-helpers/generator-helpers.ts
+// src/zod/generator-helpers.ts
 async function generateZodEnumSchemas(prismaSchemaEnum, modelSchemaEnum) {
   const enumTypes = [
     ...prismaSchemaEnum,
@@ -11196,10 +11192,10 @@ async function generateZodIndex() {
 }
 __name(generateZodIndex, "generateZodIndex");
 
-// src/zod-helpers/helpers.ts
+// src/zod/helpers.ts
 init_esm_shims();
 
-// src/zod-helpers/include-helpers.ts
+// src/zod/include-helpers.ts
 init_esm_shims();
 function addMissingInputObjectTypesForInclude(inputObjectTypes, models, isGenerateSelect) {
   const generatedIncludeInputObjectTypes = generateModelIncludeInputObjectTypes(models, isGenerateSelect);
@@ -11282,7 +11278,7 @@ function generateModelIncludeInputObjectTypes(models, isGenerateSelect) {
 }
 __name(generateModelIncludeInputObjectTypes, "generateModelIncludeInputObjectTypes");
 
-// src/zod-helpers/modelArgs-helpers.ts
+// src/zod/modelArgs-helpers.ts
 init_esm_shims();
 function addMissingInputObjectTypesForModelArgs(inputObjectTypes, models, isGenerateSelect, isGenerateInclude) {
   const modelArgsInputObjectTypes = generateModelArgsInputObjectTypes(models, isGenerateSelect, isGenerateInclude);
@@ -11343,7 +11339,7 @@ function generateModelArgsInputObjectTypes(models, isGenerateSelect, isGenerateI
 }
 __name(generateModelArgsInputObjectTypes, "generateModelArgsInputObjectTypes");
 
-// src/zod-helpers/select-helpers.ts
+// src/zod/select-helpers.ts
 init_esm_shims();
 function addMissingInputObjectTypesForSelect(inputObjectTypes, outputObjectTypes, models) {
   const modelCountOutputTypes = getModelCountOutputTypes(outputObjectTypes);
@@ -11491,7 +11487,7 @@ function generateModelSelectInputObjectTypes(models) {
 }
 __name(generateModelSelectInputObjectTypes, "generateModelSelectInputObjectTypes");
 
-// src/zod-helpers/whereUniqueInput-helpers.ts
+// src/zod/whereUniqueInput-helpers.ts
 init_esm_shims();
 function changeOptionalToRequiredFields(inputObjectTypes) {
   inputObjectTypes.map((item) => {
@@ -11510,7 +11506,7 @@ function changeOptionalToRequiredFields(inputObjectTypes) {
 }
 __name(changeOptionalToRequiredFields, "changeOptionalToRequiredFields");
 
-// src/zod-helpers/helpers.ts
+// src/zod/helpers.ts
 function addMissingZodInputObjectTypes(inputObjectTypes, outputObjectTypes, models, modelOperations, dataSourceProvider, options) {
   if (dataSourceProvider === "mongodb") {
     addMissingInputObjectTypesForMongoDbRawOpsAndQueries(modelOperations, outputObjectTypes, inputObjectTypes);

package/dist/index.cjs

@@ -6868,15 +6868,15 @@ var ModelAction = /* @__PURE__ */ function(ModelAction2) {
   ModelAction2["findRaw"] = "findRaw";
   ModelAction2["aggregateRaw"] = "aggregateRaw";
   return ModelAction2;
-}(ModelAction || {});
+}({});
 var modelActionEnum = z.nativeEnum(ModelAction);
 var configBoolean = z.string().trim().transform((value) => !value || value.toLowerCase() === "false" || value.toLowerCase() === "n" || value.toLowerCase() === "no" || value === "0" ? false : value.toLowerCase() === "true" || value.toLowerCase() === "y" || value.toLowerCase() === "yes" || value === "1" ? true : value);
 var configSchema = z.object({
   debug: configBoolean.default("false"),
-  withMiddleware: configBoolean.default("false"),
-  withShield: configBoolean.default("true"),
   withZod: configBoolean.default("true"),
   withNext: configBoolean.default("true"),
+  withMiddleware: configBoolean.default("false"),
+  withShield: configBoolean.default("false"),
   contextPath: z.string().trim().default("../context"),
   trpcOptions: configBoolean.default("true"),
   showModelNameInProcedure: configBoolean.default("true"),
@@ -7621,9 +7621,6 @@ async function generateTRPCExports(sourceFile, config, options, outputDir) {
         "permissions"
       ]
     });
-    sourceFile.formatText({
-      indentSize: 2
-    });
   }
   sourceFile.addStatements(
     /* ts */
@@ -7741,6 +7738,9 @@ export const createCallerFactory = t.createCallerFactory;`
   `
     );
   }
+  sourceFile.formatText({
+    indentSize: 2
+  });
 }
 __name(generateTRPCExports, "generateTRPCExports");
 function generateProcedure(sourceFile, name, typeName, modelName, opType, baseOpType, config) {
@@ -10351,7 +10351,7 @@ var writeIndexFile = /* @__PURE__ */ __name(async (indexPath) => {
   await writeFileSafely(indexPath, rows.join("\n"), false);
 }, "writeIndexFile");
 
-// src/zod-helpers/aggregate-helpers.ts
+// src/zod/aggregate-helpers.ts
 init_cjs_shims();
 var isAggregateOutputType = /* @__PURE__ */ __name((name) => /(?:Count|Avg|Sum|Min|Max)AggregateOutputType$/.test(name), "isAggregateOutputType");
 var isAggregateInputType = /* @__PURE__ */ __name((name) => name.endsWith("CountAggregateInput") || name.endsWith("SumAggregateInput") || name.endsWith("AvgAggregateInput") || name.endsWith("MinAggregateInput") || name.endsWith("MaxAggregateInput"), "isAggregateInputType");
@@ -10423,7 +10423,7 @@ function resolveZodAggregateOperationSupport(inputObjectTypes) {
 }
 __name(resolveZodAggregateOperationSupport, "resolveZodAggregateOperationSupport");
 
-// src/zod-helpers/comments-helpers.ts
+// src/zod/comments-helpers.ts
 init_cjs_shims();
 var modelAttributeRegex = /(?:@@Gen\.)+[A-z]+\(.+\)/;
 var attributeNameRegex = /\.+[A-Z]+\(+/i;
@@ -10513,14 +10513,14 @@ function hideZodInputObjectTypesAndRelatedFields(inputObjectTypes, hiddenModels,
 }
 __name(hideZodInputObjectTypesAndRelatedFields, "hideZodInputObjectTypesAndRelatedFields");
 
-// src/zod-helpers/generator-helpers.ts
+// src/zod/generator-helpers.ts
 init_cjs_shims();
 
-// src/zod-helpers/transformer.ts
+// src/zod/transformer.ts
 init_cjs_shims();
 var import_node_path4 = __toESM(require("node:path"), 1);
 
-// src/zod-helpers/model-helpers.ts
+// src/zod/model-helpers.ts
 init_cjs_shims();
 function checkModelHasModelRelation(model) {
   const { fields: modelFields } = model;
@@ -10558,7 +10558,7 @@ function findModelByName(models, modelName) {
 }
 __name(findModelByName, "findModelByName");
 
-// src/zod-helpers/mongodb-helpers.ts
+// src/zod/mongodb-helpers.ts
 init_cjs_shims();
 function addMissingInputObjectTypesForMongoDbRawOpsAndQueries(modelOperations, outputObjectTypes, inputObjectTypes) {
   const rawOpsMap = resolveMongoDbRawOperations(modelOperations);
@@ -10619,7 +10619,7 @@ function getMongoDbRawQueries(outputObjectTypes) {
 __name(getMongoDbRawQueries, "getMongoDbRawQueries");
 var isMongodbRawOp = /* @__PURE__ */ __name((name) => /find[\s\S]*?Raw/.test(name) || /aggregate[\s\S]*?Raw/.test(name), "isMongodbRawOp");
 
-// src/zod-helpers/transformer.ts
+// src/zod/transformer.ts
 var Transformer = class _Transformer {
   static {
     __name(this, "Transformer");
@@ -10701,7 +10701,7 @@ ${this.generateExportSchemaStatement(`${name}`, `z.enum(${JSON.stringify(values)
     if (lines.length === 0) {
       return [];
     }
-    let alternatives = lines.reduce((result, inputType) => {
+    let alternatives = lines.slice(0, 1).reduce((result, inputType) => {
       if (inputType.type === "String") {
         result.push(this.wrapWithZodValidators("z.string()", field));
       } else if (inputType.type === "Int" || inputType.type === "Float" || inputType.type === "Decimal") {
@@ -10725,7 +10725,7 @@ ${this.generateExportSchemaStatement(`${name}`, `z.enum(${JSON.stringify(values)
           if (inputType.type !== this.name && typeof inputType.type === "string") {
             this.addSchemaImport(inputType.type);
           }
-          result.push(this.generatePrismaStringLine(field, lines.length));
+          result.push(this.generatePrismaStringLine(field, inputType, lines.length));
         }
       }
       return result;
@@ -10764,11 +10764,7 @@ ${this.generateExportSchemaStatement(`${name}`, `z.enum(${JSON.stringify(values)
   addSchemaImport(name) {
     this.schemaImports.add(name);
   }
-  generatePrismaStringLine(field, inputsLength) {
-    if (field.inputTypes.length === 0 || !field.inputTypes[0]) {
-      return "";
-    }
-    const inputType = field.inputTypes[0];
+  generatePrismaStringLine(field, inputType, inputsLength) {
     const isEnum = inputType.location === "enumTypes";
     const inputTypeString = inputType.type;
     const { isModelQueryType, modelName, queryName } = this.checkIsModelQueryType(inputTypeString);
@@ -10894,8 +10890,8 @@ ${this.generateExportSchemaStatement(`${name}`, `z.enum(${JSON.stringify(values)
     };
   }
   resolveModelQuerySchemaName(modelName, queryName) {
-    const modelNameCapitalized = modelName.charAt(0).toUpperCase() + modelName.slice(1);
-    const queryNameCapitalized = queryName.charAt(0).toUpperCase() + queryName.slice(1);
+    const modelNameCapitalized = upperCaseFirst(modelName);
+    const queryNameCapitalized = upperCaseFirst(queryName);
     return `${modelNameCapitalized}${queryNameCapitalized}Schema`;
   }
   wrapWithZodUnion(zodStringFields) {
@@ -11147,7 +11143,7 @@ ${this.generateExportSchemaStatement(`${name}`, `z.enum(${JSON.stringify(values)
   }
 };
 
-// src/zod-helpers/generator-helpers.ts
+// src/zod/generator-helpers.ts
 async function generateZodEnumSchemas(prismaSchemaEnum, modelSchemaEnum) {
   const enumTypes = [
     ...prismaSchemaEnum,
@@ -11187,10 +11183,10 @@ async function generateZodIndex() {
 }
 __name(generateZodIndex, "generateZodIndex");
 
-// src/zod-helpers/helpers.ts
+// src/zod/helpers.ts
 init_cjs_shims();
 
-// src/zod-helpers/include-helpers.ts
+// src/zod/include-helpers.ts
 init_cjs_shims();
 function addMissingInputObjectTypesForInclude(inputObjectTypes, models, isGenerateSelect) {
   const generatedIncludeInputObjectTypes = generateModelIncludeInputObjectTypes(models, isGenerateSelect);
@@ -11273,7 +11269,7 @@ function generateModelIncludeInputObjectTypes(models, isGenerateSelect) {
 }
 __name(generateModelIncludeInputObjectTypes, "generateModelIncludeInputObjectTypes");
 
-// src/zod-helpers/modelArgs-helpers.ts
+// src/zod/modelArgs-helpers.ts
 init_cjs_shims();
 function addMissingInputObjectTypesForModelArgs(inputObjectTypes, models, isGenerateSelect, isGenerateInclude) {
   const modelArgsInputObjectTypes = generateModelArgsInputObjectTypes(models, isGenerateSelect, isGenerateInclude);
@@ -11334,7 +11330,7 @@ function generateModelArgsInputObjectTypes(models, isGenerateSelect, isGenerateI
 }
 __name(generateModelArgsInputObjectTypes, "generateModelArgsInputObjectTypes");
 
-// src/zod-helpers/select-helpers.ts
+// src/zod/select-helpers.ts
 init_cjs_shims();
 function addMissingInputObjectTypesForSelect(inputObjectTypes, outputObjectTypes, models) {
   const modelCountOutputTypes = getModelCountOutputTypes(outputObjectTypes);
@@ -11482,7 +11478,7 @@ function generateModelSelectInputObjectTypes(models) {
 }
 __name(generateModelSelectInputObjectTypes, "generateModelSelectInputObjectTypes");
 
-// src/zod-helpers/whereUniqueInput-helpers.ts
+// src/zod/whereUniqueInput-helpers.ts
 init_cjs_shims();
 function changeOptionalToRequiredFields(inputObjectTypes) {
   inputObjectTypes.map((item) => {
@@ -11501,7 +11497,7 @@ function changeOptionalToRequiredFields(inputObjectTypes) {
 }
 __name(changeOptionalToRequiredFields, "changeOptionalToRequiredFields");
 
-// src/zod-helpers/helpers.ts
+// src/zod/helpers.ts
 function addMissingZodInputObjectTypes(inputObjectTypes, outputObjectTypes, models, modelOperations, dataSourceProvider, options) {
   if (dataSourceProvider === "mongodb") {
     addMissingInputObjectTypesForMongoDbRawOpsAndQueries(modelOperations, outputObjectTypes, inputObjectTypes);

package/dist/index.js

@@ -6873,15 +6873,15 @@ var ModelAction = /* @__PURE__ */ function(ModelAction2) {
   ModelAction2["findRaw"] = "findRaw";
   ModelAction2["aggregateRaw"] = "aggregateRaw";
   return ModelAction2;
-}(ModelAction || {});
+}({});
 var modelActionEnum = z.nativeEnum(ModelAction);
 var configBoolean = z.string().trim().transform((value) => !value || value.toLowerCase() === "false" || value.toLowerCase() === "n" || value.toLowerCase() === "no" || value === "0" ? false : value.toLowerCase() === "true" || value.toLowerCase() === "y" || value.toLowerCase() === "yes" || value === "1" ? true : value);
 var configSchema = z.object({
   debug: configBoolean.default("false"),
-  withMiddleware: configBoolean.default("false"),
-  withShield: configBoolean.default("true"),
   withZod: configBoolean.default("true"),
   withNext: configBoolean.default("true"),
+  withMiddleware: configBoolean.default("false"),
+  withShield: configBoolean.default("false"),
   contextPath: z.string().trim().default("../context"),
   trpcOptions: configBoolean.default("true"),
   showModelNameInProcedure: configBoolean.default("true"),
@@ -7626,9 +7626,6 @@ async function generateTRPCExports(sourceFile, config, options, outputDir) {
         "permissions"
       ]
     });
-    sourceFile.formatText({
-      indentSize: 2
-    });
   }
   sourceFile.addStatements(
     /* ts */
@@ -7746,6 +7743,9 @@ export const createCallerFactory = t.createCallerFactory;`
   `
     );
   }
+  sourceFile.formatText({
+    indentSize: 2
+  });
 }
 __name(generateTRPCExports, "generateTRPCExports");
 function generateProcedure(sourceFile, name, typeName, modelName, opType, baseOpType, config) {
@@ -10356,7 +10356,7 @@ var writeIndexFile = /* @__PURE__ */ __name(async (indexPath) => {
   await writeFileSafely(indexPath, rows.join("\n"), false);
 }, "writeIndexFile");
 
-// src/zod-helpers/aggregate-helpers.ts
+// src/zod/aggregate-helpers.ts
 init_esm_shims();
 var isAggregateOutputType = /* @__PURE__ */ __name((name) => /(?:Count|Avg|Sum|Min|Max)AggregateOutputType$/.test(name), "isAggregateOutputType");
 var isAggregateInputType = /* @__PURE__ */ __name((name) => name.endsWith("CountAggregateInput") || name.endsWith("SumAggregateInput") || name.endsWith("AvgAggregateInput") || name.endsWith("MinAggregateInput") || name.endsWith("MaxAggregateInput"), "isAggregateInputType");
@@ -10428,7 +10428,7 @@ function resolveZodAggregateOperationSupport(inputObjectTypes) {
 }
 __name(resolveZodAggregateOperationSupport, "resolveZodAggregateOperationSupport");
 
-// src/zod-helpers/comments-helpers.ts
+// src/zod/comments-helpers.ts
 init_esm_shims();
 var modelAttributeRegex = /(?:@@Gen\.)+[A-z]+\(.+\)/;
 var attributeNameRegex = /\.+[A-Z]+\(+/i;
@@ -10518,14 +10518,14 @@ function hideZodInputObjectTypesAndRelatedFields(inputObjectTypes, hiddenModels,
 }
 __name(hideZodInputObjectTypesAndRelatedFields, "hideZodInputObjectTypesAndRelatedFields");
 
-// src/zod-helpers/generator-helpers.ts
+// src/zod/generator-helpers.ts
 init_esm_shims();
 
-// src/zod-helpers/transformer.ts
+// src/zod/transformer.ts
 init_esm_shims();
 import path3 from "node:path";
 
-// src/zod-helpers/model-helpers.ts
+// src/zod/model-helpers.ts
 init_esm_shims();
 function checkModelHasModelRelation(model) {
   const { fields: modelFields } = model;
@@ -10563,7 +10563,7 @@ function findModelByName(models, modelName) {
 }
 __name(findModelByName, "findModelByName");
 
-// src/zod-helpers/mongodb-helpers.ts
+// src/zod/mongodb-helpers.ts
 init_esm_shims();
 function addMissingInputObjectTypesForMongoDbRawOpsAndQueries(modelOperations, outputObjectTypes, inputObjectTypes) {
   const rawOpsMap = resolveMongoDbRawOperations(modelOperations);
@@ -10624,7 +10624,7 @@ function getMongoDbRawQueries(outputObjectTypes) {
 __name(getMongoDbRawQueries, "getMongoDbRawQueries");
 var isMongodbRawOp = /* @__PURE__ */ __name((name) => /find[\s\S]*?Raw/.test(name) || /aggregate[\s\S]*?Raw/.test(name), "isMongodbRawOp");
 
-// src/zod-helpers/transformer.ts
+// src/zod/transformer.ts
 var Transformer = class _Transformer {
   static {
     __name(this, "Transformer");
@@ -10706,7 +10706,7 @@ ${this.generateExportSchemaStatement(`${name}`, `z.enum(${JSON.stringify(values)
     if (lines.length === 0) {
       return [];
     }
-    let alternatives = lines.reduce((result, inputType) => {
+    let alternatives = lines.slice(0, 1).reduce((result, inputType) => {
       if (inputType.type === "String") {
         result.push(this.wrapWithZodValidators("z.string()", field));
       } else if (inputType.type === "Int" || inputType.type === "Float" || inputType.type === "Decimal") {
@@ -10730,7 +10730,7 @@ ${this.generateExportSchemaStatement(`${name}`, `z.enum(${JSON.stringify(values)
           if (inputType.type !== this.name && typeof inputType.type === "string") {
             this.addSchemaImport(inputType.type);
           }
-          result.push(this.generatePrismaStringLine(field, lines.length));
+          result.push(this.generatePrismaStringLine(field, inputType, lines.length));
         }
       }
       return result;
@@ -10769,11 +10769,7 @@ ${this.generateExportSchemaStatement(`${name}`, `z.enum(${JSON.stringify(values)
   addSchemaImport(name) {
     this.schemaImports.add(name);
   }
-  generatePrismaStringLine(field, inputsLength) {
-    if (field.inputTypes.length === 0 || !field.inputTypes[0]) {
-      return "";
-    }
-    const inputType = field.inputTypes[0];
+  generatePrismaStringLine(field, inputType, inputsLength) {
     const isEnum = inputType.location === "enumTypes";
     const inputTypeString = inputType.type;
     const { isModelQueryType, modelName, queryName } = this.checkIsModelQueryType(inputTypeString);
@@ -10899,8 +10895,8 @@ ${this.generateExportSchemaStatement(`${name}`, `z.enum(${JSON.stringify(values)
     };
   }
   resolveModelQuerySchemaName(modelName, queryName) {
-    const modelNameCapitalized = modelName.charAt(0).toUpperCase() + modelName.slice(1);
-    const queryNameCapitalized = queryName.charAt(0).toUpperCase() + queryName.slice(1);
+    const modelNameCapitalized = upperCaseFirst(modelName);
+    const queryNameCapitalized = upperCaseFirst(queryName);
     return `${modelNameCapitalized}${queryNameCapitalized}Schema`;
   }
   wrapWithZodUnion(zodStringFields) {
@@ -11152,7 +11148,7 @@ ${this.generateExportSchemaStatement(`${name}`, `z.enum(${JSON.stringify(values)
   }
 };
 
-// src/zod-helpers/generator-helpers.ts
+// src/zod/generator-helpers.ts
 async function generateZodEnumSchemas(prismaSchemaEnum, modelSchemaEnum) {
   const enumTypes = [
     ...prismaSchemaEnum,
@@ -11192,10 +11188,10 @@ async function generateZodIndex() {
 }
 __name(generateZodIndex, "generateZodIndex");
 
-// src/zod-helpers/helpers.ts
+// src/zod/helpers.ts
 init_esm_shims();
 
-// src/zod-helpers/include-helpers.ts
+// src/zod/include-helpers.ts
 init_esm_shims();
 function addMissingInputObjectTypesForInclude(inputObjectTypes, models, isGenerateSelect) {
   const generatedIncludeInputObjectTypes = generateModelIncludeInputObjectTypes(models, isGenerateSelect);
@@ -11278,7 +11274,7 @@ function generateModelIncludeInputObjectTypes(models, isGenerateSelect) {
 }
 __name(generateModelIncludeInputObjectTypes, "generateModelIncludeInputObjectTypes");
 
-// src/zod-helpers/modelArgs-helpers.ts
+// src/zod/modelArgs-helpers.ts
 init_esm_shims();
 function addMissingInputObjectTypesForModelArgs(inputObjectTypes, models, isGenerateSelect, isGenerateInclude) {
   const modelArgsInputObjectTypes = generateModelArgsInputObjectTypes(models, isGenerateSelect, isGenerateInclude);
@@ -11339,7 +11335,7 @@ function generateModelArgsInputObjectTypes(models, isGenerateSelect, isGenerateI
 }
 __name(generateModelArgsInputObjectTypes, "generateModelArgsInputObjectTypes");
 
-// src/zod-helpers/select-helpers.ts
+// src/zod/select-helpers.ts
 init_esm_shims();
 function addMissingInputObjectTypesForSelect(inputObjectTypes, outputObjectTypes, models) {
   const modelCountOutputTypes = getModelCountOutputTypes(outputObjectTypes);
@@ -11487,7 +11483,7 @@ function generateModelSelectInputObjectTypes(models) {
 }
 __name(generateModelSelectInputObjectTypes, "generateModelSelectInputObjectTypes");
 
-// src/zod-helpers/whereUniqueInput-helpers.ts
+// src/zod/whereUniqueInput-helpers.ts
 init_esm_shims();
 function changeOptionalToRequiredFields(inputObjectTypes) {
   inputObjectTypes.map((item) => {
@@ -11506,7 +11502,7 @@ function changeOptionalToRequiredFields(inputObjectTypes) {
 }
 __name(changeOptionalToRequiredFields, "changeOptionalToRequiredFields");
 
-// src/zod-helpers/helpers.ts
+// src/zod/helpers.ts
 function addMissingZodInputObjectTypes(inputObjectTypes, outputObjectTypes, models, modelOperations, dataSourceProvider, options) {
   if (dataSourceProvider === "mongodb") {
     addMissingInputObjectTypesForMongoDbRawOpsAndQueries(modelOperations, outputObjectTypes, inputObjectTypes);

package/package.json

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