zenstack 0.6.0-pre.9 → 1.0.0-alpha.20

package/plugins/prisma/schema-generator.js

@@ -0,0 +1,212 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.default = void 0;
+var _ast = require("@zenstackhq/language/ast");
+var _sdk = require("@zenstackhq/sdk");
+var _fs = _interopRequireDefault(require("fs"));
+var _promises = require("fs/promises");
+var _path = _interopRequireDefault(require("path"));
+var _astUtils = require("../../utils/ast-utils");
+var _execUtils = require("../../utils/exec-utils");
+var _zmodelCodeGenerator = _interopRequireDefault(require("./zmodel-code-generator"));
+var _prismaBuilder = require("./prisma-builder");
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+/**
+ * Generates Prisma schema file
+ */
+class PrismaSchemaGenerator {
+  zModelGenerator = new _zmodelCodeGenerator.default();
+  PRELUDE = `//////////////////////////////////////////////////////////////////////////////////////////////
+// DO NOT MODIFY THIS FILE                                                                  //
+// This file is automatically generated by ZenStack CLI and should not be manually updated. //
+//////////////////////////////////////////////////////////////////////////////////////////////
+
+`;
+  async generate(model, options) {
+    const prisma = new _prismaBuilder.PrismaModel();
+    for (const decl of model.declarations) {
+      switch (decl.$type) {
+        case _ast.DataSource:
+          this.generateDataSource(prisma, decl);
+          break;
+        case _ast.Enum:
+          this.generateEnum(prisma, decl);
+          break;
+        case _ast.DataModel:
+          this.generateModel(prisma, decl);
+          break;
+        case _ast.GeneratorDecl:
+          this.generateGenerator(prisma, decl);
+          break;
+      }
+    }
+    const outFile = options.output ?? './prisma/schema.prisma';
+    if (!_fs.default.existsSync(_path.default.dirname(outFile))) {
+      _fs.default.mkdirSync(_path.default.dirname(outFile), {
+        recursive: true
+      });
+    }
+    await (0, _promises.writeFile)(outFile, this.PRELUDE + prisma.toString());
+
+    // run 'prisma generate'
+    await (0, _execUtils.execSync)(`npx prisma generate --schema ${outFile}`);
+  }
+  generateDataSource(prisma, dataSource) {
+    let provider = undefined;
+    let url = undefined;
+    let shadowDatabaseUrl = undefined;
+    for (const f of dataSource.fields) {
+      switch (f.name) {
+        case 'provider':
+          {
+            if (this.isStringLiteral(f.value)) {
+              provider = f.value.value;
+            } else {
+              throw new _sdk.PluginError('Datasource provider must be set to a string');
+            }
+            break;
+          }
+        case 'url':
+          {
+            const r = this.extractDataSourceUrl(f.value);
+            if (!r) {
+              throw new _sdk.PluginError('Invalid value for datasource url');
+            }
+            url = r;
+            break;
+          }
+        case 'shadowDatabaseUrl':
+          {
+            const r = this.extractDataSourceUrl(f.value);
+            if (!r) {
+              throw new _sdk.PluginError('Invalid value for datasource url');
+            }
+            shadowDatabaseUrl = r;
+            break;
+          }
+      }
+    }
+    if (!provider) {
+      throw new _sdk.PluginError('Datasource is missing "provider" field');
+    }
+    if (!url) {
+      throw new _sdk.PluginError('Datasource is missing "url" field');
+    }
+    prisma.addDataSource(dataSource.name, provider, url, shadowDatabaseUrl);
+  }
+  extractDataSourceUrl(fieldValue) {
+    var _fieldValue$function$;
+    if (this.isStringLiteral(fieldValue)) {
+      return new _prismaBuilder.DataSourceUrl(fieldValue.value, false);
+    } else if ((0, _ast.isInvocationExpr)(fieldValue) && ((_fieldValue$function$ = fieldValue.function.ref) === null || _fieldValue$function$ === void 0 ? void 0 : _fieldValue$function$.name) === 'env' && fieldValue.args.length === 1 && this.isStringLiteral(fieldValue.args[0].value)) {
+      return new _prismaBuilder.DataSourceUrl(fieldValue.args[0].value.value, true);
+    } else {
+      return null;
+    }
+  }
+  generateGenerator(prisma, decl) {
+    prisma.addGenerator(decl.name, decl.fields.map(f => {
+      const value = (0, _ast.isArrayExpr)(f.value) ? (0, _sdk.getLiteralArray)(f.value) : (0, _sdk.getLiteral)(f.value);
+      return {
+        name: f.name,
+        value
+      };
+    }));
+  }
+  generateModel(prisma, decl) {
+    const model = prisma.addModel(decl.name);
+    for (const field of decl.fields) {
+      this.generateModelField(model, field);
+    }
+    const {
+      allowAll,
+      denyAll,
+      hasFieldValidation
+    } = (0, _astUtils.analyzePolicies)(decl);
+    if (!allowAll && !denyAll || hasFieldValidation) {
+      // generate auxiliary fields for policy check
+
+      // add an "zenstack_guard" field for dealing with pure auth() related conditions
+      model.addField(_sdk.GUARD_FIELD_NAME, 'Boolean', [new _prismaBuilder.FieldAttribute('@default', [new _prismaBuilder.AttributeArg(undefined, new _prismaBuilder.AttributeArgValue('Boolean', true))])]);
+
+      // add an "zenstack_transaction" field for tracking records created/updated with nested writes
+      model.addField(_sdk.TRANSACTION_FIELD_NAME, 'String?');
+
+      // create an index for "zenstack_transaction" field
+      model.addAttribute('@@index', [new _prismaBuilder.AttributeArg(undefined, new _prismaBuilder.AttributeArgValue('Array', [new _prismaBuilder.AttributeArgValue('FieldReference', _sdk.TRANSACTION_FIELD_NAME)]))]);
+    }
+    for (const attr of decl.attributes.filter(attr => attr.decl.ref && this.isPrismaAttribute(attr.decl.ref))) {
+      this.generateModelAttribute(model, attr);
+    }
+    decl.attributes.filter(attr => attr.decl.ref && !this.isPrismaAttribute(attr.decl.ref)).forEach(attr => model.addComment(this.zModelGenerator.generateAttribute(attr)));
+  }
+  isPrismaAttribute(attr) {
+    return !!attr.attributes.find(a => {
+      var _a$decl$ref;
+      return ((_a$decl$ref = a.decl.ref) === null || _a$decl$ref === void 0 ? void 0 : _a$decl$ref.name) === '@@@prisma';
+    });
+  }
+  generateModelField(model, field) {
+    var _field$type$reference, _field$type$reference2;
+    const fieldType = field.type.type || ((_field$type$reference = field.type.reference) === null || _field$type$reference === void 0 ? void 0 : (_field$type$reference2 = _field$type$reference.ref) === null || _field$type$reference2 === void 0 ? void 0 : _field$type$reference2.name);
+    if (!fieldType) {
+      throw new _sdk.PluginError(`Field type is not resolved: ${field.$container.name}.${field.name}`);
+    }
+    const type = new _prismaBuilder.ModelFieldType(fieldType, field.type.array, field.type.optional);
+    const attributes = field.attributes.filter(attr => attr.decl.ref && this.isPrismaAttribute(attr.decl.ref)).map(attr => this.makeFieldAttribute(attr));
+    const nonPrismaAttributes = field.attributes.filter(attr => !attr.decl.ref || !this.isPrismaAttribute(attr.decl.ref));
+    const documentations = nonPrismaAttributes.map(attr => this.zModelGenerator.generateAttribute(attr));
+    model.addField(field.name, type, attributes, documentations);
+  }
+  makeFieldAttribute(attr) {
+    return new _prismaBuilder.FieldAttribute((0, _sdk.resolved)(attr.decl).name, attr.args.map(arg => this.makeAttributeArg(arg)));
+  }
+  makeAttributeArg(arg) {
+    return new _prismaBuilder.AttributeArg(arg.name, this.makeAttributeArgValue(arg.value));
+  }
+  makeAttributeArgValue(node) {
+    if ((0, _ast.isLiteralExpr)(node)) {
+      switch (typeof node.value) {
+        case 'string':
+          return new _prismaBuilder.AttributeArgValue('String', node.value);
+        case 'number':
+          return new _prismaBuilder.AttributeArgValue('Number', node.value);
+        case 'boolean':
+          return new _prismaBuilder.AttributeArgValue('Boolean', node.value);
+        default:
+          throw new _sdk.PluginError(`Unexpected literal type: ${typeof node.value}`);
+      }
+    } else if ((0, _ast.isArrayExpr)(node)) {
+      return new _prismaBuilder.AttributeArgValue('Array', new Array(...node.items.map(item => this.makeAttributeArgValue(item))));
+    } else if ((0, _ast.isReferenceExpr)(node)) {
+      return new _prismaBuilder.AttributeArgValue('FieldReference', new _prismaBuilder.FieldReference((0, _sdk.resolved)(node.target).name, node.args.map(arg => new _prismaBuilder.FieldReferenceArg(arg.name, arg.value))));
+    } else if ((0, _ast.isInvocationExpr)(node)) {
+      // invocation
+      return new _prismaBuilder.AttributeArgValue('FunctionCall', this.makeFunctionCall(node));
+    } else {
+      throw new _sdk.PluginError(`Unsupported attribute argument expression type: ${node.$type}`);
+    }
+  }
+  makeFunctionCall(node) {
+    return new _prismaBuilder.FunctionCall((0, _sdk.resolved)(node.function).name, node.args.map(arg => {
+      if (!(0, _ast.isLiteralExpr)(arg.value)) {
+        throw new _sdk.PluginError('Function call argument must be literal');
+      }
+      return new _prismaBuilder.FunctionCallArg(arg.name, arg.value.value);
+    }));
+  }
+  generateModelAttribute(model, attr) {
+    model.attributes.push(new _prismaBuilder.ModelAttribute((0, _sdk.resolved)(attr.decl).name, attr.args.map(arg => this.makeAttributeArg(arg))));
+  }
+  generateEnum(prisma, decl) {
+    prisma.addEnum(decl.name, decl.fields.map(f => f.name));
+  }
+  isStringLiteral(node) {
+    return (0, _ast.isLiteralExpr)(node) && typeof node.value === 'string';
+  }
+}
+exports.default = PrismaSchemaGenerator;
+//# sourceMappingURL=schema-generator.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"schema-generator.js","names":["PrismaSchemaGenerator","zModelGenerator","ZModelCodeGenerator","PRELUDE","generate","model","options","prisma","PrismaModel","decl","declarations","$type","DataSource","generateDataSource","Enum","generateEnum","DataModel","generateModel","GeneratorDecl","generateGenerator","outFile","output","fs","existsSync","path","dirname","mkdirSync","recursive","writeFile","toString","execSync","dataSource","provider","undefined","url","shadowDatabaseUrl","f","fields","name","isStringLiteral","value","PluginError","r","extractDataSourceUrl","addDataSource","fieldValue","PrismaDataSourceUrl","isInvocationExpr","function","ref","args","length","addGenerator","map","isArrayExpr","getLiteralArray","getLiteral","addModel","field","generateModelField","allowAll","denyAll","hasFieldValidation","analyzePolicies","addField","GUARD_FIELD_NAME","PrismaFieldAttribute","PrismaAttributeArg","PrismaAttributeArgValue","TRANSACTION_FIELD_NAME","addAttribute","attr","attributes","filter","isPrismaAttribute","generateModelAttribute","forEach","addComment","generateAttribute","find","a","fieldType","type","reference","$container","ModelFieldType","array","optional","makeFieldAttribute","nonPrismaAttributes","documentations","resolved","arg","makeAttributeArg","makeAttributeArgValue","node","isLiteralExpr","Array","items","item","isReferenceExpr","PrismaFieldReference","target","PrismaFieldReferenceArg","makeFunctionCall","PrismaFunctionCall","PrismaFunctionCallArg","push","PrismaModelAttribute","addEnum"],"sources":["../../../src/plugins/prisma/schema-generator.ts"],"sourcesContent":["import {\n    AstNode,\n    Attribute,\n    AttributeArg,\n    DataModel,\n    DataModelAttribute,\n    DataModelField,\n    DataModelFieldAttribute,\n    DataSource,\n    Enum,\n    Expression,\n    GeneratorDecl,\n    InvocationExpr,\n    LiteralExpr,\n    Model,\n    isArrayExpr,\n    isInvocationExpr,\n    isLiteralExpr,\n    isReferenceExpr,\n} from '@zenstackhq/language/ast';\nimport {\n    GUARD_FIELD_NAME,\n    PluginError,\n    PluginOptions,\n    TRANSACTION_FIELD_NAME,\n    getLiteral,\n    getLiteralArray,\n    resolved,\n} from '@zenstackhq/sdk';\nimport fs from 'fs';\nimport { writeFile } from 'fs/promises';\nimport path from 'path';\nimport { analyzePolicies } from '../../utils/ast-utils';\nimport { execSync } from '../../utils/exec-utils';\nimport ZModelCodeGenerator from './zmodel-code-generator';\nimport {\n    ModelFieldType,\n    AttributeArg as PrismaAttributeArg,\n    AttributeArgValue as PrismaAttributeArgValue,\n    Model as PrismaDataModel,\n    DataSourceUrl as PrismaDataSourceUrl,\n    FieldAttribute as PrismaFieldAttribute,\n    FieldReference as PrismaFieldReference,\n    FieldReferenceArg as PrismaFieldReferenceArg,\n    FunctionCall as PrismaFunctionCall,\n    FunctionCallArg as PrismaFunctionCallArg,\n    PrismaModel,\n    ModelAttribute as PrismaModelAttribute,\n} from './prisma-builder';\n\n/**\n * Generates Prisma schema file\n */\nexport default class PrismaSchemaGenerator {\n    private zModelGenerator: ZModelCodeGenerator = new ZModelCodeGenerator();\n\n    private readonly PRELUDE = `//////////////////////////////////////////////////////////////////////////////////////////////\n// DO NOT MODIFY THIS FILE                                                                  //\n// This file is automatically generated by ZenStack CLI and should not be manually updated. //\n//////////////////////////////////////////////////////////////////////////////////////////////\n\n`;\n\n    async generate(model: Model, options: PluginOptions) {\n        const prisma = new PrismaModel();\n\n        for (const decl of model.declarations) {\n            switch (decl.$type) {\n                case DataSource:\n                    this.generateDataSource(prisma, decl as DataSource);\n                    break;\n\n                case Enum:\n                    this.generateEnum(prisma, decl as Enum);\n                    break;\n\n                case DataModel:\n                    this.generateModel(prisma, decl as DataModel);\n                    break;\n\n                case GeneratorDecl:\n                    this.generateGenerator(prisma, decl as GeneratorDecl);\n                    break;\n            }\n        }\n\n        const outFile = (options.output as string) ?? './prisma/schema.prisma';\n        if (!fs.existsSync(path.dirname(outFile))) {\n            fs.mkdirSync(path.dirname(outFile), { recursive: true });\n        }\n        await writeFile(outFile, this.PRELUDE + prisma.toString());\n\n        // run 'prisma generate'\n        await execSync(`npx prisma generate --schema ${outFile}`);\n    }\n\n    private generateDataSource(prisma: PrismaModel, dataSource: DataSource) {\n        let provider: string | undefined = undefined;\n        let url: PrismaDataSourceUrl | undefined = undefined;\n        let shadowDatabaseUrl: PrismaDataSourceUrl | undefined = undefined;\n\n        for (const f of dataSource.fields) {\n            switch (f.name) {\n                case 'provider': {\n                    if (this.isStringLiteral(f.value)) {\n                        provider = f.value.value as string;\n                    } else {\n                        throw new PluginError('Datasource provider must be set to a string');\n                    }\n                    break;\n                }\n\n                case 'url': {\n                    const r = this.extractDataSourceUrl(f.value);\n                    if (!r) {\n                        throw new PluginError('Invalid value for datasource url');\n                    }\n                    url = r;\n                    break;\n                }\n\n                case 'shadowDatabaseUrl': {\n                    const r = this.extractDataSourceUrl(f.value);\n                    if (!r) {\n                        throw new PluginError('Invalid value for datasource url');\n                    }\n                    shadowDatabaseUrl = r;\n                    break;\n                }\n            }\n        }\n\n        if (!provider) {\n            throw new PluginError('Datasource is missing \"provider\" field');\n        }\n        if (!url) {\n            throw new PluginError('Datasource is missing \"url\" field');\n        }\n\n        prisma.addDataSource(dataSource.name, provider, url, shadowDatabaseUrl);\n    }\n\n    private extractDataSourceUrl(fieldValue: LiteralExpr | InvocationExpr) {\n        if (this.isStringLiteral(fieldValue)) {\n            return new PrismaDataSourceUrl(fieldValue.value as string, false);\n        } else if (\n            isInvocationExpr(fieldValue) &&\n            fieldValue.function.ref?.name === 'env' &&\n            fieldValue.args.length === 1 &&\n            this.isStringLiteral(fieldValue.args[0].value)\n        ) {\n            return new PrismaDataSourceUrl(fieldValue.args[0].value.value as string, true);\n        } else {\n            return null;\n        }\n    }\n\n    private generateGenerator(prisma: PrismaModel, decl: GeneratorDecl) {\n        prisma.addGenerator(\n            decl.name,\n            decl.fields.map((f) => {\n                const value = isArrayExpr(f.value) ? getLiteralArray(f.value) : getLiteral(f.value);\n                return { name: f.name, value };\n            })\n        );\n    }\n\n    private generateModel(prisma: PrismaModel, decl: DataModel) {\n        const model = prisma.addModel(decl.name);\n        for (const field of decl.fields) {\n            this.generateModelField(model, field);\n        }\n\n        const { allowAll, denyAll, hasFieldValidation } = analyzePolicies(decl);\n\n        if ((!allowAll && !denyAll) || hasFieldValidation) {\n            // generate auxiliary fields for policy check\n\n            // add an \"zenstack_guard\" field for dealing with pure auth() related conditions\n            model.addField(GUARD_FIELD_NAME, 'Boolean', [\n                new PrismaFieldAttribute('@default', [\n                    new PrismaAttributeArg(undefined, new PrismaAttributeArgValue('Boolean', true)),\n                ]),\n            ]);\n\n            // add an \"zenstack_transaction\" field for tracking records created/updated with nested writes\n            model.addField(TRANSACTION_FIELD_NAME, 'String?');\n\n            // create an index for \"zenstack_transaction\" field\n            model.addAttribute('@@index', [\n                new PrismaAttributeArg(\n                    undefined,\n                    new PrismaAttributeArgValue('Array', [\n                        new PrismaAttributeArgValue('FieldReference', TRANSACTION_FIELD_NAME),\n                    ])\n                ),\n            ]);\n        }\n\n        for (const attr of decl.attributes.filter((attr) => attr.decl.ref && this.isPrismaAttribute(attr.decl.ref))) {\n            this.generateModelAttribute(model, attr);\n        }\n\n        decl.attributes\n            .filter((attr) => attr.decl.ref && !this.isPrismaAttribute(attr.decl.ref))\n            .forEach((attr) => model.addComment(this.zModelGenerator.generateAttribute(attr)));\n    }\n\n    private isPrismaAttribute(attr: Attribute) {\n        return !!attr.attributes.find((a) => a.decl.ref?.name === '@@@prisma');\n    }\n\n    private generateModelField(model: PrismaDataModel, field: DataModelField) {\n        const fieldType = field.type.type || field.type.reference?.ref?.name;\n        if (!fieldType) {\n            throw new PluginError(`Field type is not resolved: ${field.$container.name}.${field.name}`);\n        }\n\n        const type = new ModelFieldType(fieldType, field.type.array, field.type.optional);\n\n        const attributes = field.attributes\n            .filter((attr) => attr.decl.ref && this.isPrismaAttribute(attr.decl.ref))\n            .map((attr) => this.makeFieldAttribute(attr));\n\n        const nonPrismaAttributes = field.attributes.filter(\n            (attr) => !attr.decl.ref || !this.isPrismaAttribute(attr.decl.ref)\n        );\n\n        const documentations = nonPrismaAttributes.map((attr) => this.zModelGenerator.generateAttribute(attr));\n\n        model.addField(field.name, type, attributes, documentations);\n    }\n\n    private makeFieldAttribute(attr: DataModelFieldAttribute) {\n        return new PrismaFieldAttribute(\n            resolved(attr.decl).name,\n            attr.args.map((arg) => this.makeAttributeArg(arg))\n        );\n    }\n\n    private makeAttributeArg(arg: AttributeArg): PrismaAttributeArg {\n        return new PrismaAttributeArg(arg.name, this.makeAttributeArgValue(arg.value));\n    }\n\n    private makeAttributeArgValue(node: Expression): PrismaAttributeArgValue {\n        if (isLiteralExpr(node)) {\n            switch (typeof node.value) {\n                case 'string':\n                    return new PrismaAttributeArgValue('String', node.value);\n                case 'number':\n                    return new PrismaAttributeArgValue('Number', node.value);\n                case 'boolean':\n                    return new PrismaAttributeArgValue('Boolean', node.value);\n                default:\n                    throw new PluginError(`Unexpected literal type: ${typeof node.value}`);\n            }\n        } else if (isArrayExpr(node)) {\n            return new PrismaAttributeArgValue(\n                'Array',\n                new Array(...node.items.map((item) => this.makeAttributeArgValue(item)))\n            );\n        } else if (isReferenceExpr(node)) {\n            return new PrismaAttributeArgValue(\n                'FieldReference',\n                new PrismaFieldReference(\n                    resolved(node.target).name,\n                    node.args.map((arg) => new PrismaFieldReferenceArg(arg.name, arg.value))\n                )\n            );\n        } else if (isInvocationExpr(node)) {\n            // invocation\n            return new PrismaAttributeArgValue('FunctionCall', this.makeFunctionCall(node));\n        } else {\n            throw new PluginError(`Unsupported attribute argument expression type: ${node.$type}`);\n        }\n    }\n\n    makeFunctionCall(node: InvocationExpr): PrismaFunctionCall {\n        return new PrismaFunctionCall(\n            resolved(node.function).name,\n            node.args.map((arg) => {\n                if (!isLiteralExpr(arg.value)) {\n                    throw new PluginError('Function call argument must be literal');\n                }\n                return new PrismaFunctionCallArg(arg.name, arg.value.value);\n            })\n        );\n    }\n\n    private generateModelAttribute(model: PrismaDataModel, attr: DataModelAttribute) {\n        model.attributes.push(\n            new PrismaModelAttribute(\n                resolved(attr.decl).name,\n                attr.args.map((arg) => this.makeAttributeArg(arg))\n            )\n        );\n    }\n\n    private generateEnum(prisma: PrismaModel, decl: Enum) {\n        prisma.addEnum(\n            decl.name,\n            decl.fields.map((f) => f.name)\n        );\n    }\n\n    private isStringLiteral(node: AstNode): node is LiteralExpr {\n        return isLiteralExpr(node) && typeof node.value === 'string';\n    }\n}\n"],"mappings":";;;;;;AAAA;AAoBA;AASA;AACA;AACA;AACA;AACA;AACA;AACA;AAa0B;AAE1B;AACA;AACA;AACe,MAAMA,qBAAqB,CAAC;EAC/BC,eAAe,GAAwB,IAAIC,4BAAmB,EAAE;EAEvDC,OAAO,GAAI;AAChC;AACA;AACA;AACA;AACA,CAAC;EAEG,MAAMC,QAAQ,CAACC,KAAY,EAAEC,OAAsB,EAAE;IACjD,MAAMC,MAAM,GAAG,IAAIC,0BAAW,EAAE;IAEhC,KAAK,MAAMC,IAAI,IAAIJ,KAAK,CAACK,YAAY,EAAE;MACnC,QAAQD,IAAI,CAACE,KAAK;QACd,KAAKC,eAAU;UACX,IAAI,CAACC,kBAAkB,CAACN,MAAM,EAAEE,IAAI,CAAe;UACnD;QAEJ,KAAKK,SAAI;UACL,IAAI,CAACC,YAAY,CAACR,MAAM,EAAEE,IAAI,CAAS;UACvC;QAEJ,KAAKO,cAAS;UACV,IAAI,CAACC,aAAa,CAACV,MAAM,EAAEE,IAAI,CAAc;UAC7C;QAEJ,KAAKS,kBAAa;UACd,IAAI,CAACC,iBAAiB,CAACZ,MAAM,EAAEE,IAAI,CAAkB;UACrD;MAAM;IAElB;IAEA,MAAMW,OAAO,GAAId,OAAO,CAACe,MAAM,IAAe,wBAAwB;IACtE,IAAI,CAACC,WAAE,CAACC,UAAU,CAACC,aAAI,CAACC,OAAO,CAACL,OAAO,CAAC,CAAC,EAAE;MACvCE,WAAE,CAACI,SAAS,CAACF,aAAI,CAACC,OAAO,CAACL,OAAO,CAAC,EAAE;QAAEO,SAAS,EAAE;MAAK,CAAC,CAAC;IAC5D;IACA,MAAM,IAAAC,mBAAS,EAACR,OAAO,EAAE,IAAI,CAACjB,OAAO,GAAGI,MAAM,CAACsB,QAAQ,EAAE,CAAC;;IAE1D;IACA,MAAM,IAAAC,mBAAQ,EAAE,gCAA+BV,OAAQ,EAAC,CAAC;EAC7D;EAEQP,kBAAkB,CAACN,MAAmB,EAAEwB,UAAsB,EAAE;IACpE,IAAIC,QAA4B,GAAGC,SAAS;IAC5C,IAAIC,GAAoC,GAAGD,SAAS;IACpD,IAAIE,iBAAkD,GAAGF,SAAS;IAElE,KAAK,MAAMG,CAAC,IAAIL,UAAU,CAACM,MAAM,EAAE;MAC/B,QAAQD,CAAC,CAACE,IAAI;QACV,KAAK,UAAU;UAAE;YACb,IAAI,IAAI,CAACC,eAAe,CAACH,CAAC,CAACI,KAAK,CAAC,EAAE;cAC/BR,QAAQ,GAAGI,CAAC,CAACI,KAAK,CAACA,KAAe;YACtC,CAAC,MAAM;cACH,MAAM,IAAIC,gBAAW,CAAC,6CAA6C,CAAC;YACxE;YACA;UACJ;QAEA,KAAK,KAAK;UAAE;YACR,MAAMC,CAAC,GAAG,IAAI,CAACC,oBAAoB,CAACP,CAAC,CAACI,KAAK,CAAC;YAC5C,IAAI,CAACE,CAAC,EAAE;cACJ,MAAM,IAAID,gBAAW,CAAC,kCAAkC,CAAC;YAC7D;YACAP,GAAG,GAAGQ,CAAC;YACP;UACJ;QAEA,KAAK,mBAAmB;UAAE;YACtB,MAAMA,CAAC,GAAG,IAAI,CAACC,oBAAoB,CAACP,CAAC,CAACI,KAAK,CAAC;YAC5C,IAAI,CAACE,CAAC,EAAE;cACJ,MAAM,IAAID,gBAAW,CAAC,kCAAkC,CAAC;YAC7D;YACAN,iBAAiB,GAAGO,CAAC;YACrB;UACJ;MAAC;IAET;IAEA,IAAI,CAACV,QAAQ,EAAE;MACX,MAAM,IAAIS,gBAAW,CAAC,wCAAwC,CAAC;IACnE;IACA,IAAI,CAACP,GAAG,EAAE;MACN,MAAM,IAAIO,gBAAW,CAAC,mCAAmC,CAAC;IAC9D;IAEAlC,MAAM,CAACqC,aAAa,CAACb,UAAU,CAACO,IAAI,EAAEN,QAAQ,EAAEE,GAAG,EAAEC,iBAAiB,CAAC;EAC3E;EAEQQ,oBAAoB,CAACE,UAAwC,EAAE;IAAA;IACnE,IAAI,IAAI,CAACN,eAAe,CAACM,UAAU,CAAC,EAAE;MAClC,OAAO,IAAIC,4BAAmB,CAACD,UAAU,CAACL,KAAK,EAAY,KAAK,CAAC;IACrE,CAAC,MAAM,IACH,IAAAO,qBAAgB,EAACF,UAAU,CAAC,IAC5B,0BAAAA,UAAU,CAACG,QAAQ,CAACC,GAAG,0DAAvB,sBAAyBX,IAAI,MAAK,KAAK,IACvCO,UAAU,CAACK,IAAI,CAACC,MAAM,KAAK,CAAC,IAC5B,IAAI,CAACZ,eAAe,CAACM,UAAU,CAACK,IAAI,CAAC,CAAC,CAAC,CAACV,KAAK,CAAC,EAChD;MACE,OAAO,IAAIM,4BAAmB,CAACD,UAAU,CAACK,IAAI,CAAC,CAAC,CAAC,CAACV,KAAK,CAACA,KAAK,EAAY,IAAI,CAAC;IAClF,CAAC,MAAM;MACH,OAAO,IAAI;IACf;EACJ;EAEQrB,iBAAiB,CAACZ,MAAmB,EAAEE,IAAmB,EAAE;IAChEF,MAAM,CAAC6C,YAAY,CACf3C,IAAI,CAAC6B,IAAI,EACT7B,IAAI,CAAC4B,MAAM,CAACgB,GAAG,CAAEjB,CAAC,IAAK;MACnB,MAAMI,KAAK,GAAG,IAAAc,gBAAW,EAAClB,CAAC,CAACI,KAAK,CAAC,GAAG,IAAAe,oBAAe,EAACnB,CAAC,CAACI,KAAK,CAAC,GAAG,IAAAgB,eAAU,EAACpB,CAAC,CAACI,KAAK,CAAC;MACnF,OAAO;QAAEF,IAAI,EAAEF,CAAC,CAACE,IAAI;QAAEE;MAAM,CAAC;IAClC,CAAC,CAAC,CACL;EACL;EAEQvB,aAAa,CAACV,MAAmB,EAAEE,IAAe,EAAE;IACxD,MAAMJ,KAAK,GAAGE,MAAM,CAACkD,QAAQ,CAAChD,IAAI,CAAC6B,IAAI,CAAC;IACxC,KAAK,MAAMoB,KAAK,IAAIjD,IAAI,CAAC4B,MAAM,EAAE;MAC7B,IAAI,CAACsB,kBAAkB,CAACtD,KAAK,EAAEqD,KAAK,CAAC;IACzC;IAEA,MAAM;MAAEE,QAAQ;MAAEC,OAAO;MAAEC;IAAmB,CAAC,GAAG,IAAAC,yBAAe,EAACtD,IAAI,CAAC;IAEvE,IAAK,CAACmD,QAAQ,IAAI,CAACC,OAAO,IAAKC,kBAAkB,EAAE;MAC/C;;MAEA;MACAzD,KAAK,CAAC2D,QAAQ,CAACC,qBAAgB,EAAE,SAAS,EAAE,CACxC,IAAIC,6BAAoB,CAAC,UAAU,EAAE,CACjC,IAAIC,2BAAkB,CAAClC,SAAS,EAAE,IAAImC,gCAAuB,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,CAClF,CAAC,CACL,CAAC;;MAEF;MACA/D,KAAK,CAAC2D,QAAQ,CAACK,2BAAsB,EAAE,SAAS,CAAC;;MAEjD;MACAhE,KAAK,CAACiE,YAAY,CAAC,SAAS,EAAE,CAC1B,IAAIH,2BAAkB,CAClBlC,SAAS,EACT,IAAImC,gCAAuB,CAAC,OAAO,EAAE,CACjC,IAAIA,gCAAuB,CAAC,gBAAgB,EAAEC,2BAAsB,CAAC,CACxE,CAAC,CACL,CACJ,CAAC;IACN;IAEA,KAAK,MAAME,IAAI,IAAI9D,IAAI,CAAC+D,UAAU,CAACC,MAAM,CAAEF,IAAI,IAAKA,IAAI,CAAC9D,IAAI,CAACwC,GAAG,IAAI,IAAI,CAACyB,iBAAiB,CAACH,IAAI,CAAC9D,IAAI,CAACwC,GAAG,CAAC,CAAC,EAAE;MACzG,IAAI,CAAC0B,sBAAsB,CAACtE,KAAK,EAAEkE,IAAI,CAAC;IAC5C;IAEA9D,IAAI,CAAC+D,UAAU,CACVC,MAAM,CAAEF,IAAI,IAAKA,IAAI,CAAC9D,IAAI,CAACwC,GAAG,IAAI,CAAC,IAAI,CAACyB,iBAAiB,CAACH,IAAI,CAAC9D,IAAI,CAACwC,GAAG,CAAC,CAAC,CACzE2B,OAAO,CAAEL,IAAI,IAAKlE,KAAK,CAACwE,UAAU,CAAC,IAAI,CAAC5E,eAAe,CAAC6E,iBAAiB,CAACP,IAAI,CAAC,CAAC,CAAC;EAC1F;EAEQG,iBAAiB,CAACH,IAAe,EAAE;IACvC,OAAO,CAAC,CAACA,IAAI,CAACC,UAAU,CAACO,IAAI,CAAEC,CAAC;MAAA;MAAA,OAAK,gBAAAA,CAAC,CAACvE,IAAI,CAACwC,GAAG,gDAAV,YAAYX,IAAI,MAAK,WAAW;IAAA,EAAC;EAC1E;EAEQqB,kBAAkB,CAACtD,KAAsB,EAAEqD,KAAqB,EAAE;IAAA;IACtE,MAAMuB,SAAS,GAAGvB,KAAK,CAACwB,IAAI,CAACA,IAAI,8BAAIxB,KAAK,CAACwB,IAAI,CAACC,SAAS,oFAApB,sBAAsBlC,GAAG,2DAAzB,uBAA2BX,IAAI;IACpE,IAAI,CAAC2C,SAAS,EAAE;MACZ,MAAM,IAAIxC,gBAAW,CAAE,+BAA8BiB,KAAK,CAAC0B,UAAU,CAAC9C,IAAK,IAAGoB,KAAK,CAACpB,IAAK,EAAC,CAAC;IAC/F;IAEA,MAAM4C,IAAI,GAAG,IAAIG,6BAAc,CAACJ,SAAS,EAAEvB,KAAK,CAACwB,IAAI,CAACI,KAAK,EAAE5B,KAAK,CAACwB,IAAI,CAACK,QAAQ,CAAC;IAEjF,MAAMf,UAAU,GAAGd,KAAK,CAACc,UAAU,CAC9BC,MAAM,CAAEF,IAAI,IAAKA,IAAI,CAAC9D,IAAI,CAACwC,GAAG,IAAI,IAAI,CAACyB,iBAAiB,CAACH,IAAI,CAAC9D,IAAI,CAACwC,GAAG,CAAC,CAAC,CACxEI,GAAG,CAAEkB,IAAI,IAAK,IAAI,CAACiB,kBAAkB,CAACjB,IAAI,CAAC,CAAC;IAEjD,MAAMkB,mBAAmB,GAAG/B,KAAK,CAACc,UAAU,CAACC,MAAM,CAC9CF,IAAI,IAAK,CAACA,IAAI,CAAC9D,IAAI,CAACwC,GAAG,IAAI,CAAC,IAAI,CAACyB,iBAAiB,CAACH,IAAI,CAAC9D,IAAI,CAACwC,GAAG,CAAC,CACrE;IAED,MAAMyC,cAAc,GAAGD,mBAAmB,CAACpC,GAAG,CAAEkB,IAAI,IAAK,IAAI,CAACtE,eAAe,CAAC6E,iBAAiB,CAACP,IAAI,CAAC,CAAC;IAEtGlE,KAAK,CAAC2D,QAAQ,CAACN,KAAK,CAACpB,IAAI,EAAE4C,IAAI,EAAEV,UAAU,EAAEkB,cAAc,CAAC;EAChE;EAEQF,kBAAkB,CAACjB,IAA6B,EAAE;IACtD,OAAO,IAAIL,6BAAoB,CAC3B,IAAAyB,aAAQ,EAACpB,IAAI,CAAC9D,IAAI,CAAC,CAAC6B,IAAI,EACxBiC,IAAI,CAACrB,IAAI,CAACG,GAAG,CAAEuC,GAAG,IAAK,IAAI,CAACC,gBAAgB,CAACD,GAAG,CAAC,CAAC,CACrD;EACL;EAEQC,gBAAgB,CAACD,GAAiB,EAAsB;IAC5D,OAAO,IAAIzB,2BAAkB,CAACyB,GAAG,CAACtD,IAAI,EAAE,IAAI,CAACwD,qBAAqB,CAACF,GAAG,CAACpD,KAAK,CAAC,CAAC;EAClF;EAEQsD,qBAAqB,CAACC,IAAgB,EAA2B;IACrE,IAAI,IAAAC,kBAAa,EAACD,IAAI,CAAC,EAAE;MACrB,QAAQ,OAAOA,IAAI,CAACvD,KAAK;QACrB,KAAK,QAAQ;UACT,OAAO,IAAI4B,gCAAuB,CAAC,QAAQ,EAAE2B,IAAI,CAACvD,KAAK,CAAC;QAC5D,KAAK,QAAQ;UACT,OAAO,IAAI4B,gCAAuB,CAAC,QAAQ,EAAE2B,IAAI,CAACvD,KAAK,CAAC;QAC5D,KAAK,SAAS;UACV,OAAO,IAAI4B,gCAAuB,CAAC,SAAS,EAAE2B,IAAI,CAACvD,KAAK,CAAC;QAC7D;UACI,MAAM,IAAIC,gBAAW,CAAE,4BAA2B,OAAOsD,IAAI,CAACvD,KAAM,EAAC,CAAC;MAAC;IAEnF,CAAC,MAAM,IAAI,IAAAc,gBAAW,EAACyC,IAAI,CAAC,EAAE;MAC1B,OAAO,IAAI3B,gCAAuB,CAC9B,OAAO,EACP,IAAI6B,KAAK,CAAC,GAAGF,IAAI,CAACG,KAAK,CAAC7C,GAAG,CAAE8C,IAAI,IAAK,IAAI,CAACL,qBAAqB,CAACK,IAAI,CAAC,CAAC,CAAC,CAC3E;IACL,CAAC,MAAM,IAAI,IAAAC,oBAAe,EAACL,IAAI,CAAC,EAAE;MAC9B,OAAO,IAAI3B,gCAAuB,CAC9B,gBAAgB,EAChB,IAAIiC,6BAAoB,CACpB,IAAAV,aAAQ,EAACI,IAAI,CAACO,MAAM,CAAC,CAAChE,IAAI,EAC1ByD,IAAI,CAAC7C,IAAI,CAACG,GAAG,CAAEuC,GAAG,IAAK,IAAIW,gCAAuB,CAACX,GAAG,CAACtD,IAAI,EAAEsD,GAAG,CAACpD,KAAK,CAAC,CAAC,CAC3E,CACJ;IACL,CAAC,MAAM,IAAI,IAAAO,qBAAgB,EAACgD,IAAI,CAAC,EAAE;MAC/B;MACA,OAAO,IAAI3B,gCAAuB,CAAC,cAAc,EAAE,IAAI,CAACoC,gBAAgB,CAACT,IAAI,CAAC,CAAC;IACnF,CAAC,MAAM;MACH,MAAM,IAAItD,gBAAW,CAAE,mDAAkDsD,IAAI,CAACpF,KAAM,EAAC,CAAC;IAC1F;EACJ;EAEA6F,gBAAgB,CAACT,IAAoB,EAAsB;IACvD,OAAO,IAAIU,2BAAkB,CACzB,IAAAd,aAAQ,EAACI,IAAI,CAAC/C,QAAQ,CAAC,CAACV,IAAI,EAC5ByD,IAAI,CAAC7C,IAAI,CAACG,GAAG,CAAEuC,GAAG,IAAK;MACnB,IAAI,CAAC,IAAAI,kBAAa,EAACJ,GAAG,CAACpD,KAAK,CAAC,EAAE;QAC3B,MAAM,IAAIC,gBAAW,CAAC,wCAAwC,CAAC;MACnE;MACA,OAAO,IAAIiE,8BAAqB,CAACd,GAAG,CAACtD,IAAI,EAAEsD,GAAG,CAACpD,KAAK,CAACA,KAAK,CAAC;IAC/D,CAAC,CAAC,CACL;EACL;EAEQmC,sBAAsB,CAACtE,KAAsB,EAAEkE,IAAwB,EAAE;IAC7ElE,KAAK,CAACmE,UAAU,CAACmC,IAAI,CACjB,IAAIC,6BAAoB,CACpB,IAAAjB,aAAQ,EAACpB,IAAI,CAAC9D,IAAI,CAAC,CAAC6B,IAAI,EACxBiC,IAAI,CAACrB,IAAI,CAACG,GAAG,CAAEuC,GAAG,IAAK,IAAI,CAACC,gBAAgB,CAACD,GAAG,CAAC,CAAC,CACrD,CACJ;EACL;EAEQ7E,YAAY,CAACR,MAAmB,EAAEE,IAAU,EAAE;IAClDF,MAAM,CAACsG,OAAO,CACVpG,IAAI,CAAC6B,IAAI,EACT7B,IAAI,CAAC4B,MAAM,CAACgB,GAAG,CAAEjB,CAAC,IAAKA,CAAC,CAACE,IAAI,CAAC,CACjC;EACL;EAEQC,eAAe,CAACwD,IAAa,EAAuB;IACxD,OAAO,IAAAC,kBAAa,EAACD,IAAI,CAAC,IAAI,OAAOA,IAAI,CAACvD,KAAK,KAAK,QAAQ;EAChE;AACJ;AAAC"}

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

@@ -0,0 +1,27 @@
+import { Argument, AttributeArg, BinaryExpr, DataModelAttribute, DataModelFieldAttribute, Expression, InvocationExpr, LiteralExpr, MemberAccessExpr, 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;
+    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,110 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.default = void 0;
+var _ast = require("@zenstackhq/language/ast");
+var _sdk = require("@zenstackhq/sdk");
+class ZModelCodeGenerator {
+  constructor(options) {
+    this.options = {
+      binaryExprNumberOfSpaces: (options === null || options === void 0 ? void 0 : options.binaryExprNumberOfSpaces) ?? 1,
+      unaryExprNumberOfSpaces: (options === null || options === void 0 ? void 0 : options.unaryExprNumberOfSpaces) ?? 0
+    };
+  }
+  generateAttribute(ast) {
+    const args = ast.args.length ? `(${ast.args.map(x => this.generateAttributeArg(x)).join(', ')})` : '';
+    return `${(0, _sdk.resolved)(ast.decl).name}${args}`;
+  }
+  generateAttributeArg(ast) {
+    return this.generateExpression(ast.value);
+  }
+  generateExpression(ast) {
+    switch (ast.$type) {
+      case _ast.LiteralExpr:
+        return this.generateLiteralExpr(ast);
+      case _ast.UnaryExpr:
+        return this.generateUnaryExpr(ast);
+      case _ast.BinaryExpr:
+        return this.generateBinaryExpr(ast);
+      case _ast.ReferenceExpr:
+        return this.generateReferenceExpr(ast);
+      case _ast.MemberAccessExpr:
+        return this.generateMemberExpr(ast);
+      case _ast.InvocationExpr:
+        return this.generateInvocationExpr(ast);
+      case _ast.NullExpr:
+      case _ast.ThisExpr:
+        return ast.value;
+      default:
+        throw new Error(`Not implemented: ${ast.$type}`);
+    }
+  }
+  generateLiteralExpr(ast) {
+    return typeof ast.value === 'string' ? `'${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 _ast$target$ref;
+    const args = ast.args.length ? `(${ast.args.map(x => this.generateReferenceArg(x)).join(', ')})` : '';
+    return `${(_ast$target$ref = ast.target.ref) === null || _ast$target$ref === void 0 ? void 0 : _ast$target$ref.name}${args}`;
+  }
+  generateReferenceArg(ast) {
+    return `${ast.name}:${ast.value}`;
+  }
+  generateMemberExpr(ast) {
+    var _ast$member$ref;
+    return `${this.generateExpression(ast.operand)}.${(_ast$member$ref = ast.member.ref) === null || _ast$member$ref === void 0 ? void 0 : _ast$member$ref.name}`;
+  }
+  generateInvocationExpr(ast) {
+    var _ast$function$ref;
+    return `${(_ast$function$ref = ast.function.ref) === null || _ast$function$ref === void 0 ? void 0 : _ast$function$ref.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.BinaryExprOperatorPriority[operator];
+    if (ast.left.$type === _ast.BinaryExpr && _ast.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.BinaryExpr && _ast.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","names":["ZModelCodeGenerator","constructor","options","binaryExprNumberOfSpaces","unaryExprNumberOfSpaces","generateAttribute","ast","args","length","map","x","generateAttributeArg","join","resolved","decl","name","generateExpression","value","$type","LiteralExpr","generateLiteralExpr","UnaryExpr","generateUnaryExpr","BinaryExpr","generateBinaryExpr","ReferenceExpr","generateReferenceExpr","MemberAccessExpr","generateMemberExpr","InvocationExpr","generateInvocationExpr","NullExpr","ThisExpr","Error","toString","operator","unaryExprSpace","operand","isCollectionPredicate","isCollectionPredicateOperator","rightExpr","right","left","isLeftParenthesis","isRightParenthesis","isParenthesesNeededForBinaryExpr","binaryExprSpace","generateReferenceArg","target","ref","member","function","generateArgument","repeat","result","currentPriority","BinaryExprOperatorPriority","op","includes"],"sources":["../../../src/plugins/prisma/zmodel-code-generator.ts"],"sourcesContent":["import {\n    Argument,\n    AttributeArg,\n    BinaryExpr,\n    DataModelAttribute,\n    DataModelFieldAttribute,\n    Expression,\n    InvocationExpr,\n    LiteralExpr,\n    MemberAccessExpr,\n    NullExpr,\n    ReferenceArg,\n    ReferenceExpr,\n    ThisExpr,\n    UnaryExpr,\n    BinaryExprOperatorPriority,\n} from '@zenstackhq/language/ast';\nimport { resolved } from '@zenstackhq/sdk';\n\n/**\n * Options for the generator.\n */\nexport interface ZModelCodeOptions {\n    binaryExprNumberOfSpaces: number;\n    unaryExprNumberOfSpaces: number;\n}\n\nexport default class ZModelCodeGenerator {\n    private readonly options: ZModelCodeOptions;\n    constructor(options?: Partial<ZModelCodeOptions>) {\n        this.options = {\n            binaryExprNumberOfSpaces: options?.binaryExprNumberOfSpaces ?? 1,\n            unaryExprNumberOfSpaces: options?.unaryExprNumberOfSpaces ?? 0,\n        };\n    }\n    generateAttribute(ast: DataModelAttribute | DataModelFieldAttribute): string {\n        const args = ast.args.length ? `(${ast.args.map((x) => this.generateAttributeArg(x)).join(', ')})` : '';\n        return `${resolved(ast.decl).name}${args}`;\n    }\n\n    generateAttributeArg(ast: AttributeArg) {\n        return this.generateExpression(ast.value);\n    }\n\n    generateExpression(ast: Expression): string {\n        switch (ast.$type) {\n            case LiteralExpr:\n                return this.generateLiteralExpr(ast as LiteralExpr);\n            case UnaryExpr:\n                return this.generateUnaryExpr(ast as UnaryExpr);\n            case BinaryExpr:\n                return this.generateBinaryExpr(ast as BinaryExpr);\n            case ReferenceExpr:\n                return this.generateReferenceExpr(ast as ReferenceExpr);\n            case MemberAccessExpr:\n                return this.generateMemberExpr(ast as MemberAccessExpr);\n            case InvocationExpr:\n                return this.generateInvocationExpr(ast as InvocationExpr);\n            case NullExpr:\n            case ThisExpr:\n                return (ast as NullExpr | ThisExpr).value;\n            default:\n                throw new Error(`Not implemented: ${ast.$type}`);\n        }\n    }\n\n    generateLiteralExpr(ast: LiteralExpr) {\n        return typeof ast.value === 'string' ? `'${ast.value}'` : ast.value.toString();\n    }\n\n    generateUnaryExpr(ast: UnaryExpr) {\n        return `${ast.operator}${this.unaryExprSpace}${this.generateExpression(ast.operand)}`;\n    }\n\n    generateBinaryExpr(ast: BinaryExpr) {\n        const operator = ast.operator;\n        const isCollectionPredicate = this.isCollectionPredicateOperator(operator);\n        const rightExpr = this.generateExpression(ast.right);\n\n        const { left: isLeftParenthesis, right: isRightParenthesis } = this.isParenthesesNeededForBinaryExpr(ast);\n\n        return `${isLeftParenthesis ? '(' : ''}${this.generateExpression(ast.left)}${isLeftParenthesis ? ')' : ''}${\n            this.binaryExprSpace\n        }${operator}${this.binaryExprSpace}${isRightParenthesis ? '(' : ''}${\n            isCollectionPredicate ? `[${rightExpr}]` : rightExpr\n        }${isRightParenthesis ? ')' : ''}`;\n    }\n\n    generateReferenceExpr(ast: ReferenceExpr) {\n        const args = ast.args.length ? `(${ast.args.map((x) => this.generateReferenceArg(x)).join(', ')})` : '';\n        return `${ast.target.ref?.name}${args}`;\n    }\n\n    generateReferenceArg(ast: ReferenceArg) {\n        return `${ast.name}:${ast.value}`;\n    }\n\n    generateMemberExpr(ast: MemberAccessExpr) {\n        return `${this.generateExpression(ast.operand)}.${ast.member.ref?.name}`;\n    }\n\n    generateInvocationExpr(ast: InvocationExpr) {\n        return `${ast.function.ref?.name}(${ast.args.map((x) => this.generateArgument(x)).join(', ')})`;\n    }\n\n    generateArgument(ast: Argument) {\n        return `${ast.name && ':'} ${this.generateExpression(ast.value)}`;\n    }\n\n    private get binaryExprSpace(): string {\n        return ' '.repeat(this.options.binaryExprNumberOfSpaces);\n    }\n\n    private get unaryExprSpace(): string {\n        return ' '.repeat(this.options.unaryExprNumberOfSpaces);\n    }\n\n    private isParenthesesNeededForBinaryExpr(ast: BinaryExpr): { left: boolean; right: boolean } {\n        const result = { left: false, right: false };\n        const operator = ast.operator;\n        const isCollectionPredicate = this.isCollectionPredicateOperator(operator);\n\n        const currentPriority = BinaryExprOperatorPriority[operator];\n\n        if (\n            ast.left.$type === BinaryExpr &&\n            BinaryExprOperatorPriority[(ast.left as BinaryExpr)['operator']] < currentPriority\n        ) {\n            result.left = true;\n        }\n        /**\n         *  1 collection predicate operator has [] around the right operand, no need to add parenthesis.\n         *  2 grammar is left associative, so if the right operand has the same priority still need to add parenthesis.\n         **/\n        if (\n            !isCollectionPredicate &&\n            ast.right.$type === BinaryExpr &&\n            BinaryExprOperatorPriority[(ast.right as BinaryExpr)['operator']] <= currentPriority\n        ) {\n            result.right = true;\n        }\n\n        return result;\n    }\n\n    private isCollectionPredicateOperator(op: BinaryExpr['operator']) {\n        return ['?', '!', '^'].includes(op);\n    }\n}\n"],"mappings":";;;;;;AAAA;AAiBA;AAUe,MAAMA,mBAAmB,CAAC;EAErCC,WAAW,CAACC,OAAoC,EAAE;IAC9C,IAAI,CAACA,OAAO,GAAG;MACXC,wBAAwB,EAAE,CAAAD,OAAO,aAAPA,OAAO,uBAAPA,OAAO,CAAEC,wBAAwB,KAAI,CAAC;MAChEC,uBAAuB,EAAE,CAAAF,OAAO,aAAPA,OAAO,uBAAPA,OAAO,CAAEE,uBAAuB,KAAI;IACjE,CAAC;EACL;EACAC,iBAAiB,CAACC,GAAiD,EAAU;IACzE,MAAMC,IAAI,GAAGD,GAAG,CAACC,IAAI,CAACC,MAAM,GAAI,IAAGF,GAAG,CAACC,IAAI,CAACE,GAAG,CAAEC,CAAC,IAAK,IAAI,CAACC,oBAAoB,CAACD,CAAC,CAAC,CAAC,CAACE,IAAI,CAAC,IAAI,CAAE,GAAE,GAAG,EAAE;IACvG,OAAQ,GAAE,IAAAC,aAAQ,EAACP,GAAG,CAACQ,IAAI,CAAC,CAACC,IAAK,GAAER,IAAK,EAAC;EAC9C;EAEAI,oBAAoB,CAACL,GAAiB,EAAE;IACpC,OAAO,IAAI,CAACU,kBAAkB,CAACV,GAAG,CAACW,KAAK,CAAC;EAC7C;EAEAD,kBAAkB,CAACV,GAAe,EAAU;IACxC,QAAQA,GAAG,CAACY,KAAK;MACb,KAAKC,gBAAW;QACZ,OAAO,IAAI,CAACC,mBAAmB,CAACd,GAAG,CAAgB;MACvD,KAAKe,cAAS;QACV,OAAO,IAAI,CAACC,iBAAiB,CAAChB,GAAG,CAAc;MACnD,KAAKiB,eAAU;QACX,OAAO,IAAI,CAACC,kBAAkB,CAAClB,GAAG,CAAe;MACrD,KAAKmB,kBAAa;QACd,OAAO,IAAI,CAACC,qBAAqB,CAACpB,GAAG,CAAkB;MAC3D,KAAKqB,qBAAgB;QACjB,OAAO,IAAI,CAACC,kBAAkB,CAACtB,GAAG,CAAqB;MAC3D,KAAKuB,mBAAc;QACf,OAAO,IAAI,CAACC,sBAAsB,CAACxB,GAAG,CAAmB;MAC7D,KAAKyB,aAAQ;MACb,KAAKC,aAAQ;QACT,OAAQ1B,GAAG,CAAyBW,KAAK;MAC7C;QACI,MAAM,IAAIgB,KAAK,CAAE,oBAAmB3B,GAAG,CAACY,KAAM,EAAC,CAAC;IAAC;EAE7D;EAEAE,mBAAmB,CAACd,GAAgB,EAAE;IAClC,OAAO,OAAOA,GAAG,CAACW,KAAK,KAAK,QAAQ,GAAI,IAAGX,GAAG,CAACW,KAAM,GAAE,GAAGX,GAAG,CAACW,KAAK,CAACiB,QAAQ,EAAE;EAClF;EAEAZ,iBAAiB,CAAChB,GAAc,EAAE;IAC9B,OAAQ,GAAEA,GAAG,CAAC6B,QAAS,GAAE,IAAI,CAACC,cAAe,GAAE,IAAI,CAACpB,kBAAkB,CAACV,GAAG,CAAC+B,OAAO,CAAE,EAAC;EACzF;EAEAb,kBAAkB,CAAClB,GAAe,EAAE;IAChC,MAAM6B,QAAQ,GAAG7B,GAAG,CAAC6B,QAAQ;IAC7B,MAAMG,qBAAqB,GAAG,IAAI,CAACC,6BAA6B,CAACJ,QAAQ,CAAC;IAC1E,MAAMK,SAAS,GAAG,IAAI,CAACxB,kBAAkB,CAACV,GAAG,CAACmC,KAAK,CAAC;IAEpD,MAAM;MAAEC,IAAI,EAAEC,iBAAiB;MAAEF,KAAK,EAAEG;IAAmB,CAAC,GAAG,IAAI,CAACC,gCAAgC,CAACvC,GAAG,CAAC;IAEzG,OAAQ,GAAEqC,iBAAiB,GAAG,GAAG,GAAG,EAAG,GAAE,IAAI,CAAC3B,kBAAkB,CAACV,GAAG,CAACoC,IAAI,CAAE,GAAEC,iBAAiB,GAAG,GAAG,GAAG,EAAG,GACtG,IAAI,CAACG,eACR,GAAEX,QAAS,GAAE,IAAI,CAACW,eAAgB,GAAEF,kBAAkB,GAAG,GAAG,GAAG,EAAG,GAC/DN,qBAAqB,GAAI,IAAGE,SAAU,GAAE,GAAGA,SAC9C,GAAEI,kBAAkB,GAAG,GAAG,GAAG,EAAG,EAAC;EACtC;EAEAlB,qBAAqB,CAACpB,GAAkB,EAAE;IAAA;IACtC,MAAMC,IAAI,GAAGD,GAAG,CAACC,IAAI,CAACC,MAAM,GAAI,IAAGF,GAAG,CAACC,IAAI,CAACE,GAAG,CAAEC,CAAC,IAAK,IAAI,CAACqC,oBAAoB,CAACrC,CAAC,CAAC,CAAC,CAACE,IAAI,CAAC,IAAI,CAAE,GAAE,GAAG,EAAE;IACvG,OAAQ,sBAAEN,GAAG,CAAC0C,MAAM,CAACC,GAAG,oDAAd,gBAAgBlC,IAAK,GAAER,IAAK,EAAC;EAC3C;EAEAwC,oBAAoB,CAACzC,GAAiB,EAAE;IACpC,OAAQ,GAAEA,GAAG,CAACS,IAAK,IAAGT,GAAG,CAACW,KAAM,EAAC;EACrC;EAEAW,kBAAkB,CAACtB,GAAqB,EAAE;IAAA;IACtC,OAAQ,GAAE,IAAI,CAACU,kBAAkB,CAACV,GAAG,CAAC+B,OAAO,CAAE,IAAC,mBAAE/B,GAAG,CAAC4C,MAAM,CAACD,GAAG,oDAAd,gBAAgBlC,IAAK,EAAC;EAC5E;EAEAe,sBAAsB,CAACxB,GAAmB,EAAE;IAAA;IACxC,OAAQ,wBAAEA,GAAG,CAAC6C,QAAQ,CAACF,GAAG,sDAAhB,kBAAkBlC,IAAK,IAAGT,GAAG,CAACC,IAAI,CAACE,GAAG,CAAEC,CAAC,IAAK,IAAI,CAAC0C,gBAAgB,CAAC1C,CAAC,CAAC,CAAC,CAACE,IAAI,CAAC,IAAI,CAAE,GAAE;EACnG;EAEAwC,gBAAgB,CAAC9C,GAAa,EAAE;IAC5B,OAAQ,GAAEA,GAAG,CAACS,IAAI,IAAI,GAAI,IAAG,IAAI,CAACC,kBAAkB,CAACV,GAAG,CAACW,KAAK,CAAE,EAAC;EACrE;EAEA,IAAY6B,eAAe,GAAW;IAClC,OAAO,GAAG,CAACO,MAAM,CAAC,IAAI,CAACnD,OAAO,CAACC,wBAAwB,CAAC;EAC5D;EAEA,IAAYiC,cAAc,GAAW;IACjC,OAAO,GAAG,CAACiB,MAAM,CAAC,IAAI,CAACnD,OAAO,CAACE,uBAAuB,CAAC;EAC3D;EAEQyC,gCAAgC,CAACvC,GAAe,EAAqC;IACzF,MAAMgD,MAAM,GAAG;MAAEZ,IAAI,EAAE,KAAK;MAAED,KAAK,EAAE;IAAM,CAAC;IAC5C,MAAMN,QAAQ,GAAG7B,GAAG,CAAC6B,QAAQ;IAC7B,MAAMG,qBAAqB,GAAG,IAAI,CAACC,6BAA6B,CAACJ,QAAQ,CAAC;IAE1E,MAAMoB,eAAe,GAAGC,+BAA0B,CAACrB,QAAQ,CAAC;IAE5D,IACI7B,GAAG,CAACoC,IAAI,CAACxB,KAAK,KAAKK,eAAU,IAC7BiC,+BAA0B,CAAElD,GAAG,CAACoC,IAAI,CAAgB,UAAU,CAAC,CAAC,GAAGa,eAAe,EACpF;MACED,MAAM,CAACZ,IAAI,GAAG,IAAI;IACtB;IACA;AACR;AACA;AACA;IACQ,IACI,CAACJ,qBAAqB,IACtBhC,GAAG,CAACmC,KAAK,CAACvB,KAAK,KAAKK,eAAU,IAC9BiC,+BAA0B,CAAElD,GAAG,CAACmC,KAAK,CAAgB,UAAU,CAAC,CAAC,IAAIc,eAAe,EACtF;MACED,MAAM,CAACb,KAAK,GAAG,IAAI;IACvB;IAEA,OAAOa,MAAM;EACjB;EAEQf,6BAA6B,CAACkB,EAA0B,EAAE;IAC9D,OAAO,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAACC,QAAQ,CAACD,EAAE,CAAC;EACvC;AACJ;AAAC"}

package/res/prism-zmodel.js

@@ -2,13 +2,11 @@
 
 (function (Prism) {
     Prism.languages.zmodel = Prism.languages.extend('clike', {
-        keyword:
-            /\b(?:datasource|enum|generator|model|attribute|function|null|this)\b/,
+        keyword: /\b(?:datasource|enum|generator|model|attribute|function|null|this)\b/,
         'type-class-name': /(\b()\s+)[\w.\\]+/,
     });
 
-    Prism.languages.javascript['class-name'][0].pattern =
-        /(\b(?:model|datasource|enum|generator)\s+)[\w.\\]+/;
+    Prism.languages.javascript['class-name'][0].pattern = /(\b(?:model|datasource|enum|generator)\s+)[\w.\\]+/;
 
     Prism.languages.insertBefore('zmodel', 'function', {
         annotation: {

package/res/starter.zmodel

@@ -0,0 +1,47 @@
+// This is a sample model to get you started.
+// Learn how to model you app: https://zenstack.dev/#/modeling-your-app.
+
+/*
+ * A sample data source using local sqlite db.
+ * See how to use a different db: https://zenstack.dev/#/zmodel-data-source.
+ */
+datasource db {
+    provider = 'sqlite'
+    url = 'file:./todo.db'
+}
+
+/*
+ * User model
+ */
+model User {
+    id String @id @default(cuid())
+    email String @unique @email
+    password String @password @omit @length(8, 16)
+    posts Post[]
+
+    // everybody can signup
+    @@allow('create', true)
+
+    // full access by self
+    @@allow('all', auth() == this)
+}
+
+/*
+ * Post model
+ */
+model Post {
+    id String @id @default(cuid())
+    createdAt DateTime @default(now())
+    updatedAt DateTime @updatedAt
+    title String @length(1, 256)
+    content String
+    published Boolean @default(false)
+    author User? @relation(fields: [authorId], references: [id])
+    authorId String?
+
+    // allow read for all signin users
+    @@allow('read', auth() != null && published)
+
+    // full access by author
+    @@allow('all', author == auth())
+}

package/res/stdlib.zmodel

@@ -84,8 +84,19 @@ function autoincrement(): Int {}
  */
 function dbgenerated(expr: String): Any {}
 
+/**
+ * Gets entities value before an update. Only valid when used in a "update" policy rule.
+ */
+function future(): Any {}
+
+/**
+ * Marks an attribute to be only applicable to certain field types.
+ */
 attribute @@@targetField(targetField: AttributeTargetField[])
 
+/**
+ * Indicates an attribute is directly supported by the Prisma schema.
+ */
 attribute @@@prisma()
 
 /*
@@ -116,7 +127,7 @@ attribute @@index(_ fields: FieldReference[], map: String?) @@@prisma
 /*
  * Defines meta information about the relation.
  */
-attribute @relation(_ name: String?, fields: FieldReference[]?, references: FieldReference[]?, onDelete: ReferentialAction?, onUpdate: ReferentialAction?, map: String?) @@@prisma
+attribute @relation(_ name: String?, fields: FieldReference[]?, references: TransitiveFieldReference[]?, onDelete: ReferentialAction?, onUpdate: ReferentialAction?, map: String?) @@@prisma
 
 /*
  * Maps a field name or enum value from the schema to a column with a different name in the database.

package/telemetry.d.ts

@@ -0,0 +1,21 @@
+/**
+ * Telemetry events
+ */
+export declare type TelemetryEvents = 'cli:start' | 'cli:complete' | 'cli:error' | 'cli:command:start' | 'cli:command:complete' | 'cli:command:error' | 'cli:plugin:start' | 'cli:plugin:complete' | 'cli:plugin:error';
+/**
+ * Utility class for sending telemetry
+ */
+export declare class Telemetry {
+    private readonly mixpanel;
+    private readonly hostId;
+    private readonly sessionid;
+    private readonly trackingToken;
+    private readonly _os;
+    private readonly version;
+    private exitWait;
+    constructor();
+    track(event: TelemetryEvents, properties?: Record<string, unknown>): void;
+    trackSpan(startEvent: TelemetryEvents, completeEvent: TelemetryEvents, errorEvent: TelemetryEvents, properties: Record<string, unknown>, action: () => Promise<unknown> | void): Promise<void>;
+}
+declare const _default: Telemetry;
+export default _default;

package/telemetry.js

@@ -0,0 +1,102 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.default = exports.Telemetry = void 0;
+var _asyncExitHook = _interopRequireDefault(require("async-exit-hook"));
+var _commander = require("commander");
+var _cuid = _interopRequireDefault(require("cuid"));
+var _mixpanel = require("mixpanel");
+var _nodeMachineId = require("node-machine-id");
+var os = _interopRequireWildcard(require("os"));
+var _sleepPromise = _interopRequireDefault(require("sleep-promise"));
+var _cliError = require("./cli/cli-error");
+var _versionUtils = require("./utils/version-utils");
+function _getRequireWildcardCache(nodeInterop) { if (typeof WeakMap !== "function") return null; var cacheBabelInterop = new WeakMap(); var cacheNodeInterop = new WeakMap(); return (_getRequireWildcardCache = function (nodeInterop) { return nodeInterop ? cacheNodeInterop : cacheBabelInterop; })(nodeInterop); }
+function _interopRequireWildcard(obj, nodeInterop) { if (!nodeInterop && obj && obj.__esModule) { return obj; } if (obj === null || typeof obj !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(nodeInterop); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (key !== "default" && Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj.default = obj; if (cache) { cache.set(obj, newObj); } return newObj; }
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+/**
+ * Utility class for sending telemetry
+ */
+class Telemetry {
+  hostId = (0, _nodeMachineId.machineIdSync)();
+  sessionid = (0, _cuid.default)();
+  trackingToken = process && process.env && process.env.TELEMETRY_TRACKING_TOKEN || "74944eb779d7d3b4ce185be843fde9fc";
+  _os = os.platform();
+  version = (0, _versionUtils.getVersion)();
+  exitWait = 200;
+  constructor() {
+    if ((process && process.env && process.env.DO_NOT_TRACK || "1") !== '1' && this.trackingToken) {
+      this.mixpanel = (0, _mixpanel.init)(this.trackingToken, {
+        geolocate: true
+      });
+    }
+    (0, _asyncExitHook.default)(async callback => {
+      if (this.mixpanel) {
+        // a small delay to ensure telemetry is sent
+        await (0, _sleepPromise.default)(this.exitWait);
+      }
+      callback();
+    });
+    const errorHandler = async err => {
+      this.track('cli:error', {
+        message: err.message,
+        stack: err.stack
+      });
+      if (this.mixpanel) {
+        // a small delay to ensure telemetry is sent
+        await (0, _sleepPromise.default)(this.exitWait);
+      }
+      if (err instanceof _cliError.CliError || err instanceof _commander.CommanderError) {
+        // error already logged
+      } else {
+        console.error('\nAn unexpected error occurred:\n', err);
+      }
+      process.exit(1);
+    };
+    _asyncExitHook.default.unhandledRejectionHandler(errorHandler);
+    _asyncExitHook.default.uncaughtExceptionHandler(errorHandler);
+  }
+  track(event, properties = {}) {
+    if (this.mixpanel) {
+      const payload = {
+        distinct_id: this.hostId,
+        session: this.sessionid,
+        time: new Date(),
+        $os: this._os,
+        nodeVersion: process.version,
+        version: this.version,
+        ...properties
+      };
+      this.mixpanel.track(event, payload);
+    }
+  }
+  async trackSpan(startEvent, completeEvent, errorEvent, properties, action) {
+    this.track(startEvent, properties);
+    const start = Date.now();
+    let success = true;
+    try {
+      await Promise.resolve(action());
+      // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    } catch (err) {
+      this.track(errorEvent, {
+        message: err.message,
+        stack: err.stack,
+        ...properties
+      });
+      success = false;
+      throw err;
+    } finally {
+      this.track(completeEvent, {
+        duration: Date.now() - start,
+        success,
+        ...properties
+      });
+    }
+  }
+}
+exports.Telemetry = Telemetry;
+var _default = new Telemetry();
+exports.default = _default;
+//# sourceMappingURL=telemetry.js.map