zenstack 1.0.0 → 1.0.2

package/plugins/prisma/schema-generator.js

@@ -0,0 +1,336 @@
+"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 });
+const ast_1 = require("@zenstackhq/language/ast");
+const ts_pattern_1 = require("ts-pattern");
+const runtime_1 = require("@zenstackhq/runtime");
+const sdk_1 = require("@zenstackhq/sdk");
+const fs_1 = __importDefault(require("fs"));
+const promises_1 = require("fs/promises");
+const path_1 = __importDefault(require("path"));
+const semver_1 = __importDefault(require("semver"));
+const strip_color_1 = __importDefault(require("strip-color"));
+const _1 = require(".");
+const utils_1 = require("../../language-server/validator/utils");
+const telemetry_1 = __importDefault(require("../../telemetry"));
+const exec_utils_1 = require("../../utils/exec-utils");
+const prisma_builder_1 = require("./prisma-builder");
+const zmodel_code_generator_1 = __importDefault(require("./zmodel-code-generator"));
+const MODEL_PASSTHROUGH_ATTR = '@@prisma.passthrough';
+const FIELD_PASSTHROUGH_ATTR = '@prisma.passthrough';
+/**
+ * Generates Prisma schema file
+ */
+class PrismaSchemaGenerator {
+    constructor() {
+        this.zModelGenerator = new zmodel_code_generator_1.default();
+        this.PRELUDE = `//////////////////////////////////////////////////////////////////////////////////////////////
+// DO NOT MODIFY THIS FILE                                                                  //
+// This file is automatically generated by ZenStack CLI and should not be manually updated. //
+//////////////////////////////////////////////////////////////////////////////////////////////
+
+`;
+    }
+    generate(model, options) {
+        var _a;
+        return __awaiter(this, void 0, void 0, function* () {
+            const warnings = [];
+            const prismaVersion = (0, sdk_1.getPrismaVersion)();
+            if (prismaVersion && semver_1.default.lt(prismaVersion, runtime_1.PRISMA_MINIMUM_VERSION)) {
+                warnings.push(`ZenStack requires Prisma version "${runtime_1.PRISMA_MINIMUM_VERSION}" or higher. Detected version is "${prismaVersion}".`);
+            }
+            const prisma = new prisma_builder_1.PrismaModel();
+            for (const decl of model.declarations) {
+                switch (decl.$type) {
+                    case ast_1.DataSource:
+                        this.generateDataSource(prisma, decl);
+                        break;
+                    case ast_1.Enum:
+                        this.generateEnum(prisma, decl);
+                        break;
+                    case ast_1.DataModel:
+                        this.generateModel(prisma, decl);
+                        break;
+                    case ast_1.GeneratorDecl:
+                        this.generateGenerator(prisma, decl);
+                        break;
+                }
+            }
+            let outFile = (_a = options.output) !== null && _a !== void 0 ? _a : './prisma/schema.prisma';
+            outFile = (0, sdk_1.resolvePath)(outFile, options);
+            if (!fs_1.default.existsSync(path_1.default.dirname(outFile))) {
+                fs_1.default.mkdirSync(path_1.default.dirname(outFile), { recursive: true });
+            }
+            yield (0, promises_1.writeFile)(outFile, this.PRELUDE + prisma.toString());
+            if (options.format === true) {
+                try {
+                    // run 'prisma format'
+                    yield (0, exec_utils_1.execSync)(`npx prisma format --schema ${outFile}`);
+                }
+                catch (_b) {
+                    warnings.push(`Failed to format Prisma schema file`);
+                }
+            }
+            const generateClient = options.generateClient !== false;
+            if (generateClient) {
+                try {
+                    // run 'prisma generate'
+                    yield (0, exec_utils_1.execSync)(`npx prisma generate --schema ${outFile}`, 'ignore');
+                }
+                catch (_c) {
+                    yield this.trackPrismaSchemaError(outFile);
+                    try {
+                        // run 'prisma generate' again with output to the console
+                        yield (0, exec_utils_1.execSync)(`npx prisma generate --schema ${outFile}`);
+                    }
+                    catch (_d) {
+                        // noop
+                    }
+                    throw new sdk_1.PluginError(_1.name, `Failed to run "prisma generate"`);
+                }
+            }
+            return warnings;
+        });
+    }
+    trackPrismaSchemaError(schema) {
+        return __awaiter(this, void 0, void 0, function* () {
+            try {
+                yield (0, sdk_1.getDMMF)({ datamodel: fs_1.default.readFileSync(schema, 'utf-8') });
+            }
+            catch (err) {
+                if (err instanceof Error) {
+                    // eslint-disable-next-line @typescript-eslint/no-var-requires
+                    telemetry_1.default.track('prisma:error', { command: 'generate', message: (0, strip_color_1.default)(err.message) });
+                }
+            }
+        });
+    }
+    generateDataSource(prisma, dataSource) {
+        const fields = dataSource.fields.map((f) => ({
+            name: f.name,
+            text: this.configExprToText(f.value),
+        }));
+        prisma.addDataSource(dataSource.name, fields);
+    }
+    configExprToText(expr) {
+        if ((0, ast_1.isLiteralExpr)(expr)) {
+            return this.literalToText(expr);
+        }
+        else if ((0, ast_1.isInvocationExpr)(expr)) {
+            const fc = this.makeFunctionCall(expr);
+            return fc.toString();
+        }
+        else {
+            return this.configArrayToText(expr);
+        }
+    }
+    configArrayToText(expr) {
+        return ('[' +
+            expr.items
+                .map((item) => {
+                if ((0, ast_1.isLiteralExpr)(item)) {
+                    return this.literalToText(item);
+                }
+                else {
+                    return (item.name +
+                        (item.args.length > 0
+                            ? '(' + item.args.map((arg) => this.configInvocationArgToText(arg)).join(', ') + ')'
+                            : ''));
+                }
+            })
+                .join(', ') +
+            ']');
+    }
+    configInvocationArgToText(arg) {
+        return `${arg.name}: ${this.literalToText(arg.value)}`;
+    }
+    literalToText(expr) {
+        return JSON.stringify(expr.value);
+    }
+    generateGenerator(prisma, decl) {
+        var _a, _b;
+        const generator = prisma.addGenerator(decl.name, decl.fields.map((f) => ({ name: f.name, text: this.configExprToText(f.value) })));
+        // deal with configuring PrismaClient preview features
+        const provider = generator.fields.find((f) => f.name === 'provider');
+        if ((provider === null || provider === void 0 ? void 0 : provider.text) === JSON.stringify('prisma-client-js')) {
+            const prismaVersion = (0, sdk_1.getPrismaVersion)();
+            if (prismaVersion) {
+                const previewFeatures = JSON.parse((_b = (_a = generator.fields.find((f) => f.name === 'previewFeatures')) === null || _a === void 0 ? void 0 : _a.text) !== null && _b !== void 0 ? _b : '[]');
+                if (!Array.isArray(previewFeatures)) {
+                    throw new sdk_1.PluginError(_1.name, 'option "previewFeatures" must be an array');
+                }
+                if (semver_1.default.lt(prismaVersion, '5.0.0')) {
+                    // extendedWhereUnique feature is opt-in pre V5
+                    if (!previewFeatures.includes('extendedWhereUnique')) {
+                        previewFeatures.push('extendedWhereUnique');
+                    }
+                }
+                if (semver_1.default.lt(prismaVersion, '5.0.0')) {
+                    // fieldReference feature is opt-in pre V5
+                    if (!previewFeatures.includes('fieldReference')) {
+                        previewFeatures.push('fieldReference');
+                    }
+                }
+                if (previewFeatures.length > 0) {
+                    const curr = generator.fields.find((f) => f.name === 'previewFeatures');
+                    if (!curr) {
+                        generator.fields.push({ name: 'previewFeatures', text: JSON.stringify(previewFeatures) });
+                    }
+                    else {
+                        curr.text = JSON.stringify(previewFeatures);
+                    }
+                }
+            }
+        }
+    }
+    generateModel(prisma, decl) {
+        const model = decl.isView ? prisma.addView(decl.name) : prisma.addModel(decl.name);
+        for (const field of decl.fields) {
+            this.generateModelField(model, field);
+        }
+        for (const attr of decl.attributes.filter((attr) => this.isPrismaAttribute(attr))) {
+            this.generateContainerAttribute(model, attr);
+        }
+        decl.attributes
+            .filter((attr) => attr.decl.ref && !this.isPrismaAttribute(attr))
+            .forEach((attr) => model.addComment('/// ' + this.zModelGenerator.generateAttribute(attr)));
+        // user defined comments pass-through
+        decl.comments.forEach((c) => model.addComment(c));
+    }
+    isPrismaAttribute(attr) {
+        if (!attr.decl.ref) {
+            return false;
+        }
+        const attrDecl = (0, sdk_1.resolved)(attr.decl);
+        return (!!attrDecl.attributes.find((a) => { var _a; return ((_a = a.decl.ref) === null || _a === void 0 ? void 0 : _a.name) === '@@@prisma'; }) ||
+            // the special pass-through attribute
+            attrDecl.name === MODEL_PASSTHROUGH_ATTR ||
+            attrDecl.name === FIELD_PASSTHROUGH_ATTR);
+    }
+    getUnsupportedFieldType(fieldType) {
+        if (fieldType.unsupported) {
+            const value = (0, utils_1.getStringLiteral)(fieldType.unsupported.value);
+            if (value) {
+                return `Unsupported("${value}")`;
+            }
+            else {
+                return undefined;
+            }
+        }
+        else {
+            return undefined;
+        }
+    }
+    generateModelField(model, field) {
+        var _a, _b;
+        const fieldType = field.type.type || ((_b = (_a = field.type.reference) === null || _a === void 0 ? void 0 : _a.ref) === null || _b === void 0 ? void 0 : _b.name) || this.getUnsupportedFieldType(field.type);
+        if (!fieldType) {
+            throw new sdk_1.PluginError(_1.name, `Field type is not resolved: ${field.$container.name}.${field.name}`);
+        }
+        const type = new prisma_builder_1.ModelFieldType(fieldType, field.type.array, field.type.optional);
+        const attributes = field.attributes
+            .filter((attr) => this.isPrismaAttribute(attr))
+            .map((attr) => this.makeFieldAttribute(attr));
+        const nonPrismaAttributes = field.attributes.filter((attr) => attr.decl.ref && !this.isPrismaAttribute(attr));
+        const documentations = nonPrismaAttributes.map((attr) => '/// ' + this.zModelGenerator.generateAttribute(attr));
+        const result = model.addField(field.name, type, attributes, documentations);
+        // user defined comments pass-through
+        field.comments.forEach((c) => result.addComment(c));
+    }
+    makeFieldAttribute(attr) {
+        const attrName = (0, sdk_1.resolved)(attr.decl).name;
+        if (attrName === FIELD_PASSTHROUGH_ATTR) {
+            const text = (0, sdk_1.getLiteral)(attr.args[0].value);
+            if (text) {
+                return new prisma_builder_1.PassThroughAttribute(text);
+            }
+            else {
+                throw new sdk_1.PluginError(_1.name, `Invalid arguments for ${FIELD_PASSTHROUGH_ATTR} attribute`);
+            }
+        }
+        else {
+            return new prisma_builder_1.FieldAttribute(attrName, attr.args.map((arg) => this.makeAttributeArg(arg)));
+        }
+    }
+    makeAttributeArg(arg) {
+        return new prisma_builder_1.AttributeArg(arg.name, this.makeAttributeArgValue(arg.value));
+    }
+    makeAttributeArgValue(node) {
+        if ((0, ast_1.isLiteralExpr)(node)) {
+            const argType = (0, ts_pattern_1.match)(node.$type)
+                .with(ast_1.StringLiteral, () => 'String')
+                .with(ast_1.NumberLiteral, () => 'Number')
+                .with(ast_1.BooleanLiteral, () => 'Boolean')
+                .exhaustive();
+            return new prisma_builder_1.AttributeArgValue(argType, node.value);
+        }
+        else if ((0, ast_1.isArrayExpr)(node)) {
+            return new prisma_builder_1.AttributeArgValue('Array', new Array(...node.items.map((item) => this.makeAttributeArgValue(item))));
+        }
+        else if ((0, ast_1.isReferenceExpr)(node)) {
+            return new prisma_builder_1.AttributeArgValue('FieldReference', new prisma_builder_1.FieldReference((0, sdk_1.resolved)(node.target).name, node.args.map((arg) => new prisma_builder_1.FieldReferenceArg(arg.name, arg.value))));
+        }
+        else if ((0, ast_1.isInvocationExpr)(node)) {
+            // invocation
+            return new prisma_builder_1.AttributeArgValue('FunctionCall', this.makeFunctionCall(node));
+        }
+        else {
+            throw new sdk_1.PluginError(_1.name, `Unsupported attribute argument expression type: ${node.$type}`);
+        }
+    }
+    makeFunctionCall(node) {
+        return new prisma_builder_1.FunctionCall((0, sdk_1.resolved)(node.function).name, node.args.map((arg) => {
+            if (!(0, ast_1.isLiteralExpr)(arg.value)) {
+                throw new sdk_1.PluginError(_1.name, 'Function call argument must be literal');
+            }
+            return new prisma_builder_1.FunctionCallArg(arg.name, arg.value.value);
+        }));
+    }
+    generateContainerAttribute(container, attr) {
+        const attrName = (0, sdk_1.resolved)(attr.decl).name;
+        if (attrName === MODEL_PASSTHROUGH_ATTR) {
+            const text = (0, sdk_1.getLiteral)(attr.args[0].value);
+            if (text) {
+                container.attributes.push(new prisma_builder_1.PassThroughAttribute(text));
+            }
+        }
+        else {
+            container.attributes.push(new prisma_builder_1.ContainerAttribute(attrName, attr.args.map((arg) => this.makeAttributeArg(arg))));
+        }
+    }
+    generateEnum(prisma, decl) {
+        const _enum = prisma.addEnum(decl.name);
+        for (const field of decl.fields) {
+            this.generateEnumField(_enum, field);
+        }
+        for (const attr of decl.attributes.filter((attr) => this.isPrismaAttribute(attr))) {
+            this.generateContainerAttribute(_enum, attr);
+        }
+        decl.attributes
+            .filter((attr) => attr.decl.ref && !this.isPrismaAttribute(attr))
+            .forEach((attr) => _enum.addComment('/// ' + this.zModelGenerator.generateAttribute(attr)));
+        // user defined comments pass-through
+        decl.comments.forEach((c) => _enum.addComment(c));
+    }
+    generateEnumField(_enum, field) {
+        const attributes = field.attributes
+            .filter((attr) => this.isPrismaAttribute(attr))
+            .map((attr) => this.makeFieldAttribute(attr));
+        const nonPrismaAttributes = field.attributes.filter((attr) => attr.decl.ref && !this.isPrismaAttribute(attr));
+        const documentations = nonPrismaAttributes.map((attr) => '/// ' + this.zModelGenerator.generateAttribute(attr));
+        _enum.addField(field.name, attributes, documentations);
+    }
+}
+exports.default = PrismaSchemaGenerator;
+//# sourceMappingURL=schema-generator.js.map

package/plugins/prisma/schema-generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema-generator.js","sourceRoot":"","sources":["../../../src/plugins/prisma/schema-generator.ts"],"names":[],"mappings":";;;;;;;;;;;;;;AAAA,kDAyBkC;AAClC,2CAAmC;AAEnC,iDAA6D;AAC7D,yCAQyB;AACzB,4CAAoB;AACpB,0CAAwC;AACxC,gDAAwB;AACxB,oDAA4B;AAC5B,8DAAqC;AACrC,wBAAyB;AACzB,iEAAyE;AACzE,gEAAwC;AACxC,uDAAkD;AAClD,qDAgB0B;AAC1B,oFAA0D;AAE1D,MAAM,sBAAsB,GAAG,sBAAsB,CAAC;AACtD,MAAM,sBAAsB,GAAG,qBAAqB,CAAC;AAErD;;GAEG;AACH,MAAqB,qBAAqB;IAA1C;QACY,oBAAe,GAAwB,IAAI,+BAAmB,EAAE,CAAC;QAExD,YAAO,GAAG;;;;;CAK9B,CAAC;IA+VF,CAAC;IA7VS,QAAQ,CAAC,KAAY,EAAE,OAAsB;;;YAC/C,MAAM,QAAQ,GAAa,EAAE,CAAC;YAE9B,MAAM,aAAa,GAAG,IAAA,sBAAgB,GAAE,CAAC;YACzC,IAAI,aAAa,IAAI,gBAAM,CAAC,EAAE,CAAC,aAAa,EAAE,gCAAsB,CAAC,EAAE;gBACnE,QAAQ,CAAC,IAAI,CACT,qCAAqC,gCAAsB,qCAAqC,aAAa,IAAI,CACpH,CAAC;aACL;YAED,MAAM,MAAM,GAAG,IAAI,4BAAW,EAAE,CAAC;YAEjC,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,YAAY,EAAE;gBACnC,QAAQ,IAAI,CAAC,KAAK,EAAE;oBAChB,KAAK,gBAAU;wBACX,IAAI,CAAC,kBAAkB,CAAC,MAAM,EAAE,IAAkB,CAAC,CAAC;wBACpD,MAAM;oBAEV,KAAK,UAAI;wBACL,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,IAAY,CAAC,CAAC;wBACxC,MAAM;oBAEV,KAAK,eAAS;wBACV,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,IAAiB,CAAC,CAAC;wBAC9C,MAAM;oBAEV,KAAK,mBAAa;wBACd,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,IAAqB,CAAC,CAAC;wBACtD,MAAM;iBACb;aACJ;YAED,IAAI,OAAO,GAAG,MAAC,OAAO,CAAC,MAAiB,mCAAI,wBAAwB,CAAC;YACrE,OAAO,GAAG,IAAA,iBAAW,EAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAExC,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,cAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,EAAE;gBACvC,YAAE,CAAC,SAAS,CAAC,cAAI,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;aAC5D;YACD,MAAM,IAAA,oBAAS,EAAC,OAAO,EAAE,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;YAE3D,IAAI,OAAO,CAAC,MAAM,KAAK,IAAI,EAAE;gBACzB,IAAI;oBACA,sBAAsB;oBACtB,MAAM,IAAA,qBAAQ,EAAC,8BAA8B,OAAO,EAAE,CAAC,CAAC;iBAC3D;gBAAC,WAAM;oBACJ,QAAQ,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAC;iBACxD;aACJ;YAED,MAAM,cAAc,GAAG,OAAO,CAAC,cAAc,KAAK,KAAK,CAAC;YAExD,IAAI,cAAc,EAAE;gBAChB,IAAI;oBACA,wBAAwB;oBACxB,MAAM,IAAA,qBAAQ,EAAC,gCAAgC,OAAO,EAAE,EAAE,QAAQ,CAAC,CAAC;iBACvE;gBAAC,WAAM;oBACJ,MAAM,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC;oBAC3C,IAAI;wBACA,yDAAyD;wBACzD,MAAM,IAAA,qBAAQ,EAAC,gCAAgC,OAAO,EAAE,CAAC,CAAC;qBAC7D;oBAAC,WAAM;wBACJ,OAAO;qBACV;oBACD,MAAM,IAAI,iBAAW,CAAC,OAAI,EAAE,iCAAiC,CAAC,CAAC;iBAClE;aACJ;YAED,OAAO,QAAQ,CAAC;;KACnB;IAEa,sBAAsB,CAAC,MAAc;;YAC/C,IAAI;gBACA,MAAM,IAAA,aAAO,EAAC,EAAE,SAAS,EAAE,YAAE,CAAC,YAAY,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;aAClE;YAAC,OAAO,GAAG,EAAE;gBACV,IAAI,GAAG,YAAY,KAAK,EAAE;oBACtB,8DAA8D;oBAC9D,mBAAS,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,IAAA,qBAAU,EAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;iBAC9F;aACJ;QACL,CAAC;KAAA;IAEO,kBAAkB,CAAC,MAAmB,EAAE,UAAsB;QAClE,MAAM,MAAM,GAAkB,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACxD,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,IAAI,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,KAAK,CAAC;SACvC,CAAC,CAAC,CAAC;QACJ,MAAM,CAAC,aAAa,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAClD,CAAC;IAEO,gBAAgB,CAAC,IAAgB;QACrC,IAAI,IAAA,mBAAa,EAAC,IAAI,CAAC,EAAE;YACrB,OAAO,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;SACnC;aAAM,IAAI,IAAA,sBAAgB,EAAC,IAAI,CAAC,EAAE;YAC/B,MAAM,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;YACvC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC;SACxB;aAAM;YACH,OAAO,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;SACvC;IACL,CAAC;IAEO,iBAAiB,CAAC,IAAqB;QAC3C,OAAO,CACH,GAAG;YACH,IAAI,CAAC,KAAK;iBACL,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;gBACV,IAAI,IAAA,mBAAa,EAAC,IAAI,CAAC,EAAE;oBACrB,OAAO,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;iBACnC;qBAAM;oBACH,OAAO,CACH,IAAI,CAAC,IAAI;wBACT,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC;4BACjB,CAAC,CAAC,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,yBAAyB,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,GAAG;4BACpF,CAAC,CAAC,EAAE,CAAC,CACZ,CAAC;iBACL;YACL,CAAC,CAAC;iBACD,IAAI,CAAC,IAAI,CAAC;YACf,GAAG,CACN,CAAC;IACN,CAAC;IAEO,yBAAyB,CAAC,GAAwB;QACtD,OAAO,GAAG,GAAG,CAAC,IAAI,KAAK,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;IAC3D,CAAC;IAEO,aAAa,CAAC,IAAiB;QACnC,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACtC,CAAC;IAEO,iBAAiB,CAAC,MAAmB,EAAE,IAAmB;;QAC9D,MAAM,SAAS,GAAG,MAAM,CAAC,YAAY,CACjC,IAAI,CAAC,IAAI,EACT,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CACnF,CAAC;QAEF,sDAAsD;QACtD,MAAM,QAAQ,GAAG,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;QACrE,IAAI,CAAA,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,CAAE,IAAI,MAAK,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,EAAE;YACvD,MAAM,aAAa,GAAG,IAAA,sBAAgB,GAAE,CAAC;YACzC,IAAI,aAAa,EAAE;gBACf,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAC9B,MAAA,MAAA,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,iBAAiB,CAAC,0CAAE,IAAI,mCAAI,IAAI,CAC3E,CAAC;gBAEF,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE;oBACjC,MAAM,IAAI,iBAAW,CAAC,OAAI,EAAE,2CAA2C,CAAC,CAAC;iBAC5E;gBAED,IAAI,gBAAM,CAAC,EAAE,CAAC,aAAa,EAAE,OAAO,CAAC,EAAE;oBACnC,+CAA+C;oBAC/C,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,qBAAqB,CAAC,EAAE;wBAClD,eAAe,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;qBAC/C;iBACJ;gBAED,IAAI,gBAAM,CAAC,EAAE,CAAC,aAAa,EAAE,OAAO,CAAC,EAAE;oBACnC,0CAA0C;oBAC1C,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE;wBAC7C,eAAe,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;qBAC1C;iBACJ;gBAED,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE;oBAC5B,MAAM,IAAI,GAAG,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,iBAAiB,CAAC,CAAC;oBACxE,IAAI,CAAC,IAAI,EAAE;wBACP,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;qBAC7F;yBAAM;wBACH,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;qBAC/C;iBACJ;aACJ;SACJ;IACL,CAAC;IAEO,aAAa,CAAC,MAAmB,EAAE,IAAe;QACtD,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnF,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,EAAE;YAC7B,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;SACzC;QAED,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,EAAE;YAC/E,IAAI,CAAC,0BAA0B,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;SAChD;QAED,IAAI,CAAC,UAAU;aACV,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;aAChE,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAEhG,qCAAqC;QACrC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;IACtD,CAAC;IAEO,iBAAiB,CAAC,IAAkD;QACxE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE;YAChB,OAAO,KAAK,CAAC;SAChB;QACD,MAAM,QAAQ,GAAG,IAAA,cAAQ,EAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrC,OAAO,CACH,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,WAAC,OAAA,CAAA,MAAA,CAAC,CAAC,IAAI,CAAC,GAAG,0CAAE,IAAI,MAAK,WAAW,CAAA,EAAA,CAAC;YACnE,qCAAqC;YACrC,QAAQ,CAAC,IAAI,KAAK,sBAAsB;YACxC,QAAQ,CAAC,IAAI,KAAK,sBAAsB,CAC3C,CAAC;IACN,CAAC;IAEO,uBAAuB,CAAC,SAA6B;QACzD,IAAI,SAAS,CAAC,WAAW,EAAE;YACvB,MAAM,KAAK,GAAG,IAAA,wBAAgB,EAAC,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAC5D,IAAI,KAAK,EAAE;gBACP,OAAO,gBAAgB,KAAK,IAAI,CAAC;aACpC;iBAAM;gBACH,OAAO,SAAS,CAAC;aACpB;SACJ;aAAM;YACH,OAAO,SAAS,CAAC;SACpB;IACL,CAAC;IAEO,kBAAkB,CAAC,KAAsB,EAAE,KAAqB;;QACpE,MAAM,SAAS,GACX,KAAK,CAAC,IAAI,CAAC,IAAI,KAAI,MAAA,MAAA,KAAK,CAAC,IAAI,CAAC,SAAS,0CAAE,GAAG,0CAAE,IAAI,CAAA,IAAI,IAAI,CAAC,uBAAuB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACnG,IAAI,CAAC,SAAS,EAAE;YACZ,MAAM,IAAI,iBAAW,CAAC,OAAI,EAAE,+BAA+B,KAAK,CAAC,UAAU,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;SACrG;QAED,MAAM,IAAI,GAAG,IAAI,+BAAc,CAAC,SAAS,EAAE,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAElF,MAAM,UAAU,GAAG,KAAK,CAAC,UAAU;aAC9B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;aAC9C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC,CAAC;QAElD,MAAM,mBAAmB,GAAG,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC;QAE9G,MAAM,cAAc,GAAG,mBAAmB,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC;QAEhH,MAAM,MAAM,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,UAAU,EAAE,cAAc,CAAC,CAAC;QAE5E,qCAAqC;QACrC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;IACxD,CAAC;IAEO,kBAAkB,CAAC,IAA6B;QACpD,MAAM,QAAQ,GAAG,IAAA,cAAQ,EAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC;QAC1C,IAAI,QAAQ,KAAK,sBAAsB,EAAE;YACrC,MAAM,IAAI,GAAG,IAAA,gBAAU,EAAS,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;YACpD,IAAI,IAAI,EAAE;gBACN,OAAO,IAAI,qCAA0B,CAAC,IAAI,CAAC,CAAC;aAC/C;iBAAM;gBACH,MAAM,IAAI,iBAAW,CAAC,OAAI,EAAE,yBAAyB,sBAAsB,YAAY,CAAC,CAAC;aAC5F;SACJ;aAAM;YACH,OAAO,IAAI,+BAAoB,CAC3B,QAAQ,EACR,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CACrD,CAAC;SACL;IACL,CAAC;IAEO,gBAAgB,CAAC,GAAiB;QACtC,OAAO,IAAI,6BAAkB,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;IACnF,CAAC;IAEO,qBAAqB,CAAC,IAAgB;QAC1C,IAAI,IAAA,mBAAa,EAAC,IAAI,CAAC,EAAE;YACrB,MAAM,OAAO,GAAG,IAAA,kBAAK,EAAC,IAAI,CAAC,KAAK,CAAC;iBAC5B,IAAI,CAAC,mBAAa,EAAE,GAAG,EAAE,CAAC,QAAiB,CAAC;iBAC5C,IAAI,CAAC,mBAAa,EAAE,GAAG,EAAE,CAAC,QAAiB,CAAC;iBAC5C,IAAI,CAAC,oBAAc,EAAE,GAAG,EAAE,CAAC,SAAkB,CAAC;iBAC9C,UAAU,EAAE,CAAC;YAClB,OAAO,IAAI,kCAAuB,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;SAC3D;aAAM,IAAI,IAAA,iBAAW,EAAC,IAAI,CAAC,EAAE;YAC1B,OAAO,IAAI,kCAAuB,CAC9B,OAAO,EACP,IAAI,KAAK,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC,CAAC,CAC3E,CAAC;SACL;aAAM,IAAI,IAAA,qBAAe,EAAC,IAAI,CAAC,EAAE;YAC9B,OAAO,IAAI,kCAAuB,CAC9B,gBAAgB,EAChB,IAAI,+BAAoB,CACpB,IAAA,cAAQ,EAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,EAC1B,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,kCAAuB,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CAC3E,CACJ,CAAC;SACL;aAAM,IAAI,IAAA,sBAAgB,EAAC,IAAI,CAAC,EAAE;YAC/B,aAAa;YACb,OAAO,IAAI,kCAAuB,CAAC,cAAc,EAAE,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC;SACnF;aAAM;YACH,MAAM,IAAI,iBAAW,CAAC,OAAI,EAAE,mDAAmD,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;SAChG;IACL,CAAC;IAED,gBAAgB,CAAC,IAAoB;QACjC,OAAO,IAAI,6BAAkB,CACzB,IAAA,cAAQ,EAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,EAC5B,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YAClB,IAAI,CAAC,IAAA,mBAAa,EAAC,GAAG,CAAC,KAAK,CAAC,EAAE;gBAC3B,MAAM,IAAI,iBAAW,CAAC,OAAI,EAAE,wCAAwC,CAAC,CAAC;aACzE;YACD,OAAO,IAAI,gCAAqB,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAChE,CAAC,CAAC,CACL,CAAC;IACN,CAAC;IAEO,0BAA0B,CAAC,SAAqC,EAAE,IAAwB;QAC9F,MAAM,QAAQ,GAAG,IAAA,cAAQ,EAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC;QAC1C,IAAI,QAAQ,KAAK,sBAAsB,EAAE;YACrC,MAAM,IAAI,GAAG,IAAA,gBAAU,EAAS,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;YACpD,IAAI,IAAI,EAAE;gBACN,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,qCAA0B,CAAC,IAAI,CAAC,CAAC,CAAC;aACnE;SACJ;aAAM;YACH,SAAS,CAAC,UAAU,CAAC,IAAI,CACrB,IAAI,mCAAoB,CACpB,QAAQ,EACR,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CACrD,CACJ,CAAC;SACL;IACL,CAAC;IAEO,YAAY,CAAC,MAAmB,EAAE,IAAU;QAChD,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAExC,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,EAAE;YAC7B,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;SACxC;QAED,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,EAAE;YAC/E,IAAI,CAAC,0BAA0B,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;SAChD;QAED,IAAI,CAAC,UAAU;aACV,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;aAChE,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAEhG,qCAAqC;QACrC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;IACtD,CAAC;IAEO,iBAAiB,CAAC,KAAiB,EAAE,KAAgB;QACzD,MAAM,UAAU,GAAG,KAAK,CAAC,UAAU;aAC9B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;aAC9C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC,CAAC;QAElD,MAAM,mBAAmB,GAAG,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC;QAE9G,MAAM,cAAc,GAAG,mBAAmB,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC;QAChH,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,UAAU,EAAE,cAAc,CAAC,CAAC;IAC3D,CAAC;CACJ;AAvWD,wCAuWC"}

package/plugins/prisma/zmodel-code-generator.d.ts

@@ -0,0 +1,30 @@
+import { Argument, ArrayExpr, AttributeArg, BinaryExpr, DataModelAttribute, DataModelFieldAttribute, Expression, FieldInitializer, InvocationExpr, LiteralExpr, MemberAccessExpr, ObjectExpr, ReferenceArg, ReferenceExpr, UnaryExpr } from '@zenstackhq/language/ast';
+/**
+ * Options for the generator.
+ */
+export interface ZModelCodeOptions {
+    binaryExprNumberOfSpaces: number;
+    unaryExprNumberOfSpaces: number;
+}
+export default class ZModelCodeGenerator {
+    private readonly options;
+    constructor(options?: Partial<ZModelCodeOptions>);
+    generateAttribute(ast: DataModelAttribute | DataModelFieldAttribute): string;
+    generateAttributeArg(ast: AttributeArg): string;
+    generateExpression(ast: Expression): string;
+    generateObjectExpr(ast: ObjectExpr): string;
+    generateObjectField(field: FieldInitializer): string;
+    generateArrayExpr(ast: ArrayExpr): string;
+    generateLiteralExpr(ast: LiteralExpr): string;
+    generateUnaryExpr(ast: UnaryExpr): string;
+    generateBinaryExpr(ast: BinaryExpr): string;
+    generateReferenceExpr(ast: ReferenceExpr): string;
+    generateReferenceArg(ast: ReferenceArg): string;
+    generateMemberExpr(ast: MemberAccessExpr): string;
+    generateInvocationExpr(ast: InvocationExpr): string;
+    generateArgument(ast: Argument): string;
+    private get binaryExprSpace();
+    private get unaryExprSpace();
+    private isParenthesesNeededForBinaryExpr;
+    private isCollectionPredicateOperator;
+}

package/plugins/prisma/zmodel-code-generator.js

@@ -0,0 +1,124 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const ast_1 = require("@zenstackhq/language/ast");
+const sdk_1 = require("@zenstackhq/sdk");
+class ZModelCodeGenerator {
+    constructor(options) {
+        var _a, _b;
+        this.options = {
+            binaryExprNumberOfSpaces: (_a = options === null || options === void 0 ? void 0 : options.binaryExprNumberOfSpaces) !== null && _a !== void 0 ? _a : 1,
+            unaryExprNumberOfSpaces: (_b = options === null || options === void 0 ? void 0 : options.unaryExprNumberOfSpaces) !== null && _b !== void 0 ? _b : 0,
+        };
+    }
+    generateAttribute(ast) {
+        const args = ast.args.length ? `(${ast.args.map((x) => this.generateAttributeArg(x)).join(', ')})` : '';
+        return `${(0, sdk_1.resolved)(ast.decl).name}${args}`;
+    }
+    generateAttributeArg(ast) {
+        if (ast.name) {
+            return `${ast.name}: ${this.generateExpression(ast.value)}`;
+        }
+        else {
+            return this.generateExpression(ast.value);
+        }
+    }
+    generateExpression(ast) {
+        switch (ast.$type) {
+            case ast_1.StringLiteral:
+            case ast_1.NumberLiteral:
+            case ast_1.BooleanLiteral:
+                return this.generateLiteralExpr(ast);
+            case ast_1.UnaryExpr:
+                return this.generateUnaryExpr(ast);
+            case ast_1.ArrayExpr:
+                return this.generateArrayExpr(ast);
+            case ast_1.BinaryExpr:
+                return this.generateBinaryExpr(ast);
+            case ast_1.ReferenceExpr:
+                return this.generateReferenceExpr(ast);
+            case ast_1.MemberAccessExpr:
+                return this.generateMemberExpr(ast);
+            case ast_1.InvocationExpr:
+                return this.generateInvocationExpr(ast);
+            case ast_1.ObjectExpr:
+                return this.generateObjectExpr(ast);
+            case ast_1.NullExpr:
+            case ast_1.ThisExpr:
+                return ast.value;
+            default:
+                throw new Error(`Not implemented: ${ast}`);
+        }
+    }
+    generateObjectExpr(ast) {
+        return `{ ${ast.fields.map((field) => this.generateObjectField(field)).join(', ')} }`;
+    }
+    generateObjectField(field) {
+        return `${field.name}: ${this.generateExpression(field.value)}`;
+    }
+    generateArrayExpr(ast) {
+        return `[${ast.items.map((item) => this.generateExpression(item)).join(', ')}]`;
+    }
+    generateLiteralExpr(ast) {
+        return ast.$type === ast_1.StringLiteral ? `'${ast.value}'` : ast.value.toString();
+    }
+    generateUnaryExpr(ast) {
+        return `${ast.operator}${this.unaryExprSpace}${this.generateExpression(ast.operand)}`;
+    }
+    generateBinaryExpr(ast) {
+        const operator = ast.operator;
+        const isCollectionPredicate = this.isCollectionPredicateOperator(operator);
+        const rightExpr = this.generateExpression(ast.right);
+        const { left: isLeftParenthesis, right: isRightParenthesis } = this.isParenthesesNeededForBinaryExpr(ast);
+        return `${isLeftParenthesis ? '(' : ''}${this.generateExpression(ast.left)}${isLeftParenthesis ? ')' : ''}${this.binaryExprSpace}${operator}${this.binaryExprSpace}${isRightParenthesis ? '(' : ''}${isCollectionPredicate ? `[${rightExpr}]` : rightExpr}${isRightParenthesis ? ')' : ''}`;
+    }
+    generateReferenceExpr(ast) {
+        var _a;
+        const args = ast.args.length ? `(${ast.args.map((x) => this.generateReferenceArg(x)).join(', ')})` : '';
+        return `${(_a = ast.target.ref) === null || _a === void 0 ? void 0 : _a.name}${args}`;
+    }
+    generateReferenceArg(ast) {
+        return `${ast.name}:${ast.value}`;
+    }
+    generateMemberExpr(ast) {
+        var _a;
+        return `${this.generateExpression(ast.operand)}.${(_a = ast.member.ref) === null || _a === void 0 ? void 0 : _a.name}`;
+    }
+    generateInvocationExpr(ast) {
+        var _a;
+        return `${(_a = ast.function.ref) === null || _a === void 0 ? void 0 : _a.name}(${ast.args.map((x) => this.generateArgument(x)).join(', ')})`;
+    }
+    generateArgument(ast) {
+        return `${ast.name && ':'} ${this.generateExpression(ast.value)}`;
+    }
+    get binaryExprSpace() {
+        return ' '.repeat(this.options.binaryExprNumberOfSpaces);
+    }
+    get unaryExprSpace() {
+        return ' '.repeat(this.options.unaryExprNumberOfSpaces);
+    }
+    isParenthesesNeededForBinaryExpr(ast) {
+        const result = { left: false, right: false };
+        const operator = ast.operator;
+        const isCollectionPredicate = this.isCollectionPredicateOperator(operator);
+        const currentPriority = ast_1.BinaryExprOperatorPriority[operator];
+        if (ast.left.$type === ast_1.BinaryExpr &&
+            ast_1.BinaryExprOperatorPriority[ast.left['operator']] < currentPriority) {
+            result.left = true;
+        }
+        /**
+         *  1 collection predicate operator has [] around the right operand, no need to add parenthesis.
+         *  2 grammar is left associative, so if the right operand has the same priority still need to add parenthesis.
+         **/
+        if (!isCollectionPredicate &&
+            ast.right.$type === ast_1.BinaryExpr &&
+            ast_1.BinaryExprOperatorPriority[ast.right['operator']] <= currentPriority) {
+            result.right = true;
+        }
+        return result;
+    }
+    isCollectionPredicateOperator(op) {
+        return ['?', '!', '^'].includes(op);
+    }
+}
+exports.default = ZModelCodeGenerator;
+//# sourceMappingURL=zmodel-code-generator.js.map

package/plugins/prisma/zmodel-code-generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"zmodel-code-generator.js","sourceRoot":"","sources":["../../../src/plugins/prisma/zmodel-code-generator.ts"],"names":[],"mappings":";;AAAA,kDAsBkC;AAClC,yCAA2C;AAU3C,MAAqB,mBAAmB;IAEpC,YAAY,OAAoC;;QAC5C,IAAI,CAAC,OAAO,GAAG;YACX,wBAAwB,EAAE,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,wBAAwB,mCAAI,CAAC;YAChE,uBAAuB,EAAE,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,uBAAuB,mCAAI,CAAC;SACjE,CAAC;IACN,CAAC;IACD,iBAAiB,CAAC,GAAiD;QAC/D,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;QACxG,OAAO,GAAG,IAAA,cAAQ,EAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,GAAG,IAAI,EAAE,CAAC;IAC/C,CAAC;IAED,oBAAoB,CAAC,GAAiB;QAClC,IAAI,GAAG,CAAC,IAAI,EAAE;YACV,OAAO,GAAG,GAAG,CAAC,IAAI,KAAK,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;SAC/D;aAAM;YACH,OAAO,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;SAC7C;IACL,CAAC;IAED,kBAAkB,CAAC,GAAe;QAC9B,QAAQ,GAAG,CAAC,KAAK,EAAE;YACf,KAAK,mBAAa,CAAC;YACnB,KAAK,mBAAa,CAAC;YACnB,KAAK,oBAAc;gBACf,OAAO,IAAI,CAAC,mBAAmB,CAAC,GAAkB,CAAC,CAAC;YACxD,KAAK,eAAS;gBACV,OAAO,IAAI,CAAC,iBAAiB,CAAC,GAAgB,CAAC,CAAC;YACpD,KAAK,eAAS;gBACV,OAAO,IAAI,CAAC,iBAAiB,CAAC,GAAgB,CAAC,CAAC;YACpD,KAAK,gBAAU;gBACX,OAAO,IAAI,CAAC,kBAAkB,CAAC,GAAiB,CAAC,CAAC;YACtD,KAAK,mBAAa;gBACd,OAAO,IAAI,CAAC,qBAAqB,CAAC,GAAoB,CAAC,CAAC;YAC5D,KAAK,sBAAgB;gBACjB,OAAO,IAAI,CAAC,kBAAkB,CAAC,GAAuB,CAAC,CAAC;YAC5D,KAAK,oBAAc;gBACf,OAAO,IAAI,CAAC,sBAAsB,CAAC,GAAqB,CAAC,CAAC;YAC9D,KAAK,gBAAU;gBACX,OAAO,IAAI,CAAC,kBAAkB,CAAC,GAAiB,CAAC,CAAC;YACtD,KAAK,cAAQ,CAAC;YACd,KAAK,cAAQ;gBACT,OAAQ,GAA2B,CAAC,KAAK,CAAC;YAC9C;gBACI,MAAM,IAAI,KAAK,CAAC,oBAAoB,GAAG,EAAE,CAAC,CAAC;SAClD;IACL,CAAC;IAED,kBAAkB,CAAC,GAAe;QAC9B,OAAO,KAAK,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;IAC1F,CAAC;IAED,mBAAmB,CAAC,KAAuB;QACvC,OAAO,GAAG,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;IACpE,CAAC;IAED,iBAAiB,CAAC,GAAc;QAC5B,OAAO,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;IACpF,CAAC;IAED,mBAAmB,CAAC,GAAgB;QAChC,OAAO,GAAG,CAAC,KAAK,KAAK,mBAAa,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;IACjF,CAAC;IAED,iBAAiB,CAAC,GAAc;QAC5B,OAAO,GAAG,GAAG,CAAC,QAAQ,GAAG,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;IAC1F,CAAC;IAED,kBAAkB,CAAC,GAAe;QAC9B,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC;QAC9B,MAAM,qBAAqB,GAAG,IAAI,CAAC,6BAA6B,CAAC,QAAQ,CAAC,CAAC;QAC3E,MAAM,SAAS,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAErD,MAAM,EAAE,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,kBAAkB,EAAE,GAAG,IAAI,CAAC,gCAAgC,CAAC,GAAG,CAAC,CAAC;QAE1G,OAAO,GAAG,iBAAiB,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,iBAAiB,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GACrG,IAAI,CAAC,eACT,GAAG,QAAQ,GAAG,IAAI,CAAC,eAAe,GAAG,kBAAkB,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAC9D,qBAAqB,CAAC,CAAC,CAAC,IAAI,SAAS,GAAG,CAAC,CAAC,CAAC,SAC/C,GAAG,kBAAkB,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;IACvC,CAAC;IAED,qBAAqB,CAAC,GAAkB;;QACpC,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;QACxG,OAAO,GAAG,MAAA,GAAG,CAAC,MAAM,CAAC,GAAG,0CAAE,IAAI,GAAG,IAAI,EAAE,CAAC;IAC5C,CAAC;IAED,oBAAoB,CAAC,GAAiB;QAClC,OAAO,GAAG,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;IACtC,CAAC;IAED,kBAAkB,CAAC,GAAqB;;QACpC,OAAO,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,MAAA,GAAG,CAAC,MAAM,CAAC,GAAG,0CAAE,IAAI,EAAE,CAAC;IAC7E,CAAC;IAED,sBAAsB,CAAC,GAAmB;;QACtC,OAAO,GAAG,MAAA,GAAG,CAAC,QAAQ,CAAC,GAAG,0CAAE,IAAI,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;IACpG,CAAC;IAED,gBAAgB,CAAC,GAAa;QAC1B,OAAO,GAAG,GAAG,CAAC,IAAI,IAAI,GAAG,IAAI,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;IACtE,CAAC;IAED,IAAY,eAAe;QACvB,OAAO,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,wBAAwB,CAAC,CAAC;IAC7D,CAAC;IAED,IAAY,cAAc;QACtB,OAAO,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC;IAC5D,CAAC;IAEO,gCAAgC,CAAC,GAAe;QACpD,MAAM,MAAM,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;QAC7C,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC;QAC9B,MAAM,qBAAqB,GAAG,IAAI,CAAC,6BAA6B,CAAC,QAAQ,CAAC,CAAC;QAE3E,MAAM,eAAe,GAAG,gCAA0B,CAAC,QAAQ,CAAC,CAAC;QAE7D,IACI,GAAG,CAAC,IAAI,CAAC,KAAK,KAAK,gBAAU;YAC7B,gCAA0B,CAAE,GAAG,CAAC,IAAmB,CAAC,UAAU,CAAC,CAAC,GAAG,eAAe,EACpF;YACE,MAAM,CAAC,IAAI,GAAG,IAAI,CAAC;SACtB;QACD;;;YAGI;QACJ,IACI,CAAC,qBAAqB;YACtB,GAAG,CAAC,KAAK,CAAC,KAAK,KAAK,gBAAU;YAC9B,gCAA0B,CAAE,GAAG,CAAC,KAAoB,CAAC,UAAU,CAAC,CAAC,IAAI,eAAe,EACtF;YACE,MAAM,CAAC,KAAK,GAAG,IAAI,CAAC;SACvB;QAED,OAAO,MAAM,CAAC;IAClB,CAAC;IAEO,6BAA6B,CAAC,EAA0B;QAC5D,OAAO,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACxC,CAAC;CACJ;AA/ID,sCA+IC"}

package/plugins/zod/generator.d.ts

@@ -0,0 +1,4 @@
+import { DMMF } from '@prisma/generator-helper';
+import { PluginGlobalOptions, PluginOptions } from '@zenstackhq/sdk';
+import { Model } from '@zenstackhq/sdk/ast';
+export declare function generate(model: Model, options: PluginOptions, dmmf: DMMF.Document, globalOptions?: PluginGlobalOptions): Promise<void>;