zenstack 1.10.0 → 2.0.0-alpha.2

package/plugins/enhancer/enhance/index.js

@@ -0,0 +1,336 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generate = void 0;
+const runtime_1 = require("@zenstackhq/runtime");
+const sdk_1 = require("@zenstackhq/sdk");
+const ast_1 = require("@zenstackhq/sdk/ast");
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const ts_morph_1 = require("ts-morph");
+const __1 = require("..");
+const exec_utils_1 = require("../../../utils/exec-utils");
+const prisma_1 = require("../../prisma");
+const schema_generator_1 = require("../../prisma/schema-generator");
+const enhancer_utils_1 = require("../enhancer-utils");
+function generate(model, options, project, outDir) {
+    return __awaiter(this, void 0, void 0, function* () {
+        let logicalPrismaClientDir;
+        let dmmf;
+        if (needsLogicalClient(model)) {
+            // schema contains delegate models, need to generate a logical prisma schema
+            const result = yield generateLogicalPrisma(model, options, outDir);
+            logicalPrismaClientDir = './.logical-prisma-client';
+            dmmf = result.dmmf;
+            // create a reexport of the logical prisma client
+            const prismaDts = project.createSourceFile(path_1.default.join(outDir, 'prisma.d.ts'), `export type * from '${logicalPrismaClientDir}/index-fixed';`, { overwrite: true });
+            yield saveSourceFile(prismaDts, options);
+        }
+        else {
+            // just reexport the prisma client
+            const prismaDts = project.createSourceFile(path_1.default.join(outDir, 'prisma.d.ts'), `export type * from '${(0, sdk_1.getPrismaClientImportSpec)(outDir, options)}';`, { overwrite: true });
+            yield saveSourceFile(prismaDts, options);
+        }
+        const enhanceTs = project.createSourceFile(path_1.default.join(outDir, 'enhance.ts'), `import { createEnhancement, type EnhancementContext, type EnhancementOptions, type ZodSchemas } from '@zenstackhq/runtime';
+import modelMeta from './model-meta';
+import policy from './policy';
+${options.withZodSchemas ? "import * as zodSchemas from './zod';" : 'const zodSchemas = undefined;'}
+import { Prisma } from '${(0, sdk_1.getPrismaClientImportSpec)(outDir, options)}';
+${logicalPrismaClientDir ? `import { type PrismaClient } from '${logicalPrismaClientDir}/index-fixed';` : ``}
+
+export function enhance<DbClient extends object>(prisma: DbClient, context?: EnhancementContext, options?: EnhancementOptions) {
+    return createEnhancement(prisma, {
+        modelMeta,
+        policy,
+        zodSchemas: zodSchemas as unknown as (ZodSchemas | undefined),
+        prismaModule: Prisma,
+        ...options
+    }, context)${logicalPrismaClientDir ? ' as PrismaClient' : ''};
+}
+`, { overwrite: true });
+        yield saveSourceFile(enhanceTs, options);
+        return { dmmf };
+    });
+}
+exports.generate = generate;
+function needsLogicalClient(model) {
+    return hasDelegateModel(model) || hasAuthInDefault(model);
+}
+function hasDelegateModel(model) {
+    const dataModels = (0, sdk_1.getDataModels)(model);
+    return dataModels.some((dm) => (0, sdk_1.isDelegateModel)(dm) && dataModels.some((sub) => sub.superTypes.some((base) => base.ref === dm)));
+}
+function hasAuthInDefault(model) {
+    return (0, sdk_1.getDataModels)(model).some((dm) => dm.fields.some((f) => f.attributes.some((attr) => (0, enhancer_utils_1.isDefaultWithAuth)(attr))));
+}
+function generateLogicalPrisma(model, options, outDir) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const prismaGenerator = new schema_generator_1.PrismaSchemaGenerator(model);
+        const prismaClientOutDir = './.logical-prisma-client';
+        const logicalPrismaFile = path_1.default.join(outDir, 'logical.prisma');
+        yield prismaGenerator.generate({
+            provider: '@internal', // doesn't matter
+            schemaPath: options.schemaPath,
+            output: logicalPrismaFile,
+            overrideClientGenerationPath: prismaClientOutDir,
+            mode: 'logical',
+        });
+        // generate the prisma client
+        const generateCmd = `prisma generate --schema "${logicalPrismaFile}" --no-engine`;
+        try {
+            // run 'prisma generate'
+            yield (0, exec_utils_1.execPackage)(generateCmd, { stdio: 'ignore' });
+        }
+        catch (_a) {
+            yield (0, prisma_1.trackPrismaSchemaError)(logicalPrismaFile);
+            try {
+                // run 'prisma generate' again with output to the console
+                yield (0, exec_utils_1.execPackage)(generateCmd);
+            }
+            catch (_b) {
+                // noop
+            }
+            throw new sdk_1.PluginError(__1.name, `Failed to run "prisma generate"`);
+        }
+        // make a bunch of typing fixes to the generated prisma client
+        yield processClientTypes(model, path_1.default.join(outDir, prismaClientOutDir));
+        return {
+            prismaSchema: logicalPrismaFile,
+            // load the dmmf of the logical prisma schema
+            dmmf: yield (0, sdk_1.getDMMF)({ datamodel: fs_1.default.readFileSync(logicalPrismaFile, { encoding: 'utf-8' }) }),
+        };
+    });
+}
+function processClientTypes(model, prismaClientDir) {
+    return __awaiter(this, void 0, void 0, function* () {
+        // make necessary updates to the generated `index.d.ts` file and save it as `index-fixed.d.ts`
+        const project = new ts_morph_1.Project();
+        const sf = project.addSourceFileAtPath(path_1.default.join(prismaClientDir, 'index.d.ts'));
+        // build a map of delegate models and their sub models
+        const delegateInfo = [];
+        model.declarations
+            .filter((d) => (0, sdk_1.isDelegateModel)(d))
+            .forEach((dm) => {
+            delegateInfo.push([
+                dm,
+                model.declarations.filter((d) => (0, ast_1.isDataModel)(d) && d.superTypes.some((s) => s.ref === dm)),
+            ]);
+        });
+        const sfNew = project.createSourceFile(path_1.default.join(prismaClientDir, 'index-fixed.d.ts'), undefined, {
+            overwrite: true,
+        });
+        if (delegateInfo.length > 0) {
+            // transform types for delegated models
+            transformDelegate(sf, sfNew, delegateInfo);
+            sfNew.formatText();
+        }
+        else {
+            // just copy
+            sfNew.replaceWithText(sf.getFullText());
+        }
+        yield sfNew.save();
+    });
+}
+function transformDelegate(sf, sfNew, delegateModels) {
+    // copy toplevel imports
+    sfNew.addImportDeclarations(sf.getImportDeclarations().map((n) => n.getStructure()));
+    // copy toplevel import equals
+    sfNew.addStatements(sf.getChildrenOfKind(ts_morph_1.SyntaxKind.ImportEqualsDeclaration).map((n) => n.getFullText()));
+    // copy toplevel exports
+    sfNew.addExportAssignments(sf.getExportAssignments().map((n) => n.getStructure()));
+    // copy toplevel type aliases
+    sfNew.addTypeAliases(sf.getTypeAliases().map((n) => n.getStructure()));
+    // copy toplevel classes
+    sfNew.addClasses(sf.getClasses().map((n) => n.getStructure()));
+    // copy toplevel variables
+    sfNew.addVariableStatements(sf.getVariableStatements().map((n) => n.getStructure()));
+    // copy toplevel namespaces except for `Prisma`
+    sfNew.addModules(sf
+        .getModules()
+        .filter((n) => n.getName() !== 'Prisma')
+        .map((n) => n.getStructure()));
+    // transform the `Prisma` namespace
+    const prismaModule = sf.getModuleOrThrow('Prisma');
+    const newPrismaModule = sfNew.addModule({ name: 'Prisma', isExported: true });
+    transformPrismaModule(prismaModule, newPrismaModule, delegateModels);
+}
+function transformPrismaModule(prismaModule, newPrismaModule, delegateInfo) {
+    // module block is the direct container of declarations inside a namespace
+    const moduleBlock = prismaModule.getFirstChildByKindOrThrow(ts_morph_1.SyntaxKind.ModuleBlock);
+    // most of the toplevel constructs should be copied over
+    // here we use ts-morph batch operations for optimal performance
+    // copy imports
+    newPrismaModule.addStatements(moduleBlock.getChildrenOfKind(ts_morph_1.SyntaxKind.ImportEqualsDeclaration).map((n) => n.getFullText()));
+    // copy classes
+    newPrismaModule.addClasses(moduleBlock.getClasses().map((n) => n.getStructure()));
+    // copy functions
+    newPrismaModule.addFunctions(moduleBlock.getFunctions().map((n) => n.getStructure()));
+    // copy nested namespaces
+    newPrismaModule.addModules(moduleBlock.getModules().map((n) => n.getStructure()));
+    // transform variables
+    const newVariables = moduleBlock.getVariableStatements().map((variable) => transformVariableStatement(variable));
+    newPrismaModule.addVariableStatements(newVariables);
+    // transform interfaces
+    const newInterfaces = moduleBlock.getInterfaces().map((iface) => transformInterface(iface, delegateInfo));
+    newPrismaModule.addInterfaces(newInterfaces);
+    // transform type aliases
+    const newTypeAliases = moduleBlock.getTypeAliases().map((typeAlias) => transformTypeAlias(typeAlias, delegateInfo));
+    newPrismaModule.addTypeAliases(newTypeAliases);
+}
+function transformVariableStatement(variable) {
+    const structure = variable.getStructure();
+    // remove `delegate_aux_*` fields from the variable's typing
+    const auxFields = findAuxDecls(variable);
+    if (auxFields.length > 0) {
+        structure.declarations.forEach((variable) => {
+            var _a;
+            let source = (_a = variable.type) === null || _a === void 0 ? void 0 : _a.toString();
+            auxFields.forEach((f) => {
+                source = source === null || source === void 0 ? void 0 : source.replace(f.getText(), '');
+            });
+            variable.type = source;
+        });
+    }
+    return structure;
+}
+function transformInterface(iface, delegateInfo) {
+    var _a, _b, _c;
+    const structure = iface.getStructure();
+    // filter out aux fields
+    structure.properties = (_a = structure.properties) === null || _a === void 0 ? void 0 : _a.filter((p) => !p.name.startsWith(runtime_1.DELEGATE_AUX_RELATION_PREFIX));
+    // filter out aux methods
+    structure.methods = (_b = structure.methods) === null || _b === void 0 ? void 0 : _b.filter((m) => !m.name.startsWith(runtime_1.DELEGATE_AUX_RELATION_PREFIX));
+    if (delegateInfo.some(([delegate]) => `${delegate.name}Delegate` === iface.getName())) {
+        // delegate models cannot be created directly, remove create/createMany/upsert
+        structure.methods = (_c = structure.methods) === null || _c === void 0 ? void 0 : _c.filter((m) => !['create', 'createMany', 'upsert'].includes(m.name));
+    }
+    return structure;
+}
+function transformTypeAlias(typeAlias, delegateInfo) {
+    const structure = typeAlias.getStructure();
+    let source = structure.type;
+    // remove aux fields
+    source = removeAuxFieldsFromTypeAlias(typeAlias, source);
+    // remove discriminator field from concrete input types
+    source = removeDiscriminatorFromConcreteInput(typeAlias, delegateInfo, source);
+    // remove create/connectOrCreate/upsert fields from delegate's input types
+    source = removeCreateFromDelegateInput(typeAlias, delegateInfo, source);
+    // fix delegate payload union type
+    source = fixDelegatePayloadType(typeAlias, delegateInfo, source);
+    structure.type = source;
+    return structure;
+}
+function fixDelegatePayloadType(typeAlias, delegateInfo, source) {
+    // change the type of `$<DelegateModel>Payload` type of delegate model to a union of concrete types
+    const typeName = typeAlias.getName();
+    const payloadRecord = delegateInfo.find(([delegate]) => `$${delegate.name}Payload` === typeName);
+    if (payloadRecord) {
+        const discriminatorDecl = getDiscriminatorField(payloadRecord[0]);
+        if (discriminatorDecl) {
+            source = `${payloadRecord[1]
+                .map((concrete) => `($${concrete.name}Payload<ExtArgs> & { scalars: { ${discriminatorDecl.name}: '${concrete.name}' } })`)
+                .join(' | ')}`;
+        }
+    }
+    return source;
+}
+function removeCreateFromDelegateInput(typeAlias, delegateModels, source) {
+    // remove create/connectOrCreate/upsert fields from delegate's input types because
+    // delegate models cannot be created directly
+    const typeName = typeAlias.getName();
+    const delegateModelNames = delegateModels.map(([delegate]) => delegate.name);
+    const delegateCreateUpdateInputRegex = new RegExp(`\\${delegateModelNames.join('|')}(Unchecked)?(Create|Update).*Input`);
+    if (delegateCreateUpdateInputRegex.test(typeName)) {
+        const toRemove = typeAlias
+            .getDescendantsOfKind(ts_morph_1.SyntaxKind.PropertySignature)
+            .filter((p) => ['create', 'connectOrCreate', 'upsert'].includes(p.getName()));
+        toRemove.forEach((r) => {
+            source = source.replace(r.getText(), '');
+        });
+    }
+    return source;
+}
+function removeDiscriminatorFromConcreteInput(typeAlias, delegateInfo, source) {
+    // remove discriminator field from the create/update input of concrete models because
+    // discriminator cannot be set directly
+    const typeName = typeAlias.getName();
+    const concreteModelNames = delegateInfo.map(([, concretes]) => concretes.map((c) => c.name)).flatMap((c) => c);
+    const concreteCreateUpdateInputRegex = new RegExp(`(${concreteModelNames.join('|')})(Unchecked)?(Create|Update).*Input`);
+    const match = typeName.match(concreteCreateUpdateInputRegex);
+    if (match) {
+        const modelName = match[1];
+        const record = delegateInfo.find(([, concretes]) => concretes.some((c) => c.name === modelName));
+        if (record) {
+            // remove all discriminator fields recursively
+            const delegateOfConcrete = record[0];
+            const discriminators = getDiscriminatorFieldsRecursively(delegateOfConcrete);
+            discriminators.forEach((discriminatorDecl) => {
+                const discriminatorNode = findNamedProperty(typeAlias, discriminatorDecl.name);
+                if (discriminatorNode) {
+                    source = source.replace(discriminatorNode.getText(), '');
+                }
+            });
+        }
+    }
+    return source;
+}
+function removeAuxFieldsFromTypeAlias(typeAlias, source) {
+    // remove `delegate_aux_*` fields from the type alias
+    const auxDecls = findAuxDecls(typeAlias);
+    if (auxDecls.length > 0) {
+        auxDecls.forEach((d) => {
+            source = source.replace(d.getText(), '');
+        });
+    }
+    return source;
+}
+function findNamedProperty(typeAlias, name) {
+    return typeAlias.getFirstDescendant((d) => d.isKind(ts_morph_1.SyntaxKind.PropertySignature) && d.getName() === name);
+}
+function findAuxDecls(node) {
+    return node
+        .getDescendantsOfKind(ts_morph_1.SyntaxKind.PropertySignature)
+        .filter((n) => n.getName().startsWith(runtime_1.DELEGATE_AUX_RELATION_PREFIX));
+}
+function getDiscriminatorField(delegate) {
+    var _a;
+    const delegateAttr = (0, sdk_1.getAttribute)(delegate, '@@delegate');
+    if (!delegateAttr) {
+        return undefined;
+    }
+    const arg = (_a = delegateAttr.args[0]) === null || _a === void 0 ? void 0 : _a.value;
+    return (0, ast_1.isReferenceExpr)(arg) ? arg.target.ref : undefined;
+}
+function getDiscriminatorFieldsRecursively(delegate, result = []) {
+    if ((0, sdk_1.isDelegateModel)(delegate)) {
+        const discriminator = getDiscriminatorField(delegate);
+        if (discriminator) {
+            result.push(discriminator);
+        }
+        for (const superType of delegate.superTypes) {
+            if (superType.ref) {
+                result.push(...getDiscriminatorFieldsRecursively(superType.ref, result));
+            }
+        }
+    }
+    return result;
+}
+function saveSourceFile(sf, options) {
+    return __awaiter(this, void 0, void 0, function* () {
+        if (options.preserveTsFiles) {
+            yield sf.save();
+        }
+    });
+}
+//# sourceMappingURL=index.js.map

package/plugins/enhancer/enhance/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/plugins/enhancer/enhance/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AACA,iDAAmE;AACnE,yCAQyB;AACzB,6CAA0G;AAC1G,4CAAoB;AACpB,gDAAwB;AACxB,uCAUkB;AAClB,0BAA0B;AAC1B,0DAAwD;AACxD,yCAAsD;AACtD,oEAAsE;AACtE,sDAAsD;AAKtD,SAAsB,QAAQ,CAAC,KAAY,EAAE,OAAsB,EAAE,OAAgB,EAAE,MAAc;;QACjG,IAAI,sBAA0C,CAAC;QAC/C,IAAI,IAA+B,CAAC;QAEpC,IAAI,kBAAkB,CAAC,KAAK,CAAC,EAAE,CAAC;YAC5B,4EAA4E;YAC5E,MAAM,MAAM,GAAG,MAAM,qBAAqB,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;YAEnE,sBAAsB,GAAG,0BAA0B,CAAC;YACpD,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;YAEnB,iDAAiD;YACjD,MAAM,SAAS,GAAG,OAAO,CAAC,gBAAgB,CACtC,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,aAAa,CAAC,EAChC,uBAAuB,sBAAsB,gBAAgB,EAC7D,EAAE,SAAS,EAAE,IAAI,EAAE,CACtB,CAAC;YACF,MAAM,cAAc,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAC7C,CAAC;aAAM,CAAC;YACJ,kCAAkC;YAClC,MAAM,SAAS,GAAG,OAAO,CAAC,gBAAgB,CACtC,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,aAAa,CAAC,EAChC,uBAAuB,IAAA,+BAAyB,EAAC,MAAM,EAAE,OAAO,CAAC,IAAI,EACrE,EAAE,SAAS,EAAE,IAAI,EAAE,CACtB,CAAC;YACF,MAAM,cAAc,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAC7C,CAAC;QAED,MAAM,SAAS,GAAG,OAAO,CAAC,gBAAgB,CACtC,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,EAC/B;;;EAGN,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,sCAAsC,CAAC,CAAC,CAAC,+BAA+B;0BACzE,IAAA,+BAAyB,EAAC,MAAM,EAAE,OAAO,CAAC;EAClE,sBAAsB,CAAC,CAAC,CAAC,sCAAsC,sBAAsB,gBAAgB,CAAC,CAAC,CAAC,EAAE;;;;;;;;;iBAS3F,sBAAsB,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE;;CAEhE,EACO,EAAE,SAAS,EAAE,IAAI,EAAE,CACtB,CAAC;QAEF,MAAM,cAAc,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAEzC,OAAO,EAAE,IAAI,EAAE,CAAC;IACpB,CAAC;CAAA;AArDD,4BAqDC;AAED,SAAS,kBAAkB,CAAC,KAAY;IACpC,OAAO,gBAAgB,CAAC,KAAK,CAAC,IAAI,gBAAgB,CAAC,KAAK,CAAC,CAAC;AAC9D,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAY;IAClC,MAAM,UAAU,GAAG,IAAA,mBAAa,EAAC,KAAK,CAAC,CAAC;IACxC,OAAO,UAAU,CAAC,IAAI,CAClB,CAAC,EAAE,EAAE,EAAE,CAAC,IAAA,qBAAe,EAAC,EAAE,CAAC,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,KAAK,EAAE,CAAC,CAAC,CAC1G,CAAC;AACN,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAY;IAClC,OAAO,IAAA,mBAAa,EAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CACpC,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAA,kCAAiB,EAAC,IAAI,CAAC,CAAC,CAAC,CAC9E,CAAC;AACN,CAAC;AAED,SAAe,qBAAqB,CAAC,KAAY,EAAE,OAAsB,EAAE,MAAc;;QACrF,MAAM,eAAe,GAAG,IAAI,wCAAqB,CAAC,KAAK,CAAC,CAAC;QACzD,MAAM,kBAAkB,GAAG,0BAA0B,CAAC;QACtD,MAAM,iBAAiB,GAAG,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC;QAC9D,MAAM,eAAe,CAAC,QAAQ,CAAC;YAC3B,QAAQ,EAAE,WAAW,EAAE,iBAAiB;YACxC,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,MAAM,EAAE,iBAAiB;YACzB,4BAA4B,EAAE,kBAAkB;YAChD,IAAI,EAAE,SAAS;SAClB,CAAC,CAAC;QAEH,6BAA6B;QAC7B,MAAM,WAAW,GAAG,6BAA6B,iBAAiB,eAAe,CAAC;QAClF,IAAI,CAAC;YACD,wBAAwB;YACxB,MAAM,IAAA,wBAAW,EAAC,WAAW,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;QACxD,CAAC;QAAC,WAAM,CAAC;YACL,MAAM,IAAA,+BAAsB,EAAC,iBAAiB,CAAC,CAAC;YAChD,IAAI,CAAC;gBACD,yDAAyD;gBACzD,MAAM,IAAA,wBAAW,EAAC,WAAW,CAAC,CAAC;YACnC,CAAC;YAAC,WAAM,CAAC;gBACL,OAAO;YACX,CAAC;YACD,MAAM,IAAI,iBAAW,CAAC,QAAI,EAAE,iCAAiC,CAAC,CAAC;QACnE,CAAC;QAED,8DAA8D;QAC9D,MAAM,kBAAkB,CAAC,KAAK,EAAE,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,kBAAkB,CAAC,CAAC,CAAC;QAEvE,OAAO;YACH,YAAY,EAAE,iBAAiB;YAC/B,6CAA6C;YAC7C,IAAI,EAAE,MAAM,IAAA,aAAO,EAAC,EAAE,SAAS,EAAE,YAAE,CAAC,YAAY,CAAC,iBAAiB,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;SAChG,CAAC;IACN,CAAC;CAAA;AAED,SAAe,kBAAkB,CAAC,KAAY,EAAE,eAAuB;;QACnE,8FAA8F;QAC9F,MAAM,OAAO,GAAG,IAAI,kBAAO,EAAE,CAAC;QAC9B,MAAM,EAAE,GAAG,OAAO,CAAC,mBAAmB,CAAC,cAAI,CAAC,IAAI,CAAC,eAAe,EAAE,YAAY,CAAC,CAAC,CAAC;QAEjF,sDAAsD;QACtD,MAAM,YAAY,GAAiB,EAAE,CAAC;QACtC,KAAK,CAAC,YAAY;aACb,MAAM,CAAC,CAAC,CAAC,EAAkB,EAAE,CAAC,IAAA,qBAAe,EAAC,CAAC,CAAC,CAAC;aACjD,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE;YACZ,YAAY,CAAC,IAAI,CAAC;gBACd,EAAE;gBACF,KAAK,CAAC,YAAY,CAAC,MAAM,CACrB,CAAC,CAAC,EAAkB,EAAE,CAAC,IAAA,iBAAW,EAAC,CAAC,CAAC,IAAI,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,KAAK,EAAE,CAAC,CAClF;aACJ,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;QAEP,MAAM,KAAK,GAAG,OAAO,CAAC,gBAAgB,CAAC,cAAI,CAAC,IAAI,CAAC,eAAe,EAAE,kBAAkB,CAAC,EAAE,SAAS,EAAE;YAC9F,SAAS,EAAE,IAAI;SAClB,CAAC,CAAC;QAEH,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1B,uCAAuC;YACvC,iBAAiB,CAAC,EAAE,EAAE,KAAK,EAAE,YAAY,CAAC,CAAC;YAC3C,KAAK,CAAC,UAAU,EAAE,CAAC;QACvB,CAAC;aAAM,CAAC;YACJ,YAAY;YACZ,KAAK,CAAC,eAAe,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,CAAC;QAC5C,CAAC;QACD,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;IACvB,CAAC;CAAA;AAED,SAAS,iBAAiB,CAAC,EAAc,EAAE,KAAiB,EAAE,cAA4B;IACtF,wBAAwB;IACxB,KAAK,CAAC,qBAAqB,CAAC,EAAE,CAAC,qBAAqB,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;IAErF,8BAA8B;IAC9B,KAAK,CAAC,aAAa,CAAC,EAAE,CAAC,iBAAiB,CAAC,qBAAU,CAAC,uBAAuB,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;IAE1G,wBAAwB;IACxB,KAAK,CAAC,oBAAoB,CAAC,EAAE,CAAC,oBAAoB,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;IAEnF,6BAA6B;IAC7B,KAAK,CAAC,cAAc,CAAC,EAAE,CAAC,cAAc,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;IAEvE,wBAAwB;IACxB,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC,UAAU,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;IAE/D,0BAA0B;IAC1B,KAAK,CAAC,qBAAqB,CAAC,EAAE,CAAC,qBAAqB,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;IAErF,+CAA+C;IAC/C,KAAK,CAAC,UAAU,CACZ,EAAE;SACG,UAAU,EAAE;SACZ,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,KAAK,QAAQ,CAAC;SACvC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,EAAE,CAAC,CACpC,CAAC;IAEF,mCAAmC;IACnC,MAAM,YAAY,GAAG,EAAE,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IACnD,MAAM,eAAe,GAAG,KAAK,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;IAC9E,qBAAqB,CAAC,YAAY,EAAE,eAAe,EAAE,cAAc,CAAC,CAAC;AACzE,CAAC;AAED,SAAS,qBAAqB,CAC1B,YAA+B,EAC/B,eAAkC,EAClC,YAA0B;IAE1B,0EAA0E;IAC1E,MAAM,WAAW,GAAG,YAAY,CAAC,0BAA0B,CAAC,qBAAU,CAAC,WAAW,CAAC,CAAC;IAEpF,wDAAwD;IACxD,gEAAgE;IAEhE,eAAe;IACf,eAAe,CAAC,aAAa,CACzB,WAAW,CAAC,iBAAiB,CAAC,qBAAU,CAAC,uBAAuB,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAChG,CAAC;IAEF,eAAe;IACf,eAAe,CAAC,UAAU,CAAC,WAAW,CAAC,UAAU,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;IAElF,iBAAiB;IACjB,eAAe,CAAC,YAAY,CACxB,WAAW,CAAC,YAAY,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,EAAkC,CAAC,CAC1F,CAAC;IAEF,yBAAyB;IACzB,eAAe,CAAC,UAAU,CAAC,WAAW,CAAC,UAAU,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;IAElF,sBAAsB;IACtB,MAAM,YAAY,GAAG,WAAW,CAAC,qBAAqB,EAAE,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,0BAA0B,CAAC,QAAQ,CAAC,CAAC,CAAC;IACjH,eAAe,CAAC,qBAAqB,CAAC,YAAY,CAAC,CAAC;IAEpD,uBAAuB;IACvB,MAAM,aAAa,GAAG,WAAW,CAAC,aAAa,EAAE,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC,CAAC;IAC1G,eAAe,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC;IAE7C,yBAAyB;IACzB,MAAM,cAAc,GAAG,WAAW,CAAC,cAAc,EAAE,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,kBAAkB,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC,CAAC;IACpH,eAAe,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC;AACnD,CAAC;AAED,SAAS,0BAA0B,CAAC,QAA2B;IAC3D,MAAM,SAAS,GAAG,QAAQ,CAAC,YAAY,EAAE,CAAC;IAE1C,4DAA4D;IAC5D,MAAM,SAAS,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IACzC,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvB,SAAS,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;;YACxC,IAAI,MAAM,GAAG,MAAA,QAAQ,CAAC,IAAI,0CAAE,QAAQ,EAAE,CAAC;YACvC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;gBACpB,MAAM,GAAG,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,OAAO,CAAC,CAAC,CAAC,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;YAC9C,CAAC,CAAC,CAAC;YACH,QAAQ,CAAC,IAAI,GAAG,MAAM,CAAC;QAC3B,CAAC,CAAC,CAAC;IACP,CAAC;IAED,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,SAAS,kBAAkB,CAAC,KAA2B,EAAE,YAA0B;;IAC/E,MAAM,SAAS,GAAG,KAAK,CAAC,YAAY,EAAE,CAAC;IAEvC,wBAAwB;IACxB,SAAS,CAAC,UAAU,GAAG,MAAA,SAAS,CAAC,UAAU,0CAAE,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,sCAA4B,CAAC,CAAC,CAAC;IAE7G,yBAAyB;IACzB,SAAS,CAAC,OAAO,GAAG,MAAA,SAAS,CAAC,OAAO,0CAAE,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,sCAA4B,CAAC,CAAC,CAAC;IAEvG,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,GAAG,QAAQ,CAAC,IAAI,UAAU,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;QACpF,8EAA8E;QAC9E,SAAS,CAAC,OAAO,GAAG,MAAA,SAAS,CAAC,OAAO,0CAAE,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,YAAY,EAAE,QAAQ,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IAC/G,CAAC;IAED,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,SAAS,kBAAkB,CAAC,SAA+B,EAAE,YAA0B;IACnF,MAAM,SAAS,GAAG,SAAS,CAAC,YAAY,EAAE,CAAC;IAC3C,IAAI,MAAM,GAAG,SAAS,CAAC,IAAc,CAAC;IAEtC,oBAAoB;IACpB,MAAM,GAAG,4BAA4B,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IAEzD,uDAAuD;IACvD,MAAM,GAAG,oCAAoC,CAAC,SAAS,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;IAE/E,0EAA0E;IAC1E,MAAM,GAAG,6BAA6B,CAAC,SAAS,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;IAExE,kCAAkC;IAClC,MAAM,GAAG,sBAAsB,CAAC,SAAS,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;IAEjE,SAAS,CAAC,IAAI,GAAG,MAAM,CAAC;IACxB,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,SAAS,sBAAsB,CAAC,SAA+B,EAAE,YAA0B,EAAE,MAAc;IACvG,mGAAmG;IACnG,MAAM,QAAQ,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC;IACrC,MAAM,aAAa,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,IAAI,QAAQ,CAAC,IAAI,SAAS,KAAK,QAAQ,CAAC,CAAC;IACjG,IAAI,aAAa,EAAE,CAAC;QAChB,MAAM,iBAAiB,GAAG,qBAAqB,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;QAClE,IAAI,iBAAiB,EAAE,CAAC;YACpB,MAAM,GAAG,GAAG,aAAa,CAAC,CAAC,CAAC;iBACvB,GAAG,CACA,CAAC,QAAQ,EAAE,EAAE,CACT,KAAK,QAAQ,CAAC,IAAI,mCAAmC,iBAAiB,CAAC,IAAI,MAAM,QAAQ,CAAC,IAAI,QAAQ,CAC7G;iBACA,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QACvB,CAAC;IACL,CAAC;IACD,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,SAAS,6BAA6B,CAAC,SAA+B,EAAE,cAA4B,EAAE,MAAc;IAChH,kFAAkF;IAClF,6CAA6C;IAC7C,MAAM,QAAQ,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC;IACrC,MAAM,kBAAkB,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IAC7E,MAAM,8BAA8B,GAAG,IAAI,MAAM,CAC7C,KAAK,kBAAkB,CAAC,IAAI,CAAC,GAAG,CAAC,oCAAoC,CACxE,CAAC;IACF,IAAI,8BAA8B,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;QAChD,MAAM,QAAQ,GAAG,SAAS;aACrB,oBAAoB,CAAC,qBAAU,CAAC,iBAAiB,CAAC;aAClD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,EAAE,iBAAiB,EAAE,QAAQ,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAClF,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YACnB,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;IACP,CAAC;IACD,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,SAAS,oCAAoC,CACzC,SAA+B,EAC/B,YAA0B,EAC1B,MAAc;IAEd,qFAAqF;IACrF,uCAAuC;IACvC,MAAM,QAAQ,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC;IACrC,MAAM,kBAAkB,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IAC/G,MAAM,8BAA8B,GAAG,IAAI,MAAM,CAC7C,IAAI,kBAAkB,CAAC,IAAI,CAAC,GAAG,CAAC,qCAAqC,CACxE,CAAC;IAEF,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;IAC7D,IAAI,KAAK,EAAE,CAAC;QACR,MAAM,SAAS,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAC3B,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC,EAAE,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC;QACjG,IAAI,MAAM,EAAE,CAAC;YACT,8CAA8C;YAC9C,MAAM,kBAAkB,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,cAAc,GAAG,iCAAiC,CAAC,kBAAkB,CAAC,CAAC;YAC7E,cAAc,CAAC,OAAO,CAAC,CAAC,iBAAiB,EAAE,EAAE;gBACzC,MAAM,iBAAiB,GAAG,iBAAiB,CAAC,SAAS,EAAE,iBAAiB,CAAC,IAAI,CAAC,CAAC;gBAC/E,IAAI,iBAAiB,EAAE,CAAC;oBACpB,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;gBAC7D,CAAC;YACL,CAAC,CAAC,CAAC;QACP,CAAC;IACL,CAAC;IACD,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,SAAS,4BAA4B,CAAC,SAA+B,EAAE,MAAc;IACjF,qDAAqD;IACrD,MAAM,QAAQ,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC;IACzC,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtB,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YACnB,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;IACP,CAAC;IACD,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,SAAS,iBAAiB,CAAC,SAA+B,EAAE,IAAY;IACpE,OAAO,SAAS,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,qBAAU,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,KAAK,IAAI,CAAC,CAAC;AAC/G,CAAC;AAED,SAAS,YAAY,CAAC,IAAU;IAC5B,OAAO,IAAI;SACN,oBAAoB,CAAC,qBAAU,CAAC,iBAAiB,CAAC;SAClD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,UAAU,CAAC,sCAA4B,CAAC,CAAC,CAAC;AAC7E,CAAC;AAED,SAAS,qBAAqB,CAAC,QAAmB;;IAC9C,MAAM,YAAY,GAAG,IAAA,kBAAY,EAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;IAC1D,IAAI,CAAC,YAAY,EAAE,CAAC;QAChB,OAAO,SAAS,CAAC;IACrB,CAAC;IACD,MAAM,GAAG,GAAG,MAAA,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,0CAAE,KAAK,CAAC;IACxC,OAAO,IAAA,qBAAe,EAAC,GAAG,CAAC,CAAC,CAAC,CAAE,GAAG,CAAC,MAAM,CAAC,GAAsB,CAAC,CAAC,CAAC,SAAS,CAAC;AACjF,CAAC;AAED,SAAS,iCAAiC,CAAC,QAAmB,EAAE,SAA2B,EAAE;IACzF,IAAI,IAAA,qBAAe,EAAC,QAAQ,CAAC,EAAE,CAAC;QAC5B,MAAM,aAAa,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;QACtD,IAAI,aAAa,EAAE,CAAC;YAChB,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC/B,CAAC;QAED,KAAK,MAAM,SAAS,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC;YAC1C,IAAI,SAAS,CAAC,GAAG,EAAE,CAAC;gBAChB,MAAM,CAAC,IAAI,CAAC,GAAG,iCAAiC,CAAC,SAAS,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC,CAAC;YAC7E,CAAC;QACL,CAAC;IACL,CAAC;IACD,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,SAAe,cAAc,CAAC,EAAc,EAAE,OAAsB;;QAChE,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;YAC1B,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC;QACpB,CAAC;IACL,CAAC;CAAA"}

package/plugins/enhancer/enhancer-utils.d.ts

@@ -0,0 +1,5 @@
+import type { DataModelFieldAttribute } from '@zenstackhq/sdk/ast';
+/**
+ * Check if the given field attribute is a `@default` with `auth()` invocation
+ */
+export declare function isDefaultWithAuth(attr: DataModelFieldAttribute): boolean;

package/plugins/enhancer/enhancer-utils.js

@@ -0,0 +1,22 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.isDefaultWithAuth = void 0;
+const sdk_1 = require("@zenstackhq/sdk");
+const langium_1 = require("langium");
+/**
+ * Check if the given field attribute is a `@default` with `auth()` invocation
+ */
+function isDefaultWithAuth(attr) {
+    var _a, _b;
+    if (((_a = attr.decl.ref) === null || _a === void 0 ? void 0 : _a.name) !== '@default') {
+        return false;
+    }
+    const expr = (_b = attr.args[0]) === null || _b === void 0 ? void 0 : _b.value;
+    if (!expr) {
+        return false;
+    }
+    // find `auth()` in default value expression
+    return (0, langium_1.streamAst)(expr).some(sdk_1.isAuthInvocation);
+}
+exports.isDefaultWithAuth = isDefaultWithAuth;
+//# sourceMappingURL=enhancer-utils.js.map

package/plugins/enhancer/enhancer-utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"enhancer-utils.js","sourceRoot":"","sources":["../../../src/plugins/enhancer/enhancer-utils.ts"],"names":[],"mappings":";;;AAAA,yCAAmD;AAEnD,qCAAoC;AAEpC;;GAEG;AACH,SAAgB,iBAAiB,CAAC,IAA6B;;IAC3D,IAAI,CAAA,MAAA,IAAI,CAAC,IAAI,CAAC,GAAG,0CAAE,IAAI,MAAK,UAAU,EAAE,CAAC;QACrC,OAAO,KAAK,CAAC;IACjB,CAAC;IAED,MAAM,IAAI,GAAG,MAAA,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,0CAAE,KAAK,CAAC;IACjC,IAAI,CAAC,IAAI,EAAE,CAAC;QACR,OAAO,KAAK,CAAC;IACjB,CAAC;IAED,4CAA4C;IAC5C,OAAO,IAAA,mBAAS,EAAC,IAAI,CAAC,CAAC,IAAI,CAAC,sBAAgB,CAAC,CAAC;AAClD,CAAC;AAZD,8CAYC"}

package/plugins/enhancer/index.d.ts

@@ -0,0 +1,5 @@
+import { type PluginFunction } from '@zenstackhq/sdk';
+export declare const name = "Prisma Enhancer";
+export declare const description = "Generating PrismaClient enhancer";
+declare const run: PluginFunction;
+export default run;

package/plugins/enhancer/index.js

@@ -0,0 +1,51 @@
+"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.description = exports.name = void 0;
+const sdk_1 = require("@zenstackhq/sdk");
+const path_1 = __importDefault(require("path"));
+const plugin_utils_1 = require("../plugin-utils");
+const enhance_1 = require("./enhance");
+const model_meta_1 = require("./model-meta");
+const policy_1 = require("./policy");
+exports.name = 'Prisma Enhancer';
+exports.description = 'Generating PrismaClient enhancer';
+const run = (model, options, _dmmf, globalOptions) => __awaiter(void 0, void 0, void 0, function* () {
+    var _a;
+    let outDir = options.output ? options.output : (0, plugin_utils_1.getDefaultOutputFolder)(globalOptions);
+    if (!outDir) {
+        throw new sdk_1.PluginError(exports.name, `Unable to determine output path, not running plugin`);
+    }
+    outDir = (0, sdk_1.resolvePath)(outDir, options);
+    const project = (_a = globalOptions === null || globalOptions === void 0 ? void 0 : globalOptions.tsProject) !== null && _a !== void 0 ? _a : (0, sdk_1.createProject)();
+    yield (0, model_meta_1.generate)(model, options, project, outDir);
+    yield (0, policy_1.generate)(model, options, project, outDir);
+    const { dmmf } = yield (0, enhance_1.generate)(model, options, project, outDir);
+    let prismaClientPath;
+    if (dmmf) {
+        // a logical client is generated
+        if (typeof options.output === 'string') {
+            // get the absolute path of the logical prisma client
+            const prismaClientPathAbs = path_1.default.resolve(options.output, 'prisma');
+            // resolve it relative to the schema path
+            prismaClientPath = path_1.default.relative(path_1.default.dirname(options.schemaPath), prismaClientPathAbs);
+        }
+        else {
+            prismaClientPath = `${sdk_1.RUNTIME_PACKAGE}/prisma`;
+        }
+    }
+    return { dmmf, warnings: [], prismaClientPath };
+});
+exports.default = run;
+//# sourceMappingURL=index.js.map

package/plugins/enhancer/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/plugins/enhancer/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,yCAAgH;AAChH,gDAAwB;AACxB,kDAAyD;AACzD,uCAAyD;AACzD,6CAA6D;AAC7D,qCAAsD;AAEzC,QAAA,IAAI,GAAG,iBAAiB,CAAC;AACzB,QAAA,WAAW,GAAG,kCAAkC,CAAC;AAE9D,MAAM,GAAG,GAAmB,CAAO,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE;;IACvE,IAAI,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAE,OAAO,CAAC,MAAiB,CAAC,CAAC,CAAC,IAAA,qCAAsB,EAAC,aAAa,CAAC,CAAC;IACjG,IAAI,CAAC,MAAM,EAAE,CAAC;QACV,MAAM,IAAI,iBAAW,CAAC,YAAI,EAAE,qDAAqD,CAAC,CAAC;IACvF,CAAC;IACD,MAAM,GAAG,IAAA,iBAAW,EAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAEtC,MAAM,OAAO,GAAG,MAAA,aAAa,aAAb,aAAa,uBAAb,aAAa,CAAE,SAAS,mCAAI,IAAA,mBAAa,GAAE,CAAC;IAE5D,MAAM,IAAA,qBAAiB,EAAC,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IACzD,MAAM,IAAA,iBAAc,EAAC,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IACtD,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAA,kBAAgB,EAAC,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAEzE,IAAI,gBAAoC,CAAC;IACzC,IAAI,IAAI,EAAE,CAAC;QACP,gCAAgC;QAChC,IAAI,OAAO,OAAO,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YACrC,qDAAqD;YACrD,MAAM,mBAAmB,GAAG,cAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YAEnE,yCAAyC;YACzC,gBAAgB,GAAG,cAAI,CAAC,QAAQ,CAAC,cAAI,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,mBAAmB,CAAC,CAAC;QAC5F,CAAC;aAAM,CAAC;YACJ,gBAAgB,GAAG,GAAG,qBAAe,SAAS,CAAC;QACnD,CAAC;IACL,CAAC;IAED,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,gBAAgB,EAAE,CAAC;AACpD,CAAC,CAAA,CAAC;AAEF,kBAAe,GAAG,CAAC"}

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

@@ -0,0 +1,4 @@
+import { type PluginOptions } from '@zenstackhq/sdk';
+import type { Model } from '@zenstackhq/sdk/ast';
+import type { Project } from 'ts-morph';
+export declare function generate(model: Model, options: PluginOptions, project: Project, outDir: string): Promise<void>;

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

@@ -0,0 +1,32 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generate = void 0;
+const sdk_1 = require("@zenstackhq/sdk");
+const path_1 = __importDefault(require("path"));
+function generate(model, options, project, outDir) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const outFile = path_1.default.join(outDir, 'model-meta.ts');
+        const dataModels = (0, sdk_1.getDataModels)(model);
+        // save ts files if requested explicitly or the user provided
+        const preserveTsFiles = options.preserveTsFiles === true || !!options.output;
+        yield (0, sdk_1.generateModelMeta)(project, dataModels, {
+            output: outFile,
+            generateAttributes: true,
+            preserveTsFiles,
+        });
+    });
+}
+exports.generate = generate;
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/plugins/enhancer/model-meta/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,yCAAuF;AAEvF,gDAAwB;AAGxB,SAAsB,QAAQ,CAAC,KAAY,EAAE,OAAsB,EAAE,OAAgB,EAAE,MAAc;;QACjG,MAAM,OAAO,GAAG,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;QACnD,MAAM,UAAU,GAAG,IAAA,mBAAa,EAAC,KAAK,CAAC,CAAC;QAExC,6DAA6D;QAC7D,MAAM,eAAe,GAAG,OAAO,CAAC,eAAe,KAAK,IAAI,IAAI,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;QAC7E,MAAM,IAAA,uBAAiB,EAAC,OAAO,EAAE,UAAU,EAAE;YACzC,MAAM,EAAE,OAAO;YACf,kBAAkB,EAAE,IAAI;YACxB,eAAe;SAClB,CAAC,CAAC;IACP,CAAC;CAAA;AAXD,4BAWC"}

package/plugins/{access-policy → enhancer/policy}/expression-writer.d.ts

@@ -20,6 +20,9 @@ export declare class ExpressionWriter {
      */
     write(expr: Expression): void;
     private writeReference;
+    private writeBaseHierarchy;
+    private getDelegateBase;
+    private isFieldReferenceToDelegateModel;
     private writeMemberAccess;
     private writeExprList;
     private writeBinary;