@stryke/prisma-trpc-generator 0.13.48 → 0.13.50

package/dist/zod/model-helpers.cjs

@@ -1 +1,244 @@
-const e=require(`../_virtual/rolldown_runtime.cjs`),t=require(`../packages/string-format/src/lower-case-first.cjs`),n=require(`../utils/get-prisma-internals.cjs`),r=require(`./docs-helpers.cjs`);let i=require(`node:path`);i=e.__toESM(i);let a=require(`ts-morph`);function o(e){let{fields:t}=e;for(let e of t)if(c(e))return!0;return!1}function s(e){let{fields:t}=e;for(let e of t)if(l(e))return!0;return!1}function c(e){let{kind:t,relationName:n}=e;return t===`object`&&!!n}function l(e){return c(e)&&e.isList}function u(e,t){return e.find(({name:e})=>e===t)}const d=(e,t,n=!0)=>t.forEach(t=>e.write(t).conditionalNewLine(n)),f=({modelCase:e,modelSuffix:t,relationModel:n})=>{let r=(n,r=``)=>(e===`camelCase`&&(n=n.slice(0,1).toLowerCase()+n.slice(1)),`${r}${n}${t}`);return{modelName:e=>r(e,n===`default`?`_`:``),relatedModelName:e=>r(n===`default`?e.toString():`Related${e.toString()}`)}},p=e=>{let t=e.replace(/^\\\\\?\\/,``).replace(/\\/g,`/`).replace(/\/{2,}/g,`/`);return t.includes(`/node_modules/`)?t.split(`/node_modules/`).slice(-1)[0]:t.startsWith(`../`)?t:`./${t}`},m=(e,t)=>e.reduce((e,n,r)=>r%t===0?[...e,[n]]:[...e.slice(0,-1),[...e.slice(-1)[0],n]],[]),h=(e,t)=>e.fields.some(e=>e.kind===`object`)&&t.relationModel!==!1,g=async(e,t,r,o)=>{let s=(await n.getPrismaInternals()).parseEnvValue(o.generator.output),{relatedModelName:c}=f(r),l=[{kind:a.StructureKind.ImportDeclaration,namespaceImport:`z`,moduleSpecifier:`zod`}];r.imports&&l.push({kind:a.StructureKind.ImportDeclaration,namespaceImport:`imports`,moduleSpecifier:p(i.default.relative(s,i.default.resolve(i.default.dirname(o.schemaPath),r.imports)))}),r.useDecimalJs&&e.fields.some(e=>e.type===`Decimal`)&&l.push({kind:a.StructureKind.ImportDeclaration,namedImports:[`Decimal`],moduleSpecifier:`decimal.js`});let u=e.fields.filter(e=>e.kind===`enum`),d=e.fields.filter(e=>e.kind===`object`),m=o.otherGenerators.find(e=>e.provider.value===`prisma-client-js`).output.value,h=i.default.relative(s,m);if(u.length>0&&l.push({kind:a.StructureKind.ImportDeclaration,isTypeOnly:u.length===0,moduleSpecifier:p(h),namedImports:u.map(e=>e.type)}),r.relationModel!==!1&&d.length>0){let t=d.filter(t=>t.type!==e.name);t.length>0&&l.push({kind:a.StructureKind.ImportDeclaration,moduleSpecifier:`./index`,namedImports:Array.from(new Set(t.flatMap(e=>[`${e.type}`,c(e.type)])))})}t.addImportDeclarations(l)},_=e=>r.getZodDocElements(e).find(e=>e.startsWith(`custom(`))?.slice(7).slice(0,-1),v=e=>r.getZodDocElements(e).filter(e=>!e.startsWith(`custom(`)),y=(e,t=e=>e.toString())=>{let n=`z.unknown()`,r=[``];if(e.kind===`scalar`)switch(e.type){case`String`:n=`z.string()`;break;case`Int`:n=`z.number()`,r.push(`int()`);break;case`BigInt`:n=`z.bigint()`;break;case`DateTime`:n=`z.date()`;break;case`Float`:n=`z.number()`;break;case`Decimal`:n=`z.number()`;break;case`Json`:n=`jsonSchema`;break;case`Boolean`:n=`z.boolean()`;break;case`Bytes`:n=`z.unknown()`;break}else e.kind===`enum`?n=`z.nativeEnum(${e.type})`:e.kind===`object`&&(n=t(e.type));return e.isList&&r.push(`array()`),e.documentation&&(n=_(e.documentation)??n,r.push(...v(e.documentation))),!e.isRequired&&e.type!==`Json`&&r.push(`nullish()`),`${n}${r.join(`.`)}`},b=(e,t,n)=>{e.fields.some(e=>e.type===`Json`)&&t.addStatements(e=>{e.newLine(),d(e,[`// Helper schema for JSON fields`,`type Literal = boolean | number | string${n.prismaJsonNullability?``:`| null`}`,`type Json = Literal | { [key: string]: Json } | Json[]`,`const literalSchema = z.union([z.string(), z.number(), z.boolean()${n.prismaJsonNullability?``:`, z.null()`}])`,`const jsonSchema: z.ZodSchema<Json> = z.lazy(() => z.union([literalSchema, z.array(jsonSchema), z.record(jsonSchema)]))`])}),n.useDecimalJs&&e.fields.some(e=>e.type===`Decimal`)&&t.addStatements(e=>{e.newLine(),d(e,[`// Helper schema for Decimal fields`,`z`,`.instanceof(Decimal)`,`.or(z.string())`,`.or(z.number())`,`.refine((value) => {`,`  try {`,`    return new Decimal(value);`,`  } catch (error) {`,`    return false;`,`  }`,`})`,`.transform((value) => new Decimal(value));`])})},x=(e,t,n)=>{let{modelName:i}=f(n);t.addVariableStatement({declarationKind:a.VariableDeclarationKind.Const,isExported:!0,leadingTrivia:e=>e.blankLineIfLastNot(),declarations:[{name:i(e.name),initializer(t){t.write(`z.object(`).inlineBlock(()=>{e.fields.filter(e=>e.kind!==`object`).forEach(e=>{d(t,r.getJSDocs(e.documentation)),t.write(`${e.name}: ${y(e)}`).write(`,`).newLine()})}).write(`)`)}}]})},S=(e,t,n)=>{let{modelName:i,relatedModelName:o}=f(n),s=e.fields.filter(e=>e.kind===`object`);t.addInterface({name:`${e.name}`,isExported:!0,extends:[`z.infer<typeof ${i(e.name)}>`],properties:s.map(e=>({hasQuestionToken:!e.isRequired,name:e.name,type:`${e.type}${e.isList?`[]`:``}${e.isRequired?``:` | null`}`}))}),t.addStatements(t=>d(t,[``,`/**`,` * ${o(e.name)} contains all relations on your model in addition to the scalars`,` *`,` * NOTE: Lazy required in case of potential circular dependencies within schema`,` */`])),t.addVariableStatement({declarationKind:a.VariableDeclarationKind.Const,isExported:!0,declarations:[{name:o(e.name),type:`z.ZodSchema<${e.name}>`,initializer(t){t.write(`z.lazy(() => ${i(e.name)}.extend(`).inlineBlock(()=>{s.forEach(e=>{d(t,r.getJSDocs(e.documentation)),t.write(`${e.name}: ${y(e,o)}`).write(`,`).newLine()})}).write(`))`)}}]})},C=async(e,t,n,r)=>{await g(e,t,n,r),b(e,t,n),x(e,t,n),h(e,n)&&S(e,t,n)},w=(e,n)=>{e.forEach(e=>n.addExportDeclaration({moduleSpecifier:`./${t.lowerCaseFirst(e.name)}.schema`}))};exports.checkIsModelRelationField=c,exports.checkModelHasManyModelRelation=s,exports.checkModelHasModelRelation=o,exports.chunk=m,exports.findModelByName=u,exports.generateBarrelFile=w,exports.populateModelFile=C;
+const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
+const require_lower_case_first = require('../packages/string-format/src/lower-case-first.cjs');
+const require_get_prisma_internals = require('../utils/get-prisma-internals.cjs');
+const require_docs_helpers = require('./docs-helpers.cjs');
+let node_path = require("node:path");
+node_path = require_rolldown_runtime.__toESM(node_path);
+let ts_morph = require("ts-morph");
+
+//#region src/zod/model-helpers.ts
+function checkModelHasModelRelation(model) {
+	const { fields: modelFields } = model;
+	for (const modelField of modelFields) if (checkIsModelRelationField(modelField)) return true;
+	return false;
+}
+function checkModelHasManyModelRelation(model) {
+	const { fields: modelFields } = model;
+	for (const modelField of modelFields) if (checkIsManyModelRelationField(modelField)) return true;
+	return false;
+}
+function checkIsModelRelationField(modelField) {
+	const { kind, relationName } = modelField;
+	return kind === "object" && !!relationName;
+}
+function checkIsManyModelRelationField(modelField) {
+	return checkIsModelRelationField(modelField) && modelField.isList;
+}
+function findModelByName(models, modelName) {
+	return models.find(({ name }) => name === modelName);
+}
+const writeArray = (writer, array, newLine = true) => array.forEach((line) => writer.write(line).conditionalNewLine(newLine));
+const useModelNames = ({ modelCase, modelSuffix, relationModel }) => {
+	const formatModelName = (name, prefix = "") => {
+		if (modelCase === "camelCase") name = name.slice(0, 1).toLowerCase() + name.slice(1);
+		return `${prefix}${name}${modelSuffix}`;
+	};
+	return {
+		modelName: (name) => formatModelName(name, relationModel === "default" ? "_" : ""),
+		relatedModelName: (name) => formatModelName(relationModel === "default" ? name.toString() : `Related${name.toString()}`)
+	};
+};
+const dotSlash = (input) => {
+	const converted = input.replace(/^\\\\\?\\/, "").replace(/\\/g, "/").replace(/\/{2,}/g, "/");
+	if (converted.includes(`/node_modules/`)) return converted.split(`/node_modules/`).slice(-1)[0];
+	if (converted.startsWith(`../`)) return converted;
+	return `./${converted}`;
+};
+const chunk = (input, size) => {
+	return input.reduce((arr, item, idx) => {
+		return idx % size === 0 ? [...arr, [item]] : [...arr.slice(0, -1), [...arr.slice(-1)[0], item]];
+	}, []);
+};
+const needsRelatedModel = (model, config) => model.fields.some((field) => field.kind === "object") && config.relationModel !== false;
+const writeImportsForModel = async (model, sourceFile, config, options) => {
+	const outputPath = (await require_get_prisma_internals.getPrismaInternals()).parseEnvValue(options.generator.output);
+	const { relatedModelName } = useModelNames(config);
+	const importList = [{
+		kind: ts_morph.StructureKind.ImportDeclaration,
+		namespaceImport: "z",
+		moduleSpecifier: "zod"
+	}];
+	if (config.imports) importList.push({
+		kind: ts_morph.StructureKind.ImportDeclaration,
+		namespaceImport: "imports",
+		moduleSpecifier: dotSlash(node_path.default.relative(outputPath, node_path.default.resolve(node_path.default.dirname(options.schemaPath), config.imports)))
+	});
+	if (config.useDecimalJs && model.fields.some((f) => f.type === "Decimal")) importList.push({
+		kind: ts_morph.StructureKind.ImportDeclaration,
+		namedImports: ["Decimal"],
+		moduleSpecifier: "decimal.js"
+	});
+	const enumFields = model.fields.filter((f) => f.kind === "enum");
+	const relationFields = model.fields.filter((f) => f.kind === "object");
+	const clientPath = options.otherGenerators.find((each) => each.provider.value === "prisma-client-js").output.value;
+	const relativePath = node_path.default.relative(outputPath, clientPath);
+	if (enumFields.length > 0) importList.push({
+		kind: ts_morph.StructureKind.ImportDeclaration,
+		isTypeOnly: enumFields.length === 0,
+		moduleSpecifier: dotSlash(relativePath),
+		namedImports: enumFields.map((f) => f.type)
+	});
+	if (config.relationModel !== false && relationFields.length > 0) {
+		const filteredFields = relationFields.filter((f) => f.type !== model.name);
+		if (filteredFields.length > 0) importList.push({
+			kind: ts_morph.StructureKind.ImportDeclaration,
+			moduleSpecifier: "./index",
+			namedImports: Array.from(new Set(filteredFields.flatMap((f) => [`${f.type}`, relatedModelName(f.type)])))
+		});
+	}
+	sourceFile.addImportDeclarations(importList);
+};
+const computeCustomSchema = (docString) => {
+	return require_docs_helpers.getZodDocElements(docString).find((modifier) => modifier.startsWith("custom("))?.slice(7).slice(0, -1);
+};
+const computeModifiers = (docString) => {
+	return require_docs_helpers.getZodDocElements(docString).filter((each) => !each.startsWith("custom("));
+};
+const getZodConstructor = (field, getRelatedModelName = (name) => name.toString()) => {
+	let zodType = "z.unknown()";
+	const extraModifiers = [""];
+	if (field.kind === "scalar") switch (field.type) {
+		case "String":
+			zodType = "z.string()";
+			break;
+		case "Int":
+			zodType = "z.number()";
+			extraModifiers.push("int()");
+			break;
+		case "BigInt":
+			zodType = "z.bigint()";
+			break;
+		case "DateTime":
+			zodType = "z.date()";
+			break;
+		case "Float":
+			zodType = "z.number()";
+			break;
+		case "Decimal":
+			zodType = "z.number()";
+			break;
+		case "Json":
+			zodType = "jsonSchema";
+			break;
+		case "Boolean":
+			zodType = "z.boolean()";
+			break;
+		case "Bytes":
+			zodType = "z.unknown()";
+			break;
+	}
+	else if (field.kind === "enum") zodType = `z.nativeEnum(${field.type})`;
+	else if (field.kind === "object") zodType = getRelatedModelName(field.type);
+	if (field.isList) extraModifiers.push("array()");
+	if (field.documentation) {
+		zodType = computeCustomSchema(field.documentation) ?? zodType;
+		extraModifiers.push(...computeModifiers(field.documentation));
+	}
+	if (!field.isRequired && field.type !== "Json") extraModifiers.push("nullish()");
+	return `${zodType}${extraModifiers.join(".")}`;
+};
+const writeTypeSpecificSchemas = (model, sourceFile, config) => {
+	if (model.fields.some((f) => f.type === "Json")) sourceFile.addStatements((writer) => {
+		writer.newLine();
+		writeArray(writer, [
+			"// Helper schema for JSON fields",
+			`type Literal = boolean | number | string${config.prismaJsonNullability ? "" : "| null"}`,
+			"type Json = Literal | { [key: string]: Json } | Json[]",
+			`const literalSchema = z.union([z.string(), z.number(), z.boolean()${config.prismaJsonNullability ? "" : ", z.null()"}])`,
+			"const jsonSchema: z.ZodSchema<Json> = z.lazy(() => z.union([literalSchema, z.array(jsonSchema), z.record(jsonSchema)]))"
+		]);
+	});
+	if (config.useDecimalJs && model.fields.some((f) => f.type === "Decimal")) sourceFile.addStatements((writer) => {
+		writer.newLine();
+		writeArray(writer, [
+			"// Helper schema for Decimal fields",
+			"z",
+			".instanceof(Decimal)",
+			".or(z.string())",
+			".or(z.number())",
+			".refine((value) => {",
+			"  try {",
+			"    return new Decimal(value);",
+			"  } catch (error) {",
+			"    return false;",
+			"  }",
+			"})",
+			".transform((value) => new Decimal(value));"
+		]);
+	});
+};
+const generateSchemaForModel = (model, sourceFile, config) => {
+	const { modelName } = useModelNames(config);
+	sourceFile.addVariableStatement({
+		declarationKind: ts_morph.VariableDeclarationKind.Const,
+		isExported: true,
+		leadingTrivia: (writer) => writer.blankLineIfLastNot(),
+		declarations: [{
+			name: modelName(model.name),
+			initializer(writer) {
+				writer.write("z.object(").inlineBlock(() => {
+					model.fields.filter((f) => f.kind !== "object").forEach((field) => {
+						writeArray(writer, require_docs_helpers.getJSDocs(field.documentation));
+						writer.write(`${field.name}: ${getZodConstructor(field)}`).write(",").newLine();
+					});
+				}).write(")");
+			}
+		}]
+	});
+};
+const generateRelatedSchemaForModel = (model, sourceFile, config) => {
+	const { modelName, relatedModelName } = useModelNames(config);
+	const relationFields = model.fields.filter((f) => f.kind === "object");
+	sourceFile.addInterface({
+		name: `${model.name}`,
+		isExported: true,
+		extends: [`z.infer<typeof ${modelName(model.name)}>`],
+		properties: relationFields.map((f) => ({
+			hasQuestionToken: !f.isRequired,
+			name: f.name,
+			type: `${f.type}${f.isList ? "[]" : ""}${!f.isRequired ? " | null" : ""}`
+		}))
+	});
+	sourceFile.addStatements((writer) => writeArray(writer, [
+		"",
+		"/**",
+		` * ${relatedModelName(model.name)} contains all relations on your model in addition to the scalars`,
+		" *",
+		" * NOTE: Lazy required in case of potential circular dependencies within schema",
+		" */"
+	]));
+	sourceFile.addVariableStatement({
+		declarationKind: ts_morph.VariableDeclarationKind.Const,
+		isExported: true,
+		declarations: [{
+			name: relatedModelName(model.name),
+			type: `z.ZodSchema<${model.name}>`,
+			initializer(writer) {
+				writer.write(`z.lazy(() => ${modelName(model.name)}.extend(`).inlineBlock(() => {
+					relationFields.forEach((field) => {
+						writeArray(writer, require_docs_helpers.getJSDocs(field.documentation));
+						writer.write(`${field.name}: ${getZodConstructor(field, relatedModelName)}`).write(",").newLine();
+					});
+				}).write("))");
+			}
+		}]
+	});
+};
+const populateModelFile = async (model, sourceFile, config, options) => {
+	await writeImportsForModel(model, sourceFile, config, options);
+	writeTypeSpecificSchemas(model, sourceFile, config);
+	generateSchemaForModel(model, sourceFile, config);
+	if (needsRelatedModel(model, config)) generateRelatedSchemaForModel(model, sourceFile, config);
+};
+const generateBarrelFile = (models, indexFile) => {
+	models.forEach((model) => indexFile.addExportDeclaration({ moduleSpecifier: `./${require_lower_case_first.lowerCaseFirst(model.name)}.schema` }));
+};
+
+//#endregion
+exports.checkIsModelRelationField = checkIsModelRelationField;
+exports.checkModelHasManyModelRelation = checkModelHasManyModelRelation;
+exports.checkModelHasModelRelation = checkModelHasModelRelation;
+exports.chunk = chunk;
+exports.findModelByName = findModelByName;
+exports.generateBarrelFile = generateBarrelFile;
+exports.populateModelFile = populateModelFile;

package/dist/zod/model-helpers.mjs

@@ -1,2 +1,237 @@
-import{lowerCaseFirst as e}from"../packages/string-format/src/lower-case-first.mjs";import{getPrismaInternals as t}from"../utils/get-prisma-internals.mjs";import{getJSDocs as n,getZodDocElements as r}from"./docs-helpers.mjs";import i from"node:path";import{StructureKind as a,VariableDeclarationKind as o}from"ts-morph";function s(e){let{fields:t}=e;for(let e of t)if(l(e))return!0;return!1}function c(e){let{fields:t}=e;for(let e of t)if(u(e))return!0;return!1}function l(e){let{kind:t,relationName:n}=e;return t===`object`&&!!n}function u(e){return l(e)&&e.isList}function d(e,t){return e.find(({name:e})=>e===t)}const f=(e,t,n=!0)=>t.forEach(t=>e.write(t).conditionalNewLine(n)),p=({modelCase:e,modelSuffix:t,relationModel:n})=>{let r=(n,r=``)=>(e===`camelCase`&&(n=n.slice(0,1).toLowerCase()+n.slice(1)),`${r}${n}${t}`);return{modelName:e=>r(e,n===`default`?`_`:``),relatedModelName:e=>r(n===`default`?e.toString():`Related${e.toString()}`)}},m=e=>{let t=e.replace(/^\\\\\?\\/,``).replace(/\\/g,`/`).replace(/\/{2,}/g,`/`);return t.includes(`/node_modules/`)?t.split(`/node_modules/`).slice(-1)[0]:t.startsWith(`../`)?t:`./${t}`},h=(e,t)=>e.reduce((e,n,r)=>r%t===0?[...e,[n]]:[...e.slice(0,-1),[...e.slice(-1)[0],n]],[]),g=(e,t)=>e.fields.some(e=>e.kind===`object`)&&t.relationModel!==!1,_=async(e,n,r,o)=>{let s=(await t()).parseEnvValue(o.generator.output),{relatedModelName:c}=p(r),l=[{kind:a.ImportDeclaration,namespaceImport:`z`,moduleSpecifier:`zod`}];r.imports&&l.push({kind:a.ImportDeclaration,namespaceImport:`imports`,moduleSpecifier:m(i.relative(s,i.resolve(i.dirname(o.schemaPath),r.imports)))}),r.useDecimalJs&&e.fields.some(e=>e.type===`Decimal`)&&l.push({kind:a.ImportDeclaration,namedImports:[`Decimal`],moduleSpecifier:`decimal.js`});let u=e.fields.filter(e=>e.kind===`enum`),d=e.fields.filter(e=>e.kind===`object`),f=o.otherGenerators.find(e=>e.provider.value===`prisma-client-js`).output.value,h=i.relative(s,f);if(u.length>0&&l.push({kind:a.ImportDeclaration,isTypeOnly:u.length===0,moduleSpecifier:m(h),namedImports:u.map(e=>e.type)}),r.relationModel!==!1&&d.length>0){let t=d.filter(t=>t.type!==e.name);t.length>0&&l.push({kind:a.ImportDeclaration,moduleSpecifier:`./index`,namedImports:Array.from(new Set(t.flatMap(e=>[`${e.type}`,c(e.type)])))})}n.addImportDeclarations(l)},v=e=>r(e).find(e=>e.startsWith(`custom(`))?.slice(7).slice(0,-1),y=e=>r(e).filter(e=>!e.startsWith(`custom(`)),b=(e,t=e=>e.toString())=>{let n=`z.unknown()`,r=[``];if(e.kind===`scalar`)switch(e.type){case`String`:n=`z.string()`;break;case`Int`:n=`z.number()`,r.push(`int()`);break;case`BigInt`:n=`z.bigint()`;break;case`DateTime`:n=`z.date()`;break;case`Float`:n=`z.number()`;break;case`Decimal`:n=`z.number()`;break;case`Json`:n=`jsonSchema`;break;case`Boolean`:n=`z.boolean()`;break;case`Bytes`:n=`z.unknown()`;break}else e.kind===`enum`?n=`z.nativeEnum(${e.type})`:e.kind===`object`&&(n=t(e.type));return e.isList&&r.push(`array()`),e.documentation&&(n=v(e.documentation)??n,r.push(...y(e.documentation))),!e.isRequired&&e.type!==`Json`&&r.push(`nullish()`),`${n}${r.join(`.`)}`},x=(e,t,n)=>{e.fields.some(e=>e.type===`Json`)&&t.addStatements(e=>{e.newLine(),f(e,[`// Helper schema for JSON fields`,`type Literal = boolean | number | string${n.prismaJsonNullability?``:`| null`}`,`type Json = Literal | { [key: string]: Json } | Json[]`,`const literalSchema = z.union([z.string(), z.number(), z.boolean()${n.prismaJsonNullability?``:`, z.null()`}])`,`const jsonSchema: z.ZodSchema<Json> = z.lazy(() => z.union([literalSchema, z.array(jsonSchema), z.record(jsonSchema)]))`])}),n.useDecimalJs&&e.fields.some(e=>e.type===`Decimal`)&&t.addStatements(e=>{e.newLine(),f(e,[`// Helper schema for Decimal fields`,`z`,`.instanceof(Decimal)`,`.or(z.string())`,`.or(z.number())`,`.refine((value) => {`,`  try {`,`    return new Decimal(value);`,`  } catch (error) {`,`    return false;`,`  }`,`})`,`.transform((value) => new Decimal(value));`])})},S=(e,t,r)=>{let{modelName:i}=p(r);t.addVariableStatement({declarationKind:o.Const,isExported:!0,leadingTrivia:e=>e.blankLineIfLastNot(),declarations:[{name:i(e.name),initializer(t){t.write(`z.object(`).inlineBlock(()=>{e.fields.filter(e=>e.kind!==`object`).forEach(e=>{f(t,n(e.documentation)),t.write(`${e.name}: ${b(e)}`).write(`,`).newLine()})}).write(`)`)}}]})},C=(e,t,r)=>{let{modelName:i,relatedModelName:a}=p(r),s=e.fields.filter(e=>e.kind===`object`);t.addInterface({name:`${e.name}`,isExported:!0,extends:[`z.infer<typeof ${i(e.name)}>`],properties:s.map(e=>({hasQuestionToken:!e.isRequired,name:e.name,type:`${e.type}${e.isList?`[]`:``}${e.isRequired?``:` | null`}`}))}),t.addStatements(t=>f(t,[``,`/**`,` * ${a(e.name)} contains all relations on your model in addition to the scalars`,` *`,` * NOTE: Lazy required in case of potential circular dependencies within schema`,` */`])),t.addVariableStatement({declarationKind:o.Const,isExported:!0,declarations:[{name:a(e.name),type:`z.ZodSchema<${e.name}>`,initializer(t){t.write(`z.lazy(() => ${i(e.name)}.extend(`).inlineBlock(()=>{s.forEach(e=>{f(t,n(e.documentation)),t.write(`${e.name}: ${b(e,a)}`).write(`,`).newLine()})}).write(`))`)}}]})},w=async(e,t,n,r)=>{await _(e,t,n,r),x(e,t,n),S(e,t,n),g(e,n)&&C(e,t,n)},T=(t,n)=>{t.forEach(t=>n.addExportDeclaration({moduleSpecifier:`./${e(t.name)}.schema`}))};export{l as checkIsModelRelationField,c as checkModelHasManyModelRelation,s as checkModelHasModelRelation,h as chunk,d as findModelByName,T as generateBarrelFile,w as populateModelFile};
+import { lowerCaseFirst } from "../packages/string-format/src/lower-case-first.mjs";
+import { getPrismaInternals } from "../utils/get-prisma-internals.mjs";
+import { getJSDocs, getZodDocElements } from "./docs-helpers.mjs";
+import path from "node:path";
+import { StructureKind, VariableDeclarationKind } from "ts-morph";
+
+//#region src/zod/model-helpers.ts
+function checkModelHasModelRelation(model) {
+	const { fields: modelFields } = model;
+	for (const modelField of modelFields) if (checkIsModelRelationField(modelField)) return true;
+	return false;
+}
+function checkModelHasManyModelRelation(model) {
+	const { fields: modelFields } = model;
+	for (const modelField of modelFields) if (checkIsManyModelRelationField(modelField)) return true;
+	return false;
+}
+function checkIsModelRelationField(modelField) {
+	const { kind, relationName } = modelField;
+	return kind === "object" && !!relationName;
+}
+function checkIsManyModelRelationField(modelField) {
+	return checkIsModelRelationField(modelField) && modelField.isList;
+}
+function findModelByName(models, modelName) {
+	return models.find(({ name }) => name === modelName);
+}
+const writeArray = (writer, array, newLine = true) => array.forEach((line) => writer.write(line).conditionalNewLine(newLine));
+const useModelNames = ({ modelCase, modelSuffix, relationModel }) => {
+	const formatModelName = (name, prefix = "") => {
+		if (modelCase === "camelCase") name = name.slice(0, 1).toLowerCase() + name.slice(1);
+		return `${prefix}${name}${modelSuffix}`;
+	};
+	return {
+		modelName: (name) => formatModelName(name, relationModel === "default" ? "_" : ""),
+		relatedModelName: (name) => formatModelName(relationModel === "default" ? name.toString() : `Related${name.toString()}`)
+	};
+};
+const dotSlash = (input) => {
+	const converted = input.replace(/^\\\\\?\\/, "").replace(/\\/g, "/").replace(/\/{2,}/g, "/");
+	if (converted.includes(`/node_modules/`)) return converted.split(`/node_modules/`).slice(-1)[0];
+	if (converted.startsWith(`../`)) return converted;
+	return `./${converted}`;
+};
+const chunk = (input, size) => {
+	return input.reduce((arr, item, idx) => {
+		return idx % size === 0 ? [...arr, [item]] : [...arr.slice(0, -1), [...arr.slice(-1)[0], item]];
+	}, []);
+};
+const needsRelatedModel = (model, config) => model.fields.some((field) => field.kind === "object") && config.relationModel !== false;
+const writeImportsForModel = async (model, sourceFile, config, options) => {
+	const outputPath = (await getPrismaInternals()).parseEnvValue(options.generator.output);
+	const { relatedModelName } = useModelNames(config);
+	const importList = [{
+		kind: StructureKind.ImportDeclaration,
+		namespaceImport: "z",
+		moduleSpecifier: "zod"
+	}];
+	if (config.imports) importList.push({
+		kind: StructureKind.ImportDeclaration,
+		namespaceImport: "imports",
+		moduleSpecifier: dotSlash(path.relative(outputPath, path.resolve(path.dirname(options.schemaPath), config.imports)))
+	});
+	if (config.useDecimalJs && model.fields.some((f) => f.type === "Decimal")) importList.push({
+		kind: StructureKind.ImportDeclaration,
+		namedImports: ["Decimal"],
+		moduleSpecifier: "decimal.js"
+	});
+	const enumFields = model.fields.filter((f) => f.kind === "enum");
+	const relationFields = model.fields.filter((f) => f.kind === "object");
+	const clientPath = options.otherGenerators.find((each) => each.provider.value === "prisma-client-js").output.value;
+	const relativePath = path.relative(outputPath, clientPath);
+	if (enumFields.length > 0) importList.push({
+		kind: StructureKind.ImportDeclaration,
+		isTypeOnly: enumFields.length === 0,
+		moduleSpecifier: dotSlash(relativePath),
+		namedImports: enumFields.map((f) => f.type)
+	});
+	if (config.relationModel !== false && relationFields.length > 0) {
+		const filteredFields = relationFields.filter((f) => f.type !== model.name);
+		if (filteredFields.length > 0) importList.push({
+			kind: StructureKind.ImportDeclaration,
+			moduleSpecifier: "./index",
+			namedImports: Array.from(new Set(filteredFields.flatMap((f) => [`${f.type}`, relatedModelName(f.type)])))
+		});
+	}
+	sourceFile.addImportDeclarations(importList);
+};
+const computeCustomSchema = (docString) => {
+	return getZodDocElements(docString).find((modifier) => modifier.startsWith("custom("))?.slice(7).slice(0, -1);
+};
+const computeModifiers = (docString) => {
+	return getZodDocElements(docString).filter((each) => !each.startsWith("custom("));
+};
+const getZodConstructor = (field, getRelatedModelName = (name) => name.toString()) => {
+	let zodType = "z.unknown()";
+	const extraModifiers = [""];
+	if (field.kind === "scalar") switch (field.type) {
+		case "String":
+			zodType = "z.string()";
+			break;
+		case "Int":
+			zodType = "z.number()";
+			extraModifiers.push("int()");
+			break;
+		case "BigInt":
+			zodType = "z.bigint()";
+			break;
+		case "DateTime":
+			zodType = "z.date()";
+			break;
+		case "Float":
+			zodType = "z.number()";
+			break;
+		case "Decimal":
+			zodType = "z.number()";
+			break;
+		case "Json":
+			zodType = "jsonSchema";
+			break;
+		case "Boolean":
+			zodType = "z.boolean()";
+			break;
+		case "Bytes":
+			zodType = "z.unknown()";
+			break;
+	}
+	else if (field.kind === "enum") zodType = `z.nativeEnum(${field.type})`;
+	else if (field.kind === "object") zodType = getRelatedModelName(field.type);
+	if (field.isList) extraModifiers.push("array()");
+	if (field.documentation) {
+		zodType = computeCustomSchema(field.documentation) ?? zodType;
+		extraModifiers.push(...computeModifiers(field.documentation));
+	}
+	if (!field.isRequired && field.type !== "Json") extraModifiers.push("nullish()");
+	return `${zodType}${extraModifiers.join(".")}`;
+};
+const writeTypeSpecificSchemas = (model, sourceFile, config) => {
+	if (model.fields.some((f) => f.type === "Json")) sourceFile.addStatements((writer) => {
+		writer.newLine();
+		writeArray(writer, [
+			"// Helper schema for JSON fields",
+			`type Literal = boolean | number | string${config.prismaJsonNullability ? "" : "| null"}`,
+			"type Json = Literal | { [key: string]: Json } | Json[]",
+			`const literalSchema = z.union([z.string(), z.number(), z.boolean()${config.prismaJsonNullability ? "" : ", z.null()"}])`,
+			"const jsonSchema: z.ZodSchema<Json> = z.lazy(() => z.union([literalSchema, z.array(jsonSchema), z.record(jsonSchema)]))"
+		]);
+	});
+	if (config.useDecimalJs && model.fields.some((f) => f.type === "Decimal")) sourceFile.addStatements((writer) => {
+		writer.newLine();
+		writeArray(writer, [
+			"// Helper schema for Decimal fields",
+			"z",
+			".instanceof(Decimal)",
+			".or(z.string())",
+			".or(z.number())",
+			".refine((value) => {",
+			"  try {",
+			"    return new Decimal(value);",
+			"  } catch (error) {",
+			"    return false;",
+			"  }",
+			"})",
+			".transform((value) => new Decimal(value));"
+		]);
+	});
+};
+const generateSchemaForModel = (model, sourceFile, config) => {
+	const { modelName } = useModelNames(config);
+	sourceFile.addVariableStatement({
+		declarationKind: VariableDeclarationKind.Const,
+		isExported: true,
+		leadingTrivia: (writer) => writer.blankLineIfLastNot(),
+		declarations: [{
+			name: modelName(model.name),
+			initializer(writer) {
+				writer.write("z.object(").inlineBlock(() => {
+					model.fields.filter((f) => f.kind !== "object").forEach((field) => {
+						writeArray(writer, getJSDocs(field.documentation));
+						writer.write(`${field.name}: ${getZodConstructor(field)}`).write(",").newLine();
+					});
+				}).write(")");
+			}
+		}]
+	});
+};
+const generateRelatedSchemaForModel = (model, sourceFile, config) => {
+	const { modelName, relatedModelName } = useModelNames(config);
+	const relationFields = model.fields.filter((f) => f.kind === "object");
+	sourceFile.addInterface({
+		name: `${model.name}`,
+		isExported: true,
+		extends: [`z.infer<typeof ${modelName(model.name)}>`],
+		properties: relationFields.map((f) => ({
+			hasQuestionToken: !f.isRequired,
+			name: f.name,
+			type: `${f.type}${f.isList ? "[]" : ""}${!f.isRequired ? " | null" : ""}`
+		}))
+	});
+	sourceFile.addStatements((writer) => writeArray(writer, [
+		"",
+		"/**",
+		` * ${relatedModelName(model.name)} contains all relations on your model in addition to the scalars`,
+		" *",
+		" * NOTE: Lazy required in case of potential circular dependencies within schema",
+		" */"
+	]));
+	sourceFile.addVariableStatement({
+		declarationKind: VariableDeclarationKind.Const,
+		isExported: true,
+		declarations: [{
+			name: relatedModelName(model.name),
+			type: `z.ZodSchema<${model.name}>`,
+			initializer(writer) {
+				writer.write(`z.lazy(() => ${modelName(model.name)}.extend(`).inlineBlock(() => {
+					relationFields.forEach((field) => {
+						writeArray(writer, getJSDocs(field.documentation));
+						writer.write(`${field.name}: ${getZodConstructor(field, relatedModelName)}`).write(",").newLine();
+					});
+				}).write("))");
+			}
+		}]
+	});
+};
+const populateModelFile = async (model, sourceFile, config, options) => {
+	await writeImportsForModel(model, sourceFile, config, options);
+	writeTypeSpecificSchemas(model, sourceFile, config);
+	generateSchemaForModel(model, sourceFile, config);
+	if (needsRelatedModel(model, config)) generateRelatedSchemaForModel(model, sourceFile, config);
+};
+const generateBarrelFile = (models, indexFile) => {
+	models.forEach((model) => indexFile.addExportDeclaration({ moduleSpecifier: `./${lowerCaseFirst(model.name)}.schema` }));
+};
+
+//#endregion
+export { checkIsModelRelationField, checkModelHasManyModelRelation, checkModelHasModelRelation, chunk, findModelByName, generateBarrelFile, populateModelFile };
 //# sourceMappingURL=model-helpers.mjs.map

package/dist/zod/model-helpers.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"model-helpers.mjs","names":["importList: ImportDeclarationStructure[]","extraModifiers: string[]"],"sources":["../../src/zod/model-helpers.ts"],"sourcesContent":["/* -------------------------------------------------------------------\n\n                       ⚡ Storm Software - Stryke\n\n This code was released as part of the Stryke project. Stryke\n is maintained by Storm Software under the Apache-2.0 License, and is\n free for commercial and private use. For more information, please visit\n our licensing page.\n\n Website:         https://stormsoftware.com\n Repository:      https://github.com/storm-software/stryke\n Documentation:   https://stormsoftware.com/projects/stryke/docs\n Contact:         https://stormsoftware.com/contact\n License:         https://stormsoftware.com/projects/stryke/license\n\n ------------------------------------------------------------------- */\n\nimport type {\n  DMMF,\n  EnvValue,\n  GeneratorOptions\n} from \"@prisma/generator-helper\";\nimport { lowerCaseFirst } from \"@stryke/string-format/lower-case-first\";\nimport path from \"node:path\";\nimport type {\n  CodeBlockWriter,\n  ImportDeclarationStructure,\n  SourceFile\n} from \"ts-morph\";\nimport { StructureKind, VariableDeclarationKind } from \"ts-morph\";\nimport type { Config } from \"../config\";\nimport { getPrismaInternals } from \"../utils/get-prisma-internals\";\nimport { getJSDocs, getZodDocElements } from \"./docs-helpers\";\n\nexport function checkModelHasModelRelation(model: DMMF.Model) {\n  const { fields: modelFields } = model;\n  for (const modelField of modelFields) {\n    const isRelationField = checkIsModelRelationField(modelField);\n    if (isRelationField) {\n      return true;\n    }\n  }\n  return false;\n}\n\nexport function checkModelHasManyModelRelation(model: DMMF.Model) {\n  const { fields: modelFields } = model;\n  for (const modelField of modelFields) {\n    const isManyRelationField = checkIsManyModelRelationField(modelField);\n    if (isManyRelationField) {\n      return true;\n    }\n  }\n  return false;\n}\n\nexport function checkIsModelRelationField(modelField: DMMF.Field) {\n  const { kind, relationName } = modelField;\n\n  return kind === \"object\" && !!relationName;\n}\n\nexport function checkIsManyModelRelationField(modelField: DMMF.Field) {\n  return checkIsModelRelationField(modelField) && modelField.isList;\n}\n\nexport function findModelByName(models: DMMF.Model[], modelName: string) {\n  return models.find(({ name }) => name === modelName);\n}\n\nexport const writeArray = (\n  writer: CodeBlockWriter,\n  array: string[],\n  newLine = true\n) => array.forEach(line => writer.write(line).conditionalNewLine(newLine));\n\nexport const useModelNames = ({\n  modelCase,\n  modelSuffix,\n  relationModel\n}: Config) => {\n  const formatModelName = (name: string, prefix = \"\") => {\n    if (modelCase === \"camelCase\") {\n      name = name.slice(0, 1).toLowerCase() + name.slice(1);\n    }\n    return `${prefix}${name}${modelSuffix}`;\n  };\n\n  return {\n    modelName: (name: string) =>\n      formatModelName(name, relationModel === \"default\" ? \"_\" : \"\"),\n    relatedModelName: (\n      name: string | DMMF.SchemaEnum | DMMF.OutputType | DMMF.SchemaArg\n    ) =>\n      formatModelName(\n        relationModel === \"default\"\n          ? name.toString()\n          : `Related${name.toString()}`\n      )\n  };\n};\n\nexport const dotSlash = (input: string) => {\n  const converted = input\n    .replace(/^\\\\\\\\\\?\\\\/, \"\")\n    .replace(/\\\\/g, \"/\")\n    .replace(/\\/{2,}/g, \"/\");\n\n  if (converted.includes(`/node_modules/`))\n    return converted.split(`/node_modules/`).slice(-1)[0];\n\n  if (converted.startsWith(`../`)) return converted;\n\n  return `./${converted}`;\n};\n\nexport const chunk = <T extends any[]>(input: T, size: number): T[] => {\n  return input.reduce((arr, item, idx) => {\n    return idx % size === 0\n      ? [...arr, [item]]\n      : [...(arr as T).slice(0, -1), [...(arr as T).slice(-1)[0], item]];\n  }, []);\n};\n\nexport const needsRelatedModel = (model: DMMF.Model, config: Config) =>\n  model.fields.some(field => field.kind === \"object\") &&\n  config.relationModel !== false;\n\nexport const writeImportsForModel = async (\n  model: DMMF.Model,\n  sourceFile: SourceFile,\n  config: Config,\n  options: GeneratorOptions\n) => {\n  const internals = await getPrismaInternals();\n\n  const outputPath = internals.parseEnvValue(\n    options.generator.output as EnvValue\n  );\n\n  const { relatedModelName } = useModelNames(config);\n  const importList: ImportDeclarationStructure[] = [\n    {\n      kind: StructureKind.ImportDeclaration,\n      namespaceImport: \"z\",\n      moduleSpecifier: \"zod\"\n    }\n  ];\n\n  if (config.imports) {\n    importList.push({\n      kind: StructureKind.ImportDeclaration,\n      namespaceImport: \"imports\",\n      moduleSpecifier: dotSlash(\n        path.relative(\n          outputPath,\n          path.resolve(path.dirname(options.schemaPath), config.imports)\n        )\n      )!\n    });\n  }\n\n  if (config.useDecimalJs && model.fields.some(f => f.type === \"Decimal\")) {\n    importList.push({\n      kind: StructureKind.ImportDeclaration,\n      namedImports: [\"Decimal\"],\n      moduleSpecifier: \"decimal.js\"\n    });\n  }\n\n  const enumFields = model.fields.filter(f => f.kind === \"enum\");\n  const relationFields = model.fields.filter(f => f.kind === \"object\");\n\n  const clientPath = options.otherGenerators.find(\n    each => each.provider.value === \"prisma-client-js\"\n  )!.output!.value!;\n\n  const relativePath = path.relative(outputPath, clientPath);\n\n  if (enumFields.length > 0) {\n    importList.push({\n      kind: StructureKind.ImportDeclaration,\n      isTypeOnly: enumFields.length === 0,\n      moduleSpecifier: dotSlash(relativePath)!,\n      namedImports: enumFields.map(f => f.type)\n    });\n  }\n\n  if (config.relationModel !== false && relationFields.length > 0) {\n    const filteredFields = relationFields.filter(f => f.type !== model.name);\n\n    if (filteredFields.length > 0) {\n      importList.push({\n        kind: StructureKind.ImportDeclaration,\n        moduleSpecifier: \"./index\",\n        namedImports: Array.from(\n          new Set(\n            filteredFields.flatMap(f => [`${f.type}`, relatedModelName(f.type)])\n          )\n        )\n      });\n    }\n  }\n\n  sourceFile.addImportDeclarations(importList);\n};\n\nexport const computeCustomSchema = (docString: string) => {\n  return getZodDocElements(docString)\n    .find(modifier => modifier.startsWith(\"custom(\"))\n    ?.slice(7)\n    .slice(0, -1);\n};\n\nexport const computeModifiers = (docString: string) => {\n  return getZodDocElements(docString).filter(\n    each => !each.startsWith(\"custom(\")\n  );\n};\n\nexport const getZodConstructor = (\n  field: DMMF.Field,\n  getRelatedModelName = (\n    name: string | DMMF.SchemaEnum | DMMF.OutputType | DMMF.SchemaArg\n  ) => name.toString()\n) => {\n  let zodType = \"z.unknown()\";\n  const extraModifiers: string[] = [\"\"];\n  if (field.kind === \"scalar\") {\n    switch (field.type) {\n      case \"String\":\n        zodType = \"z.string()\";\n        break;\n      case \"Int\":\n        zodType = \"z.number()\";\n        extraModifiers.push(\"int()\");\n        break;\n      case \"BigInt\":\n        zodType = \"z.bigint()\";\n        break;\n      case \"DateTime\":\n        zodType = \"z.date()\";\n        break;\n      case \"Float\":\n        zodType = \"z.number()\";\n        break;\n      case \"Decimal\":\n        zodType = \"z.number()\";\n        break;\n      case \"Json\":\n        zodType = \"jsonSchema\";\n        break;\n      case \"Boolean\":\n        zodType = \"z.boolean()\";\n        break;\n      // TODO: Proper type for bytes fields\n      case \"Bytes\":\n        zodType = \"z.unknown()\";\n        break;\n    }\n  } else if (field.kind === \"enum\") {\n    zodType = `z.nativeEnum(${field.type})`;\n  } else if (field.kind === \"object\") {\n    zodType = getRelatedModelName(field.type);\n  }\n\n  if (field.isList) extraModifiers.push(\"array()\");\n  if (field.documentation) {\n    zodType = computeCustomSchema(field.documentation) ?? zodType;\n    extraModifiers.push(...computeModifiers(field.documentation));\n  }\n  if (!field.isRequired && field.type !== \"Json\")\n    extraModifiers.push(\"nullish()\");\n  // if (field.hasDefaultValue) extraModifiers.push('optional()')\n\n  return `${zodType}${extraModifiers.join(\".\")}`;\n};\n\nexport const writeTypeSpecificSchemas = (\n  model: DMMF.Model,\n  sourceFile: SourceFile,\n  config: Config\n) => {\n  if (model.fields.some(f => f.type === \"Json\")) {\n    sourceFile.addStatements(writer => {\n      writer.newLine();\n      writeArray(writer, [\n        \"// Helper schema for JSON fields\",\n        `type Literal = boolean | number | string${\n          config.prismaJsonNullability ? \"\" : \"| null\"\n        }`,\n        \"type Json = Literal | { [key: string]: Json } | Json[]\",\n        `const literalSchema = z.union([z.string(), z.number(), z.boolean()${\n          config.prismaJsonNullability ? \"\" : \", z.null()\"\n        }])`,\n        \"const jsonSchema: z.ZodSchema<Json> = z.lazy(() => z.union([literalSchema, z.array(jsonSchema), z.record(jsonSchema)]))\"\n      ]);\n    });\n  }\n\n  if (config.useDecimalJs && model.fields.some(f => f.type === \"Decimal\")) {\n    sourceFile.addStatements(writer => {\n      writer.newLine();\n      writeArray(writer, [\n        \"// Helper schema for Decimal fields\",\n        \"z\",\n        \".instanceof(Decimal)\",\n        \".or(z.string())\",\n        \".or(z.number())\",\n        \".refine((value) => {\",\n        \"  try {\",\n        \"    return new Decimal(value);\",\n        \"  } catch (error) {\",\n        \"    return false;\",\n        \"  }\",\n        \"})\",\n        \".transform((value) => new Decimal(value));\"\n      ]);\n    });\n  }\n};\n\nexport const generateSchemaForModel = (\n  model: DMMF.Model,\n  sourceFile: SourceFile,\n  config: Config\n) => {\n  const { modelName } = useModelNames(config);\n\n  sourceFile.addVariableStatement({\n    declarationKind: VariableDeclarationKind.Const,\n    isExported: true,\n    leadingTrivia: writer => writer.blankLineIfLastNot(),\n    declarations: [\n      {\n        name: modelName(model.name),\n        initializer(writer) {\n          writer\n            .write(\"z.object(\")\n            .inlineBlock(() => {\n              model.fields\n                .filter(f => f.kind !== \"object\")\n                .forEach(field => {\n                  writeArray(writer, getJSDocs(field.documentation));\n                  writer\n                    .write(`${field.name}: ${getZodConstructor(field)}`)\n                    .write(\",\")\n                    .newLine();\n                });\n            })\n            .write(\")\");\n        }\n      }\n    ]\n  });\n};\n\nexport const generateRelatedSchemaForModel = (\n  model: DMMF.Model,\n  sourceFile: SourceFile,\n  config: Config\n) => {\n  const { modelName, relatedModelName } = useModelNames(config);\n\n  const relationFields = model.fields.filter(f => f.kind === \"object\");\n\n  sourceFile.addInterface({\n    name: `${model.name}`,\n    isExported: true,\n    extends: [`z.infer<typeof ${modelName(model.name)}>`],\n    properties: relationFields.map(f => ({\n      hasQuestionToken: !f.isRequired,\n      name: f.name,\n      type: `${f.type}${f.isList ? \"[]\" : \"\"}${!f.isRequired ? \" | null\" : \"\"}`\n    }))\n  });\n\n  sourceFile.addStatements(writer =>\n    writeArray(writer, [\n      \"\",\n      \"/**\",\n      ` * ${relatedModelName(\n        model.name\n      )} contains all relations on your model in addition to the scalars`,\n      \" *\",\n      \" * NOTE: Lazy required in case of potential circular dependencies within schema\",\n      \" */\"\n    ])\n  );\n\n  sourceFile.addVariableStatement({\n    declarationKind: VariableDeclarationKind.Const,\n    isExported: true,\n    declarations: [\n      {\n        name: relatedModelName(model.name),\n        type: `z.ZodSchema<${model.name}>`,\n        initializer(writer) {\n          writer\n            .write(`z.lazy(() => ${modelName(model.name)}.extend(`)\n            .inlineBlock(() => {\n              relationFields.forEach(field => {\n                writeArray(writer, getJSDocs(field.documentation));\n\n                writer\n                  .write(\n                    `${field.name}: ${getZodConstructor(\n                      field,\n                      relatedModelName\n                    )}`\n                  )\n                  .write(\",\")\n                  .newLine();\n              });\n            })\n            .write(\"))\");\n        }\n      }\n    ]\n  });\n};\n\nexport const populateModelFile = async (\n  model: DMMF.Model,\n  sourceFile: SourceFile,\n  config: Config,\n  options: GeneratorOptions\n) => {\n  await writeImportsForModel(model, sourceFile, config, options);\n  writeTypeSpecificSchemas(model, sourceFile, config);\n  generateSchemaForModel(model, sourceFile, config);\n  if (needsRelatedModel(model, config))\n    generateRelatedSchemaForModel(model, sourceFile, config);\n};\n\nexport const generateBarrelFile = (\n  models: DMMF.Model[],\n  indexFile: SourceFile\n) => {\n  models.forEach(model =>\n    indexFile.addExportDeclaration({\n      moduleSpecifier: `./${lowerCaseFirst(model.name)}.schema`\n    })\n  );\n};\n"],"mappings":"gUAkCA,SAAgB,EAA2B,EAAmB,CAC5D,GAAM,CAAE,OAAQ,GAAgB,EAChC,IAAK,IAAM,KAAc,EAEvB,GADwB,EAA0B,EAAW,CAE3D,MAAO,GAGX,MAAO,GAGT,SAAgB,EAA+B,EAAmB,CAChE,GAAM,CAAE,OAAQ,GAAgB,EAChC,IAAK,IAAM,KAAc,EAEvB,GAD4B,EAA8B,EAAW,CAEnE,MAAO,GAGX,MAAO,GAGT,SAAgB,EAA0B,EAAwB,CAChE,GAAM,CAAE,OAAM,gBAAiB,EAE/B,OAAO,IAAS,UAAY,CAAC,CAAC,EAGhC,SAAgB,EAA8B,EAAwB,CACpE,OAAO,EAA0B,EAAW,EAAI,EAAW,OAG7D,SAAgB,EAAgB,EAAsB,EAAmB,CACvE,OAAO,EAAO,MAAM,CAAE,UAAW,IAAS,EAAU,CAGtD,MAAa,GACX,EACA,EACA,EAAU,KACP,EAAM,QAAQ,GAAQ,EAAO,MAAM,EAAK,CAAC,mBAAmB,EAAQ,CAAC,CAE7D,GAAiB,CAC5B,YACA,cACA,mBACY,CACZ,IAAM,GAAmB,EAAc,EAAS,MAC1C,IAAc,cAChB,EAAO,EAAK,MAAM,EAAG,EAAE,CAAC,aAAa,CAAG,EAAK,MAAM,EAAE,EAEhD,GAAG,IAAS,IAAO,KAG5B,MAAO,CACL,UAAY,GACV,EAAgB,EAAM,IAAkB,UAAY,IAAM,GAAG,CAC/D,iBACE,GAEA,EACE,IAAkB,UACd,EAAK,UAAU,CACf,UAAU,EAAK,UAAU,GAC9B,CACJ,EAGU,EAAY,GAAkB,CACzC,IAAM,EAAY,EACf,QAAQ,YAAa,GAAG,CACxB,QAAQ,MAAO,IAAI,CACnB,QAAQ,UAAW,IAAI,CAO1B,OALI,EAAU,SAAS,iBAAiB,CAC/B,EAAU,MAAM,iBAAiB,CAAC,MAAM,GAAG,CAAC,GAEjD,EAAU,WAAW,MAAM,CAAS,EAEjC,KAAK,KAGD,GAA0B,EAAU,IACxC,EAAM,QAAQ,EAAK,EAAM,IACvB,EAAM,IAAS,EAClB,CAAC,GAAG,EAAK,CAAC,EAAK,CAAC,CAChB,CAAC,GAAI,EAAU,MAAM,EAAG,GAAG,CAAE,CAAC,GAAI,EAAU,MAAM,GAAG,CAAC,GAAI,EAAK,CAAC,CACnE,EAAE,CAAC,CAGK,GAAqB,EAAmB,IACnD,EAAM,OAAO,KAAK,GAAS,EAAM,OAAS,SAAS,EACnD,EAAO,gBAAkB,GAEd,EAAuB,MAClC,EACA,EACA,EACA,IACG,CAGH,IAAM,GAFY,MAAM,GAAoB,EAEf,cAC3B,EAAQ,UAAU,OACnB,CAEK,CAAE,oBAAqB,EAAc,EAAO,CAC5CA,EAA2C,CAC/C,CACE,KAAM,EAAc,kBACpB,gBAAiB,IACjB,gBAAiB,MAClB,CACF,CAEG,EAAO,SACT,EAAW,KAAK,CACd,KAAM,EAAc,kBACpB,gBAAiB,UACjB,gBAAiB,EACf,EAAK,SACH,EACA,EAAK,QAAQ,EAAK,QAAQ,EAAQ,WAAW,CAAE,EAAO,QAAQ,CAC/D,CACF,CACF,CAAC,CAGA,EAAO,cAAgB,EAAM,OAAO,KAAK,GAAK,EAAE,OAAS,UAAU,EACrE,EAAW,KAAK,CACd,KAAM,EAAc,kBACpB,aAAc,CAAC,UAAU,CACzB,gBAAiB,aAClB,CAAC,CAGJ,IAAM,EAAa,EAAM,OAAO,OAAO,GAAK,EAAE,OAAS,OAAO,CACxD,EAAiB,EAAM,OAAO,OAAO,GAAK,EAAE,OAAS,SAAS,CAE9D,EAAa,EAAQ,gBAAgB,KACzC,GAAQ,EAAK,SAAS,QAAU,mBACjC,CAAE,OAAQ,MAEL,EAAe,EAAK,SAAS,EAAY,EAAW,CAW1D,GATI,EAAW,OAAS,GACtB,EAAW,KAAK,CACd,KAAM,EAAc,kBACpB,WAAY,EAAW,SAAW,EAClC,gBAAiB,EAAS,EAAa,CACvC,aAAc,EAAW,IAAI,GAAK,EAAE,KAAK,CAC1C,CAAC,CAGA,EAAO,gBAAkB,IAAS,EAAe,OAAS,EAAG,CAC/D,IAAM,EAAiB,EAAe,OAAO,GAAK,EAAE,OAAS,EAAM,KAAK,CAEpE,EAAe,OAAS,GAC1B,EAAW,KAAK,CACd,KAAM,EAAc,kBACpB,gBAAiB,UACjB,aAAc,MAAM,KAClB,IAAI,IACF,EAAe,QAAQ,GAAK,CAAC,GAAG,EAAE,OAAQ,EAAiB,EAAE,KAAK,CAAC,CAAC,CACrE,CACF,CACF,CAAC,CAIN,EAAW,sBAAsB,EAAW,EAGjC,EAAuB,GAC3B,EAAkB,EAAU,CAChC,KAAK,GAAY,EAAS,WAAW,UAAU,CAAC,EAC/C,MAAM,EAAE,CACT,MAAM,EAAG,GAAG,CAGJ,EAAoB,GACxB,EAAkB,EAAU,CAAC,OAClC,GAAQ,CAAC,EAAK,WAAW,UAAU,CACpC,CAGU,GACX,EACA,EACE,GACG,EAAK,UAAU,GACjB,CACH,IAAI,EAAU,cACRC,EAA2B,CAAC,GAAG,CACrC,GAAI,EAAM,OAAS,SACjB,OAAQ,EAAM,KAAd,CACE,IAAK,SACH,EAAU,aACV,MACF,IAAK,MACH,EAAU,aACV,EAAe,KAAK,QAAQ,CAC5B,MACF,IAAK,SACH,EAAU,aACV,MACF,IAAK,WACH,EAAU,WACV,MACF,IAAK,QACH,EAAU,aACV,MACF,IAAK,UACH,EAAU,aACV,MACF,IAAK,OACH,EAAU,aACV,MACF,IAAK,UACH,EAAU,cACV,MAEF,IAAK,QACH,EAAU,cACV,WAEK,EAAM,OAAS,OACxB,EAAU,gBAAgB,EAAM,KAAK,GAC5B,EAAM,OAAS,WACxB,EAAU,EAAoB,EAAM,KAAK,EAY3C,OATI,EAAM,QAAQ,EAAe,KAAK,UAAU,CAC5C,EAAM,gBACR,EAAU,EAAoB,EAAM,cAAc,EAAI,EACtD,EAAe,KAAK,GAAG,EAAiB,EAAM,cAAc,CAAC,EAE3D,CAAC,EAAM,YAAc,EAAM,OAAS,QACtC,EAAe,KAAK,YAAY,CAG3B,GAAG,IAAU,EAAe,KAAK,IAAI,IAGjC,GACX,EACA,EACA,IACG,CACC,EAAM,OAAO,KAAK,GAAK,EAAE,OAAS,OAAO,EAC3C,EAAW,cAAc,GAAU,CACjC,EAAO,SAAS,CAChB,EAAW,EAAQ,CACjB,mCACA,2CACE,EAAO,sBAAwB,GAAK,WAEtC,yDACA,qEACE,EAAO,sBAAwB,GAAK,aACrC,IACD,0HACD,CAAC,EACF,CAGA,EAAO,cAAgB,EAAM,OAAO,KAAK,GAAK,EAAE,OAAS,UAAU,EACrE,EAAW,cAAc,GAAU,CACjC,EAAO,SAAS,CAChB,EAAW,EAAQ,CACjB,sCACA,IACA,uBACA,kBACA,kBACA,uBACA,UACA,iCACA,sBACA,oBACA,MACA,KACA,6CACD,CAAC,EACF,EAIO,GACX,EACA,EACA,IACG,CACH,GAAM,CAAE,aAAc,EAAc,EAAO,CAE3C,EAAW,qBAAqB,CAC9B,gBAAiB,EAAwB,MACzC,WAAY,GACZ,cAAe,GAAU,EAAO,oBAAoB,CACpD,aAAc,CACZ,CACE,KAAM,EAAU,EAAM,KAAK,CAC3B,YAAY,EAAQ,CAClB,EACG,MAAM,YAAY,CAClB,gBAAkB,CACjB,EAAM,OACH,OAAO,GAAK,EAAE,OAAS,SAAS,CAChC,QAAQ,GAAS,CAChB,EAAW,EAAQ,EAAU,EAAM,cAAc,CAAC,CAClD,EACG,MAAM,GAAG,EAAM,KAAK,IAAI,EAAkB,EAAM,GAAG,CACnD,MAAM,IAAI,CACV,SAAS,EACZ,EACJ,CACD,MAAM,IAAI,EAEhB,CACF,CACF,CAAC,EAGS,GACX,EACA,EACA,IACG,CACH,GAAM,CAAE,YAAW,oBAAqB,EAAc,EAAO,CAEvD,EAAiB,EAAM,OAAO,OAAO,GAAK,EAAE,OAAS,SAAS,CAEpE,EAAW,aAAa,CACtB,KAAM,GAAG,EAAM,OACf,WAAY,GACZ,QAAS,CAAC,kBAAkB,EAAU,EAAM,KAAK,CAAC,GAAG,CACrD,WAAY,EAAe,IAAI,IAAM,CACnC,iBAAkB,CAAC,EAAE,WACrB,KAAM,EAAE,KACR,KAAM,GAAG,EAAE,OAAO,EAAE,OAAS,KAAO,KAAM,EAAE,WAAyB,GAAZ,YAC1D,EAAE,CACJ,CAAC,CAEF,EAAW,cAAc,GACvB,EAAW,EAAQ,CACjB,GACA,MACA,MAAM,EACJ,EAAM,KACP,CAAC,kEACF,KACA,kFACA,MACD,CAAC,CACH,CAED,EAAW,qBAAqB,CAC9B,gBAAiB,EAAwB,MACzC,WAAY,GACZ,aAAc,CACZ,CACE,KAAM,EAAiB,EAAM,KAAK,CAClC,KAAM,eAAe,EAAM,KAAK,GAChC,YAAY,EAAQ,CAClB,EACG,MAAM,gBAAgB,EAAU,EAAM,KAAK,CAAC,UAAU,CACtD,gBAAkB,CACjB,EAAe,QAAQ,GAAS,CAC9B,EAAW,EAAQ,EAAU,EAAM,cAAc,CAAC,CAElD,EACG,MACC,GAAG,EAAM,KAAK,IAAI,EAChB,EACA,EACD,GACF,CACA,MAAM,IAAI,CACV,SAAS,EACZ,EACF,CACD,MAAM,KAAK,EAEjB,CACF,CACF,CAAC,EAGS,EAAoB,MAC/B,EACA,EACA,EACA,IACG,CACH,MAAM,EAAqB,EAAO,EAAY,EAAQ,EAAQ,CAC9D,EAAyB,EAAO,EAAY,EAAO,CACnD,EAAuB,EAAO,EAAY,EAAO,CAC7C,EAAkB,EAAO,EAAO,EAClC,EAA8B,EAAO,EAAY,EAAO,EAG/C,GACX,EACA,IACG,CACH,EAAO,QAAQ,GACb,EAAU,qBAAqB,CAC7B,gBAAiB,KAAK,EAAe,EAAM,KAAK,CAAC,SAClD,CAAC,CACH"}
+{"version":3,"file":"model-helpers.mjs","names":["importList: ImportDeclarationStructure[]","extraModifiers: string[]"],"sources":["../../src/zod/model-helpers.ts"],"sourcesContent":["/* -------------------------------------------------------------------\n\n                       ⚡ Storm Software - Stryke\n\n This code was released as part of the Stryke project. Stryke\n is maintained by Storm Software under the Apache-2.0 License, and is\n free for commercial and private use. For more information, please visit\n our licensing page.\n\n Website:         https://stormsoftware.com\n Repository:      https://github.com/storm-software/stryke\n Documentation:   https://stormsoftware.com/projects/stryke/docs\n Contact:         https://stormsoftware.com/contact\n License:         https://stormsoftware.com/projects/stryke/license\n\n ------------------------------------------------------------------- */\n\nimport type {\n  DMMF,\n  EnvValue,\n  GeneratorOptions\n} from \"@prisma/generator-helper\";\nimport { lowerCaseFirst } from \"@stryke/string-format/lower-case-first\";\nimport path from \"node:path\";\nimport type {\n  CodeBlockWriter,\n  ImportDeclarationStructure,\n  SourceFile\n} from \"ts-morph\";\nimport { StructureKind, VariableDeclarationKind } from \"ts-morph\";\nimport type { Config } from \"../config\";\nimport { getPrismaInternals } from \"../utils/get-prisma-internals\";\nimport { getJSDocs, getZodDocElements } from \"./docs-helpers\";\n\nexport function checkModelHasModelRelation(model: DMMF.Model) {\n  const { fields: modelFields } = model;\n  for (const modelField of modelFields) {\n    const isRelationField = checkIsModelRelationField(modelField);\n    if (isRelationField) {\n      return true;\n    }\n  }\n  return false;\n}\n\nexport function checkModelHasManyModelRelation(model: DMMF.Model) {\n  const { fields: modelFields } = model;\n  for (const modelField of modelFields) {\n    const isManyRelationField = checkIsManyModelRelationField(modelField);\n    if (isManyRelationField) {\n      return true;\n    }\n  }\n  return false;\n}\n\nexport function checkIsModelRelationField(modelField: DMMF.Field) {\n  const { kind, relationName } = modelField;\n\n  return kind === \"object\" && !!relationName;\n}\n\nexport function checkIsManyModelRelationField(modelField: DMMF.Field) {\n  return checkIsModelRelationField(modelField) && modelField.isList;\n}\n\nexport function findModelByName(models: DMMF.Model[], modelName: string) {\n  return models.find(({ name }) => name === modelName);\n}\n\nexport const writeArray = (\n  writer: CodeBlockWriter,\n  array: string[],\n  newLine = true\n) => array.forEach(line => writer.write(line).conditionalNewLine(newLine));\n\nexport const useModelNames = ({\n  modelCase,\n  modelSuffix,\n  relationModel\n}: Config) => {\n  const formatModelName = (name: string, prefix = \"\") => {\n    if (modelCase === \"camelCase\") {\n      name = name.slice(0, 1).toLowerCase() + name.slice(1);\n    }\n    return `${prefix}${name}${modelSuffix}`;\n  };\n\n  return {\n    modelName: (name: string) =>\n      formatModelName(name, relationModel === \"default\" ? \"_\" : \"\"),\n    relatedModelName: (\n      name: string | DMMF.SchemaEnum | DMMF.OutputType | DMMF.SchemaArg\n    ) =>\n      formatModelName(\n        relationModel === \"default\"\n          ? name.toString()\n          : `Related${name.toString()}`\n      )\n  };\n};\n\nexport const dotSlash = (input: string) => {\n  const converted = input\n    .replace(/^\\\\\\\\\\?\\\\/, \"\")\n    .replace(/\\\\/g, \"/\")\n    .replace(/\\/{2,}/g, \"/\");\n\n  if (converted.includes(`/node_modules/`))\n    return converted.split(`/node_modules/`).slice(-1)[0];\n\n  if (converted.startsWith(`../`)) return converted;\n\n  return `./${converted}`;\n};\n\nexport const chunk = <T extends any[]>(input: T, size: number): T[] => {\n  return input.reduce((arr, item, idx) => {\n    return idx % size === 0\n      ? [...arr, [item]]\n      : [...(arr as T).slice(0, -1), [...(arr as T).slice(-1)[0], item]];\n  }, []);\n};\n\nexport const needsRelatedModel = (model: DMMF.Model, config: Config) =>\n  model.fields.some(field => field.kind === \"object\") &&\n  config.relationModel !== false;\n\nexport const writeImportsForModel = async (\n  model: DMMF.Model,\n  sourceFile: SourceFile,\n  config: Config,\n  options: GeneratorOptions\n) => {\n  const internals = await getPrismaInternals();\n\n  const outputPath = internals.parseEnvValue(\n    options.generator.output as EnvValue\n  );\n\n  const { relatedModelName } = useModelNames(config);\n  const importList: ImportDeclarationStructure[] = [\n    {\n      kind: StructureKind.ImportDeclaration,\n      namespaceImport: \"z\",\n      moduleSpecifier: \"zod\"\n    }\n  ];\n\n  if (config.imports) {\n    importList.push({\n      kind: StructureKind.ImportDeclaration,\n      namespaceImport: \"imports\",\n      moduleSpecifier: dotSlash(\n        path.relative(\n          outputPath,\n          path.resolve(path.dirname(options.schemaPath), config.imports)\n        )\n      )!\n    });\n  }\n\n  if (config.useDecimalJs && model.fields.some(f => f.type === \"Decimal\")) {\n    importList.push({\n      kind: StructureKind.ImportDeclaration,\n      namedImports: [\"Decimal\"],\n      moduleSpecifier: \"decimal.js\"\n    });\n  }\n\n  const enumFields = model.fields.filter(f => f.kind === \"enum\");\n  const relationFields = model.fields.filter(f => f.kind === \"object\");\n\n  const clientPath = options.otherGenerators.find(\n    each => each.provider.value === \"prisma-client-js\"\n  )!.output!.value!;\n\n  const relativePath = path.relative(outputPath, clientPath);\n\n  if (enumFields.length > 0) {\n    importList.push({\n      kind: StructureKind.ImportDeclaration,\n      isTypeOnly: enumFields.length === 0,\n      moduleSpecifier: dotSlash(relativePath)!,\n      namedImports: enumFields.map(f => f.type)\n    });\n  }\n\n  if (config.relationModel !== false && relationFields.length > 0) {\n    const filteredFields = relationFields.filter(f => f.type !== model.name);\n\n    if (filteredFields.length > 0) {\n      importList.push({\n        kind: StructureKind.ImportDeclaration,\n        moduleSpecifier: \"./index\",\n        namedImports: Array.from(\n          new Set(\n            filteredFields.flatMap(f => [`${f.type}`, relatedModelName(f.type)])\n          )\n        )\n      });\n    }\n  }\n\n  sourceFile.addImportDeclarations(importList);\n};\n\nexport const computeCustomSchema = (docString: string) => {\n  return getZodDocElements(docString)\n    .find(modifier => modifier.startsWith(\"custom(\"))\n    ?.slice(7)\n    .slice(0, -1);\n};\n\nexport const computeModifiers = (docString: string) => {\n  return getZodDocElements(docString).filter(\n    each => !each.startsWith(\"custom(\")\n  );\n};\n\nexport const getZodConstructor = (\n  field: DMMF.Field,\n  getRelatedModelName = (\n    name: string | DMMF.SchemaEnum | DMMF.OutputType | DMMF.SchemaArg\n  ) => name.toString()\n) => {\n  let zodType = \"z.unknown()\";\n  const extraModifiers: string[] = [\"\"];\n  if (field.kind === \"scalar\") {\n    switch (field.type) {\n      case \"String\":\n        zodType = \"z.string()\";\n        break;\n      case \"Int\":\n        zodType = \"z.number()\";\n        extraModifiers.push(\"int()\");\n        break;\n      case \"BigInt\":\n        zodType = \"z.bigint()\";\n        break;\n      case \"DateTime\":\n        zodType = \"z.date()\";\n        break;\n      case \"Float\":\n        zodType = \"z.number()\";\n        break;\n      case \"Decimal\":\n        zodType = \"z.number()\";\n        break;\n      case \"Json\":\n        zodType = \"jsonSchema\";\n        break;\n      case \"Boolean\":\n        zodType = \"z.boolean()\";\n        break;\n      // TODO: Proper type for bytes fields\n      case \"Bytes\":\n        zodType = \"z.unknown()\";\n        break;\n    }\n  } else if (field.kind === \"enum\") {\n    zodType = `z.nativeEnum(${field.type})`;\n  } else if (field.kind === \"object\") {\n    zodType = getRelatedModelName(field.type);\n  }\n\n  if (field.isList) extraModifiers.push(\"array()\");\n  if (field.documentation) {\n    zodType = computeCustomSchema(field.documentation) ?? zodType;\n    extraModifiers.push(...computeModifiers(field.documentation));\n  }\n  if (!field.isRequired && field.type !== \"Json\")\n    extraModifiers.push(\"nullish()\");\n  // if (field.hasDefaultValue) extraModifiers.push('optional()')\n\n  return `${zodType}${extraModifiers.join(\".\")}`;\n};\n\nexport const writeTypeSpecificSchemas = (\n  model: DMMF.Model,\n  sourceFile: SourceFile,\n  config: Config\n) => {\n  if (model.fields.some(f => f.type === \"Json\")) {\n    sourceFile.addStatements(writer => {\n      writer.newLine();\n      writeArray(writer, [\n        \"// Helper schema for JSON fields\",\n        `type Literal = boolean | number | string${\n          config.prismaJsonNullability ? \"\" : \"| null\"\n        }`,\n        \"type Json = Literal | { [key: string]: Json } | Json[]\",\n        `const literalSchema = z.union([z.string(), z.number(), z.boolean()${\n          config.prismaJsonNullability ? \"\" : \", z.null()\"\n        }])`,\n        \"const jsonSchema: z.ZodSchema<Json> = z.lazy(() => z.union([literalSchema, z.array(jsonSchema), z.record(jsonSchema)]))\"\n      ]);\n    });\n  }\n\n  if (config.useDecimalJs && model.fields.some(f => f.type === \"Decimal\")) {\n    sourceFile.addStatements(writer => {\n      writer.newLine();\n      writeArray(writer, [\n        \"// Helper schema for Decimal fields\",\n        \"z\",\n        \".instanceof(Decimal)\",\n        \".or(z.string())\",\n        \".or(z.number())\",\n        \".refine((value) => {\",\n        \"  try {\",\n        \"    return new Decimal(value);\",\n        \"  } catch (error) {\",\n        \"    return false;\",\n        \"  }\",\n        \"})\",\n        \".transform((value) => new Decimal(value));\"\n      ]);\n    });\n  }\n};\n\nexport const generateSchemaForModel = (\n  model: DMMF.Model,\n  sourceFile: SourceFile,\n  config: Config\n) => {\n  const { modelName } = useModelNames(config);\n\n  sourceFile.addVariableStatement({\n    declarationKind: VariableDeclarationKind.Const,\n    isExported: true,\n    leadingTrivia: writer => writer.blankLineIfLastNot(),\n    declarations: [\n      {\n        name: modelName(model.name),\n        initializer(writer) {\n          writer\n            .write(\"z.object(\")\n            .inlineBlock(() => {\n              model.fields\n                .filter(f => f.kind !== \"object\")\n                .forEach(field => {\n                  writeArray(writer, getJSDocs(field.documentation));\n                  writer\n                    .write(`${field.name}: ${getZodConstructor(field)}`)\n                    .write(\",\")\n                    .newLine();\n                });\n            })\n            .write(\")\");\n        }\n      }\n    ]\n  });\n};\n\nexport const generateRelatedSchemaForModel = (\n  model: DMMF.Model,\n  sourceFile: SourceFile,\n  config: Config\n) => {\n  const { modelName, relatedModelName } = useModelNames(config);\n\n  const relationFields = model.fields.filter(f => f.kind === \"object\");\n\n  sourceFile.addInterface({\n    name: `${model.name}`,\n    isExported: true,\n    extends: [`z.infer<typeof ${modelName(model.name)}>`],\n    properties: relationFields.map(f => ({\n      hasQuestionToken: !f.isRequired,\n      name: f.name,\n      type: `${f.type}${f.isList ? \"[]\" : \"\"}${!f.isRequired ? \" | null\" : \"\"}`\n    }))\n  });\n\n  sourceFile.addStatements(writer =>\n    writeArray(writer, [\n      \"\",\n      \"/**\",\n      ` * ${relatedModelName(\n        model.name\n      )} contains all relations on your model in addition to the scalars`,\n      \" *\",\n      \" * NOTE: Lazy required in case of potential circular dependencies within schema\",\n      \" */\"\n    ])\n  );\n\n  sourceFile.addVariableStatement({\n    declarationKind: VariableDeclarationKind.Const,\n    isExported: true,\n    declarations: [\n      {\n        name: relatedModelName(model.name),\n        type: `z.ZodSchema<${model.name}>`,\n        initializer(writer) {\n          writer\n            .write(`z.lazy(() => ${modelName(model.name)}.extend(`)\n            .inlineBlock(() => {\n              relationFields.forEach(field => {\n                writeArray(writer, getJSDocs(field.documentation));\n\n                writer\n                  .write(\n                    `${field.name}: ${getZodConstructor(\n                      field,\n                      relatedModelName\n                    )}`\n                  )\n                  .write(\",\")\n                  .newLine();\n              });\n            })\n            .write(\"))\");\n        }\n      }\n    ]\n  });\n};\n\nexport const populateModelFile = async (\n  model: DMMF.Model,\n  sourceFile: SourceFile,\n  config: Config,\n  options: GeneratorOptions\n) => {\n  await writeImportsForModel(model, sourceFile, config, options);\n  writeTypeSpecificSchemas(model, sourceFile, config);\n  generateSchemaForModel(model, sourceFile, config);\n  if (needsRelatedModel(model, config))\n    generateRelatedSchemaForModel(model, sourceFile, config);\n};\n\nexport const generateBarrelFile = (\n  models: DMMF.Model[],\n  indexFile: SourceFile\n) => {\n  models.forEach(model =>\n    indexFile.addExportDeclaration({\n      moduleSpecifier: `./${lowerCaseFirst(model.name)}.schema`\n    })\n  );\n};\n"],"mappings":";;;;;;;AAkCA,SAAgB,2BAA2B,OAAmB;CAC5D,MAAM,EAAE,QAAQ,gBAAgB;AAChC,MAAK,MAAM,cAAc,YAEvB,KADwB,0BAA0B,WAAW,CAE3D,QAAO;AAGX,QAAO;;AAGT,SAAgB,+BAA+B,OAAmB;CAChE,MAAM,EAAE,QAAQ,gBAAgB;AAChC,MAAK,MAAM,cAAc,YAEvB,KAD4B,8BAA8B,WAAW,CAEnE,QAAO;AAGX,QAAO;;AAGT,SAAgB,0BAA0B,YAAwB;CAChE,MAAM,EAAE,MAAM,iBAAiB;AAE/B,QAAO,SAAS,YAAY,CAAC,CAAC;;AAGhC,SAAgB,8BAA8B,YAAwB;AACpE,QAAO,0BAA0B,WAAW,IAAI,WAAW;;AAG7D,SAAgB,gBAAgB,QAAsB,WAAmB;AACvE,QAAO,OAAO,MAAM,EAAE,WAAW,SAAS,UAAU;;AAGtD,MAAa,cACX,QACA,OACA,UAAU,SACP,MAAM,SAAQ,SAAQ,OAAO,MAAM,KAAK,CAAC,mBAAmB,QAAQ,CAAC;AAE1E,MAAa,iBAAiB,EAC5B,WACA,aACA,oBACY;CACZ,MAAM,mBAAmB,MAAc,SAAS,OAAO;AACrD,MAAI,cAAc,YAChB,QAAO,KAAK,MAAM,GAAG,EAAE,CAAC,aAAa,GAAG,KAAK,MAAM,EAAE;AAEvD,SAAO,GAAG,SAAS,OAAO;;AAG5B,QAAO;EACL,YAAY,SACV,gBAAgB,MAAM,kBAAkB,YAAY,MAAM,GAAG;EAC/D,mBACE,SAEA,gBACE,kBAAkB,YACd,KAAK,UAAU,GACf,UAAU,KAAK,UAAU,GAC9B;EACJ;;AAGH,MAAa,YAAY,UAAkB;CACzC,MAAM,YAAY,MACf,QAAQ,aAAa,GAAG,CACxB,QAAQ,OAAO,IAAI,CACnB,QAAQ,WAAW,IAAI;AAE1B,KAAI,UAAU,SAAS,iBAAiB,CACtC,QAAO,UAAU,MAAM,iBAAiB,CAAC,MAAM,GAAG,CAAC;AAErD,KAAI,UAAU,WAAW,MAAM,CAAE,QAAO;AAExC,QAAO,KAAK;;AAGd,MAAa,SAA0B,OAAU,SAAsB;AACrE,QAAO,MAAM,QAAQ,KAAK,MAAM,QAAQ;AACtC,SAAO,MAAM,SAAS,IAClB,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,GAChB,CAAC,GAAI,IAAU,MAAM,GAAG,GAAG,EAAE,CAAC,GAAI,IAAU,MAAM,GAAG,CAAC,IAAI,KAAK,CAAC;IACnE,EAAE,CAAC;;AAGR,MAAa,qBAAqB,OAAmB,WACnD,MAAM,OAAO,MAAK,UAAS,MAAM,SAAS,SAAS,IACnD,OAAO,kBAAkB;AAE3B,MAAa,uBAAuB,OAClC,OACA,YACA,QACA,YACG;CAGH,MAAM,cAFY,MAAM,oBAAoB,EAEf,cAC3B,QAAQ,UAAU,OACnB;CAED,MAAM,EAAE,qBAAqB,cAAc,OAAO;CAClD,MAAMA,aAA2C,CAC/C;EACE,MAAM,cAAc;EACpB,iBAAiB;EACjB,iBAAiB;EAClB,CACF;AAED,KAAI,OAAO,QACT,YAAW,KAAK;EACd,MAAM,cAAc;EACpB,iBAAiB;EACjB,iBAAiB,SACf,KAAK,SACH,YACA,KAAK,QAAQ,KAAK,QAAQ,QAAQ,WAAW,EAAE,OAAO,QAAQ,CAC/D,CACF;EACF,CAAC;AAGJ,KAAI,OAAO,gBAAgB,MAAM,OAAO,MAAK,MAAK,EAAE,SAAS,UAAU,CACrE,YAAW,KAAK;EACd,MAAM,cAAc;EACpB,cAAc,CAAC,UAAU;EACzB,iBAAiB;EAClB,CAAC;CAGJ,MAAM,aAAa,MAAM,OAAO,QAAO,MAAK,EAAE,SAAS,OAAO;CAC9D,MAAM,iBAAiB,MAAM,OAAO,QAAO,MAAK,EAAE,SAAS,SAAS;CAEpE,MAAM,aAAa,QAAQ,gBAAgB,MACzC,SAAQ,KAAK,SAAS,UAAU,mBACjC,CAAE,OAAQ;CAEX,MAAM,eAAe,KAAK,SAAS,YAAY,WAAW;AAE1D,KAAI,WAAW,SAAS,EACtB,YAAW,KAAK;EACd,MAAM,cAAc;EACpB,YAAY,WAAW,WAAW;EAClC,iBAAiB,SAAS,aAAa;EACvC,cAAc,WAAW,KAAI,MAAK,EAAE,KAAK;EAC1C,CAAC;AAGJ,KAAI,OAAO,kBAAkB,SAAS,eAAe,SAAS,GAAG;EAC/D,MAAM,iBAAiB,eAAe,QAAO,MAAK,EAAE,SAAS,MAAM,KAAK;AAExE,MAAI,eAAe,SAAS,EAC1B,YAAW,KAAK;GACd,MAAM,cAAc;GACpB,iBAAiB;GACjB,cAAc,MAAM,KAClB,IAAI,IACF,eAAe,SAAQ,MAAK,CAAC,GAAG,EAAE,QAAQ,iBAAiB,EAAE,KAAK,CAAC,CAAC,CACrE,CACF;GACF,CAAC;;AAIN,YAAW,sBAAsB,WAAW;;AAG9C,MAAa,uBAAuB,cAAsB;AACxD,QAAO,kBAAkB,UAAU,CAChC,MAAK,aAAY,SAAS,WAAW,UAAU,CAAC,EAC/C,MAAM,EAAE,CACT,MAAM,GAAG,GAAG;;AAGjB,MAAa,oBAAoB,cAAsB;AACrD,QAAO,kBAAkB,UAAU,CAAC,QAClC,SAAQ,CAAC,KAAK,WAAW,UAAU,CACpC;;AAGH,MAAa,qBACX,OACA,uBACE,SACG,KAAK,UAAU,KACjB;CACH,IAAI,UAAU;CACd,MAAMC,iBAA2B,CAAC,GAAG;AACrC,KAAI,MAAM,SAAS,SACjB,SAAQ,MAAM,MAAd;EACE,KAAK;AACH,aAAU;AACV;EACF,KAAK;AACH,aAAU;AACV,kBAAe,KAAK,QAAQ;AAC5B;EACF,KAAK;AACH,aAAU;AACV;EACF,KAAK;AACH,aAAU;AACV;EACF,KAAK;AACH,aAAU;AACV;EACF,KAAK;AACH,aAAU;AACV;EACF,KAAK;AACH,aAAU;AACV;EACF,KAAK;AACH,aAAU;AACV;EAEF,KAAK;AACH,aAAU;AACV;;UAEK,MAAM,SAAS,OACxB,WAAU,gBAAgB,MAAM,KAAK;UAC5B,MAAM,SAAS,SACxB,WAAU,oBAAoB,MAAM,KAAK;AAG3C,KAAI,MAAM,OAAQ,gBAAe,KAAK,UAAU;AAChD,KAAI,MAAM,eAAe;AACvB,YAAU,oBAAoB,MAAM,cAAc,IAAI;AACtD,iBAAe,KAAK,GAAG,iBAAiB,MAAM,cAAc,CAAC;;AAE/D,KAAI,CAAC,MAAM,cAAc,MAAM,SAAS,OACtC,gBAAe,KAAK,YAAY;AAGlC,QAAO,GAAG,UAAU,eAAe,KAAK,IAAI;;AAG9C,MAAa,4BACX,OACA,YACA,WACG;AACH,KAAI,MAAM,OAAO,MAAK,MAAK,EAAE,SAAS,OAAO,CAC3C,YAAW,eAAc,WAAU;AACjC,SAAO,SAAS;AAChB,aAAW,QAAQ;GACjB;GACA,2CACE,OAAO,wBAAwB,KAAK;GAEtC;GACA,qEACE,OAAO,wBAAwB,KAAK,aACrC;GACD;GACD,CAAC;GACF;AAGJ,KAAI,OAAO,gBAAgB,MAAM,OAAO,MAAK,MAAK,EAAE,SAAS,UAAU,CACrE,YAAW,eAAc,WAAU;AACjC,SAAO,SAAS;AAChB,aAAW,QAAQ;GACjB;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACD,CAAC;GACF;;AAIN,MAAa,0BACX,OACA,YACA,WACG;CACH,MAAM,EAAE,cAAc,cAAc,OAAO;AAE3C,YAAW,qBAAqB;EAC9B,iBAAiB,wBAAwB;EACzC,YAAY;EACZ,gBAAe,WAAU,OAAO,oBAAoB;EACpD,cAAc,CACZ;GACE,MAAM,UAAU,MAAM,KAAK;GAC3B,YAAY,QAAQ;AAClB,WACG,MAAM,YAAY,CAClB,kBAAkB;AACjB,WAAM,OACH,QAAO,MAAK,EAAE,SAAS,SAAS,CAChC,SAAQ,UAAS;AAChB,iBAAW,QAAQ,UAAU,MAAM,cAAc,CAAC;AAClD,aACG,MAAM,GAAG,MAAM,KAAK,IAAI,kBAAkB,MAAM,GAAG,CACnD,MAAM,IAAI,CACV,SAAS;OACZ;MACJ,CACD,MAAM,IAAI;;GAEhB,CACF;EACF,CAAC;;AAGJ,MAAa,iCACX,OACA,YACA,WACG;CACH,MAAM,EAAE,WAAW,qBAAqB,cAAc,OAAO;CAE7D,MAAM,iBAAiB,MAAM,OAAO,QAAO,MAAK,EAAE,SAAS,SAAS;AAEpE,YAAW,aAAa;EACtB,MAAM,GAAG,MAAM;EACf,YAAY;EACZ,SAAS,CAAC,kBAAkB,UAAU,MAAM,KAAK,CAAC,GAAG;EACrD,YAAY,eAAe,KAAI,OAAM;GACnC,kBAAkB,CAAC,EAAE;GACrB,MAAM,EAAE;GACR,MAAM,GAAG,EAAE,OAAO,EAAE,SAAS,OAAO,KAAK,CAAC,EAAE,aAAa,YAAY;GACtE,EAAE;EACJ,CAAC;AAEF,YAAW,eAAc,WACvB,WAAW,QAAQ;EACjB;EACA;EACA,MAAM,iBACJ,MAAM,KACP,CAAC;EACF;EACA;EACA;EACD,CAAC,CACH;AAED,YAAW,qBAAqB;EAC9B,iBAAiB,wBAAwB;EACzC,YAAY;EACZ,cAAc,CACZ;GACE,MAAM,iBAAiB,MAAM,KAAK;GAClC,MAAM,eAAe,MAAM,KAAK;GAChC,YAAY,QAAQ;AAClB,WACG,MAAM,gBAAgB,UAAU,MAAM,KAAK,CAAC,UAAU,CACtD,kBAAkB;AACjB,oBAAe,SAAQ,UAAS;AAC9B,iBAAW,QAAQ,UAAU,MAAM,cAAc,CAAC;AAElD,aACG,MACC,GAAG,MAAM,KAAK,IAAI,kBAChB,OACA,iBACD,GACF,CACA,MAAM,IAAI,CACV,SAAS;OACZ;MACF,CACD,MAAM,KAAK;;GAEjB,CACF;EACF,CAAC;;AAGJ,MAAa,oBAAoB,OAC/B,OACA,YACA,QACA,YACG;AACH,OAAM,qBAAqB,OAAO,YAAY,QAAQ,QAAQ;AAC9D,0BAAyB,OAAO,YAAY,OAAO;AACnD,wBAAuB,OAAO,YAAY,OAAO;AACjD,KAAI,kBAAkB,OAAO,OAAO,CAClC,+BAA8B,OAAO,YAAY,OAAO;;AAG5D,MAAa,sBACX,QACA,cACG;AACH,QAAO,SAAQ,UACb,UAAU,qBAAqB,EAC7B,iBAAiB,KAAK,eAAe,MAAM,KAAK,CAAC,UAClD,CAAC,CACH"}

package/dist/zod/modelArgs-helpers.cjs

@@ -1 +1,56 @@
-const e=require(`./model-helpers.cjs`);function t(e,t,r,i){let a=n(t,r,i);for(let t of a)e.push(t)}function n(t,n,r){let i=[];for(let a of t){let{name:t}=a,o=[];if(n){let e={name:`select`,isRequired:!1,isNullable:!1,inputTypes:[{isList:!1,type:`${t}Select`,location:`inputObjectTypes`,namespace:`prisma`}]};o.push(e)}let s=e.checkModelHasModelRelation(a);if(r&&s){let e={name:`include`,isRequired:!1,isNullable:!1,inputTypes:[{isList:!1,type:`${t}Include`,location:`inputObjectTypes`,namespace:`prisma`}]};o.push(e)}let c={name:`${t}DefaultArgs`,constraints:{maxNumFields:null,minNumFields:null},fields:o};i.push(c)}return i}exports.addMissingInputObjectTypesForModelArgs=t;
+const require_model_helpers = require('./model-helpers.cjs');
+
+//#region src/zod/modelArgs-helpers.ts
+function addMissingInputObjectTypesForModelArgs(inputObjectTypes, models, isGenerateSelect, isGenerateInclude) {
+	const modelArgsInputObjectTypes = generateModelArgsInputObjectTypes(models, isGenerateSelect, isGenerateInclude);
+	for (const modelArgsInputObjectType of modelArgsInputObjectTypes) inputObjectTypes.push(modelArgsInputObjectType);
+}
+function generateModelArgsInputObjectTypes(models, isGenerateSelect, isGenerateInclude) {
+	const modelArgsInputObjectTypes = [];
+	for (const model of models) {
+		const { name: modelName } = model;
+		const fields = [];
+		if (isGenerateSelect) {
+			const selectField = {
+				name: "select",
+				isRequired: false,
+				isNullable: false,
+				inputTypes: [{
+					isList: false,
+					type: `${modelName}Select`,
+					location: "inputObjectTypes",
+					namespace: "prisma"
+				}]
+			};
+			fields.push(selectField);
+		}
+		const hasRelationToAnotherModel = require_model_helpers.checkModelHasModelRelation(model);
+		if (isGenerateInclude && hasRelationToAnotherModel) {
+			const includeField = {
+				name: "include",
+				isRequired: false,
+				isNullable: false,
+				inputTypes: [{
+					isList: false,
+					type: `${modelName}Include`,
+					location: "inputObjectTypes",
+					namespace: "prisma"
+				}]
+			};
+			fields.push(includeField);
+		}
+		const modelArgsInputObjectType = {
+			name: `${modelName}DefaultArgs`,
+			constraints: {
+				maxNumFields: null,
+				minNumFields: null
+			},
+			fields
+		};
+		modelArgsInputObjectTypes.push(modelArgsInputObjectType);
+	}
+	return modelArgsInputObjectTypes;
+}
+
+//#endregion
+exports.addMissingInputObjectTypesForModelArgs = addMissingInputObjectTypesForModelArgs;