@danielhritcu/zenstack-orm 3.5.13 → 3.5.15

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 ZenStack
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/dist/index.cjs

@@ -3251,6 +3251,7 @@ var BaseOperationHandler = class {
       if (vr.kind === "filtered") {
         const sourceField = vr.relation;
         const isSelfReferencing = field === sourceField;
+        const { value: rewrittenValue, plan: nestedPlan } = this.rewriteNestedVirtualRelationArgs(vr.targetModel, value, virtualRelations);
         if (isSelfReferencing) {
           plans.push({
             kind: "filtered",
@@ -3259,11 +3260,12 @@ var BaseOperationHandler = class {
             sourceModel: vr.targetModel,
             single: vr.single,
             discriminatorWhere: vr.where,
-            introducedSourceField: false
+            introducedSourceField: false,
+            nestedPlan
           });
         } else {
-          const nestedArgs = value === true ? {} : {
-            ...value
+          const nestedArgs = rewrittenValue === true ? {} : {
+            ...rewrittenValue
           };
           const where = nestedArgs.where ? {
             ...nestedArgs.where,
@@ -3284,15 +3286,17 @@ var BaseOperationHandler = class {
             sourceField,
             sourceModel: vr.targetModel,
             single: vr.single,
-            introducedSourceField: selection[sourceField] === void 0
+            introducedSourceField: selection[sourceField] === void 0,
+            nestedPlan
           });
         }
       } else if (vr.kind === "through") {
         const path = vr.path;
         const [sourceField, ...restPath] = path;
         if (!sourceField) continue;
-        let throughValue = value === true ? true : {
-          ...value
+        const { value: rewrittenValue, plan: nestedPlan } = this.rewriteNestedVirtualRelationArgs(vr.targetModel, value, virtualRelations);
+        let throughValue = rewrittenValue === true ? true : {
+          ...rewrittenValue
         };
         for (let i = restPath.length - 1; i >= 0; i--) {
           throughValue = {
@@ -3309,7 +3313,27 @@ var BaseOperationHandler = class {
           sourceField,
           sourceModel: vr.targetModel,
           path: restPath,
-          introducedSourceField: selection[sourceField] === void 0
+          introducedSourceField: selection[sourceField] === void 0,
+          nestedPlan
+        });
+      }
+    }
+    for (const [field, value] of Object.entries(nextSelection)) {
+      const fieldDef = getField(this.schema, model, field);
+      if (!fieldDef?.relation || value === true || !value || typeof value !== "object") {
+        continue;
+      }
+      const { value: rewrittenValue, plan: nestedPlan } = this.rewriteNestedVirtualRelationArgs(fieldDef.type, value, virtualRelations);
+      if (rewrittenValue !== value) {
+        nextSelection[field] = rewrittenValue;
+      }
+      if (nestedPlan) {
+        plans.push({
+          kind: "actual",
+          field,
+          sourceField: field,
+          sourceModel: fieldDef.type,
+          nestedPlan
         });
       }
     }
@@ -3320,6 +3344,46 @@ var BaseOperationHandler = class {
       } : void 0
     };
   }
+  rewriteNestedVirtualRelationArgs(model, value, virtualRelations) {
+    if (value === true || !value || typeof value !== "object" || Array.isArray(value)) {
+      return {
+        value
+      };
+    }
+    let changed = false;
+    const nextValue = {
+      ...value
+    };
+    let combinedPlan;
+    const mergePlan = /* @__PURE__ */ __name((plan) => {
+      if (!plan) return;
+      if (!combinedPlan) {
+        combinedPlan = {
+          selections: [
+            ...plan.selections
+          ]
+        };
+      } else {
+        combinedPlan.selections.push(...plan.selections);
+      }
+    }, "mergePlan");
+    if (nextValue["select"] && typeof nextValue["select"] === "object") {
+      const { selection, plan } = this.rewriteSelectionMap(model, nextValue["select"], virtualRelations);
+      nextValue["select"] = selection;
+      changed = true;
+      mergePlan(plan);
+    }
+    if (nextValue["include"] && typeof nextValue["include"] === "object") {
+      const { selection, plan } = this.rewriteSelectionMap(model, nextValue["include"], virtualRelations);
+      nextValue["include"] = selection;
+      changed = true;
+      mergePlan(plan);
+    }
+    return {
+      value: changed ? nextValue : value,
+      plan: combinedPlan
+    };
+  }
   applyVirtualRelationPlan(data, plan) {
     if (!plan || data == null) return data;
     if (Array.isArray(data)) return data.map((item) => this.applyVirtualRelationPlan(item, plan));
@@ -3329,15 +3393,24 @@ var BaseOperationHandler = class {
     };
     for (const sel of plan.selections) {
       const sourceValue = this.normalizeVirtualRelationValue(result[sel.sourceField]);
-      if (sel.kind === "filtered") {
+      if (sel.kind === "actual") {
+        result[sel.field] = sel.nestedPlan ? this.applyVirtualRelationPlan(sourceValue, sel.nestedPlan) : sourceValue;
+      } else if (sel.kind === "filtered") {
         let value = sel.single ? Array.isArray(sourceValue) ? sourceValue[0] ?? null : sourceValue ?? null : sourceValue;
         if (sel.discriminatorWhere && value != null) {
           const matches = Object.entries(sel.discriminatorWhere).every(([key, val]) => result[key] === val);
           if (!matches) value = null;
         }
+        if (sel.nestedPlan) {
+          value = this.applyVirtualRelationPlan(value, sel.nestedPlan);
+        }
         result[sel.field] = value;
       } else if (sel.kind === "through") {
-        result[sel.field] = this.extractThroughValue(sourceValue, sel.path ?? []);
+        let value = this.extractThroughValue(sourceValue, sel.path ?? []);
+        if (sel.nestedPlan) {
+          value = this.applyVirtualRelationPlan(value, sel.nestedPlan);
+        }
+        result[sel.field] = value;
       }
       if (sel.introducedSourceField) {
         delete result[sel.sourceField];