zenstack 1.0.0 → 1.0.1

package/plugins/zod/generator.js

@@ -0,0 +1,254 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generate = void 0;
+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 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, globalOptions) {
+    var _a, _b;
+    return __awaiter(this, void 0, void 0, function* () {
+        let output = options.output;
+        if (!output) {
+            const defaultOutputFolder = (0, plugin_utils_1.getDefaultOutputFolder)(globalOptions);
+            if (defaultOutputFolder) {
+                output = path_1.default.join(defaultOutputFolder, 'zod');
+            }
+            else {
+                output = './generated/zod';
+            }
+        }
+        output = (0, sdk_1.resolvePath)(output, options);
+        yield handleGeneratorOutputValue(output);
+        const prismaClientDmmf = dmmf;
+        const modelOperations = prismaClientDmmf.mappings.modelOperations;
+        const inputObjectTypes = prismaClientDmmf.schema.inputObjectTypes.prisma;
+        const outputObjectTypes = prismaClientDmmf.schema.outputObjectTypes.prisma;
+        const models = prismaClientDmmf.datamodel.models;
+        // whether Prisma's Unchecked* series of input types should be generated
+        const generateUnchecked = options.noUncheckedInput !== true;
+        const project = (0, sdk_1.createProject)();
+        // 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);
+        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, generateUnchecked);
+            // input schemas
+            const transformer = new transformer_1.default({
+                models,
+                modelOperations,
+                aggregateOperationSupport,
+                project,
+                zmodel: model,
+                inputObjectTypes,
+            });
+            yield transformer.generateInputSchemas(generateUnchecked);
+        }
+        // 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
+        let shouldCompile = true;
+        if (typeof options.compile === 'boolean') {
+            // explicit override
+            shouldCompile = options.compile;
+        }
+        else if (globalOptions) {
+            // from CLI or config file
+            shouldCompile = globalOptions.compile;
+        }
+        if (!shouldCompile || options.preserveTsFiles === true) {
+            // save ts files
+            yield (0, sdk_1.saveProject)(project);
+        }
+        if (shouldCompile) {
+            yield (0, sdk_1.emitProject)(project);
+        }
+    });
+}
+exports.generate = generate;
+function handleGeneratorOutputValue(output) {
+    return __awaiter(this, void 0, void 0, function* () {
+        // create the output directory and delete contents that might exist from a previous run
+        yield fs_1.promises.mkdir(output, { recursive: true });
+        const isRemoveContentsOnly = true;
+        yield (0, removeDir_1.default)(output, isRemoveContentsOnly);
+        transformer_1.default.setOutputPath(output);
+    });
+}
+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()}).passthrough()]);
+`, { 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) => (0, upper_case_first_1.upperCaseFirst)(enumItem.name));
+        transformer_1.default.enumNames = enumNames !== null && enumNames !== void 0 ? enumNames : [];
+        const transformer = new transformer_1.default({
+            enumTypes,
+            project,
+            zmodel,
+            inputObjectTypes: [],
+        });
+        yield transformer.generateEnumSchemas();
+    });
+}
+function generateObjectSchemas(inputObjectTypes, project, output, zmodel, generateUnchecked) {
+    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;
+            if (!generateUnchecked && name.includes('Unchecked')) {
+                continue;
+            }
+            const transformer = new transformer_1.default({ name, fields, project, zmodel, inputObjectTypes });
+            const moduleName = transformer.generateObjectSchema(generateUnchecked);
+            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(project, zmodel, output) {
+    return __awaiter(this, void 0, void 0, function* () {
+        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; 
+            // scalar fields only
+            return !(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, sdk_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
+            const importedEnumSchemas = new Set();
+            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);
+                    if (!importedEnumSchemas.has(name)) {
+                        writer.writeLine(`import { ${name}Schema } from '../enums/${name}.schema';`);
+                        importedEnumSchemas.add(name);
+                    }
+                }
+            }
+            // import Decimal
+            if (fields.some((field) => field.type.type === 'Decimal')) {
+                writer.writeLine(`import { DecimalSchema } from '../common';`);
+                writer.writeLine(`import { Decimal } from 'decimal.js';`);
+            }
+            // 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<T, D extends z.ZodTypeDef>(schema: z.ZodType<T, D, T>) { 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};`);
+        });
+        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

@@ -0,0 +1 @@
+{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../../src/plugins/zod/generator.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AACA,yCAcyB;AACzB,6CAAiH;AACjH,+DAA4G;AAC5G,2BAAoC;AACpC,qCAA4C;AAC5C,gDAAwB;AAExB,uDAAkD;AAClD,kDAAyD;AACzD,gEAAwC;AACxC,kEAA0C;AAC1C,mDAAgF;AAEhF,SAAsB,QAAQ,CAC1B,KAAY,EACZ,OAAsB,EACtB,IAAmB,EACnB,aAAmC;;;QAEnC,IAAI,MAAM,GAAG,OAAO,CAAC,MAAgB,CAAC;QACtC,IAAI,CAAC,MAAM,EAAE;YACT,MAAM,mBAAmB,GAAG,IAAA,qCAAsB,EAAC,aAAa,CAAC,CAAC;YAClE,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,wEAAwE;QACxE,MAAM,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,KAAK,IAAI,CAAC;QAE5D,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,EAAE,iBAAiB,CAAC,CAAC;YAEzF,gBAAgB;YAChB,MAAM,WAAW,GAAG,IAAI,qBAAW,CAAC;gBAChC,MAAM;gBACN,eAAe;gBACf,yBAAyB;gBACzB,OAAO;gBACP,MAAM,EAAE,KAAK;gBACb,gBAAgB;aACnB,CAAC,CAAC;YACH,MAAM,WAAW,CAAC,oBAAoB,CAAC,iBAAiB,CAAC,CAAC;SAC7D;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,IAAI,aAAa,GAAG,IAAI,CAAC;QACzB,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,SAAS,EAAE;YACtC,oBAAoB;YACpB,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC;SACnC;aAAM,IAAI,aAAa,EAAE;YACtB,0BAA0B;YAC1B,aAAa,GAAG,aAAa,CAAC,OAAO,CAAC;SACzC;QAED,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;AA5FD,4BA4FC;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;;;CAGP,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,IAAA,iCAAc,EAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;QAC7E,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;YACN,gBAAgB,EAAE,EAAE;SACvB,CAAC,CAAC;QACH,MAAM,WAAW,CAAC,mBAAmB,EAAE,CAAC;IAC5C,CAAC;CAAA;AAED,SAAe,qBAAqB,CAChC,gBAAkC,EAClC,OAAgB,EAChB,MAAc,EACd,MAAa,EACb,iBAA0B;;;QAE1B,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,IAAI,CAAC,iBAAiB,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE;gBAClD,SAAS;aACZ;YACD,MAAM,WAAW,GAAG,IAAI,qBAAW,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,gBAAgB,EAAE,CAAC,CAAC;YACzF,MAAM,UAAU,GAAG,WAAW,CAAC,oBAAoB,CAAC,iBAAiB,CAAC,CAAC;YACvE,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;YACN,qBAAqB;YACrB,OAAA,CAAC,IAAA,iBAAW,EAAC,MAAA,KAAK,CAAC,IAAI,CAAC,SAAS,0CAAE,GAAG,CAAC,IAAI,CAAC,IAAA,uBAAiB,EAAC,KAAK,CAAC,CAAA,EAAA,CAC3E,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,kBAAY,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,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAU,CAAC;YAC9C,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,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;wBAChC,MAAM,CAAC,SAAS,CAAC,YAAY,IAAI,2BAA2B,IAAI,WAAW,CAAC,CAAC;wBAC7E,mBAAmB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;qBACjC;iBACJ;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;gBAC/D,MAAM,CAAC,SAAS,CAAC,uCAAuC,CAAC,CAAC;aAC7D;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,CACZ,yFAAyF,WAAW,CAAC,IAAI,CACrG,IAAI,CACP,KAAK,CACT,CAAC;aACL;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

@@ -0,0 +1,4 @@
+import { PluginFunction } from '@zenstackhq/sdk';
+export declare const name = "Zod";
+declare const run: PluginFunction;
+export default run;

package/plugins/zod/index.js

@@ -0,0 +1,24 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.name = void 0;
+const tiny_invariant_1 = __importDefault(require("tiny-invariant"));
+const generator_1 = require("./generator");
+exports.name = 'Zod';
+const run = (model, options, dmmf, globalOptions) => __awaiter(void 0, void 0, void 0, function* () {
+    (0, tiny_invariant_1.default)(dmmf);
+    return (0, generator_1.generate)(model, options, dmmf, globalOptions);
+});
+exports.default = run;
+//# sourceMappingURL=index.js.map

package/plugins/zod/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/plugins/zod/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AACA,oEAAuC;AACvC,2CAAuC;AAE1B,QAAA,IAAI,GAAG,KAAK,CAAC;AAE1B,MAAM,GAAG,GAAmB,CAAO,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE;IACtE,IAAA,wBAAS,EAAC,IAAI,CAAC,CAAC;IAChB,OAAO,IAAA,oBAAQ,EAAC,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,CAAC,CAAC;AACzD,CAAC,CAAA,CAAC;AAEF,kBAAe,GAAG,CAAC"}

package/plugins/zod/transformer.d.ts

@@ -0,0 +1,68 @@
+import type { DMMF as PrismaDMMF } from '@prisma/generator-helper';
+import { AggregateOperationSupport, TransformerParams } from './types';
+export default class Transformer {
+    name: string;
+    originalName: string;
+    fields: PrismaDMMF.SchemaArg[];
+    schemaImports: Set<string>;
+    models: PrismaDMMF.Model[];
+    modelOperations: PrismaDMMF.ModelMapping[];
+    aggregateOperationSupport: AggregateOperationSupport;
+    enumTypes: PrismaDMMF.SchemaEnum[];
+    static enumNames: string[];
+    static rawOpsMap: {
+        [name: string]: string;
+    };
+    static provider: string;
+    private static outputPath;
+    private hasJson;
+    private hasDecimal;
+    private project;
+    private zmodel;
+    private inputObjectTypes;
+    constructor(params: TransformerParams);
+    static setOutputPath(outPath: string): void;
+    static getOutputPath(): string;
+    generateEnumSchemas(): Promise<void>;
+    generateImportZodStatement(): string;
+    generateExportSchemaStatement(name: string, schema: string): string;
+    generateObjectSchema(generateUnchecked: boolean): string;
+    generateObjectSchemaFields(generateUnchecked: boolean): string[];
+    generateObjectSchemaField(field: PrismaDMMF.SchemaArg, generateUnchecked: boolean): [string, PrismaDMMF.SchemaArg, boolean][];
+    wrapWithZodValidators(mainValidators: string | string[], field: PrismaDMMF.SchemaArg, inputType: PrismaDMMF.SchemaArgInputType): string;
+    addSchemaImport(name: string): void;
+    generatePrismaStringLine(field: PrismaDMMF.SchemaArg, inputType: PrismaDMMF.SchemaArgInputType, inputsLength: number): string;
+    generateFieldValidators(zodStringWithMainType: string, field: PrismaDMMF.SchemaArg): string;
+    prepareObjectSchema(zodObjectSchemaFields: string[]): string;
+    generateExportObjectSchemaStatement(schema: string): string;
+    addFinalWrappers({ zodStringFields }: {
+        zodStringFields: string[];
+    }): string;
+    generateImportPrismaStatement(): string;
+    generateJsonSchemaImplementation(): string;
+    generateObjectSchemaImportStatements(): string;
+    generateSchemaImports(): string;
+    private generateCommonImport;
+    checkIsModelQueryType(type: string): {
+        isModelQueryType: boolean;
+        modelName: string;
+        queryName: string;
+    } | {
+        isModelQueryType: boolean;
+        modelName?: undefined;
+        queryName?: undefined;
+    };
+    resolveModelQuerySchemaName(modelName: string, queryName: string): string;
+    wrapWithZodUnion(zodStringFields: string[]): string;
+    wrapWithZodObject(zodStringFields: string | string[]): string;
+    generateInputSchemas(generateUnchecked: boolean): Promise<void>;
+    generateImportStatements(imports: (string | undefined)[]): string;
+    resolveSelectIncludeImportAndZodSchemaLine(model: PrismaDMMF.Model): {
+        selectImport: string;
+        includeImport: string;
+        selectZodSchemaLine: string;
+        includeZodSchemaLine: string;
+        selectZodSchemaLineLazy: string;
+        includeZodSchemaLineLazy: string;
+    };
+}