@zenstackhq/runtime 1.0.0-beta.13 → 1.0.0-beta.16

package/enhancements/policy/policy-utils.js

@@ -14,86 +14,133 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.PolicyUtil = void 0;
-const cuid2_1 = require("@paralleldrive/cuid2");
 const deepcopy_1 = __importDefault(require("deepcopy"));
 const lower_case_first_1 = require("lower-case-first");
-const pluralize_1 = __importDefault(require("pluralize"));
 const upper_case_first_1 = require("upper-case-first");
 const zod_validation_error_1 = require("zod-validation-error");
 const constants_1 = require("../../constants");
-const error_1 = require("../../error");
 const version_1 = require("../../version");
 const model_meta_1 = require("../model-meta");
-const nested_write_vistor_1 = require("../nested-write-vistor");
 const utils_1 = require("../utils");
 const logger_1 = require("./logger");
-const semver_1 = __importDefault(require("semver"));
 /**
  * Access policy enforcement utilities
  */
 class PolicyUtil {
-    constructor(db, modelMeta, policy, zodSchemas, user, logPrismaQuery) {
+    constructor(db, modelMeta, policy, zodSchemas, user, shouldLogQuery = false) {
         this.db = db;
         this.modelMeta = modelMeta;
         this.policy = policy;
         this.zodSchemas = zodSchemas;
         this.user = user;
-        this.logPrismaQuery = logPrismaQuery;
-        this.supportNestedToOneFilter = false;
+        this.shouldLogQuery = shouldLogQuery;
         this.logger = new logger_1.Logger(db);
-        // use Prisma version to detect if we can filter when nested-fetching to-one relation
-        const prismaVersion = (0, version_1.getPrismaVersion)();
-        this.supportNestedToOneFilter = prismaVersion ? semver_1.default.gte(prismaVersion, '4.8.0') : false;
     }
+    //#region Logical operators
     /**
      * Creates a conjunction of a list of query conditions.
      */
     and(...conditions) {
-        if (conditions.includes(false)) {
-            // always false
-            return { [constants_1.GUARD_FIELD_NAME]: false };
-        }
-        const filtered = conditions.filter((c) => typeof c === 'object' && !!c && Object.keys(c).length > 0);
-        if (filtered.length === 0) {
-            return undefined;
-        }
-        else if (filtered.length === 1) {
-            return filtered[0];
-        }
-        else {
-            return { AND: filtered };
-        }
+        return this.reduce({ AND: conditions });
     }
     /**
      * Creates a disjunction of a list of query conditions.
      */
     or(...conditions) {
-        if (conditions.includes(true)) {
-            // always true
-            return { [constants_1.GUARD_FIELD_NAME]: true };
+        return this.reduce({ OR: conditions });
+    }
+    /**
+     * Creates a negation of a query condition.
+     */
+    not(condition) {
+        if (condition === undefined) {
+            return this.makeTrue();
         }
-        const filtered = conditions.filter((c) => typeof c === 'object' && !!c);
-        if (filtered.length === 0) {
-            return undefined;
+        else if (typeof condition === 'boolean') {
+            return this.reduce(!condition);
         }
-        else if (filtered.length === 1) {
-            return filtered[0];
+        else {
+            return this.reduce({ NOT: condition });
+        }
+    }
+    // Static True/False conditions
+    // https://www.prisma.io/docs/concepts/components/prisma-client/null-and-undefined#the-effect-of-null-and-undefined-on-conditionals
+    isTrue(condition) {
+        if (condition === null || condition === undefined) {
+            return false;
         }
         else {
-            return { OR: filtered };
+            return ((typeof condition === 'object' && Object.keys(condition).length === 0) ||
+                ('AND' in condition && Array.isArray(condition.AND) && condition.AND.length === 0));
         }
     }
-    /**
-     * Creates a negation of a query condition.
-     */
-    not(condition) {
-        if (typeof condition === 'boolean') {
-            return !condition;
+    isFalse(condition) {
+        if (condition === null || condition === undefined) {
+            return false;
         }
         else {
-            return { NOT: condition };
+            return 'OR' in condition && Array.isArray(condition.OR) && condition.OR.length === 0;
+        }
+    }
+    makeTrue() {
+        return { AND: [] };
+    }
+    makeFalse() {
+        return { OR: [] };
+    }
+    reduce(condition) {
+        if (condition === true || condition === undefined) {
+            return this.makeTrue();
+        }
+        if (condition === false) {
+            return this.makeFalse();
+        }
+        if ('AND' in condition && Array.isArray(condition.AND)) {
+            const children = condition.AND.map((c) => this.reduce(c)).filter((c) => c !== undefined && !this.isTrue(c));
+            if (children.length === 0) {
+                return this.makeTrue();
+            }
+            else if (children.some((c) => this.isFalse(c))) {
+                return this.makeFalse();
+            }
+            else if (children.length === 1) {
+                return children[0];
+            }
+            else {
+                return { AND: children };
+            }
+        }
+        if ('OR' in condition && Array.isArray(condition.OR)) {
+            const children = condition.OR.map((c) => this.reduce(c)).filter((c) => c !== undefined && !this.isFalse(c));
+            if (children.length === 0) {
+                return this.makeFalse();
+            }
+            else if (children.some((c) => this.isTrue(c))) {
+                return this.makeTrue();
+            }
+            else if (children.length === 1) {
+                return children[0];
+            }
+            else {
+                return { OR: children };
+            }
         }
+        if ('NOT' in condition && condition.NOT !== null && typeof condition.NOT === 'object') {
+            const child = this.reduce(condition.NOT);
+            if (this.isTrue(child)) {
+                return this.makeFalse();
+            }
+            else if (this.isFalse(child)) {
+                return this.makeTrue();
+            }
+            else {
+                return { NOT: child };
+            }
+        }
+        return condition;
     }
+    //#endregion
+    //# Auth guard
     /**
      * Gets pregenerated authorization guard object for a given model and operation.
      *
@@ -107,46 +154,65 @@ class PolicyUtil {
         }
         const provider = guard[operation];
         if (typeof provider === 'boolean') {
-            return provider;
+            return this.reduce(provider);
         }
         if (!provider) {
             throw this.unknownError(`zenstack: unable to load authorization guard for ${model}`);
         }
-        return provider({ user: this.user, preValue });
-    }
-    hasValidation(model) {
-        var _a, _b;
-        return ((_b = (_a = this.policy.validation) === null || _a === void 0 ? void 0 : _a[(0, lower_case_first_1.lowerCaseFirst)(model)]) === null || _b === void 0 ? void 0 : _b.hasValidation) === true;
+        const r = provider({ user: this.user, preValue });
+        return this.reduce(r);
     }
-    getPreValueSelect(model) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const guard = this.policy.guard[(0, lower_case_first_1.lowerCaseFirst)(model)];
-            if (!guard) {
-                throw this.unknownError(`unable to load policy guard for ${model}`);
-            }
-            return guard.preValueSelect;
-        });
+    /**
+     * Checks if the given model has a policy guard for the given operation.
+     */
+    hasAuthGuard(model, operation) {
+        const guard = this.policy.guard[(0, lower_case_first_1.lowerCaseFirst)(model)];
+        if (!guard) {
+            return false;
+        }
+        const provider = guard[operation];
+        return typeof provider !== 'boolean' || provider !== true;
     }
-    getModelSchema(model) {
-        var _a, _b;
-        return this.hasValidation(model) && ((_b = (_a = this.zodSchemas) === null || _a === void 0 ? void 0 : _a.models) === null || _b === void 0 ? void 0 : _b[`${(0, upper_case_first_1.upperCaseFirst)(model)}Schema`]);
+    /**
+     * Checks model creation policy based on static analysis to the input args.
+     *
+     * @returns boolean if static analysis is enough to determine the result, undefined if not
+     */
+    checkInputGuard(model, args, operation) {
+        const guard = this.policy.guard[(0, lower_case_first_1.lowerCaseFirst)(model)];
+        if (!guard) {
+            return undefined;
+        }
+        const provider = guard[`${operation}_input`];
+        if (typeof provider === 'boolean') {
+            return provider;
+        }
+        if (!provider) {
+            return undefined;
+        }
+        return provider(args, { user: this.user });
     }
     /**
      * Injects model auth guard as where clause.
      */
     injectAuthGuard(args, model, operation) {
         return __awaiter(this, void 0, void 0, function* () {
+            const guard = this.getAuthGuard(model, operation);
+            if (this.isFalse(guard)) {
+                args.where = this.makeFalse();
+                return false;
+            }
             if (args.where) {
                 // inject into relation fields:
                 //   to-many: some/none/every
                 //   to-one: direct-conditions/is/isNot
-                yield this.injectGuardForFields(model, args.where, operation);
+                yield this.injectGuardForRelationFields(model, args.where, operation);
             }
-            const guard = this.getAuthGuard(model, operation);
             args.where = this.and(args.where, guard);
+            return true;
         });
     }
-    injectGuardForFields(model, payload, operation) {
+    injectGuardForRelationFields(model, payload, operation) {
         return __awaiter(this, void 0, void 0, function* () {
             for (const [field, subPayload] of Object.entries(payload)) {
                 if (!subPayload) {
@@ -169,12 +235,12 @@ class PolicyUtil {
         return __awaiter(this, void 0, void 0, function* () {
             const guard = this.getAuthGuard(fieldInfo.type, operation);
             if (payload.some) {
-                yield this.injectGuardForFields(fieldInfo.type, payload.some, operation);
+                yield this.injectGuardForRelationFields(fieldInfo.type, payload.some, operation);
                 // turn "some" into: { some: { AND: [guard, payload.some] } }
                 payload.some = this.and(payload.some, guard);
             }
             if (payload.none) {
-                yield this.injectGuardForFields(fieldInfo.type, payload.none, operation);
+                yield this.injectGuardForRelationFields(fieldInfo.type, payload.none, operation);
                 // turn none into: { none: { AND: [guard, payload.none] } }
                 payload.none = this.and(payload.none, guard);
             }
@@ -182,7 +248,7 @@ class PolicyUtil {
                 typeof payload.every === 'object' &&
                 // ignore empty every clause
                 Object.keys(payload.every).length > 0) {
-                yield this.injectGuardForFields(fieldInfo.type, payload.every, operation);
+                yield this.injectGuardForRelationFields(fieldInfo.type, payload.every, operation);
                 // turn "every" into: { none: { AND: [guard, { NOT: payload.every }] } }
                 if (!payload.none) {
                     payload.none = {};
@@ -197,19 +263,19 @@ class PolicyUtil {
             const guard = this.getAuthGuard(fieldInfo.type, operation);
             if (payload.is || payload.isNot) {
                 if (payload.is) {
-                    yield this.injectGuardForFields(fieldInfo.type, payload.is, operation);
+                    yield this.injectGuardForRelationFields(fieldInfo.type, payload.is, operation);
                     // turn "is" into: { is: { AND: [ originalIs, guard ] }
                     payload.is = this.and(payload.is, guard);
                 }
                 if (payload.isNot) {
-                    yield this.injectGuardForFields(fieldInfo.type, payload.isNot, operation);
+                    yield this.injectGuardForRelationFields(fieldInfo.type, payload.isNot, operation);
                     // turn "isNot" into: { isNot: { AND: [ originalIsNot, { NOT: guard } ] } }
                     payload.isNot = this.and(payload.isNot, this.not(guard));
                     delete payload.isNot;
                 }
             }
             else {
-                yield this.injectGuardForFields(fieldInfo.type, payload, operation);
+                yield this.injectGuardForRelationFields(fieldInfo.type, payload, operation);
                 // turn direct conditions into: { is: { AND: [ originalConditions, guard ] } }
                 const combined = this.and((0, deepcopy_1.default)(payload), guard);
                 Object.keys(payload).forEach((key) => delete payload[key]);
@@ -218,63 +284,120 @@ class PolicyUtil {
         });
     }
     /**
-     * Read model entities w.r.t the given query args. The result list
-     * are guaranteed to fully satisfy 'read' policy rules recursively.
-     *
-     * For to-many relations involved, items not satisfying policy are
-     * silently trimmed. For to-one relation, if relation data fails policy
-     * an error is thrown.
+     * Injects auth guard for read operations.
      */
-    readWithCheck(model, args) {
+    injectForRead(model, args) {
+        var _a, _b;
         return __awaiter(this, void 0, void 0, function* () {
-            args = this.clone(args);
+            const injected = {};
+            if (!(yield this.injectAuthGuard(injected, model, 'read'))) {
+                return false;
+            }
             if (args.where) {
-                // query args will be used with findMany, so we need to
-                // translate unique constraint filters into a flat filter
-                // e.g.: { a_b: { a: '1', b: '1' } } => { a: '1', b: '1' }
-                yield this.flattenGeneratedUniqueField(model, args.where);
-            }
-            yield this.injectAuthGuard(args, model, 'read');
-            // recursively inject read guard conditions into the query args
-            yield this.injectNestedReadConditions(model, args);
-            if (this.shouldLogQuery) {
-                this.logger.info(`[withPolicy] \`findMany\`:\n${(0, utils_1.formatObject)(args)}`);
+                // inject into relation fields:
+                //   to-many: some/none/every
+                //   to-one: direct-conditions/is/isNot
+                yield this.injectGuardForRelationFields(model, args.where, 'read');
             }
-            const result = yield this.db[model].findMany(args);
-            yield this.postProcessForRead(result, model, args, 'read');
-            return result;
+            if (injected.where && Object.keys(injected.where).length > 0 && !this.isTrue(injected.where)) {
+                args.where = (_a = args.where) !== null && _a !== void 0 ? _a : {};
+                Object.assign(args.where, injected.where);
+            }
+            // recursively inject read guard conditions into nested select, include, and _count
+            const hoistedConditions = yield this.injectNestedReadConditions(model, args);
+            // the injection process may generate conditions that need to be hoisted to the toplevel,
+            // if so, merge it with the existing where
+            if (hoistedConditions.length > 0) {
+                args.where = (_b = args.where) !== null && _b !== void 0 ? _b : {};
+                Object.assign(args.where, ...hoistedConditions);
+            }
+            return true;
         });
     }
     // flatten unique constraint filters
     flattenGeneratedUniqueField(model, args) {
         var _a;
+        // e.g.: { a_b: { a: '1', b: '1' } } => { a: '1', b: '1' }
+        const uniqueConstraints = (_a = this.modelMeta.uniqueConstraints) === null || _a === void 0 ? void 0 : _a[(0, lower_case_first_1.lowerCaseFirst)(model)];
+        if (uniqueConstraints && Object.keys(uniqueConstraints).length > 0) {
+            for (const [field, value] of Object.entries(args)) {
+                if (uniqueConstraints[field] &&
+                    uniqueConstraints[field].fields.length > 1 &&
+                    typeof value === 'object') {
+                    // multi-field unique constraint, flatten it
+                    delete args[field];
+                    for (const [f, v] of Object.entries(value)) {
+                        args[f] = v;
+                    }
+                }
+            }
+        }
+    }
+    /**
+     * Gets unique constraints for the given model.
+     */
+    getUniqueConstraints(model) {
+        var _a, _b;
+        return (_b = (_a = this.modelMeta.uniqueConstraints) === null || _a === void 0 ? void 0 : _a[(0, lower_case_first_1.lowerCaseFirst)(model)]) !== null && _b !== void 0 ? _b : {};
+    }
+    /**
+     * Builds a reversed query for the given nested path.
+     */
+    buildReversedQuery(context) {
         return __awaiter(this, void 0, void 0, function* () {
-            // e.g.: { a_b: { a: '1', b: '1' } } => { a: '1', b: '1' }
-            const uniqueConstraints = (_a = this.modelMeta.uniqueConstraints) === null || _a === void 0 ? void 0 : _a[(0, lower_case_first_1.lowerCaseFirst)(model)];
-            let flattened = false;
-            if (uniqueConstraints && Object.keys(uniqueConstraints).length > 0) {
-                for (const [field, value] of Object.entries(args)) {
-                    if (uniqueConstraints[field] && typeof value === 'object') {
-                        for (const [f, v] of Object.entries(value)) {
-                            args[f] = v;
+            let result, currQuery;
+            let currField;
+            for (let i = context.nestingPath.length - 1; i >= 0; i--) {
+                const { field, model, where } = context.nestingPath[i];
+                // never modify the original where because it's shared in the structure
+                const visitWhere = Object.assign({}, where);
+                if (model && where) {
+                    // make sure composite unique condition is flattened
+                    this.flattenGeneratedUniqueField(model, visitWhere);
+                }
+                if (!result) {
+                    // first segment (bottom), just use its where clause
+                    result = currQuery = Object.assign({}, visitWhere);
+                    currField = field;
+                }
+                else {
+                    if (!currField) {
+                        throw this.unknownError(`missing field in nested path`);
+                    }
+                    if (!currField.backLink) {
+                        throw this.unknownError(`field ${currField.type}.${currField.name} doesn't have a backLink`);
+                    }
+                    const backLinkField = this.getModelField(currField.type, currField.backLink);
+                    if (backLinkField === null || backLinkField === void 0 ? void 0 : backLinkField.isArray) {
+                        // many-side of relationship, wrap with "some" query
+                        currQuery[currField.backLink] = { some: Object.assign({}, visitWhere) };
+                    }
+                    else {
+                        if (where && backLinkField.isRelationOwner && backLinkField.foreignKeyMapping) {
+                            for (const [r, fk] of Object.entries(backLinkField.foreignKeyMapping)) {
+                                currQuery[fk] = visitWhere[r];
+                            }
+                            if (i > 0) {
+                                currQuery[currField.backLink] = {};
+                            }
+                        }
+                        else {
+                            currQuery[currField.backLink] = Object.assign({}, visitWhere);
                         }
-                        delete args[field];
-                        flattened = true;
                     }
+                    currQuery = currQuery[currField.backLink];
+                    currField = field;
                 }
             }
-            if (flattened) {
-                // DEBUG
-                // this.logger.info(`Filter flattened: ${JSON.stringify(args)}`);
-            }
+            return result;
         });
     }
     injectNestedReadConditions(model, args) {
-        var _a, _b;
+        var _a;
         return __awaiter(this, void 0, void 0, function* () {
             const injectTarget = (_a = args.select) !== null && _a !== void 0 ? _a : args.include;
             if (!injectTarget) {
-                return;
+                return [];
             }
             if (injectTarget._count !== undefined) {
                 // _count needs to respect read policies of related models
@@ -305,424 +428,143 @@ class PolicyUtil {
                     yield this.injectAuthGuard(injectTarget._count.select[field], fieldInfo.type, 'read');
                 }
             }
-            const idFields = this.getIdFields(model);
+            // collect filter conditions that should be hoisted to the toplevel
+            const hoistedConditions = [];
             for (const field of (0, utils_1.getModelFields)(injectTarget)) {
                 const fieldInfo = (0, model_meta_1.resolveField)(this.modelMeta, model, field);
                 if (!fieldInfo || !fieldInfo.isDataModel) {
                     // only care about relation fields
                     continue;
                 }
+                let hoisted;
                 if (fieldInfo.isArray ||
-                    // if Prisma version is high enough to support filtering directly when
-                    // fetching a nullable to-one relation, let's do it that way
+                    // Injecting where at include/select level for nullable to-one relation is supported since Prisma 4.8.0
                     // https://github.com/prisma/prisma/discussions/20350
-                    (this.supportNestedToOneFilter && fieldInfo.isOptional)) {
+                    fieldInfo.isOptional) {
                     if (typeof injectTarget[field] !== 'object') {
                         injectTarget[field] = {};
                     }
                     // inject extra condition for to-many or nullable to-one relation
                     yield this.injectAuthGuard(injectTarget[field], fieldInfo.type, 'read');
-                    // recurse
-                    yield this.injectNestedReadConditions(fieldInfo.type, injectTarget[field]);
                 }
                 else {
-                    // there's no way of injecting condition for to-one relation, so if there's
-                    // "select" clause we make sure 'id' fields are selected and check them against
-                    // query result; nothing needs to be done for "include" clause because all
-                    // fields are already selected
-                    if ((_b = injectTarget[field]) === null || _b === void 0 ? void 0 : _b.select) {
-                        for (const idField of idFields) {
-                            if (injectTarget[field].select[idField.name] !== true) {
-                                injectTarget[field].select[idField.name] = true;
-                            }
-                        }
-                    }
+                    // hoist non-nullable to-one filter to the parent level
+                    hoisted = this.getAuthGuard(fieldInfo.type, 'read');
+                }
+                // recurse
+                const subHoisted = yield this.injectNestedReadConditions(fieldInfo.type, injectTarget[field]);
+                if (subHoisted.length > 0) {
+                    hoisted = this.and(hoisted, ...subHoisted);
+                }
+                if (hoisted && !this.isTrue(hoisted)) {
+                    hoistedConditions.push({ [field]: hoisted });
                 }
             }
+            return hoistedConditions;
         });
     }
     /**
-     * Post processing checks for read model entities. Validates to-one relations
-     * (which can't be trimmed at query time) and removes fields that should be
-     * omitted.
+     * Given a model and a unique filter, checks the operation is allowed by policies and field validations.
+     * Rejects with an error if not allowed.
      */
-    postProcessForRead(data, model, args, operation) {
+    checkPolicyForUnique(model, uniqueFilter, operation, db, preValue) {
         return __awaiter(this, void 0, void 0, function* () {
-            yield Promise.all((0, utils_1.enumerate)(data).map((entityData) => this.postProcessSingleEntityForRead(entityData, model, args, operation)));
-        });
-    }
-    postProcessSingleEntityForRead(data, model, args, operation) {
-        var _a;
-        return __awaiter(this, void 0, void 0, function* () {
-            if (typeof data !== 'object' || !data) {
+            const guard = this.getAuthGuard(model, operation, preValue);
+            if (this.isFalse(guard)) {
+                throw this.deniedByPolicy(model, operation, `entity ${(0, utils_1.formatObject)(uniqueFilter)} failed policy check`);
+            }
+            // Zod schema is to be checked for "create" and "postUpdate"
+            const schema = ['create', 'postUpdate'].includes(operation) ? this.getZodSchema(model) : undefined;
+            if (this.isTrue(guard) && !schema) {
+                // unconditionally allowed
                 return;
             }
-            // strip auxiliary fields
-            for (const auxField of constants_1.AUXILIARY_FIELDS) {
-                if (auxField in data) {
-                    delete data[auxField];
-                }
+            const select = schema
+                ? // need to validate against schema, need to fetch all fields
+                    undefined
+                : // only fetch id fields
+                    this.makeIdSelection(model);
+            let where = this.clone(uniqueFilter);
+            // query args may have be of combined-id form, need to flatten it to call findFirst
+            this.flattenGeneratedUniqueField(model, where);
+            // query with policy guard
+            where = this.and(where, guard);
+            const query = { select, where };
+            if (this.shouldLogQuery) {
+                this.logger.info(`[policy] checking ${model} for ${operation}, \`findFirst\`:\n${(0, utils_1.formatObject)(query)}`);
             }
-            const injectTarget = (_a = args.select) !== null && _a !== void 0 ? _a : args.include;
-            if (!injectTarget) {
-                return;
+            const result = yield db[model].findFirst(query);
+            if (!result) {
+                throw this.deniedByPolicy(model, operation, `entity ${(0, utils_1.formatObject)(uniqueFilter)} failed policy check`);
             }
-            // recurse into nested entities
-            for (const field of Object.keys(injectTarget)) {
-                const fieldData = data[field];
-                if (typeof fieldData !== 'object' || !fieldData) {
-                    continue;
-                }
-                const fieldInfo = (0, model_meta_1.resolveField)(this.modelMeta, model, field);
-                if (fieldInfo) {
-                    if (fieldInfo.isDataModel &&
-                        !fieldInfo.isArray &&
-                        // if Prisma version supports filtering nullable to-one relation, no need to further check
-                        !(this.supportNestedToOneFilter && fieldInfo.isOptional)) {
-                        // to-one relation data cannot be trimmed by injected guards, we have to
-                        // post-check them
-                        const ids = this.getEntityIds(fieldInfo.type, fieldData);
-                        if (Object.keys(ids).length !== 0) {
-                            if (this.logger.enabled('info')) {
-                                this.logger.info(`Validating read of to-one relation: ${fieldInfo.type}#${(0, utils_1.formatObject)(ids)}`);
-                            }
-                            try {
-                                yield this.checkPolicyForFilter(fieldInfo.type, ids, operation, this.db);
-                            }
-                            catch (err) {
-                                if ((0, error_1.isPrismaClientKnownRequestError)(err) &&
-                                    err.code === constants_1.PrismaErrorCode.CONSTRAINED_FAILED) {
-                                    // denied by policy
-                                    if (fieldInfo.isOptional) {
-                                        // if the relation is optional, just nullify it
-                                        data[field] = null;
-                                    }
-                                    else {
-                                        // otherwise reject
-                                        throw err;
-                                    }
-                                }
-                                else {
-                                    // unknown error
-                                    throw err;
-                                }
-                            }
-                        }
+            if (schema) {
+                // TODO: push down schema check to the database
+                const parseResult = schema.safeParse(result);
+                if (!parseResult.success) {
+                    const error = (0, zod_validation_error_1.fromZodError)(parseResult.error);
+                    if (this.logger.enabled('info')) {
+                        this.logger.info(`entity ${model} failed validation for operation ${operation}: ${error}`);
                     }
-                    // recurse
-                    yield this.postProcessForRead(fieldData, fieldInfo.type, injectTarget[field], operation);
+                    throw this.deniedByPolicy(model, operation, `entities ${JSON.stringify(uniqueFilter)} failed validation: [${error}]`, constants_1.CrudFailureReason.DATA_VALIDATION_VIOLATION);
                 }
             }
         });
     }
     /**
-     * Process Prisma write actions.
+     * Tries rejecting a request based on static "false" policy.
+     */
+    tryReject(model, operation) {
+        const guard = this.getAuthGuard(model, operation);
+        if (this.isFalse(guard)) {
+            throw this.deniedByPolicy(model, operation);
+        }
+    }
+    /**
+     * Checks if a model exists given a unique filter.
      */
-    processWrite(model, action, args, writeAction) {
+    checkExistence(db, model, uniqueFilter, throwIfNotFound = false) {
         return __awaiter(this, void 0, void 0, function* () {
-            // record model types for which new entities are created
-            // so we can post-check if they satisfy 'create' policies
-            const createdModels = new Set();
-            // record model entities that are updated, together with their
-            // values before update, so we can post-check if they satisfy
-            //     model => { ids, entity value }
-            const updatedModels = new Map();
-            function addUpdatedEntity(model, ids, entity) {
-                let modelEntities = updatedModels.get(model);
-                if (!modelEntities) {
-                    modelEntities = [];
-                    updatedModels.set(model, modelEntities);
-                }
-                modelEntities.push({ ids, value: entity });
-            }
-            const idFields = this.getIdFields(model);
-            if (args.select) {
-                // make sure id fields are selected, we need it to
-                // read back the updated entity
-                for (const idField of idFields) {
-                    if (!args.select[idField.name]) {
-                        args.select[idField.name] = true;
-                    }
-                }
+            uniqueFilter = this.clone(uniqueFilter);
+            this.flattenGeneratedUniqueField(model, uniqueFilter);
+            if (this.shouldLogQuery) {
+                this.logger.info(`[policy] checking ${model} existence, \`findFirst\`:\n${(0, utils_1.formatObject)(uniqueFilter)}`);
             }
-            // use a transaction to conduct write, so in case any create or nested create
-            // fails access policies, we can roll back the entire operation
-            const transactionId = (0, cuid2_1.createId)();
-            // args processor for create
-            const processCreate = (model, args) => __awaiter(this, void 0, void 0, function* () {
-                const guard = this.getAuthGuard(model, 'create');
-                const schema = this.getModelSchema(model);
-                if (guard === false) {
-                    throw this.deniedByPolicy(model, 'create');
-                }
-                else if (guard !== true || schema) {
-                    // mark the create with a transaction tag so we can check them later
-                    args[constants_1.TRANSACTION_FIELD_NAME] = `${transactionId}:create`;
-                    createdModels.add(model);
-                }
-            });
-            // build a reversed query for fetching entities affected by nested updates
-            const buildReversedQuery = (context) => __awaiter(this, void 0, void 0, function* () {
-                let result, currQuery;
-                let currField;
-                for (let i = context.nestingPath.length - 1; i >= 0; i--) {
-                    const { field, model, where, unique } = context.nestingPath[i];
-                    // never modify the original where because it's shared in the structure
-                    const visitWhere = Object.assign({}, where);
-                    if (model && where) {
-                        // make sure composite unique condition is flattened
-                        yield this.flattenGeneratedUniqueField(model, visitWhere);
-                    }
-                    if (!result) {
-                        // first segment (bottom), just use its where clause
-                        result = currQuery = Object.assign({}, visitWhere);
-                        currField = field;
-                    }
-                    else {
-                        if (!currField) {
-                            throw this.unknownError(`missing field in nested path`);
-                        }
-                        if (!currField.backLink) {
-                            throw this.unknownError(`field ${currField.type}.${currField.name} doesn't have a backLink`);
-                        }
-                        const backLinkField = this.getModelField(currField.type, currField.backLink);
-                        if (backLinkField === null || backLinkField === void 0 ? void 0 : backLinkField.isArray) {
-                            // many-side of relationship, wrap with "some" query
-                            currQuery[currField.backLink] = { some: Object.assign({}, visitWhere) };
-                        }
-                        else {
-                            currQuery[currField.backLink] = Object.assign({}, visitWhere);
-                        }
-                        currQuery = currQuery[currField.backLink];
-                        currField = field;
-                    }
-                    if (unique) {
-                        // hit a unique filter, no need to traverse further up
-                        break;
-                    }
-                }
-                return result;
-            });
-            // args processor for update/upsert
-            const processUpdate = (model, where, context) => __awaiter(this, void 0, void 0, function* () {
-                const preGuard = this.getAuthGuard(model, 'update');
-                if (preGuard === false) {
-                    throw this.deniedByPolicy(model, 'update');
-                }
-                else if (preGuard !== true) {
-                    if (this.isToOneRelation(context.field)) {
-                        // To-one relation field is complicated because there's no way to
-                        // filter it during update (args doesn't carry a 'where' clause).
-                        //
-                        // We need to recursively walk up its hierarcy in the query args
-                        // to construct a reversed query to identify the nested entity
-                        // under update, and then check if it satisfies policy.
-                        //
-                        // E.g.:
-                        // A - B - C
-                        //
-                        // update A with:
-                        // {
-                        //   where: { id: 'aId' },
-                        //   data: {
-                        //     b: {
-                        //       c: { value: 1 }
-                        //     }
-                        //   }
-                        // }
-                        //
-                        // To check if the update to 'c' field is permitted, we
-                        // reverse the query stack into a filter for C model, like:
-                        // {
-                        //   where: {
-                        //     b: { a: { id: 'aId' } }
-                        //   }
-                        // }
-                        // , and with this we can filter out the C entity that's going
-                        // to be nestedly updated, and check if it's allowed.
-                        //
-                        // The same logic applies to nested delete.
-                        const subQuery = yield buildReversedQuery(context);
-                        yield this.checkPolicyForFilter(model, subQuery, 'update', this.db);
-                    }
-                    else {
-                        if (!where) {
-                            throw this.unknownError(`Missing 'where' parameter`);
-                        }
-                        yield this.checkPolicyForFilter(model, where, 'update', this.db);
-                    }
-                }
-                yield preparePostUpdateCheck(model, context);
-            });
-            // args processor for updateMany
-            const processUpdateMany = (model, args, context) => __awaiter(this, void 0, void 0, function* () {
-                const guard = this.getAuthGuard(model, 'update');
-                if (guard === false) {
-                    throw this.deniedByPolicy(model, 'update');
-                }
-                else if (guard !== true) {
-                    // inject policy filter
-                    yield this.injectAuthGuard(args, model, 'update');
-                }
-                yield preparePostUpdateCheck(model, context);
-            });
-            // for models with post-update rules, we need to read and store
-            // entity values before the update for post-update check
-            const preparePostUpdateCheck = (model, context) => __awaiter(this, void 0, void 0, function* () {
-                const postGuard = this.getAuthGuard(model, 'postUpdate');
-                const schema = this.getModelSchema(model);
-                // post-update check is needed if there's post-update rule or validation schema
-                if (postGuard !== true || schema) {
-                    // fetch preValue selection (analyzed from the post-update rules)
-                    const preValueSelect = yield this.getPreValueSelect(model);
-                    const filter = yield buildReversedQuery(context);
-                    // query args will be used with findMany, so we need to
-                    // translate unique constraint filters into a flat filter
-                    // e.g.: { a_b: { a: '1', b: '1' } } => { a: '1', b: '1' }
-                    yield this.flattenGeneratedUniqueField(model, filter);
-                    const idFields = this.getIdFields(model);
-                    // eslint-disable-next-line @typescript-eslint/no-unused-vars
-                    const select = Object.assign({}, preValueSelect);
-                    for (const idField of idFields) {
-                        select[idField.name] = true;
-                    }
-                    const query = { where: filter, select };
-                    if (this.shouldLogQuery) {
-                        this.logger.info(`[withPolicy] \`findMany\` for fetching pre-update entities:\n${(0, utils_1.formatObject)(args)}`);
-                    }
-                    const entities = yield this.db[model].findMany(query);
-                    entities.forEach((entity) => {
-                        addUpdatedEntity(model, this.getEntityIds(model, entity), entity);
-                    });
-                }
-            });
-            // args processor for delete
-            const processDelete = (model, args, context) => __awaiter(this, void 0, void 0, function* () {
-                const guard = this.getAuthGuard(model, 'delete');
-                if (guard === false) {
-                    throw this.deniedByPolicy(model, 'delete');
-                }
-                else if (guard !== true) {
-                    if (this.isToOneRelation(context.field)) {
-                        // see comments in processUpdate
-                        const subQuery = yield buildReversedQuery(context);
-                        yield this.checkPolicyForFilter(model, subQuery, 'delete', this.db);
-                    }
-                    else {
-                        yield this.checkPolicyForFilter(model, args, 'delete', this.db);
-                    }
-                }
-            });
-            // process relation updates: connect, connectOrCreate, and disconnect
-            const processRelationUpdate = (model, args, context) => __awaiter(this, void 0, void 0, function* () {
-                // CHECK ME: equire the entity being connected readable?
-                // await this.checkPolicyForFilter(model, args, 'read', this.db);
-                var _a;
-                if ((_a = context.field) === null || _a === void 0 ? void 0 : _a.backLink) {
-                    // fetch the backlink field of the model being connected
-                    const backLinkField = (0, model_meta_1.resolveField)(this.modelMeta, model, context.field.backLink);
-                    if (backLinkField.isRelationOwner) {
-                        // the target side of relation owns the relation,
-                        // mark it as updated
-                        yield processUpdate(model, args, context);
-                    }
-                }
+            const existing = yield db[model].findFirst({
+                where: uniqueFilter,
+                select: this.makeIdSelection(model),
             });
-            // use a visitor to process args before conducting the write action
-            const visitor = new nested_write_vistor_1.NestedWriteVisitor(this.modelMeta, {
-                create: (model, args) => __awaiter(this, void 0, void 0, function* () {
-                    yield processCreate(model, args);
-                }),
-                connectOrCreate: (model, args, context) => __awaiter(this, void 0, void 0, function* () {
-                    if (args.create) {
-                        yield processCreate(model, args.create);
-                    }
-                    if (args.where) {
-                        yield processRelationUpdate(model, args.where, context);
-                    }
-                }),
-                connect: (model, args, context) => __awaiter(this, void 0, void 0, function* () {
-                    yield processRelationUpdate(model, args, context);
-                }),
-                disconnect: (model, args, context) => __awaiter(this, void 0, void 0, function* () {
-                    yield processRelationUpdate(model, args, context);
-                }),
-                update: (model, args, context) => __awaiter(this, void 0, void 0, function* () {
-                    yield processUpdate(model, args.where, context);
-                }),
-                updateMany: (model, args, context) => __awaiter(this, void 0, void 0, function* () {
-                    yield processUpdateMany(model, args, context);
-                }),
-                upsert: (model, args, context) => __awaiter(this, void 0, void 0, function* () {
-                    if (args.create) {
-                        yield processCreate(model, args.create);
-                    }
-                    if (args.update) {
-                        yield processUpdate(model, args.where, context);
-                    }
-                }),
-                delete: (model, args, context) => __awaiter(this, void 0, void 0, function* () {
-                    yield processDelete(model, args, context);
-                }),
-                // eslint-disable-next-line @typescript-eslint/no-unused-vars
-                deleteMany: (model, args, _context) => __awaiter(this, void 0, void 0, function* () {
-                    const guard = this.getAuthGuard(model, 'delete');
-                    if (guard === false) {
-                        throw this.deniedByPolicy(model, 'delete');
-                    }
-                    else if (guard !== true) {
-                        if (args.where) {
-                            args.where = this.and(args.where, guard);
-                        }
-                        else {
-                            const copy = (0, deepcopy_1.default)(args);
-                            for (const key of Object.keys(args)) {
-                                delete args[key];
-                            }
-                            const combined = this.and(copy, guard);
-                            Object.assign(args, combined);
-                        }
-                    }
-                }),
-            });
-            yield visitor.visit(model, action, args);
-            if (createdModels.size === 0 && updatedModels.size === 0) {
-                // no post-check needed, we can proceed with the write without transaction
-                return yield writeAction(this.db[model], args);
-            }
-            else {
-                return yield this.transaction(this.db, (tx) => __awaiter(this, void 0, void 0, function* () {
-                    // proceed with the update (with args processed)
-                    const result = yield writeAction(tx[model], args);
-                    if (createdModels.size > 0) {
-                        // do post-check on created entities
-                        yield Promise.all([...createdModels].map((model) => this.checkPolicyForFilter(model, { [constants_1.TRANSACTION_FIELD_NAME]: `${transactionId}:create` }, 'create', tx)));
-                    }
-                    if (updatedModels.size > 0) {
-                        // do post-check on updated entities
-                        yield Promise.all([...updatedModels.entries()]
-                            .map(([model, modelEntities]) => modelEntities.map(({ ids, value: preValue }) => __awaiter(this, void 0, void 0, function* () { return this.checkPostUpdate(model, ids, tx, preValue); })))
-                            .flat());
-                    }
-                    return result;
-                }));
+            if (!existing && throwIfNotFound) {
+                throw this.notFound(model);
             }
+            return existing;
         });
     }
-    getModelField(model, field) {
-        var _a;
-        model = (0, lower_case_first_1.lowerCaseFirst)(model);
-        return (_a = this.modelMeta.fields[model]) === null || _a === void 0 ? void 0 : _a[field];
-    }
-    transaction(db, action) {
-        if (db[constants_1.PRISIMA_TX_FLAG]) {
-            // already in transaction, don't nest
-            return action(db);
-        }
-        else {
-            return db.$transaction((tx) => action(tx));
-        }
+    /**
+     * Returns an entity given a unique filter with read policy checked. Reject if not readable.
+     */
+    readBack(db, model, operation, selectInclude, uniqueFilter) {
+        return __awaiter(this, void 0, void 0, function* () {
+            uniqueFilter = this.clone(uniqueFilter);
+            this.flattenGeneratedUniqueField(model, uniqueFilter);
+            const readArgs = { select: selectInclude.select, include: selectInclude.include, where: uniqueFilter };
+            const error = this.deniedByPolicy(model, operation, 'result is not allowed to be read back', constants_1.CrudFailureReason.RESULT_NOT_READABLE);
+            const injectResult = yield this.injectForRead(model, readArgs);
+            if (!injectResult) {
+                return { error, result: undefined };
+            }
+            if (this.shouldLogQuery) {
+                this.logger.info(`[policy] checking read-back, \`findFirst\` ${model}:\n${(0, utils_1.formatObject)(readArgs)}`);
+            }
+            const result = yield db[model].findFirst(readArgs);
+            if (!result) {
+                return { error, result: undefined };
+            }
+            this.postProcessForRead(result);
+            return { result, error: undefined };
+        });
     }
+    //#endregion
+    //#region Errors
     deniedByPolicy(model, operation, extra, reason) {
         return (0, utils_1.prismaClientKnownRequestError)(this.db, `denied by policy: ${model} entities failed '${operation}' check${extra ? ', ' + extra : ''}`, { clientVersion: (0, version_1.getVersion)(), code: constants_1.PrismaErrorCode.CONSTRAINED_FAILED, meta: { reason } });
     }
@@ -732,122 +574,77 @@ class PolicyUtil {
             code: 'P2025',
         });
     }
+    validationError(message) {
+        return (0, utils_1.prismaClientValidationError)(this.db, message, {
+            clientVersion: (0, version_1.getVersion)(),
+        });
+    }
     unknownError(message) {
         return (0, utils_1.prismaClientUnknownRequestError)(this.db, message, {
             clientVersion: (0, version_1.getVersion)(),
         });
     }
+    //#endregion
+    //#region Misc
     /**
-     * Given a filter, check if applying access policy filtering will result
-     * in data being trimmed, and if so, throw an error.
+     * Gets field selection for fetching pre-update entity values for the given model.
      */
-    checkPolicyForFilter(model, filter, operation, db) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const guard = this.getAuthGuard(model, operation);
-            const schema = (operation === 'create' || operation === 'update') && this.getModelSchema(model);
-            if (guard === true && !schema) {
-                // unconditionally allowed
-                return;
-            }
-            // if (this.logger.enabled('info')) {
-            //     this.logger.info(`Checking policy for ${model}#${JSON.stringify(filter)} for ${operation}`);
-            // }
-            const queryFilter = (0, deepcopy_1.default)(filter);
-            // query args will be used with findMany, so we need to
-            // translate unique constraint filters into a flat filter
-            // e.g.: { a_b: { a: '1', b: '1' } } => { a: '1', b: '1' }
-            yield this.flattenGeneratedUniqueField(model, queryFilter);
-            const countArgs = { where: queryFilter };
-            // if (this.shouldLogQuery) {
-            //     this.logger.info(
-            //         `[withPolicy] \`count\` for policy check without guard:\n${formatObject(countArgs)}`
-            //     );
-            // }
-            const count = (yield db[model].count(countArgs));
-            if (count === 0) {
-                // there's nothing to filter out
+    getPreValueSelect(model) {
+        const guard = this.policy.guard[(0, lower_case_first_1.lowerCaseFirst)(model)];
+        if (!guard) {
+            throw this.unknownError(`unable to load policy guard for ${model}`);
+        }
+        return guard.preValueSelect;
+    }
+    hasFieldValidation(model) {
+        var _a, _b;
+        return ((_b = (_a = this.policy.validation) === null || _a === void 0 ? void 0 : _a[(0, lower_case_first_1.lowerCaseFirst)(model)]) === null || _b === void 0 ? void 0 : _b.hasValidation) === true;
+    }
+    /**
+     * Gets Zod schema for the given model and access kind.
+     *
+     * @param kind If undefined, returns the full schema.
+     */
+    getZodSchema(model, kind = undefined) {
+        var _a, _b;
+        if (!this.hasFieldValidation(model)) {
+            return undefined;
+        }
+        const schemaKey = `${(0, upper_case_first_1.upperCaseFirst)(model)}${kind ? (0, upper_case_first_1.upperCaseFirst)(kind) : ''}Schema`;
+        return (_b = (_a = this.zodSchemas) === null || _a === void 0 ? void 0 : _a.models) === null || _b === void 0 ? void 0 : _b[schemaKey];
+    }
+    /**
+     * Post processing checks and clean-up for read model entities.
+     */
+    postProcessForRead(data) {
+        if (data === null || data === undefined) {
+            return;
+        }
+        for (const entityData of (0, utils_1.enumerate)(data)) {
+            if (typeof entityData !== 'object' || !entityData) {
                 return;
             }
-            if (guard === false) {
-                // unconditionally denied
-                throw this.deniedByPolicy(model, operation, `${count} ${(0, pluralize_1.default)('entity', count)} failed policy check`);
-            }
-            // build a query condition with policy injected
-            const guardedQuery = { where: this.and(queryFilter, guard) };
-            if (schema) {
-                // we've got schemas, so have to fetch entities and validate them
-                // if (this.shouldLogQuery) {
-                //     this.logger.info(
-                //         `[withPolicy] \`findMany\` for policy check with guard:\n${formatObject(countArgs)}`
-                //     );
-                // }
-                const entities = yield db[model].findMany(guardedQuery);
-                if (entities.length < count) {
-                    if (this.logger.enabled('info')) {
-                        this.logger.info(`entity ${model} failed policy check for operation ${operation}`);
-                    }
-                    throw this.deniedByPolicy(model, operation, `${count - entities.length} ${(0, pluralize_1.default)('entity', count - entities.length)} failed policy check`);
-                }
-                // TODO: push down schema check to the database
-                const schemaCheckErrors = entities.map((entity) => schema.safeParse(entity)).filter((r) => !r.success);
-                if (schemaCheckErrors.length > 0) {
-                    const error = schemaCheckErrors.map((r) => !r.success && (0, zod_validation_error_1.fromZodError)(r.error).message).join(', ');
-                    if (this.logger.enabled('info')) {
-                        this.logger.info(`entity ${model} failed schema check for operation ${operation}: ${error}`);
-                    }
-                    throw this.deniedByPolicy(model, operation, `entities failed schema check: [${error}]`, constants_1.CrudFailureReason.DATA_VALIDATION_VIOLATION);
-                }
-            }
-            else {
-                // count entities with policy injected and see if any of them are filtered out
-                // if (this.shouldLogQuery) {
-                //     this.logger.info(
-                //         `[withPolicy] \`count\` for policy check with guard:\n${formatObject(guardedQuery)}`
-                //     );
-                // }
-                const guardedCount = (yield db[model].count(guardedQuery));
-                if (guardedCount < count) {
-                    if (this.logger.enabled('info')) {
-                        this.logger.info(`entity ${model} failed policy check for operation ${operation}`);
-                    }
-                    throw this.deniedByPolicy(model, operation, `${count - guardedCount} ${(0, pluralize_1.default)('entity', count - guardedCount)} failed policy check`);
-                }
-            }
-        });
-    }
-    checkPostUpdate(model, ids, db, preValue) {
-        return __awaiter(this, void 0, void 0, function* () {
-            // if (this.logger.enabled('info')) {
-            //     this.logger.info(`Checking post-update policy for ${model}#${ids}, preValue: ${formatObject(preValue)}`);
-            // }
-            const guard = this.getAuthGuard(model, 'postUpdate', preValue);
-            // build a query condition with policy injected
-            const guardedQuery = { where: this.and(ids, guard) };
-            // query with policy injected
-            const entity = yield db[model].findFirst(guardedQuery);
-            // see if we get fewer items with policy, if so, reject with an throw
-            if (!entity) {
-                if (this.logger.enabled('info')) {
-                    this.logger.info(`entity ${model} failed policy check for operation postUpdate`);
+            // strip auxiliary fields
+            for (const auxField of constants_1.AUXILIARY_FIELDS) {
+                if (auxField in entityData) {
+                    delete entityData[auxField];
                 }
-                throw this.deniedByPolicy(model, 'postUpdate');
             }
-            // TODO: push down schema check to the database
-            const schema = this.getModelSchema(model);
-            if (schema) {
-                const schemaCheckResult = schema.safeParse(entity);
-                if (!schemaCheckResult.success) {
-                    const error = (0, zod_validation_error_1.fromZodError)(schemaCheckResult.error).message;
-                    if (this.logger.enabled('info')) {
-                        this.logger.info(`entity ${model} failed schema check for operation postUpdate: ${error}`);
-                    }
-                    throw this.deniedByPolicy(model, 'postUpdate', `entity failed schema check: ${error}`);
+            for (const fieldData of Object.values(entityData)) {
+                if (typeof fieldData !== 'object' || !fieldData) {
+                    continue;
                 }
+                this.postProcessForRead(fieldData);
             }
-        });
+        }
     }
-    isToOneRelation(field) {
-        return !!field && field.isDataModel && !field.isArray;
+    /**
+     * Gets information for a specific model field.
+     */
+    getModelField(model, field) {
+        var _a;
+        model = (0, lower_case_first_1.lowerCaseFirst)(model);
+        return (_a = this.modelMeta.fields[model]) === null || _a === void 0 ? void 0 : _a[field];
     }
     /**
      * Clones an object and makes sure it's not empty.
@@ -856,13 +653,13 @@ class PolicyUtil {
         return value ? (0, deepcopy_1.default)(value) : {};
     }
     /**
-     * Gets "id" field for a given model.
+     * Gets "id" fields for a given model.
      */
     getIdFields(model) {
         return (0, utils_1.getIdFields)(this.modelMeta, model, true);
     }
     /**
-     * Gets id field value from an entity.
+     * Gets id field values from an entity.
      */
     getEntityIds(model, entityData) {
         const idFields = this.getIdFields(model);
@@ -872,8 +669,12 @@ class PolicyUtil {
         }
         return result;
     }
-    get shouldLogQuery() {
-        return this.logPrismaQuery && this.logger.enabled('info');
+    /**
+     * Creates a selection object for id fields for the given model.
+     */
+    makeIdSelection(model) {
+        const idFields = this.getIdFields(model);
+        return Object.assign({}, ...idFields.map((f) => ({ [f.name]: true })));
     }
 }
 exports.PolicyUtil = PolicyUtil;