zenstack 1.8.2 → 2.0.0-alpha.1

package/plugins/enhancer/enhance/index.js

@@ -0,0 +1,202 @@
+"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 path_1 = __importDefault(require("path"));
+const ts_morph_1 = require("ts-morph");
+const schema_generator_1 = require("../../prisma/schema-generator");
+function generate(model, options, project, outDir) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const outFile = path_1.default.join(outDir, 'enhance.ts');
+        let logicalPrismaClientDir;
+        if (hasDelegateModel(model)) {
+            logicalPrismaClientDir = yield generateLogicalPrisma(model, options, outDir);
+        }
+        project.createSourceFile(outFile, `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)(model, outDir)}';
+${logicalPrismaClientDir ? `import { PrismaClient as EnhancedPrismaClient } from '${logicalPrismaClientDir}';` : ''}
+
+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 EnhancedPrismaClient' : ''};
+}
+`, { overwrite: true });
+    });
+}
+exports.generate = generate;
+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 generateLogicalPrisma(model, options, outDir) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const prismaGenerator = new schema_generator_1.PrismaSchemaGenerator();
+        const prismaClientOutDir = './.delegate';
+        yield prismaGenerator.generate(model, {
+            provider: '@internal',
+            schemaPath: options.schemaPath,
+            output: path_1.default.join(outDir, 'delegate.prisma'),
+            overrideClientGenerationPath: prismaClientOutDir,
+            mode: 'logical',
+        });
+        yield processClientTypes(model, path_1.default.join(outDir, prismaClientOutDir));
+        return prismaClientOutDir;
+    });
+}
+function processClientTypes(model, prismaClientDir) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const project = new ts_morph_1.Project();
+        const sf = project.addSourceFileAtPath(path_1.default.join(prismaClientDir, 'index.d.ts'));
+        const delegateModels = [];
+        model.declarations
+            .filter((d) => (0, sdk_1.isDelegateModel)(d))
+            .forEach((dm) => {
+            delegateModels.push([
+                dm,
+                model.declarations.filter((d) => (0, ast_1.isDataModel)(d) && d.superTypes.some((s) => s.ref === dm)),
+            ]);
+        });
+        const toRemove = [];
+        const toReplaceText = [];
+        sf.forEachDescendant((desc, traversal) => {
+            removeAuxRelationFields(desc, toRemove, traversal);
+            fixDelegateUnionType(desc, delegateModels, toReplaceText, traversal);
+            removeCreateFromDelegateInputTypes(desc, delegateModels, toRemove, traversal);
+            removeDelegateToplevelCreates(desc, delegateModels, toRemove, traversal);
+            removeDiscriminatorFromConcreteInputTypes(desc, delegateModels, toRemove);
+        });
+        toRemove.forEach((n) => n.remove());
+        toReplaceText.forEach(([node, text]) => node.replaceWithText(text));
+        yield project.save();
+    });
+}
+function removeAuxRelationFields(desc, toRemove, traversal) {
+    if (desc.isKind(ts_morph_1.SyntaxKind.PropertySignature) || desc.isKind(ts_morph_1.SyntaxKind.MethodSignature)) {
+        // remove aux fields
+        const name = desc.getName();
+        if (name.startsWith(runtime_1.DELEGATE_AUX_RELATION_PREFIX)) {
+            toRemove.push(desc);
+            traversal.skip();
+        }
+    }
+}
+function fixDelegateUnionType(desc, delegateModels, toReplaceText, traversal) {
+    if (!desc.isKind(ts_morph_1.SyntaxKind.TypeAliasDeclaration)) {
+        return;
+    }
+    const name = desc.getName();
+    delegateModels.forEach(([delegate, concreteModels]) => {
+        if (name === `$${delegate.name}Payload`) {
+            const discriminator = getDiscriminatorField(delegate);
+            if (discriminator) {
+                toReplaceText.push([
+                    desc,
+                    `export type ${name}<ExtArgs extends $Extensions.InternalArgs = $Extensions.DefaultArgs> = 
+        ${concreteModels
+                        .map((m) => `($${m.name}Payload<ExtArgs> & { scalars: { ${discriminator.name}: '${m.name}' } })`)
+                        .join(' | ')};`,
+                ]);
+                traversal.skip();
+            }
+        }
+    });
+}
+function removeCreateFromDelegateInputTypes(desc, delegateModels, toRemove, traversal) {
+    if (!desc.isKind(ts_morph_1.SyntaxKind.TypeAliasDeclaration)) {
+        return;
+    }
+    const name = desc.getName();
+    delegateModels.forEach(([delegate]) => {
+        // remove create related sub-payload from delegate's input types since they cannot be created directly
+        const regex = new RegExp(`\\${delegate.name}(Unchecked)?(Create|Update).*Input`);
+        if (regex.test(name)) {
+            desc.forEachDescendant((d, innerTraversal) => {
+                if (d.isKind(ts_morph_1.SyntaxKind.PropertySignature) &&
+                    ['create', 'upsert', 'connectOrCreate'].includes(d.getName())) {
+                    toRemove.push(d);
+                    innerTraversal.skip();
+                }
+            });
+            traversal.skip();
+        }
+    });
+}
+function removeDiscriminatorFromConcreteInputTypes(desc, delegateModels, toRemove) {
+    if (!desc.isKind(ts_morph_1.SyntaxKind.TypeAliasDeclaration)) {
+        return;
+    }
+    const name = desc.getName();
+    delegateModels.forEach(([delegate, concretes]) => {
+        const discriminator = getDiscriminatorField(delegate);
+        if (!discriminator) {
+            return;
+        }
+        concretes.forEach((concrete) => {
+            // remove discriminator field from the create/update input of concrete models
+            const regex = new RegExp(`\\${concrete.name}(Unchecked)?(Create|Update).*Input`);
+            if (regex.test(name)) {
+                desc.forEachDescendant((d, innerTraversal) => {
+                    if (d.isKind(ts_morph_1.SyntaxKind.PropertySignature)) {
+                        if (d.getName() === discriminator.name) {
+                            toRemove.push(d);
+                        }
+                        innerTraversal.skip();
+                    }
+                });
+            }
+        });
+    });
+}
+function removeDelegateToplevelCreates(desc, delegateModels, toRemove, traversal) {
+    if (desc.isKind(ts_morph_1.SyntaxKind.InterfaceDeclaration)) {
+        // remove create and upsert methods from delegate interfaces since they cannot be created directly
+        const name = desc.getName();
+        if (delegateModels.map(([dm]) => `${dm.name}Delegate`).includes(name)) {
+            const createMethod = desc.getMethod('create');
+            if (createMethod) {
+                toRemove.push(createMethod);
+            }
+            const createManyMethod = desc.getMethod('createMany');
+            if (createManyMethod) {
+                toRemove.push(createManyMethod);
+            }
+            const upsertMethod = desc.getMethod('upsert');
+            if (upsertMethod) {
+                toRemove.push(upsertMethod);
+            }
+            traversal.skip();
+        }
+    }
+}
+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;
+}
+//# 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":";;;;;;;;;;;;;;;AAAA,iDAAmE;AACnE,yCAMyB;AACzB,6CAA+G;AAC/G,gDAAwB;AACxB,uCAQkB;AAClB,oEAAsE;AAEtE,SAAsB,QAAQ,CAAC,KAAY,EAAE,OAAsB,EAAE,OAAgB,EAAE,MAAc;;QACjG,MAAM,OAAO,GAAG,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAChD,IAAI,sBAA0C,CAAC;QAE/C,IAAI,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1B,sBAAsB,GAAG,MAAM,qBAAqB,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QACjF,CAAC;QAED,OAAO,CAAC,gBAAgB,CACpB,OAAO,EACP;;;EAGN,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,sCAAsC,CAAC,CAAC,CAAC,+BAA+B;0BACzE,IAAA,+BAAyB,EAAC,KAAK,EAAE,MAAM,CAAC;EAChE,sBAAsB,CAAC,CAAC,CAAC,yDAAyD,sBAAsB,IAAI,CAAC,CAAC,CAAC,EAAE;;;;;;;;;iBASlG,sBAAsB,CAAC,CAAC,CAAC,0BAA0B,CAAC,CAAC,CAAC,EAAE;;CAExE,EACO,EAAE,SAAS,EAAE,IAAI,EAAE,CACtB,CAAC;IACN,CAAC;CAAA;AA7BD,4BA6BC;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,SAAe,qBAAqB,CAAC,KAAY,EAAE,OAAsB,EAAE,MAAc;;QACrF,MAAM,eAAe,GAAG,IAAI,wCAAqB,EAAE,CAAC;QACpD,MAAM,kBAAkB,GAAG,aAAa,CAAC;QACzC,MAAM,eAAe,CAAC,QAAQ,CAAC,KAAK,EAAE;YAClC,QAAQ,EAAE,WAAW;YACrB,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,MAAM,EAAE,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,iBAAiB,CAAC;YAC5C,4BAA4B,EAAE,kBAAkB;YAChD,IAAI,EAAE,SAAS;SAClB,CAAC,CAAC;QAEH,MAAM,kBAAkB,CAAC,KAAK,EAAE,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,kBAAkB,CAAC,CAAC,CAAC;QACvE,OAAO,kBAAkB,CAAC;IAC9B,CAAC;CAAA;AAED,SAAe,kBAAkB,CAAC,KAAY,EAAE,eAAuB;;QACnE,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,MAAM,cAAc,GAA+B,EAAE,CAAC;QACtD,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,cAAc,CAAC,IAAI,CAAC;gBAChB,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,QAAQ,GAA4C,EAAE,CAAC;QAC7D,MAAM,aAAa,GAAqC,EAAE,CAAC;QAE3D,EAAE,CAAC,iBAAiB,CAAC,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE;YACrC,uBAAuB,CAAC,IAAI,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;YACnD,oBAAoB,CAAC,IAAI,EAAE,cAAc,EAAE,aAAa,EAAE,SAAS,CAAC,CAAC;YACrE,kCAAkC,CAAC,IAAI,EAAE,cAAc,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;YAC9E,6BAA6B,CAAC,IAAI,EAAE,cAAc,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;YACzE,yCAAyC,CAAC,IAAI,EAAE,cAAc,EAAE,QAAQ,CAAC,CAAC;QAC9E,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;QACpC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC;QAEpE,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;IACzB,CAAC;CAAA;AAED,SAAS,uBAAuB,CAC5B,IAAU,EACV,QAAiD,EACjD,SAA4C;IAE5C,IAAI,IAAI,CAAC,MAAM,CAAC,qBAAU,CAAC,iBAAiB,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,qBAAU,CAAC,eAAe,CAAC,EAAE,CAAC;QACvF,oBAAoB;QACpB,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;QAE5B,IAAI,IAAI,CAAC,UAAU,CAAC,sCAA4B,CAAC,EAAE,CAAC;YAChD,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACpB,SAAS,CAAC,IAAI,EAAE,CAAC;QACrB,CAAC;IACL,CAAC;AACL,CAAC;AAED,SAAS,oBAAoB,CACzB,IAAU,EACV,cAA0C,EAC1C,aAA+C,EAC/C,SAA4C;IAE5C,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,qBAAU,CAAC,oBAAoB,CAAC,EAAE,CAAC;QAChD,OAAO;IACX,CAAC;IAED,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;IAC5B,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,EAAE,cAAc,CAAC,EAAE,EAAE;QAClD,IAAI,IAAI,KAAK,IAAI,QAAQ,CAAC,IAAI,SAAS,EAAE,CAAC;YACtC,MAAM,aAAa,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;YACtD,IAAI,aAAa,EAAE,CAAC;gBAChB,aAAa,CAAC,IAAI,CAAC;oBACf,IAAI;oBACJ,eAAe,IAAI;UAC7B,cAAc;yBACX,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,mCAAmC,aAAa,CAAC,IAAI,MAAM,CAAC,CAAC,IAAI,QAAQ,CAAC;yBAChG,IAAI,CAAC,KAAK,CAAC,GAAG;iBACV,CAAC,CAAC;gBACH,SAAS,CAAC,IAAI,EAAE,CAAC;YACrB,CAAC;QACL,CAAC;IACL,CAAC,CAAC,CAAC;AACP,CAAC;AAED,SAAS,kCAAkC,CACvC,IAAU,EACV,cAA0C,EAC1C,QAAiD,EACjD,SAA4C;IAE5C,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,qBAAU,CAAC,oBAAoB,CAAC,EAAE,CAAC;QAChD,OAAO;IACX,CAAC;IAED,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;IAC5B,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,EAAE;QAClC,sGAAsG;QACtG,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,KAAK,QAAQ,CAAC,IAAI,oCAAoC,CAAC,CAAC;QACjF,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YACnB,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,EAAE;gBACzC,IACI,CAAC,CAAC,MAAM,CAAC,qBAAU,CAAC,iBAAiB,CAAC;oBACtC,CAAC,QAAQ,EAAE,QAAQ,EAAE,iBAAiB,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,EAC/D,CAAC;oBACC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;oBACjB,cAAc,CAAC,IAAI,EAAE,CAAC;gBAC1B,CAAC;YACL,CAAC,CAAC,CAAC;YACH,SAAS,CAAC,IAAI,EAAE,CAAC;QACrB,CAAC;IACL,CAAC,CAAC,CAAC;AACP,CAAC;AAED,SAAS,yCAAyC,CAC9C,IAAU,EACV,cAA0C,EAC1C,QAAiD;IAEjD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,qBAAU,CAAC,oBAAoB,CAAC,EAAE,CAAC;QAChD,OAAO;IACX,CAAC;IAED,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;IAC5B,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,EAAE,SAAS,CAAC,EAAE,EAAE;QAC7C,MAAM,aAAa,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;QACtD,IAAI,CAAC,aAAa,EAAE,CAAC;YACjB,OAAO;QACX,CAAC;QAED,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;YAC3B,6EAA6E;YAC7E,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,KAAK,QAAQ,CAAC,IAAI,oCAAoC,CAAC,CAAC;YACjF,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBACnB,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,EAAE;oBACzC,IAAI,CAAC,CAAC,MAAM,CAAC,qBAAU,CAAC,iBAAiB,CAAC,EAAE,CAAC;wBACzC,IAAI,CAAC,CAAC,OAAO,EAAE,KAAK,aAAa,CAAC,IAAI,EAAE,CAAC;4BACrC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;wBACrB,CAAC;wBACD,cAAc,CAAC,IAAI,EAAE,CAAC;oBAC1B,CAAC;gBACL,CAAC,CAAC,CAAC;YACP,CAAC;QACL,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;AACP,CAAC;AAED,SAAS,6BAA6B,CAClC,IAAU,EACV,cAA0C,EAC1C,QAAiD,EACjD,SAA4C;IAE5C,IAAI,IAAI,CAAC,MAAM,CAAC,qBAAU,CAAC,oBAAoB,CAAC,EAAE,CAAC;QAC/C,kGAAkG;QAClG,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;QAC5B,IAAI,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,UAAU,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YACpE,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YAC9C,IAAI,YAAY,EAAE,CAAC;gBACf,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAChC,CAAC;YACD,MAAM,gBAAgB,GAAG,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;YACtD,IAAI,gBAAgB,EAAE,CAAC;gBACnB,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACpC,CAAC;YACD,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YAC9C,IAAI,YAAY,EAAE,CAAC;gBACf,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAChC,CAAC;YACD,SAAS,CAAC,IAAI,EAAE,CAAC;QACrB,CAAC;IACL,CAAC;AACL,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"}

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/{model-meta → enhancer}/index.js

@@ -8,24 +8,25 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.name = void 0;
+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");
-exports.name = 'Model Metadata';
+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* () {
-    let output = options.output ? options.output : (0, plugin_utils_1.getDefaultOutputFolder)(globalOptions);
-    if (!output) {
-        throw new sdk_1.PluginError(options.name, `Unable to determine output path, not running plugin`);
+    let ourDir = options.output ? options.output : (0, plugin_utils_1.getDefaultOutputFolder)(globalOptions);
+    if (!ourDir) {
+        throw new sdk_1.PluginError(exports.name, `Unable to determine output path, not running plugin`);
     }
-    output = (0, sdk_1.resolvePath)(output, options);
-    const outFile = path_1.default.join(output, 'model-meta.ts');
-    const dataModels = (0, sdk_1.getDataModels)(model);
+    ourDir = (0, sdk_1.resolvePath)(ourDir, options);
     const project = (0, sdk_1.createProject)();
+    yield (0, model_meta_1.generate)(model, options, project, ourDir);
+    yield (0, policy_1.generate)(model, options, project, ourDir);
+    yield (0, enhance_1.generate)(model, options, project, ourDir);
     let shouldCompile = true;
     if (typeof options.compile === 'boolean') {
         // explicit override
@@ -35,12 +36,12 @@ const run = (model, options, _dmmf, globalOptions) => __awaiter(void 0, void 0,
         // from CLI or config file
         shouldCompile = globalOptions.compile;
     }
-    yield (0, sdk_1.generateModelMeta)(project, dataModels, {
-        output: outFile,
-        compile: shouldCompile,
-        preserveTsFiles: options.preserveTsFiles === true,
-        generateAttributes: true,
-    });
+    if (!shouldCompile || options.preserveTsFiles === true) {
+        yield (0, sdk_1.saveProject)(project);
+    }
+    if (shouldCompile) {
+        yield (0, sdk_1.emitProject)(project);
+    }
 });
 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,yCAOyB;AACzB,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,IAAA,mBAAa,GAAE,CAAC;IAEhC,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,IAAA,kBAAgB,EAAC,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAExD,IAAI,aAAa,GAAG,IAAI,CAAC;IACzB,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;QACvC,oBAAoB;QACpB,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC;IACpC,CAAC;SAAM,IAAI,aAAa,EAAE,CAAC;QACvB,0BAA0B;QAC1B,aAAa,GAAG,aAAa,CAAC,OAAO,CAAC;IAC1C,CAAC;IAED,IAAI,CAAC,aAAa,IAAI,OAAO,CAAC,eAAe,KAAK,IAAI,EAAE,CAAC;QACrD,MAAM,IAAA,iBAAW,EAAC,OAAO,CAAC,CAAC;IAC/B,CAAC;IAED,IAAI,aAAa,EAAE,CAAC;QAChB,MAAM,IAAA,iBAAW,EAAC,OAAO,CAAC,CAAC;IAC/B,CAAC;AACL,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/{access-policy → enhancer/model-meta}/index.js

@@ -12,11 +12,18 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.name = void 0;
-const policy_guard_generator_1 = __importDefault(require("./policy-guard-generator"));
-exports.name = 'Access Policy';
-const run = (model, options, _dmmf, globalOptions) => __awaiter(void 0, void 0, void 0, function* () {
-    return new policy_guard_generator_1.default().generate(model, options, globalOptions);
-});
-exports.default = run;
+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);
+        yield (0, sdk_1.generateModelMeta)(project, dataModels, {
+            output: outFile,
+            generateAttributes: true,
+        });
+    });
+}
+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;QACxC,MAAM,IAAA,uBAAiB,EAAC,OAAO,EAAE,UAAU,EAAE;YACzC,MAAM,EAAE,OAAO;YACf,kBAAkB,EAAE,IAAI;SAC3B,CAAC,CAAC;IACP,CAAC;CAAA;AAPD,4BAOC"}

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;

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

@@ -1,12 +1,15 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ExpressionWriter = exports.FALSE = exports.TRUE = void 0;
 const ast_1 = require("@zenstackhq/language/ast");
+const runtime_1 = require("@zenstackhq/runtime");
 const sdk_1 = require("@zenstackhq/sdk");
 const lower_case_first_1 = require("lower-case-first");
-const _1 = require(".");
-const ast_utils_1 = require("../../utils/ast-utils");
-const typescript_expression_transformer_1 = require("../../utils/typescript-expression-transformer");
+const tiny_invariant_1 = __importDefault(require("tiny-invariant"));
+const __1 = require("..");
 // { OR: [] } filters to nothing, { AND: [] } includes everything
 // https://www.prisma.io/docs/concepts/components/prisma-client/null-and-undefined#the-effect-of-null-and-undefined-on-conditionals
 exports.TRUE = '{ AND: [] }';
@@ -23,7 +26,7 @@ class ExpressionWriter {
     constructor(writer, isPostGuard = false) {
         this.writer = writer;
         this.isPostGuard = isPostGuard;
-        this.plainExprBuilder = new typescript_expression_transformer_1.TypeScriptExpressionTransformer({
+        this.plainExprBuilder = new sdk_1.TypeScriptExpressionTransformer({
             context: sdk_1.ExpressionContext.AccessPolicy,
             isPostGuard: this.isPostGuard,
         });
@@ -63,11 +66,40 @@ class ExpressionWriter {
         }
         else {
             this.block(() => {
-                var _a;
-                this.writer.write(`${(_a = expr.target.ref) === null || _a === void 0 ? void 0 : _a.name}: true`);
+                const ref = expr.target.ref;
+                (0, tiny_invariant_1.default)(ref);
+                if (this.isFieldReferenceToDelegateModel(ref)) {
+                    const thisModel = ref.$container;
+                    const targetBase = ref.$inheritedFrom;
+                    this.writeBaseHierarchy(thisModel, targetBase, () => this.writer.write(`${ref.name}: true`));
+                }
+                else {
+                    this.writer.write(`${ref.name}: true`);
+                }
             });
         }
     }
+    writeBaseHierarchy(thisModel, targetBase, conditionWriter) {
+        if (!targetBase || thisModel === targetBase) {
+            conditionWriter();
+            return;
+        }
+        const base = this.getDelegateBase(thisModel);
+        if (!base) {
+            throw new sdk_1.PluginError(__1.name, `Failed to resolve delegate base model for "${thisModel.name}"`);
+        }
+        this.writer.write(`${`${runtime_1.DELEGATE_AUX_RELATION_PREFIX}_${(0, lower_case_first_1.lowerCaseFirst)(base.name)}`}: `);
+        this.writer.block(() => {
+            this.writeBaseHierarchy(base, targetBase, conditionWriter);
+        });
+    }
+    getDelegateBase(model) {
+        var _a;
+        return (_a = model.superTypes.map((t) => t.ref).filter((t) => t && (0, sdk_1.isDelegateModel)(t))) === null || _a === void 0 ? void 0 : _a[0];
+    }
+    isFieldReferenceToDelegateModel(ref) {
+        return (0, ast_1.isDataModelField)(ref) && !!ref.$inheritedFrom && (0, sdk_1.isDelegateModel)(ref.$inheritedFrom);
+    }
     writeMemberAccess(expr) {
         if (this.isAuthOrAuthMemberAccess(expr)) {
             // member access of `auth()`, generate plain expression
@@ -151,7 +183,7 @@ class ExpressionWriter {
                     }, 'in');
                 }
                 else {
-                    throw new sdk_1.PluginError(_1.name, '"in" operator cannot be used with field references on both sides');
+                    throw new sdk_1.PluginError(__1.name, '"in" operator cannot be used with field references on both sides');
                 }
             });
         }
@@ -213,8 +245,8 @@ class ExpressionWriter {
             this.writer.write(this.plainExprBuilder.transform(expr));
         }
         catch (err) {
-            if (err instanceof typescript_expression_transformer_1.TypeScriptExpressionTransformerError) {
-                throw new sdk_1.PluginError(_1.name, err.message);
+            if (err instanceof sdk_1.TypeScriptExpressionTransformerError) {
+                throw new sdk_1.PluginError(__1.name, err.message);
             }
             else {
                 throw err;
@@ -275,8 +307,8 @@ class ExpressionWriter {
                 operator === '!=' ? exports.TRUE : exports.FALSE} : `);
             }
             catch (err) {
-                if (err instanceof typescript_expression_transformer_1.TypeScriptExpressionTransformerError) {
-                    throw new sdk_1.PluginError(_1.name, err.message);
+                if (err instanceof sdk_1.TypeScriptExpressionTransformerError) {
+                    throw new sdk_1.PluginError(__1.name, err.message);
                 }
                 else {
                     throw err;
@@ -288,11 +320,11 @@ class ExpressionWriter {
                 this.block(() => {
                     var _a;
                     const dataModel = this.isModelTyped(fieldAccess);
-                    if (dataModel && (0, ast_utils_1.isAuthInvocation)(operand)) {
+                    if (dataModel && (0, sdk_1.isAuthInvocation)(operand)) {
                         // right now this branch only serves comparison with `auth`, like
                         //     @@allow('all', owner == auth())
                         if (operator !== '==' && operator !== '!=') {
-                            throw new sdk_1.PluginError(_1.name, 'Only == and != operators are allowed');
+                            throw new sdk_1.PluginError(__1.name, 'Only == and != operators are allowed');
                         }
                         if (!(0, ast_1.isThisExpr)(fieldAccess)) {
                             this.writer.writeLine(operator === '==' ? 'is:' : 'isNot:');
@@ -350,9 +382,9 @@ class ExpressionWriter {
         return (0, ast_1.isMemberAccessExpr)(expr) && (0, sdk_1.isFutureExpr)(expr.operand);
     }
     requireIdFields(dataModel) {
-        const idFields = (0, ast_utils_1.getIdFields)(dataModel);
+        const idFields = (0, sdk_1.getIdFields)(dataModel);
         if (!idFields || idFields.length === 0) {
-            throw new sdk_1.PluginError(_1.name, `Data model ${dataModel.name} does not have an id field`);
+            throw new sdk_1.PluginError(__1.name, `Data model ${dataModel.name} does not have an id field`);
         }
         return idFields;
     }
@@ -373,14 +405,14 @@ class ExpressionWriter {
     // https://www.prisma.io/docs/reference/api-reference/prisma-client-reference#compare-columns-in-the-same-table
     writeFieldReference(expr) {
         if (!expr.target.ref) {
-            throw new sdk_1.PluginError(_1.name, `Unresolved reference "${expr.target.$refText}"`);
+            throw new sdk_1.PluginError(__1.name, `Unresolved reference "${expr.target.$refText}"`);
         }
         const containingModel = expr.target.ref.$container;
         this.writer.write(`db.${(0, lower_case_first_1.lowerCaseFirst)(containingModel.name)}.fields.${expr.target.ref.name}`);
     }
     isAuthOrAuthMemberAccess(expr) {
         // recursive check for auth().x.y.z
-        return (0, ast_utils_1.isAuthInvocation)(expr) || ((0, ast_1.isMemberAccessExpr)(expr) && this.isAuthOrAuthMemberAccess(expr.operand));
+        return (0, sdk_1.isAuthInvocation)(expr) || ((0, ast_1.isMemberAccessExpr)(expr) && this.isAuthOrAuthMemberAccess(expr.operand));
     }
     writeOperator(operator, fieldAccess, writeOperand) {
         var _a;
@@ -392,7 +424,7 @@ class ExpressionWriter {
                 this.writer.write('isNot: ');
             }
             else {
-                throw new sdk_1.PluginError(_1.name, 'Only == and != operators are allowed for data model comparison');
+                throw new sdk_1.PluginError(__1.name, 'Only == and != operators are allowed for data model comparison');
             }
             writeOperand();
         }
@@ -412,50 +444,68 @@ class ExpressionWriter {
         }
     }
     writeFieldCondition(fieldAccess, writeCondition, filterOp, extraArgs) {
-        var _a, _b, _c;
-        let selector;
+        // let selector: string | undefined;
         let operand;
+        let fieldWriter;
         if ((0, ast_1.isThisExpr)(fieldAccess)) {
             // pass on
             writeCondition();
             return;
         }
         else if ((0, ast_1.isReferenceExpr)(fieldAccess)) {
-            selector = (_a = fieldAccess.target.ref) === null || _a === void 0 ? void 0 : _a.name;
+            const ref = fieldAccess.target.ref;
+            (0, tiny_invariant_1.default)(ref);
+            if (this.isFieldReferenceToDelegateModel(ref)) {
+                const thisModel = ref.$container;
+                const targetBase = ref.$inheritedFrom;
+                fieldWriter = (conditionWriter) => this.writeBaseHierarchy(thisModel, targetBase, () => {
+                    this.writer.write(`${ref.name}: `);
+                    conditionWriter();
+                });
+            }
+            else {
+                fieldWriter = (conditionWriter) => {
+                    this.writer.write(`${ref.name}: `);
+                    conditionWriter();
+                };
+            }
         }
         else if ((0, ast_1.isMemberAccessExpr)(fieldAccess)) {
-            if ((0, sdk_1.isFutureExpr)(fieldAccess.operand)) {
+            if (!(0, sdk_1.isFutureExpr)(fieldAccess.operand)) {
                 // future().field should be treated as the "field"
-                selector = (_b = fieldAccess.member.ref) === null || _b === void 0 ? void 0 : _b.name;
-            }
-            else {
-                selector = (_c = fieldAccess.member.ref) === null || _c === void 0 ? void 0 : _c.name;
                 operand = fieldAccess.operand;
             }
+            fieldWriter = (conditionWriter) => {
+                var _a;
+                this.writer.write(`${(_a = fieldAccess.member.ref) === null || _a === void 0 ? void 0 : _a.name}: `);
+                conditionWriter();
+            };
         }
         else {
-            throw new sdk_1.PluginError(_1.name, `Unsupported expression type: ${fieldAccess.$type}`);
+            throw new sdk_1.PluginError(__1.name, `Unsupported expression type: ${fieldAccess.$type}`);
         }
-        if (!selector) {
-            throw new sdk_1.PluginError(_1.name, `Failed to write FieldAccess expression`);
+        if (!fieldWriter) {
+            throw new sdk_1.PluginError(__1.name, `Failed to write FieldAccess expression`);
         }
         const writerFilterOutput = () => {
-            this.writer.write(selector + ': ');
-            if (filterOp) {
-                this.block(() => {
-                    this.writer.write(`${filterOp}: `);
-                    writeCondition();
-                    if (extraArgs) {
-                        for (const [k, v] of Object.entries(extraArgs)) {
-                            this.writer.write(`,\n${k}: `);
-                            this.plain(v);
+            // this.writer.write(selector + ': ');
+            fieldWriter(() => {
+                if (filterOp) {
+                    this.block(() => {
+                        this.writer.write(`${filterOp}: `);
+                        writeCondition();
+                        if (extraArgs) {
+                            for (const [k, v] of Object.entries(extraArgs)) {
+                                this.writer.write(`,\n${k}: `);
+                                this.plain(v);
+                            }
                         }
-                    }
-                });
-            }
-            else {
-                writeCondition();
-            }
+                    });
+                }
+                else {
+                    writeCondition();
+                }
+            });
         };
         if (operand) {
             // member access expression
@@ -547,7 +597,7 @@ class ExpressionWriter {
     }
     writeUnary(expr) {
         if (expr.operator !== '!') {
-            throw new sdk_1.PluginError(_1.name, `Unary operator "${expr.operator}" is not supported`);
+            throw new sdk_1.PluginError(__1.name, `Unary operator "${expr.operator}" is not supported`);
         }
         this.block(() => {
             this.writer.write('NOT: ');
@@ -571,7 +621,7 @@ class ExpressionWriter {
         var _a, _b;
         const funcDecl = expr.function.ref;
         if (!funcDecl) {
-            throw new sdk_1.PluginError(_1.name, `Failed to resolve function declaration`);
+            throw new sdk_1.PluginError(__1.name, `Failed to resolve function declaration`);
         }
         const functionAllowedContext = (0, sdk_1.getFunctionExpressionContext)(funcDecl);
         if (functionAllowedContext.includes(sdk_1.ExpressionContext.AccessPolicy) ||
@@ -600,7 +650,7 @@ class ExpressionWriter {
             });
         }
         else {
-            throw new sdk_1.PluginError(_1.name, `Unsupported function ${funcDecl.name}`);
+            throw new sdk_1.PluginError(__1.name, `Unsupported function ${funcDecl.name}`);
         }
     }
 }