zenstack 1.0.0-beta.1 → 1.0.0-beta.11

package/plugins/zod/generator.js

@@ -17,10 +17,14 @@ const sdk_1 = require("@zenstackhq/sdk");
 const ast_1 = require("@zenstackhq/sdk/ast");
 const dmmf_helpers_1 = require("@zenstackhq/sdk/dmmf-helpers");
 const fs_1 = require("fs");
+const langium_1 = require("langium");
 const path_1 = __importDefault(require("path"));
+const upper_case_first_1 = require("upper-case-first");
+const utils_1 = require("../../language-server/utils");
 const plugin_utils_1 = require("../plugin-utils");
 const transformer_1 = __importDefault(require("./transformer"));
 const removeDir_1 = __importDefault(require("./utils/removeDir"));
+const schema_gen_1 = require("./utils/schema-gen");
 function generate(model, options, dmmf) {
     var _a, _b;
     return __awaiter(this, void 0, void 0, function* () {
@@ -42,16 +46,36 @@ function generate(model, options, dmmf) {
         const outputObjectTypes = prismaClientDmmf.schema.outputObjectTypes.prisma;
         const models = prismaClientDmmf.datamodel.models;
         const project = (0, sdk_1.createProject)();
-        yield generateEnumSchemas(prismaClientDmmf.schema.enumTypes.prisma, (_a = prismaClientDmmf.schema.enumTypes.model) !== null && _a !== void 0 ? _a : [], project);
+        // common schemas
+        yield generateCommonSchemas(project, output);
+        // enums
+        yield generateEnumSchemas(prismaClientDmmf.schema.enumTypes.prisma, (_a = prismaClientDmmf.schema.enumTypes.model) !== null && _a !== void 0 ? _a : [], project, model);
         const dataSource = model.declarations.find((d) => (0, ast_1.isDataSource)(d));
         const dataSourceProvider = (0, sdk_1.getLiteral)((_b = dataSource === null || dataSource === void 0 ? void 0 : dataSource.fields.find((f) => f.name === 'provider')) === null || _b === void 0 ? void 0 : _b.value);
-        transformer_1.default.provider = dataSourceProvider;
-        const generatorConfigOptions = {};
-        Object.entries(options).forEach(([k, v]) => (generatorConfigOptions[k] = v));
-        (0, dmmf_helpers_1.addMissingInputObjectTypes)(inputObjectTypes, outputObjectTypes, models);
-        const aggregateOperationSupport = (0, dmmf_helpers_1.resolveAggregateOperationSupport)(inputObjectTypes);
-        yield generateObjectSchemas(inputObjectTypes, project, output);
-        yield generateModelSchemas(models, modelOperations, aggregateOperationSupport, project);
+        yield generateModelSchemas(project, model, output);
+        if (options.modelOnly !== true) {
+            // detailed object schemas referenced from input schemas
+            transformer_1.default.provider = dataSourceProvider;
+            (0, dmmf_helpers_1.addMissingInputObjectTypes)(inputObjectTypes, outputObjectTypes, models);
+            const aggregateOperationSupport = (0, dmmf_helpers_1.resolveAggregateOperationSupport)(inputObjectTypes);
+            yield generateObjectSchemas(inputObjectTypes, project, output, model);
+            // input schemas
+            const transformer = new transformer_1.default({
+                models,
+                modelOperations,
+                aggregateOperationSupport,
+                project,
+                zmodel: model,
+            });
+            yield transformer.generateInputSchemas();
+        }
+        // create barrel file
+        const exports = [`export * as models from './models'`, `export * as enums from './enums'`];
+        if (options.modelOnly !== true) {
+            exports.push(`export * as input from './input'`, `export * as objects from './objects'`);
+        }
+        project.createSourceFile(path_1.default.join(output, 'index.ts'), exports.join(';\n'), { overwrite: true });
+        // emit
         const shouldCompile = options.compile !== false;
         if (!shouldCompile || options.preserveTsFiles === true) {
             // save ts files
@@ -72,7 +96,28 @@ function handleGeneratorOutputValue(output) {
         transformer_1.default.setOutputPath(output);
     });
 }
-function generateEnumSchemas(prismaSchemaEnum, modelSchemaEnum, project) {
+function generateCommonSchemas(project, output) {
+    return __awaiter(this, void 0, void 0, function* () {
+        // Decimal
+        project.createSourceFile(path_1.default.join(output, 'common', 'index.ts'), `
+import { z } from 'zod';
+export const DecimalSchema = z.union([z.number(), z.string(), z.object({d: z.number().array(), e: z.number(), s: z.number()})]);
+
+// https://stackoverflow.com/a/54487392/20415796
+type OmitDistributive<T, K extends PropertyKey> = T extends any ? (T extends object ? OmitRecursively<T, K> : T) : never;
+type OmitRecursively<T extends any, K extends PropertyKey> = Omit<
+    { [P in keyof T]: OmitDistributive<T[P], K> },
+    K
+>;
+
+/**
+ * Strips auxiliary fields recursively
+ */
+export type Purge<T> = OmitRecursively<T, ${sdk_1.AUXILIARY_FIELDS.map((f) => "'" + f + "'").join('|')}>;
+`, { overwrite: true });
+    });
+}
+function generateEnumSchemas(prismaSchemaEnum, modelSchemaEnum, project, zmodel) {
     return __awaiter(this, void 0, void 0, function* () {
         const enumTypes = [...prismaSchemaEnum, ...modelSchemaEnum];
         const enumNames = enumTypes.map((enumItem) => enumItem.name);
@@ -80,33 +125,127 @@ function generateEnumSchemas(prismaSchemaEnum, modelSchemaEnum, project) {
         const transformer = new transformer_1.default({
             enumTypes,
             project,
+            zmodel,
         });
         yield transformer.generateEnumSchemas();
     });
 }
-function generateObjectSchemas(inputObjectTypes, project, output) {
+function generateObjectSchemas(inputObjectTypes, project, output, zmodel) {
     var _a, _b;
     return __awaiter(this, void 0, void 0, function* () {
         const moduleNames = [];
         for (let i = 0; i < inputObjectTypes.length; i += 1) {
             const fields = (_a = inputObjectTypes[i]) === null || _a === void 0 ? void 0 : _a.fields;
             const name = (_b = inputObjectTypes[i]) === null || _b === void 0 ? void 0 : _b.name;
-            const transformer = new transformer_1.default({ name, fields, project });
+            const transformer = new transformer_1.default({ name, fields, project, zmodel });
             const moduleName = transformer.generateObjectSchema();
             moduleNames.push(moduleName);
         }
         project.createSourceFile(path_1.default.join(output, 'objects/index.ts'), moduleNames.map((name) => `export * from './${name}';`).join('\n'), { overwrite: true });
     });
 }
-function generateModelSchemas(models, modelOperations, aggregateOperationSupport, project) {
+function generateModelSchemas(project, zmodel, output) {
     return __awaiter(this, void 0, void 0, function* () {
-        const transformer = new transformer_1.default({
-            models,
-            modelOperations,
-            aggregateOperationSupport,
-            project,
+        const schemaNames = [];
+        for (const dm of (0, sdk_1.getDataModels)(zmodel)) {
+            schemaNames.push(yield generateModelSchema(dm, project, output));
+        }
+        project.createSourceFile(path_1.default.join(output, 'models', 'index.ts'), schemaNames.map((name) => `export * from './${name}';`).join('\n'), { overwrite: true });
+    });
+}
+function generateModelSchema(model, project, output) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const schemaName = `${(0, upper_case_first_1.upperCaseFirst)(model.name)}.schema`;
+        const sf = project.createSourceFile(path_1.default.join(output, 'models', `${schemaName}.ts`), undefined, {
+            overwrite: true,
+        });
+        sf.replaceWithText((writer) => {
+            var _a, _b, _c;
+            const fields = model.fields.filter((field) => {
+                var _a;
+                return !sdk_1.AUXILIARY_FIELDS.includes(field.name) &&
+                    // scalar fields only
+                    !(0, ast_1.isDataModel)((_a = field.type.reference) === null || _a === void 0 ? void 0 : _a.ref) &&
+                    !(0, sdk_1.isForeignKeyField)(field);
+            });
+            writer.writeLine('/* eslint-disable */');
+            writer.writeLine(`import { z } from 'zod';`);
+            // import user-defined enums from Prisma as they might be referenced in the expressions
+            const importEnums = new Set();
+            for (const node of (0, langium_1.streamAllContents)(model)) {
+                if ((0, sdk_1.isEnumFieldReference)(node)) {
+                    const field = node.target.ref;
+                    if (!(0, utils_1.isFromStdlib)(field.$container)) {
+                        importEnums.add(field.$container.name);
+                    }
+                }
+            }
+            if (importEnums.size > 0) {
+                const prismaImport = (0, sdk_1.getPrismaClientImportSpec)(model.$container, path_1.default.join(output, 'models'));
+                writer.writeLine(`import { ${[...importEnums].join(', ')} } from '${prismaImport}';`);
+            }
+            // import enum schemas
+            for (const field of fields) {
+                if (((_a = field.type.reference) === null || _a === void 0 ? void 0 : _a.ref) && (0, ast_1.isEnum)((_b = field.type.reference) === null || _b === void 0 ? void 0 : _b.ref)) {
+                    const name = (0, upper_case_first_1.upperCaseFirst)((_c = field.type.reference) === null || _c === void 0 ? void 0 : _c.ref.name);
+                    writer.writeLine(`import { ${name}Schema } from '../enums/${name}.schema';`);
+                }
+            }
+            // import Decimal
+            if (fields.some((field) => field.type.type === 'Decimal')) {
+                writer.writeLine(`import { DecimalSchema } from '../common';`);
+            }
+            // create base schema
+            writer.write(`const baseSchema = z.object(`);
+            writer.inlineBlock(() => {
+                fields.forEach((field) => {
+                    writer.writeLine(`${field.name}: ${(0, schema_gen_1.makeFieldSchema)(field)},`);
+                });
+            });
+            writer.writeLine(');');
+            // compile "@@validate" to ".refine"
+            const refinements = (0, schema_gen_1.makeValidationRefinements)(model);
+            if (refinements.length > 0) {
+                writer.writeLine(`function refine(schema: z.ZodType) { return schema${refinements.join('\n')}; }`);
+            }
+            // model schema
+            let modelSchema = 'baseSchema';
+            const fieldsToOmit = fields.filter((field) => (0, sdk_1.hasAttribute)(field, '@omit'));
+            if (fieldsToOmit.length > 0) {
+                modelSchema = makeOmit(modelSchema, fieldsToOmit.map((f) => f.name));
+            }
+            if (refinements.length > 0) {
+                modelSchema = `refine(${modelSchema})`;
+            }
+            writer.writeLine(`export const ${(0, upper_case_first_1.upperCaseFirst)(model.name)}Schema = ${modelSchema};`);
+            // create schema
+            let createSchema = 'baseSchema';
+            const fieldsWithDefault = fields.filter((field) => (0, sdk_1.hasAttribute)(field, '@default') || (0, sdk_1.hasAttribute)(field, '@updatedAt') || field.type.array);
+            if (fieldsWithDefault.length > 0) {
+                createSchema = makePartial(createSchema, fieldsWithDefault.map((f) => f.name));
+            }
+            if (refinements.length > 0) {
+                createSchema = `refine(${createSchema})`;
+            }
+            writer.writeLine(`export const ${(0, upper_case_first_1.upperCaseFirst)(model.name)}CreateSchema = ${createSchema};`);
+            // update schema
+            let updateSchema = 'baseSchema.partial()';
+            if (refinements.length > 0) {
+                updateSchema = `refine(${updateSchema})`;
+            }
+            writer.writeLine(`export const ${(0, upper_case_first_1.upperCaseFirst)(model.name)}UpdateSchema = ${updateSchema};`);
         });
-        yield transformer.generateModelSchemas();
+        return schemaName;
     });
 }
+function makePartial(schema, fields) {
+    return `${schema}.partial({
+        ${fields.map((f) => `${f}: true`).join(', ')},
+    })`;
+}
+function makeOmit(schema, fields) {
+    return `${schema}.omit({
+        ${fields.map((f) => `${f}: true`).join(', ')},
+    })`;
+}
 //# sourceMappingURL=generator.js.map

package/plugins/zod/generator.js.map

@@ -1 +1 @@
-{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../../src/plugins/zod/generator.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAEA,yCAAkH;AAClH,6CAAsE;AACtE,+DAIsC;AACtC,2BAAoC;AACpC,gDAAwB;AAExB,kDAAyD;AACzD,gEAAwC;AACxC,kEAA0C;AAE1C,SAAsB,QAAQ,CAAC,KAAY,EAAE,OAAsB,EAAE,IAAmB;;;QACpF,IAAI,MAAM,GAAG,OAAO,CAAC,MAAgB,CAAC;QACtC,IAAI,CAAC,MAAM,EAAE;YACT,MAAM,mBAAmB,GAAG,IAAA,qCAAsB,GAAE,CAAC;YACrD,IAAI,mBAAmB,EAAE;gBACrB,MAAM,GAAG,cAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE,KAAK,CAAC,CAAC;aAClD;iBAAM;gBACH,MAAM,GAAG,iBAAiB,CAAC;aAC9B;SACJ;QACD,MAAM,GAAG,IAAA,iBAAW,EAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACtC,MAAM,0BAA0B,CAAC,MAAM,CAAC,CAAC;QAEzC,MAAM,gBAAgB,GAAG,IAAI,CAAC;QAE9B,MAAM,eAAe,GAAG,gBAAgB,CAAC,QAAQ,CAAC,eAAe,CAAC;QAClE,MAAM,gBAAgB,GAAG,gBAAgB,CAAC,MAAM,CAAC,gBAAgB,CAAC,MAAM,CAAC;QACzE,MAAM,iBAAiB,GAAG,gBAAgB,CAAC,MAAM,CAAC,iBAAiB,CAAC,MAAM,CAAC;QAC3E,MAAM,MAAM,GAAiB,gBAAgB,CAAC,SAAS,CAAC,MAAM,CAAC;QAE/D,MAAM,OAAO,GAAG,IAAA,mBAAa,GAAE,CAAC;QAEhC,MAAM,mBAAmB,CACrB,gBAAgB,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,EACxC,MAAA,gBAAgB,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,mCAAI,EAAE,EAC7C,OAAO,CACV,CAAC;QAEF,MAAM,UAAU,GAAG,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAmB,EAAE,CAAC,IAAA,kBAAY,EAAC,CAAC,CAAC,CAAC,CAAC;QAEpF,MAAM,kBAAkB,GAAG,IAAA,gBAAU,EACjC,MAAA,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,0CAAE,KAAK,CAC9C,CAAC;QAEnB,qBAAW,CAAC,QAAQ,GAAG,kBAAkB,CAAC;QAE1C,MAAM,sBAAsB,GAAuB,EAAE,CAAC;QACtD,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,GAAG,CAAW,CAAC,CAAC,CAAC;QAEvF,IAAA,yCAA0B,EAAC,gBAAgB,EAAE,iBAAiB,EAAE,MAAM,CAAC,CAAC;QAExE,MAAM,yBAAyB,GAAG,IAAA,+CAAgC,EAAC,gBAAgB,CAAC,CAAC;QAErF,MAAM,qBAAqB,CAAC,gBAAgB,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QAC/D,MAAM,oBAAoB,CAAC,MAAM,EAAE,eAAe,EAAE,yBAAyB,EAAE,OAAO,CAAC,CAAC;QAExF,MAAM,aAAa,GAAG,OAAO,CAAC,OAAO,KAAK,KAAK,CAAC;QAChD,IAAI,CAAC,aAAa,IAAI,OAAO,CAAC,eAAe,KAAK,IAAI,EAAE;YACpD,gBAAgB;YAChB,MAAM,IAAA,iBAAW,EAAC,OAAO,CAAC,CAAC;SAC9B;QACD,IAAI,aAAa,EAAE;YACf,MAAM,IAAA,iBAAW,EAAC,OAAO,CAAC,CAAC;SAC9B;;CACJ;AAtDD,4BAsDC;AAED,SAAe,0BAA0B,CAAC,MAAc;;QACpD,uFAAuF;QACvF,MAAM,aAAE,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5C,MAAM,oBAAoB,GAAG,IAAI,CAAC;QAClC,MAAM,IAAA,mBAAS,EAAC,MAAM,EAAE,oBAAoB,CAAC,CAAC;QAE9C,qBAAW,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;IACtC,CAAC;CAAA;AAED,SAAe,mBAAmB,CAC9B,gBAAmC,EACnC,eAAkC,EAClC,OAAgB;;QAEhB,MAAM,SAAS,GAAG,CAAC,GAAG,gBAAgB,EAAE,GAAG,eAAe,CAAC,CAAC;QAC5D,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC7D,qBAAW,CAAC,SAAS,GAAG,SAAS,aAAT,SAAS,cAAT,SAAS,GAAI,EAAE,CAAC;QACxC,MAAM,WAAW,GAAG,IAAI,qBAAW,CAAC;YAChC,SAAS;YACT,OAAO;SACV,CAAC,CAAC;QACH,MAAM,WAAW,CAAC,mBAAmB,EAAE,CAAC;IAC5C,CAAC;CAAA;AAED,SAAe,qBAAqB,CAAC,gBAAkC,EAAE,OAAgB,EAAE,MAAc;;;QACrG,MAAM,WAAW,GAAa,EAAE,CAAC;QACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,gBAAgB,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE;YACjD,MAAM,MAAM,GAAG,MAAA,gBAAgB,CAAC,CAAC,CAAC,0CAAE,MAAM,CAAC;YAC3C,MAAM,IAAI,GAAG,MAAA,gBAAgB,CAAC,CAAC,CAAC,0CAAE,IAAI,CAAC;YACvC,MAAM,WAAW,GAAG,IAAI,qBAAW,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;YAC/D,MAAM,UAAU,GAAG,WAAW,CAAC,oBAAoB,EAAE,CAAC;YACtD,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;SAChC;QACD,OAAO,CAAC,gBAAgB,CACpB,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,kBAAkB,CAAC,EACrC,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,oBAAoB,IAAI,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAClE,EAAE,SAAS,EAAE,IAAI,EAAE,CACtB,CAAC;;CACL;AAED,SAAe,oBAAoB,CAC/B,MAAoB,EACpB,eAAoC,EACpC,yBAAoD,EACpD,OAAgB;;QAEhB,MAAM,WAAW,GAAG,IAAI,qBAAW,CAAC;YAChC,MAAM;YACN,eAAe;YACf,yBAAyB;YACzB,OAAO;SACV,CAAC,CAAC;QACH,MAAM,WAAW,CAAC,oBAAoB,EAAE,CAAC;IAC7C,CAAC;CAAA"}
+{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../../src/plugins/zod/generator.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AACA,yCAayB;AACzB,6CAAiH;AACjH,+DAA4G;AAC5G,2BAAoC;AACpC,qCAA4C;AAC5C,gDAAwB;AAExB,uDAAkD;AAClD,uDAA2D;AAC3D,kDAAyD;AACzD,gEAAwC;AACxC,kEAA0C;AAC1C,mDAAgF;AAEhF,SAAsB,QAAQ,CAAC,KAAY,EAAE,OAAsB,EAAE,IAAmB;;;QACpF,IAAI,MAAM,GAAG,OAAO,CAAC,MAAgB,CAAC;QACtC,IAAI,CAAC,MAAM,EAAE;YACT,MAAM,mBAAmB,GAAG,IAAA,qCAAsB,GAAE,CAAC;YACrD,IAAI,mBAAmB,EAAE;gBACrB,MAAM,GAAG,cAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE,KAAK,CAAC,CAAC;aAClD;iBAAM;gBACH,MAAM,GAAG,iBAAiB,CAAC;aAC9B;SACJ;QACD,MAAM,GAAG,IAAA,iBAAW,EAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACtC,MAAM,0BAA0B,CAAC,MAAM,CAAC,CAAC;QAEzC,MAAM,gBAAgB,GAAG,IAAI,CAAC;QAE9B,MAAM,eAAe,GAAG,gBAAgB,CAAC,QAAQ,CAAC,eAAe,CAAC;QAClE,MAAM,gBAAgB,GAAG,gBAAgB,CAAC,MAAM,CAAC,gBAAgB,CAAC,MAAM,CAAC;QACzE,MAAM,iBAAiB,GAAG,gBAAgB,CAAC,MAAM,CAAC,iBAAiB,CAAC,MAAM,CAAC;QAC3E,MAAM,MAAM,GAAiB,gBAAgB,CAAC,SAAS,CAAC,MAAM,CAAC;QAE/D,MAAM,OAAO,GAAG,IAAA,mBAAa,GAAE,CAAC;QAEhC,iBAAiB;QACjB,MAAM,qBAAqB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAE7C,QAAQ;QACR,MAAM,mBAAmB,CACrB,gBAAgB,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,EACxC,MAAA,gBAAgB,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,mCAAI,EAAE,EAC7C,OAAO,EACP,KAAK,CACR,CAAC;QAEF,MAAM,UAAU,GAAG,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAmB,EAAE,CAAC,IAAA,kBAAY,EAAC,CAAC,CAAC,CAAC,CAAC;QAEpF,MAAM,kBAAkB,GAAG,IAAA,gBAAU,EACjC,MAAA,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,0CAAE,KAAK,CAC9C,CAAC;QAEnB,MAAM,oBAAoB,CAAC,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;QAEnD,IAAI,OAAO,CAAC,SAAS,KAAK,IAAI,EAAE;YAC5B,wDAAwD;YACxD,qBAAW,CAAC,QAAQ,GAAG,kBAAkB,CAAC;YAC1C,IAAA,yCAA0B,EAAC,gBAAgB,EAAE,iBAAiB,EAAE,MAAM,CAAC,CAAC;YACxE,MAAM,yBAAyB,GAAG,IAAA,+CAAgC,EAAC,gBAAgB,CAAC,CAAC;YACrF,MAAM,qBAAqB,CAAC,gBAAgB,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;YAEtE,gBAAgB;YAChB,MAAM,WAAW,GAAG,IAAI,qBAAW,CAAC;gBAChC,MAAM;gBACN,eAAe;gBACf,yBAAyB;gBACzB,OAAO;gBACP,MAAM,EAAE,KAAK;aAChB,CAAC,CAAC;YACH,MAAM,WAAW,CAAC,oBAAoB,EAAE,CAAC;SAC5C;QAED,qBAAqB;QACrB,MAAM,OAAO,GAAG,CAAC,oCAAoC,EAAE,kCAAkC,CAAC,CAAC;QAC3F,IAAI,OAAO,CAAC,SAAS,KAAK,IAAI,EAAE;YAC5B,OAAO,CAAC,IAAI,CAAC,kCAAkC,EAAE,sCAAsC,CAAC,CAAC;SAC5F;QACD,OAAO,CAAC,gBAAgB,CAAC,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,CAAC,EAAE,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAElG,OAAO;QACP,MAAM,aAAa,GAAG,OAAO,CAAC,OAAO,KAAK,KAAK,CAAC;QAChD,IAAI,CAAC,aAAa,IAAI,OAAO,CAAC,eAAe,KAAK,IAAI,EAAE;YACpD,gBAAgB;YAChB,MAAM,IAAA,iBAAW,EAAC,OAAO,CAAC,CAAC;SAC9B;QACD,IAAI,aAAa,EAAE;YACf,MAAM,IAAA,iBAAW,EAAC,OAAO,CAAC,CAAC;SAC9B;;CACJ;AA3ED,4BA2EC;AAED,SAAe,0BAA0B,CAAC,MAAc;;QACpD,uFAAuF;QACvF,MAAM,aAAE,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5C,MAAM,oBAAoB,GAAG,IAAI,CAAC;QAClC,MAAM,IAAA,mBAAS,EAAC,MAAM,EAAE,oBAAoB,CAAC,CAAC;QAE9C,qBAAW,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;IACtC,CAAC;CAAA;AAED,SAAe,qBAAqB,CAAC,OAAgB,EAAE,MAAc;;QACjE,UAAU;QACV,OAAO,CAAC,gBAAgB,CACpB,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,UAAU,CAAC,EACvC;;;;;;;;;;;;;;4CAcoC,sBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;CAC/F,EACO,EAAE,SAAS,EAAE,IAAI,EAAE,CACtB,CAAC;IACN,CAAC;CAAA;AAED,SAAe,mBAAmB,CAC9B,gBAAmC,EACnC,eAAkC,EAClC,OAAgB,EAChB,MAAa;;QAEb,MAAM,SAAS,GAAG,CAAC,GAAG,gBAAgB,EAAE,GAAG,eAAe,CAAC,CAAC;QAC5D,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC7D,qBAAW,CAAC,SAAS,GAAG,SAAS,aAAT,SAAS,cAAT,SAAS,GAAI,EAAE,CAAC;QACxC,MAAM,WAAW,GAAG,IAAI,qBAAW,CAAC;YAChC,SAAS;YACT,OAAO;YACP,MAAM;SACT,CAAC,CAAC;QACH,MAAM,WAAW,CAAC,mBAAmB,EAAE,CAAC;IAC5C,CAAC;CAAA;AAED,SAAe,qBAAqB,CAChC,gBAAkC,EAClC,OAAgB,EAChB,MAAc,EACd,MAAa;;;QAEb,MAAM,WAAW,GAAa,EAAE,CAAC;QACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,gBAAgB,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE;YACjD,MAAM,MAAM,GAAG,MAAA,gBAAgB,CAAC,CAAC,CAAC,0CAAE,MAAM,CAAC;YAC3C,MAAM,IAAI,GAAG,MAAA,gBAAgB,CAAC,CAAC,CAAC,0CAAE,IAAI,CAAC;YACvC,MAAM,WAAW,GAAG,IAAI,qBAAW,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;YACvE,MAAM,UAAU,GAAG,WAAW,CAAC,oBAAoB,EAAE,CAAC;YACtD,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;SAChC;QACD,OAAO,CAAC,gBAAgB,CACpB,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,kBAAkB,CAAC,EACrC,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,oBAAoB,IAAI,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAClE,EAAE,SAAS,EAAE,IAAI,EAAE,CACtB,CAAC;;CACL;AAED,SAAe,oBAAoB,CAAC,OAAgB,EAAE,MAAa,EAAE,MAAc;;QAC/E,MAAM,WAAW,GAAa,EAAE,CAAC;QACjC,KAAK,MAAM,EAAE,IAAI,IAAA,mBAAa,EAAC,MAAM,CAAC,EAAE;YACpC,WAAW,CAAC,IAAI,CAAC,MAAM,mBAAmB,CAAC,EAAE,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC;SACpE;QAED,OAAO,CAAC,gBAAgB,CACpB,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,UAAU,CAAC,EACvC,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,oBAAoB,IAAI,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAClE,EAAE,SAAS,EAAE,IAAI,EAAE,CACtB,CAAC;IACN,CAAC;CAAA;AAED,SAAe,mBAAmB,CAAC,KAAgB,EAAE,OAAgB,EAAE,MAAc;;QACjF,MAAM,UAAU,GAAG,GAAG,IAAA,iCAAc,EAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC;QAC1D,MAAM,EAAE,GAAG,OAAO,CAAC,gBAAgB,CAAC,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,GAAG,UAAU,KAAK,CAAC,EAAE,SAAS,EAAE;YAC5F,SAAS,EAAE,IAAI;SAClB,CAAC,CAAC;QACH,EAAE,CAAC,eAAe,CAAC,CAAC,MAAM,EAAE,EAAE;;YAC1B,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,CAC9B,CAAC,KAAK,EAAE,EAAE;;gBACN,OAAA,CAAC,sBAAgB,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC;oBACtC,qBAAqB;oBACrB,CAAC,IAAA,iBAAW,EAAC,MAAA,KAAK,CAAC,IAAI,CAAC,SAAS,0CAAE,GAAG,CAAC;oBACvC,CAAC,IAAA,uBAAiB,EAAC,KAAK,CAAC,CAAA;aAAA,CAChC,CAAC;YAEF,MAAM,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;YACzC,MAAM,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAC;YAE7C,uFAAuF;YACvF,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;YACtC,KAAK,MAAM,IAAI,IAAI,IAAA,2BAAiB,EAAC,KAAK,CAAC,EAAE;gBACzC,IAAI,IAAA,0BAAoB,EAAC,IAAI,CAAC,EAAE;oBAC5B,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAgB,CAAC;oBAC3C,IAAI,CAAC,IAAA,oBAAY,EAAC,KAAK,CAAC,UAAU,CAAC,EAAE;wBACjC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;qBAC1C;iBACJ;aACJ;YACD,IAAI,WAAW,CAAC,IAAI,GAAG,CAAC,EAAE;gBACtB,MAAM,YAAY,GAAG,IAAA,+BAAyB,EAAC,KAAK,CAAC,UAAU,EAAE,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC;gBAC9F,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,GAAG,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,YAAY,IAAI,CAAC,CAAC;aACzF;YAED,sBAAsB;YACtB,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE;gBACxB,IAAI,CAAA,MAAA,KAAK,CAAC,IAAI,CAAC,SAAS,0CAAE,GAAG,KAAI,IAAA,YAAM,EAAC,MAAA,KAAK,CAAC,IAAI,CAAC,SAAS,0CAAE,GAAG,CAAC,EAAE;oBAChE,MAAM,IAAI,GAAG,IAAA,iCAAc,EAAC,MAAA,KAAK,CAAC,IAAI,CAAC,SAAS,0CAAE,GAAG,CAAC,IAAI,CAAC,CAAC;oBAC5D,MAAM,CAAC,SAAS,CAAC,YAAY,IAAI,2BAA2B,IAAI,WAAW,CAAC,CAAC;iBAChF;aACJ;YAED,iBAAiB;YACjB,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,KAAK,SAAS,CAAC,EAAE;gBACvD,MAAM,CAAC,SAAS,CAAC,4CAA4C,CAAC,CAAC;aAClE;YAED,qBAAqB;YACrB,MAAM,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;YAC7C,MAAM,CAAC,WAAW,CAAC,GAAG,EAAE;gBACpB,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;oBACrB,MAAM,CAAC,SAAS,CAAC,GAAG,KAAK,CAAC,IAAI,KAAK,IAAA,4BAAe,EAAC,KAAK,CAAC,GAAG,CAAC,CAAC;gBAClE,CAAC,CAAC,CAAC;YACP,CAAC,CAAC,CAAC;YACH,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YAEvB,oCAAoC;YACpC,MAAM,WAAW,GAAG,IAAA,sCAAyB,EAAC,KAAK,CAAC,CAAC;YACrD,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE;gBACxB,MAAM,CAAC,SAAS,CAAC,qDAAqD,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;aACtG;YAED,eAAe;YACf,IAAI,WAAW,GAAG,YAAY,CAAC;YAC/B,MAAM,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAA,kBAAY,EAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC;YAC5E,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE;gBACzB,WAAW,GAAG,QAAQ,CAClB,WAAW,EACX,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAClC,CAAC;aACL;YACD,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE;gBACxB,WAAW,GAAG,UAAU,WAAW,GAAG,CAAC;aAC1C;YACD,MAAM,CAAC,SAAS,CAAC,gBAAgB,IAAA,iCAAc,EAAC,KAAK,CAAC,IAAI,CAAC,YAAY,WAAW,GAAG,CAAC,CAAC;YAEvF,gBAAgB;YAChB,IAAI,YAAY,GAAG,YAAY,CAAC;YAChC,MAAM,iBAAiB,GAAG,MAAM,CAAC,MAAM,CACnC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAA,kBAAY,EAAC,KAAK,EAAE,UAAU,CAAC,IAAI,IAAA,kBAAY,EAAC,KAAK,EAAE,YAAY,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,CACtG,CAAC;YACF,IAAI,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC9B,YAAY,GAAG,WAAW,CACtB,YAAY,EACZ,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CACvC,CAAC;aACL;YACD,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE;gBACxB,YAAY,GAAG,UAAU,YAAY,GAAG,CAAC;aAC5C;YACD,MAAM,CAAC,SAAS,CAAC,gBAAgB,IAAA,iCAAc,EAAC,KAAK,CAAC,IAAI,CAAC,kBAAkB,YAAY,GAAG,CAAC,CAAC;YAE9F,gBAAgB;YAChB,IAAI,YAAY,GAAG,sBAAsB,CAAC;YAC1C,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE;gBACxB,YAAY,GAAG,UAAU,YAAY,GAAG,CAAC;aAC5C;YACD,MAAM,CAAC,SAAS,CAAC,gBAAgB,IAAA,iCAAc,EAAC,KAAK,CAAC,IAAI,CAAC,kBAAkB,YAAY,GAAG,CAAC,CAAC;QAClG,CAAC,CAAC,CAAC;QACH,OAAO,UAAU,CAAC;IACtB,CAAC;CAAA;AAED,SAAS,WAAW,CAAC,MAAc,EAAE,MAAgB;IACjD,OAAO,GAAG,MAAM;UACV,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;OAC7C,CAAC;AACR,CAAC;AAED,SAAS,QAAQ,CAAC,MAAc,EAAE,MAAgB;IAC9C,OAAO,GAAG,MAAM;UACV,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;OAC7C,CAAC;AACR,CAAC"}

package/plugins/zod/index.d.ts

@@ -1,4 +1,4 @@
-import { DMMF } from '@prisma/generator-helper';
+import type { DMMF } from '@prisma/generator-helper';
 import { PluginOptions } from '@zenstackhq/sdk';
 import { Model } from '@zenstackhq/sdk/ast';
 export declare const name = "Zod";

package/plugins/zod/transformer.d.ts

@@ -15,13 +15,12 @@ export default class Transformer {
     static provider: string;
     private static outputPath;
     private hasJson;
-    private static prismaClientOutputPath;
-    private static isCustomPrismaClientOutputPath;
+    private hasDecimal;
     private project;
+    private zmodel;
     constructor(params: TransformerParams);
     static setOutputPath(outPath: string): void;
     static getOutputPath(): string;
-    static setPrismaClientOutputPath(prismaClientCustomPath: string): void;
     generateEnumSchemas(): Promise<void>;
     generateImportZodStatement(): string;
     generateExportSchemaStatement(name: string, schema: string): string;
@@ -41,6 +40,7 @@ export default class Transformer {
     generateJsonSchemaImplementation(): string;
     generateObjectSchemaImportStatements(): string;
     generateSchemaImports(): string;
+    private generateCommonImport;
     checkIsModelQueryType(type: string): {
         isModelQueryType: boolean;
         modelName: string;
@@ -54,7 +54,7 @@ export default class Transformer {
     wrapWithZodUnion(zodStringFields: string[]): string;
     wrapWithZodObject(zodStringFields: string | string[]): string;
     resolveObjectSchemaName(): string;
-    generateModelSchemas(): Promise<void>;
+    generateInputSchemas(): Promise<void>;
     generateImportStatements(imports: (string | undefined)[]): string;
     resolveSelectIncludeImportAndZodSchemaLine(model: PrismaDMMF.Model): {
         selectImport: string;