zenstack 1.0.0 → 1.0.16

package/plugins/access-policy/zod-schema-generator.js

@@ -0,0 +1,158 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ZodSchemaGenerator = void 0;
+const ast_1 = require("@zenstackhq/language/ast");
+const sdk_1 = require("@zenstackhq/sdk");
+const change_case_1 = require("change-case");
+const ast_utils_1 = require("../../utils/ast-utils");
+/**
+ * Writes Zod schema for data models.
+ */
+class ZodSchemaGenerator {
+    generate(writer, models) {
+        writer.inlineBlock(() => {
+            models.forEach((model) => {
+                const fields = model.fields.filter((field) => {
+                    var _a;
+                    return !sdk_1.AUXILIARY_FIELDS.includes(field.name) &&
+                        // scalar fields only
+                        !(0, ast_1.isDataModelField)((_a = field.type.reference) === null || _a === void 0 ? void 0 : _a.ref) &&
+                        this.hasValidationAttributes(field);
+                });
+                if (fields.length === 0) {
+                    return;
+                }
+                writer.write(`${(0, change_case_1.camelCase)(model.name)}: z.object(`);
+                writer.inlineBlock(() => {
+                    fields.forEach((field) => {
+                        writer.writeLine(`${field.name}: ${this.makeFieldValidator(field)},`);
+                    });
+                });
+                writer.writeLine(').partial(),');
+            });
+        });
+    }
+    hasValidationAttributes(field) {
+        return field.attributes.some((attr) => ast_utils_1.VALIDATION_ATTRIBUTES.includes(attr.decl.$refText));
+    }
+    makeFieldValidator(field) {
+        var _a;
+        let schema = this.makeZodSchema(field);
+        // translate field constraint attributes to zod schema
+        for (const attr of field.attributes) {
+            switch ((_a = attr.decl.ref) === null || _a === void 0 ? void 0 : _a.name) {
+                case '@length': {
+                    const min = this.getAttrLiteralArg(attr, 'min');
+                    if (min) {
+                        schema += `.min(${min})`;
+                    }
+                    const max = this.getAttrLiteralArg(attr, 'max');
+                    if (max) {
+                        schema += `.max(${max})`;
+                    }
+                    break;
+                }
+                case '@regex': {
+                    const expr = this.getAttrLiteralArg(attr, 'regex');
+                    if (expr) {
+                        schema += `.regex(/${expr}/)`;
+                    }
+                    break;
+                }
+                case '@startsWith': {
+                    const text = this.getAttrLiteralArg(attr, 'text');
+                    if (text) {
+                        schema += `.startsWith(${JSON.stringify(text)})`;
+                    }
+                    break;
+                }
+                case '@endsWith': {
+                    const text = this.getAttrLiteralArg(attr, 'text');
+                    if (text) {
+                        schema += `.endsWith(${JSON.stringify(text)})`;
+                    }
+                    break;
+                }
+                case '@email': {
+                    schema += `.email()`;
+                    break;
+                }
+                case '@url': {
+                    schema += `.url()`;
+                    break;
+                }
+                case '@datetime': {
+                    schema += `.datetime({ offset: true })`;
+                    break;
+                }
+                case '@gt': {
+                    const value = this.getAttrLiteralArg(attr, 'value');
+                    if (value !== undefined) {
+                        schema += `.gt(${value})`;
+                    }
+                    break;
+                }
+                case '@gte': {
+                    const value = this.getAttrLiteralArg(attr, 'value');
+                    if (value !== undefined) {
+                        schema += `.gte(${value})`;
+                    }
+                    break;
+                }
+                case '@lt': {
+                    const value = this.getAttrLiteralArg(attr, 'value');
+                    if (value !== undefined) {
+                        schema += `.lt(${value})`;
+                    }
+                    break;
+                }
+                case '@lte': {
+                    const value = this.getAttrLiteralArg(attr, 'value');
+                    if (value !== undefined) {
+                        schema += `.lte(${value})`;
+                    }
+                    break;
+                }
+            }
+        }
+        if (field.type.optional) {
+            schema += '.nullable()';
+        }
+        return schema;
+    }
+    makeZodSchema(field) {
+        let schema;
+        switch (field.type.type) {
+            case 'Int':
+            case 'Float':
+            case 'Decimal':
+                schema = 'z.number()';
+                break;
+            case 'BigInt':
+                schema = 'z.bigint()';
+                break;
+            case 'String':
+                schema = 'z.string()';
+                break;
+            case 'Boolean':
+                schema = 'z.boolean()';
+                break;
+            case 'DateTime':
+                schema = 'z.date()';
+                break;
+            default:
+                schema = 'z.any()';
+                break;
+        }
+        if (field.type.array) {
+            schema = `z.array(${schema})`;
+        }
+        return schema;
+    }
+    getAttrLiteralArg(attr, paramName) {
+        const arg = attr.args.find((arg) => { var _a; return ((_a = arg.$resolvedParam) === null || _a === void 0 ? void 0 : _a.name) === paramName; });
+        return arg && (0, sdk_1.getLiteral)(arg.value);
+    }
+}
+exports.ZodSchemaGenerator = ZodSchemaGenerator;
+//# sourceMappingURL=zod-schema-generator.js.map

package/plugins/access-policy/zod-schema-generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"zod-schema-generator.js","sourceRoot":"","sources":["../../../src/plugins/access-policy/zod-schema-generator.ts"],"names":[],"mappings":";;;AAAA,kDAAgH;AAChH,yCAA+D;AAC/D,6CAAwC;AAExC,qDAA8D;AAE9D;;GAEG;AACH,MAAa,kBAAkB;IAC3B,QAAQ,CAAC,MAAuB,EAAE,MAAmB;QACjD,MAAM,CAAC,WAAW,CAAC,GAAG,EAAE;YACpB,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;gBACrB,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,CAC9B,CAAC,KAAK,EAAE,EAAE;;oBACN,OAAA,CAAC,sBAAgB,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC;wBACtC,qBAAqB;wBACrB,CAAC,IAAA,sBAAgB,EAAC,MAAA,KAAK,CAAC,IAAI,CAAC,SAAS,0CAAE,GAAG,CAAC;wBAC5C,IAAI,CAAC,uBAAuB,CAAC,KAAK,CAAC,CAAA;iBAAA,CAC1C,CAAC;gBAEF,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE;oBACrB,OAAO;iBACV;gBAED,MAAM,CAAC,KAAK,CAAC,GAAG,IAAA,uBAAS,EAAC,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;gBACpD,MAAM,CAAC,WAAW,CAAC,GAAG,EAAE;oBACpB,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;wBACrB,MAAM,CAAC,SAAS,CAAC,GAAG,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;oBAC1E,CAAC,CAAC,CAAC;gBACP,CAAC,CAAC,CAAC;gBACH,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YACrC,CAAC,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;IACP,CAAC;IAEO,uBAAuB,CAAC,KAAqB;QACjD,OAAO,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,iCAAqB,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC/F,CAAC;IAEO,kBAAkB,CAAC,KAAqB;;QAC5C,IAAI,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QACvC,sDAAsD;QACtD,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,UAAU,EAAE;YACjC,QAAQ,MAAA,IAAI,CAAC,IAAI,CAAC,GAAG,0CAAE,IAAI,EAAE;gBACzB,KAAK,SAAS,CAAC,CAAC;oBACZ,MAAM,GAAG,GAAG,IAAI,CAAC,iBAAiB,CAAS,IAAI,EAAE,KAAK,CAAC,CAAC;oBACxD,IAAI,GAAG,EAAE;wBACL,MAAM,IAAI,QAAQ,GAAG,GAAG,CAAC;qBAC5B;oBACD,MAAM,GAAG,GAAG,IAAI,CAAC,iBAAiB,CAAS,IAAI,EAAE,KAAK,CAAC,CAAC;oBACxD,IAAI,GAAG,EAAE;wBACL,MAAM,IAAI,QAAQ,GAAG,GAAG,CAAC;qBAC5B;oBACD,MAAM;iBACT;gBACD,KAAK,QAAQ,CAAC,CAAC;oBACX,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAS,IAAI,EAAE,OAAO,CAAC,CAAC;oBAC3D,IAAI,IAAI,EAAE;wBACN,MAAM,IAAI,WAAW,IAAI,IAAI,CAAC;qBACjC;oBACD,MAAM;iBACT;gBACD,KAAK,aAAa,CAAC,CAAC;oBAChB,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAS,IAAI,EAAE,MAAM,CAAC,CAAC;oBAC1D,IAAI,IAAI,EAAE;wBACN,MAAM,IAAI,eAAe,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC;qBACpD;oBACD,MAAM;iBACT;gBACD,KAAK,WAAW,CAAC,CAAC;oBACd,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAS,IAAI,EAAE,MAAM,CAAC,CAAC;oBAC1D,IAAI,IAAI,EAAE;wBACN,MAAM,IAAI,aAAa,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC;qBAClD;oBACD,MAAM;iBACT;gBACD,KAAK,QAAQ,CAAC,CAAC;oBACX,MAAM,IAAI,UAAU,CAAC;oBACrB,MAAM;iBACT;gBACD,KAAK,MAAM,CAAC,CAAC;oBACT,MAAM,IAAI,QAAQ,CAAC;oBACnB,MAAM;iBACT;gBACD,KAAK,WAAW,CAAC,CAAC;oBACd,MAAM,IAAI,6BAA6B,CAAC;oBACxC,MAAM;iBACT;gBACD,KAAK,KAAK,CAAC,CAAC;oBACR,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAS,IAAI,EAAE,OAAO,CAAC,CAAC;oBAC5D,IAAI,KAAK,KAAK,SAAS,EAAE;wBACrB,MAAM,IAAI,OAAO,KAAK,GAAG,CAAC;qBAC7B;oBACD,MAAM;iBACT;gBACD,KAAK,MAAM,CAAC,CAAC;oBACT,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAS,IAAI,EAAE,OAAO,CAAC,CAAC;oBAC5D,IAAI,KAAK,KAAK,SAAS,EAAE;wBACrB,MAAM,IAAI,QAAQ,KAAK,GAAG,CAAC;qBAC9B;oBACD,MAAM;iBACT;gBACD,KAAK,KAAK,CAAC,CAAC;oBACR,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAS,IAAI,EAAE,OAAO,CAAC,CAAC;oBAC5D,IAAI,KAAK,KAAK,SAAS,EAAE;wBACrB,MAAM,IAAI,OAAO,KAAK,GAAG,CAAC;qBAC7B;oBACD,MAAM;iBACT;gBACD,KAAK,MAAM,CAAC,CAAC;oBACT,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAS,IAAI,EAAE,OAAO,CAAC,CAAC;oBAC5D,IAAI,KAAK,KAAK,SAAS,EAAE;wBACrB,MAAM,IAAI,QAAQ,KAAK,GAAG,CAAC;qBAC9B;oBACD,MAAM;iBACT;aACJ;SACJ;QAED,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE;YACrB,MAAM,IAAI,aAAa,CAAC;SAC3B;QAED,OAAO,MAAM,CAAC;IAClB,CAAC;IAEO,aAAa,CAAC,KAAqB;QACvC,IAAI,MAAc,CAAC;QACnB,QAAQ,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE;YACrB,KAAK,KAAK,CAAC;YACX,KAAK,OAAO,CAAC;YACb,KAAK,SAAS;gBACV,MAAM,GAAG,YAAY,CAAC;gBACtB,MAAM;YACV,KAAK,QAAQ;gBACT,MAAM,GAAG,YAAY,CAAC;gBACtB,MAAM;YACV,KAAK,QAAQ;gBACT,MAAM,GAAG,YAAY,CAAC;gBACtB,MAAM;YACV,KAAK,SAAS;gBACV,MAAM,GAAG,aAAa,CAAC;gBACvB,MAAM;YACV,KAAK,UAAU;gBACX,MAAM,GAAG,UAAU,CAAC;gBACpB,MAAM;YACV;gBACI,MAAM,GAAG,SAAS,CAAC;gBACnB,MAAM;SACb;QAED,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE;YAClB,MAAM,GAAG,WAAW,MAAM,GAAG,CAAC;SACjC;QAED,OAAO,MAAM,CAAC;IAClB,CAAC;IAEO,iBAAiB,CAA4B,IAA6B,EAAE,SAAiB;QACjG,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,WAAC,OAAA,CAAA,MAAA,GAAG,CAAC,cAAc,0CAAE,IAAI,MAAK,SAAS,CAAA,EAAA,CAAC,CAAC;QAC5E,OAAO,GAAG,IAAI,IAAA,gBAAU,EAAI,GAAG,CAAC,KAAK,CAAC,CAAC;IAC3C,CAAC;CACJ;AA1JD,gDA0JC"}

package/plugins/model-meta/index.d.ts

@@ -0,0 +1,4 @@
+import { Model } from '@zenstackhq/language/ast';
+import { PluginOptions } from '@zenstackhq/sdk';
+export declare const name = "Model Metadata";
+export default function run(model: Model, options: PluginOptions): Promise<void>;

package/plugins/model-meta/index.js

@@ -0,0 +1,168 @@
+"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 ast_1 = require("@zenstackhq/language/ast");
+const sdk_1 = require("@zenstackhq/sdk");
+const change_case_1 = require("change-case");
+const path_1 = __importDefault(require("path"));
+const ts_morph_1 = require("ts-morph");
+const utils_1 = require("../../language-server/utils");
+const plugin_utils_1 = require("../plugin-utils");
+exports.name = 'Model Metadata';
+function run(model, options) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const output = options.output ? options.output : (0, plugin_utils_1.getDefaultOutputFolder)();
+        if (!output) {
+            console.error(`Unable to determine output path, not running plugin ${exports.name}`);
+            return;
+        }
+        const dataModels = model.declarations.filter((d) => (0, ast_1.isDataModel)(d));
+        const project = new ts_morph_1.Project();
+        if (!options.output) {
+            (0, plugin_utils_1.ensureNodeModuleFolder)(output);
+        }
+        const sf = project.createSourceFile(path_1.default.join(output, 'model-meta.ts'), undefined, { overwrite: true });
+        sf.addVariableStatement({
+            declarationKind: ts_morph_1.VariableDeclarationKind.Const,
+            declarations: [{ name: 'metadata', initializer: (writer) => generateModelMetadata(dataModels, writer) }],
+        });
+        sf.addStatements('export default metadata;');
+        sf.formatText();
+        yield project.save();
+        yield project.emit();
+    });
+}
+exports.default = run;
+function generateModelMetadata(dataModels, writer) {
+    writer.block(() => {
+        writer.write('fields:');
+        writer.block(() => {
+            for (const model of dataModels) {
+                writer.write(`${(0, change_case_1.camelCase)(model.name)}:`);
+                writer.block(() => {
+                    var _a;
+                    for (const f of model.fields) {
+                        const backlink = getBackLink(f);
+                        writer.write(`${f.name}: {
+                    name: "${f.name}",
+                    type: "${f.type.reference
+                            ? f.type.reference.$refText
+                            : // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+                                f.type.type}",
+                    isId: ${isIdField(f)},
+                    isDataModel: ${(0, ast_1.isDataModel)((_a = f.type.reference) === null || _a === void 0 ? void 0 : _a.ref)},
+                    isArray: ${f.type.array},
+                    isOptional: ${f.type.optional},
+                    attributes: ${JSON.stringify(getFieldAttributes(f))},
+                    backLink: ${backlink ? "'" + backlink + "'" : 'undefined'}   
+                },`);
+                    }
+                });
+                writer.write(',');
+            }
+        });
+        writer.write(',');
+        writer.write('uniqueConstraints:');
+        writer.block(() => {
+            for (const model of dataModels) {
+                writer.write(`${(0, change_case_1.camelCase)(model.name)}:`);
+                writer.block(() => {
+                    for (const constraint of getUniqueConstraints(model)) {
+                        writer.write(`${constraint.name}: {
+                    name: "${constraint.name}",
+                    fields: ${JSON.stringify(constraint.fields)}
+                },`);
+                    }
+                });
+                writer.write(',');
+            }
+        });
+        writer.write(',');
+    });
+}
+function getBackLink(field) {
+    var _a, _b, _c;
+    if (!((_a = field.type.reference) === null || _a === void 0 ? void 0 : _a.ref) || !(0, ast_1.isDataModel)((_b = field.type.reference) === null || _b === void 0 ? void 0 : _b.ref)) {
+        return undefined;
+    }
+    const relName = getRelationName(field);
+    const sourceModel = field.$container;
+    const targetModel = field.type.reference.ref;
+    for (const otherField of targetModel.fields) {
+        if (((_c = otherField.type.reference) === null || _c === void 0 ? void 0 : _c.ref) === sourceModel) {
+            if (relName) {
+                const otherRelName = getRelationName(otherField);
+                if (relName === otherRelName) {
+                    return otherField.name;
+                }
+            }
+            else {
+                return otherField.name;
+            }
+        }
+    }
+    return undefined;
+}
+function getRelationName(field) {
+    var _a, _b;
+    const relAttr = field.attributes.find((attr) => { var _a; return ((_a = attr.decl.ref) === null || _a === void 0 ? void 0 : _a.name) === 'relation'; });
+    const relName = relAttr && ((_a = relAttr.args) === null || _a === void 0 ? void 0 : _a[0]) && (0, sdk_1.getLiteral)((_b = relAttr.args) === null || _b === void 0 ? void 0 : _b[0].value);
+    return relName;
+}
+function getFieldAttributes(field) {
+    return field.attributes
+        .map((attr) => {
+        const args = [];
+        for (const arg of attr.args) {
+            if (!(0, ast_1.isLiteralExpr)(arg.value)) {
+                // attributes with non-literal args are skipped
+                return undefined;
+            }
+            args.push({ name: arg.name, value: arg.value.value });
+        }
+        return { name: (0, sdk_1.resolved)(attr.decl).name, args };
+    })
+        .filter((d) => !!d);
+}
+function isIdField(field) {
+    // field-level @id attribute
+    if (field.attributes.some((attr) => { var _a; return ((_a = attr.decl.ref) === null || _a === void 0 ? void 0 : _a.name) === '@id'; })) {
+        return true;
+    }
+    // model-level @@id attribute with a list of fields
+    const model = field.$container;
+    const modelLevelIds = (0, utils_1.getIdFields)(model);
+    if (modelLevelIds.includes(field)) {
+        return true;
+    }
+    return false;
+}
+function getUniqueConstraints(model) {
+    const constraints = [];
+    for (const attr of model.attributes.filter((attr) => { var _a, _b; return ((_a = attr.decl.ref) === null || _a === void 0 ? void 0 : _a.name) === '@@unique' || ((_b = attr.decl.ref) === null || _b === void 0 ? void 0 : _b.name) === '@@id'; })) {
+        const argsMap = (0, sdk_1.getAttributeArgs)(attr);
+        if (argsMap.fields) {
+            const fieldNames = argsMap.fields.items.map((item) => (0, sdk_1.resolved)(item.target).name);
+            let constraintName = argsMap.name && (0, sdk_1.getLiteral)(argsMap.name);
+            if (!constraintName) {
+                // default constraint name is fields concatenated with underscores
+                constraintName = fieldNames.join('_');
+            }
+            constraints.push({ name: constraintName, fields: fieldNames });
+        }
+    }
+    return constraints;
+}
+//# sourceMappingURL=index.js.map

package/plugins/model-meta/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/plugins/model-meta/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,kDAQkC;AAElC,yCAAwF;AACxF,6CAAwC;AACxC,gDAAwB;AACxB,uCAA6E;AAC7E,uDAA0D;AAC1D,kDAAiF;AAEpE,QAAA,IAAI,GAAG,gBAAgB,CAAC;AAErC,SAA8B,GAAG,CAAC,KAAY,EAAE,OAAsB;;QAClE,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAE,OAAO,CAAC,MAAiB,CAAC,CAAC,CAAC,IAAA,qCAAsB,GAAE,CAAC;QACtF,IAAI,CAAC,MAAM,EAAE;YACT,OAAO,CAAC,KAAK,CAAC,uDAAuD,YAAI,EAAE,CAAC,CAAC;YAC7E,OAAO;SACV;QAED,MAAM,UAAU,GAAG,KAAK,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAkB,EAAE,CAAC,IAAA,iBAAW,EAAC,CAAC,CAAC,CAAC,CAAC;QAEpF,MAAM,OAAO,GAAG,IAAI,kBAAO,EAAE,CAAC;QAE9B,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE;YACjB,IAAA,qCAAsB,EAAC,MAAM,CAAC,CAAC;SAClC;QAED,MAAM,EAAE,GAAG,OAAO,CAAC,gBAAgB,CAAC,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,eAAe,CAAC,EAAE,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACxG,EAAE,CAAC,oBAAoB,CAAC;YACpB,eAAe,EAAE,kCAAuB,CAAC,KAAK;YAC9C,YAAY,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,qBAAqB,CAAC,UAAU,EAAE,MAAM,CAAC,EAAE,CAAC;SAC3G,CAAC,CAAC;QACH,EAAE,CAAC,aAAa,CAAC,0BAA0B,CAAC,CAAC;QAE7C,EAAE,CAAC,UAAU,EAAE,CAAC;QAEhB,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;QACrB,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;IACzB,CAAC;CAAA;AA1BD,sBA0BC;AAED,SAAS,qBAAqB,CAAC,UAAuB,EAAE,MAAuB;IAC3E,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE;QACd,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QACxB,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE;YACd,KAAK,MAAM,KAAK,IAAI,UAAU,EAAE;gBAC5B,MAAM,CAAC,KAAK,CAAC,GAAG,IAAA,uBAAS,EAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAC1C,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE;;oBACd,KAAK,MAAM,CAAC,IAAI,KAAK,CAAC,MAAM,EAAE;wBAC1B,MAAM,QAAQ,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;wBAChC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI;6BACjB,CAAC,CAAC,IAAI;6BAEX,CAAC,CAAC,IAAI,CAAC,SAAS;4BACZ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ;4BAC3B,CAAC,CAAC,oEAAoE;gCACpE,CAAC,CAAC,IAAI,CAAC,IACjB;4BACQ,SAAS,CAAC,CAAC,CAAC;mCACL,IAAA,iBAAW,EAAC,MAAA,CAAC,CAAC,IAAI,CAAC,SAAS,0CAAE,GAAG,CAAC;+BACtC,CAAC,CAAC,IAAI,CAAC,KAAK;kCACT,CAAC,CAAC,IAAI,CAAC,QAAQ;kCACf,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC;gCACvC,QAAQ,CAAC,CAAC,CAAC,GAAG,GAAG,QAAQ,GAAG,GAAG,CAAC,CAAC,CAAC,WAAW;mBAC1D,CAAC,CAAC;qBACA;gBACL,CAAC,CAAC,CAAC;gBACH,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;aACrB;QACL,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAElB,MAAM,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE;YACd,KAAK,MAAM,KAAK,IAAI,UAAU,EAAE;gBAC5B,MAAM,CAAC,KAAK,CAAC,GAAG,IAAA,uBAAS,EAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAC1C,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE;oBACd,KAAK,MAAM,UAAU,IAAI,oBAAoB,CAAC,KAAK,CAAC,EAAE;wBAClD,MAAM,CAAC,KAAK,CAAC,GAAG,UAAU,CAAC,IAAI;6BAC1B,UAAU,CAAC,IAAI;8BACd,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC;mBAC5C,CAAC,CAAC;qBACA;gBACL,CAAC,CAAC,CAAC;gBACH,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;aACrB;QACL,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACtB,CAAC,CAAC,CAAC;AACP,CAAC;AAED,SAAS,WAAW,CAAC,KAAqB;;IACtC,IAAI,CAAC,CAAA,MAAA,KAAK,CAAC,IAAI,CAAC,SAAS,0CAAE,GAAG,CAAA,IAAI,CAAC,IAAA,iBAAW,EAAC,MAAA,KAAK,CAAC,IAAI,CAAC,SAAS,0CAAE,GAAG,CAAC,EAAE;QACvE,OAAO,SAAS,CAAC;KACpB;IAED,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IAEvC,MAAM,WAAW,GAAG,KAAK,CAAC,UAAuB,CAAC;IAClD,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,GAAgB,CAAC;IAE1D,KAAK,MAAM,UAAU,IAAI,WAAW,CAAC,MAAM,EAAE;QACzC,IAAI,CAAA,MAAA,UAAU,CAAC,IAAI,CAAC,SAAS,0CAAE,GAAG,MAAK,WAAW,EAAE;YAChD,IAAI,OAAO,EAAE;gBACT,MAAM,YAAY,GAAG,eAAe,CAAC,UAAU,CAAC,CAAC;gBACjD,IAAI,OAAO,KAAK,YAAY,EAAE;oBAC1B,OAAO,UAAU,CAAC,IAAI,CAAC;iBAC1B;aACJ;iBAAM;gBACH,OAAO,UAAU,CAAC,IAAI,CAAC;aAC1B;SACJ;KACJ;IACD,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,SAAS,eAAe,CAAC,KAAqB;;IAC1C,MAAM,OAAO,GAAG,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,WAAC,OAAA,CAAA,MAAA,IAAI,CAAC,IAAI,CAAC,GAAG,0CAAE,IAAI,MAAK,UAAU,CAAA,EAAA,CAAC,CAAC;IACpF,MAAM,OAAO,GAAG,OAAO,KAAI,MAAA,OAAO,CAAC,IAAI,0CAAG,CAAC,CAAC,CAAA,IAAI,IAAA,gBAAU,EAAS,MAAA,OAAO,CAAC,IAAI,0CAAG,CAAC,EAAE,KAAK,CAAC,CAAC;IAC5F,OAAO,OAAO,CAAC;AACnB,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAqB;IAC7C,OAAO,KAAK,CAAC,UAAU;SAClB,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACV,MAAM,IAAI,GAA6C,EAAE,CAAC;QAC1D,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,IAAI,EAAE;YACzB,IAAI,CAAC,IAAA,mBAAa,EAAC,GAAG,CAAC,KAAK,CAAC,EAAE;gBAC3B,+CAA+C;gBAC/C,OAAO,SAAS,CAAC;aACpB;YACD,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;SACzD;QACD,OAAO,EAAE,IAAI,EAAE,IAAA,cAAQ,EAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC;IACpD,CAAC,CAAC;SACD,MAAM,CAAC,CAAC,CAAC,EAAyB,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AACnD,CAAC;AAED,SAAS,SAAS,CAAC,KAAqB;IACpC,4BAA4B;IAC5B,IAAI,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,WAAC,OAAA,CAAA,MAAA,IAAI,CAAC,IAAI,CAAC,GAAG,0CAAE,IAAI,MAAK,KAAK,CAAA,EAAA,CAAC,EAAE;QAChE,OAAO,IAAI,CAAC;KACf;IAED,mDAAmD;IACnD,MAAM,KAAK,GAAG,KAAK,CAAC,UAAuB,CAAC;IAC5C,MAAM,aAAa,GAAG,IAAA,mBAAW,EAAC,KAAK,CAAC,CAAC;IACzC,IAAI,aAAa,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE;QAC/B,OAAO,IAAI,CAAC;KACf;IACD,OAAO,KAAK,CAAC;AACjB,CAAC;AAED,SAAS,oBAAoB,CAAC,KAAgB;IAC1C,MAAM,WAAW,GAA8C,EAAE,CAAC;IAClE,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,UAAU,CAAC,MAAM,CACtC,CAAC,IAAI,EAAE,EAAE,eAAC,OAAA,CAAA,MAAA,IAAI,CAAC,IAAI,CAAC,GAAG,0CAAE,IAAI,MAAK,UAAU,IAAI,CAAA,MAAA,IAAI,CAAC,IAAI,CAAC,GAAG,0CAAE,IAAI,MAAK,MAAM,CAAA,EAAA,CACjF,EAAE;QACC,MAAM,OAAO,GAAG,IAAA,sBAAgB,EAAC,IAAI,CAAC,CAAC;QACvC,IAAI,OAAO,CAAC,MAAM,EAAE;YAChB,MAAM,UAAU,GAAI,OAAO,CAAC,MAAoB,CAAC,KAAK,CAAC,GAAG,CACtD,CAAC,IAAI,EAAE,EAAE,CAAC,IAAA,cAAQ,EAAE,IAAsB,CAAC,MAAM,CAAC,CAAC,IAAI,CAC1D,CAAC;YACF,IAAI,cAAc,GAAG,OAAO,CAAC,IAAI,IAAI,IAAA,gBAAU,EAAS,OAAO,CAAC,IAAI,CAAC,CAAC;YACtE,IAAI,CAAC,cAAc,EAAE;gBACjB,kEAAkE;gBAClE,cAAc,GAAG,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;aACzC;YACD,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;SAClE;KACJ;IACD,OAAO,WAAW,CAAC;AACvB,CAAC"}

package/plugins/plugin-utils.d.ts

@@ -0,0 +1,16 @@
+import { PolicyOperationKind } from '@zenstackhq/runtime';
+export declare const RUNTIME_PACKAGE = "@zenstackhq/runtime";
+export declare const ALL_OPERATION_KINDS: PolicyOperationKind[];
+/**
+ * Gets the nearest "node_modules" folder by walking up from start path.
+ */
+export declare function getNodeModulesFolder(startPath?: string): string | undefined;
+/**
+ * Gets the default node_modules/.zenstack output folder for plugins.
+ * @returns
+ */
+export declare function getDefaultOutputFolder(): string | undefined;
+/**
+ * Ensure a folder exists and has a package.json in it.
+ */
+export declare function ensureNodeModuleFolder(folder: string): void;

package/plugins/plugin-utils.js

@@ -0,0 +1,54 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ensureNodeModuleFolder = exports.getDefaultOutputFolder = exports.getNodeModulesFolder = exports.ALL_OPERATION_KINDS = exports.RUNTIME_PACKAGE = void 0;
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+exports.RUNTIME_PACKAGE = '@zenstackhq/runtime';
+exports.ALL_OPERATION_KINDS = ['create', 'update', 'postUpdate', 'read', 'delete'];
+/**
+ * Gets the nearest "node_modules" folder by walking up from start path.
+ */
+function getNodeModulesFolder(startPath) {
+    startPath = startPath !== null && startPath !== void 0 ? startPath : process.cwd();
+    if (fs_1.default.existsSync(path_1.default.join(startPath, 'node_modules'))) {
+        return path_1.default.join(startPath, 'node_modules');
+    }
+    else if (startPath !== '/') {
+        const parent = path_1.default.join(startPath, '..');
+        return getNodeModulesFolder(parent);
+    }
+    else {
+        return undefined;
+    }
+}
+exports.getNodeModulesFolder = getNodeModulesFolder;
+/**
+ * Gets the default node_modules/.zenstack output folder for plugins.
+ * @returns
+ */
+function getDefaultOutputFolder() {
+    // Find the real runtime module path, it might be a symlink in pnpm
+    const runtimeModulePath = require.resolve('@zenstackhq/runtime');
+    const modulesFolder = getNodeModulesFolder(runtimeModulePath);
+    return modulesFolder ? path_1.default.join(modulesFolder, '.zenstack') : undefined;
+}
+exports.getDefaultOutputFolder = getDefaultOutputFolder;
+/**
+ * Ensure a folder exists and has a package.json in it.
+ */
+function ensureNodeModuleFolder(folder) {
+    if (!fs_1.default.existsSync(folder)) {
+        fs_1.default.mkdirSync(folder, { recursive: true });
+    }
+    if (!fs_1.default.existsSync(path_1.default.join(folder, 'package.json'))) {
+        fs_1.default.writeFileSync(path_1.default.join(folder, 'package.json'), JSON.stringify({
+            name: '.zenstack',
+            version: '1.0.0',
+        }));
+    }
+}
+exports.ensureNodeModuleFolder = ensureNodeModuleFolder;
+//# sourceMappingURL=plugin-utils.js.map

package/plugins/plugin-utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"plugin-utils.js","sourceRoot":"","sources":["../../src/plugins/plugin-utils.ts"],"names":[],"mappings":";;;;;;AACA,4CAAoB;AACpB,gDAAwB;AAEX,QAAA,eAAe,GAAG,qBAAqB,CAAC;AACxC,QAAA,mBAAmB,GAA0B,CAAC,QAAQ,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;AAE/G;;GAEG;AACH,SAAgB,oBAAoB,CAAC,SAAkB;IACnD,SAAS,GAAG,SAAS,aAAT,SAAS,cAAT,SAAS,GAAI,OAAO,CAAC,GAAG,EAAE,CAAC;IACvC,IAAI,YAAE,CAAC,UAAU,CAAC,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC,EAAE;QACrD,OAAO,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC;KAC/C;SAAM,IAAI,SAAS,KAAK,GAAG,EAAE;QAC1B,MAAM,MAAM,GAAG,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QAC1C,OAAO,oBAAoB,CAAC,MAAM,CAAC,CAAC;KACvC;SAAM;QACH,OAAO,SAAS,CAAC;KACpB;AACL,CAAC;AAVD,oDAUC;AAED;;;GAGG;AACH,SAAgB,sBAAsB;IAClC,mEAAmE;IACnE,MAAM,iBAAiB,GAAG,OAAO,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC;IACjE,MAAM,aAAa,GAAG,oBAAoB,CAAC,iBAAiB,CAAC,CAAC;IAC9D,OAAO,aAAa,CAAC,CAAC,CAAC,cAAI,CAAC,IAAI,CAAC,aAAa,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AAC7E,CAAC;AALD,wDAKC;AAED;;GAEG;AACH,SAAgB,sBAAsB,CAAC,MAAc;IACjD,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE;QACxB,YAAE,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;KAC7C;IACD,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC,EAAE;QACnD,YAAE,CAAC,aAAa,CACZ,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,cAAc,CAAC,EACjC,IAAI,CAAC,SAAS,CAAC;YACX,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,OAAO;SACnB,CAAC,CACL,CAAC;KACL;AACL,CAAC;AAbD,wDAaC"}

package/plugins/prisma/indent-string.d.ts

@@ -0,0 +1,4 @@
+/**
+ * Utility for indenting strings
+ */
+export default function indentString(string: string, count?: number): string;

package/plugins/prisma/indent-string.js

@@ -0,0 +1,12 @@
+"use strict";
+// https://github.com/sindresorhus/indent-string
+Object.defineProperty(exports, "__esModule", { value: true });
+/**
+ * Utility for indenting strings
+ */
+function indentString(string, count = 4) {
+    const indent = ' ';
+    return string.replace(/^(?!\s*$)/gm, indent.repeat(count));
+}
+exports.default = indentString;
+//# sourceMappingURL=indent-string.js.map

package/plugins/prisma/indent-string.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"indent-string.js","sourceRoot":"","sources":["../../../src/plugins/prisma/indent-string.ts"],"names":[],"mappings":";AAAA,gDAAgD;;AAEhD;;GAEG;AACH,SAAwB,YAAY,CAAC,MAAc,EAAE,KAAK,GAAG,CAAC;IAC1D,MAAM,MAAM,GAAG,GAAG,CAAC;IACnB,OAAO,MAAM,CAAC,OAAO,CAAC,aAAa,EAAE,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;AAC/D,CAAC;AAHD,+BAGC"}

package/plugins/prisma/index.d.ts

@@ -0,0 +1,4 @@
+import { Model } from '@zenstackhq/language/ast';
+import { PluginOptions } from '@zenstackhq/sdk';
+export declare const name = "Prisma";
+export default function run(model: Model, options: PluginOptions): Promise<void>;

package/plugins/prisma/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 schema_generator_1 = __importDefault(require("./schema-generator"));
+exports.name = 'Prisma';
+function run(model, options) {
+    return __awaiter(this, void 0, void 0, function* () {
+        return new schema_generator_1.default().generate(model, options);
+    });
+}
+exports.default = run;
+//# sourceMappingURL=index.js.map

package/plugins/prisma/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/plugins/prisma/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAEA,0EAAuD;AAE1C,QAAA,IAAI,GAAG,QAAQ,CAAC;AAE7B,SAA8B,GAAG,CAAC,KAAY,EAAE,OAAsB;;QAClE,OAAO,IAAI,0BAAqB,EAAE,CAAC,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAChE,CAAC;CAAA;AAFD,sBAEC"}

package/plugins/prisma/prisma-builder.d.ts

@@ -0,0 +1,152 @@
+/**
+ * Field used by datasource and generator declarations.
+ */
+export declare type SimpleField = {
+    name: string;
+    value: string | string[];
+};
+/**
+ * Prisma schema builder
+ */
+export declare class PrismaModel {
+    private datasources;
+    private generators;
+    private models;
+    private enums;
+    addDataSource(name: string, provider: string, url: DataSourceUrl, shadowDatabaseUrl?: DataSourceUrl, restFields?: SimpleField[]): DataSource;
+    addGenerator(name: string, fields: SimpleField[]): Generator;
+    addModel(name: string): Model;
+    addEnum(name: string): Enum;
+    toString(): string;
+}
+export declare class DataSource {
+    name: string;
+    provider: string;
+    url: DataSourceUrl;
+    shadowDatabaseUrl?: DataSourceUrl | undefined;
+    restFields: SimpleField[];
+    constructor(name: string, provider: string, url: DataSourceUrl, shadowDatabaseUrl?: DataSourceUrl | undefined, restFields?: SimpleField[]);
+    toString(): string;
+}
+export declare class DataSourceUrl {
+    value: string;
+    isEnv: boolean;
+    constructor(value: string, isEnv: boolean);
+    toString(): string;
+}
+export declare class Generator {
+    name: string;
+    fields: SimpleField[];
+    constructor(name: string, fields: SimpleField[]);
+    toString(): string;
+}
+export declare class DeclarationBase {
+    documentations: string[];
+    constructor(documentations?: string[]);
+    addComment(name: string): string;
+    toString(): string;
+}
+export declare class ContainerDeclaration extends DeclarationBase {
+    attributes: (ContainerAttribute | PassThroughAttribute)[];
+    constructor(documentations?: string[], attributes?: (ContainerAttribute | PassThroughAttribute)[]);
+}
+export declare class FieldDeclaration extends DeclarationBase {
+    attributes: (FieldAttribute | PassThroughAttribute)[];
+    constructor(documentations?: string[], attributes?: (FieldAttribute | PassThroughAttribute)[]);
+}
+export declare class Model extends ContainerDeclaration {
+    name: string;
+    fields: ModelField[];
+    constructor(name: string, documentations?: string[]);
+    addField(name: string, type: ModelFieldType | string, attributes?: (FieldAttribute | PassThroughAttribute)[], documentations?: string[]): ModelField;
+    addAttribute(name: string, args?: AttributeArg[]): ContainerAttribute;
+    toString(): string;
+}
+export declare type ScalarTypes = 'String' | 'Boolean' | 'Int' | 'BigInt' | 'Float' | 'Decimal' | 'DateTime' | 'Json' | 'Bytes' | 'Unsupported';
+export declare class ModelFieldType {
+    type: ScalarTypes | string;
+    array?: boolean | undefined;
+    optional?: boolean | undefined;
+    constructor(type: ScalarTypes | string, array?: boolean | undefined, optional?: boolean | undefined);
+    toString(): string;
+}
+export declare class ModelField extends FieldDeclaration {
+    name: string;
+    type: ModelFieldType | string;
+    constructor(name: string, type: ModelFieldType | string, attributes?: (FieldAttribute | PassThroughAttribute)[], documentations?: string[]);
+    addAttribute(name: string, args?: AttributeArg[]): FieldAttribute;
+    toString(): string;
+}
+export declare class FieldAttribute {
+    name: string;
+    args: AttributeArg[];
+    constructor(name: string, args?: AttributeArg[]);
+    toString(): string;
+}
+export declare class ContainerAttribute {
+    name: string;
+    args: AttributeArg[];
+    constructor(name: string, args?: AttributeArg[]);
+    toString(): string;
+}
+/**
+ * Represents @@prisma.passthrough and @prisma.passthrough
+ */
+export declare class PassThroughAttribute {
+    text: string;
+    constructor(text: string);
+    toString(): string;
+}
+export declare class AttributeArg {
+    name: string | undefined;
+    value: AttributeArgValue;
+    constructor(name: string | undefined, value: AttributeArgValue);
+    toString(): string;
+}
+export declare class AttributeArgValue {
+    type: 'String' | 'FieldReference' | 'Number' | 'Boolean' | 'Array' | 'FunctionCall';
+    value: string | number | boolean | FieldReference | FunctionCall | AttributeArgValue[];
+    constructor(type: 'String' | 'FieldReference' | 'Number' | 'Boolean' | 'Array' | 'FunctionCall', value: string | number | boolean | FieldReference | FunctionCall | AttributeArgValue[]);
+    toString(): string;
+}
+export declare class FieldReference {
+    field: string;
+    args: FieldReferenceArg[];
+    constructor(field: string, args?: FieldReferenceArg[]);
+}
+export declare class FieldReferenceArg {
+    name: 'sort';
+    value: 'Asc' | 'Desc';
+    constructor(name: 'sort', value: 'Asc' | 'Desc');
+    toString(): string;
+}
+export declare class FunctionCall {
+    func: string;
+    args: FunctionCallArg[];
+    constructor(func: string, args?: FunctionCallArg[]);
+    toString(): string;
+}
+export declare class FunctionCallArg {
+    name: string | undefined;
+    value: any;
+    constructor(name: string | undefined, value: any);
+    toString(): string;
+}
+export declare class Enum extends ContainerDeclaration {
+    name: string;
+    documentations: string[];
+    fields: EnumField[];
+    constructor(name: string, documentations?: string[]);
+    addField(name: string, attributes?: (FieldAttribute | PassThroughAttribute)[], documentations?: string[]): EnumField;
+    addAttribute(name: string, args?: AttributeArg[]): ContainerAttribute;
+    addComment(name: string): string;
+    toString(): string;
+}
+export declare class EnumField extends DeclarationBase {
+    name: string;
+    attributes: (FieldAttribute | PassThroughAttribute)[];
+    documentations: string[];
+    constructor(name: string, attributes?: (FieldAttribute | PassThroughAttribute)[], documentations?: string[]);
+    addAttribute(name: string, args?: AttributeArg[]): FieldAttribute;
+    toString(): string;
+}