@zenstackhq/runtime 3.0.0-alpha.30 → 3.0.0-alpha.31

package/dist/index.d.cts

@@ -1,5 +1,5 @@
-import { C as ClientConstructor } from './contract-DUtKB39c.cjs';
-export { K as AggregateArgs, L as AggregateResult, B as BatchResult, i as BooleanFilter, h as BytesFilter, c as ClientContract, d as ClientOptions, G as CountArgs, H as CountResult, u as CreateArgs, w as CreateManyAndReturnArgs, v as CreateManyArgs, D as DateTimeFilter, A as DeleteArgs, E as DeleteManyArgs, F as FindArgs, s as FindFirstArgs, r as FindManyArgs, t as FindUniqueArgs, P as GroupByArgs, Q as GroupByResult, I as IncludeInput, J as JsonArray, a as JsonObject, b as JsonValue, M as ModelResult, k as NullsOrder, N as NumberFilter, m as OmitInput, O as OrderBy, n as SelectIncludeOmit, o as SelectInput, q as SelectSubset, S as SimplifiedModelResult, j as SortOrder, g as StringFilter, p as Subset, T as ToKysely, f as TypeDefResult, U as UpdateArgs, y as UpdateManyAndReturnArgs, x as UpdateManyArgs, z as UpsertArgs, W as WhereInput, l as WhereUniqueInput, Z as ZenStackPromise, e as definePlugin } from './contract-DUtKB39c.cjs';
+import { C as ClientConstructor } from './contract-5Wcmlo5v.cjs';
+export { K as AggregateArgs, L as AggregateResult, B as BatchResult, i as BooleanFilter, h as BytesFilter, c as ClientContract, d as ClientOptions, G as CountArgs, H as CountResult, u as CreateArgs, w as CreateManyAndReturnArgs, v as CreateManyArgs, D as DateTimeFilter, A as DeleteArgs, E as DeleteManyArgs, F as FindArgs, s as FindFirstArgs, r as FindManyArgs, t as FindUniqueArgs, P as GroupByArgs, Q as GroupByResult, I as IncludeInput, J as JsonArray, a as JsonObject, b as JsonValue, M as ModelResult, k as NullsOrder, N as NumberFilter, m as OmitInput, O as OrderBy, n as SelectIncludeOmit, o as SelectInput, q as SelectSubset, S as SimplifiedModelResult, j as SortOrder, g as StringFilter, p as Subset, T as ToKysely, f as TypeDefResult, U as UpdateArgs, y as UpdateManyAndReturnArgs, x as UpdateManyArgs, z as UpsertArgs, W as WhereInput, l as WhereUniqueInput, Z as ZenStackPromise, e as definePlugin } from './contract-5Wcmlo5v.cjs';
 export { sql } from 'kysely';
 import 'decimal.js';
 import '@zenstackhq/sdk/schema';

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-import { C as ClientConstructor } from './contract-DUtKB39c.js';
-export { K as AggregateArgs, L as AggregateResult, B as BatchResult, i as BooleanFilter, h as BytesFilter, c as ClientContract, d as ClientOptions, G as CountArgs, H as CountResult, u as CreateArgs, w as CreateManyAndReturnArgs, v as CreateManyArgs, D as DateTimeFilter, A as DeleteArgs, E as DeleteManyArgs, F as FindArgs, s as FindFirstArgs, r as FindManyArgs, t as FindUniqueArgs, P as GroupByArgs, Q as GroupByResult, I as IncludeInput, J as JsonArray, a as JsonObject, b as JsonValue, M as ModelResult, k as NullsOrder, N as NumberFilter, m as OmitInput, O as OrderBy, n as SelectIncludeOmit, o as SelectInput, q as SelectSubset, S as SimplifiedModelResult, j as SortOrder, g as StringFilter, p as Subset, T as ToKysely, f as TypeDefResult, U as UpdateArgs, y as UpdateManyAndReturnArgs, x as UpdateManyArgs, z as UpsertArgs, W as WhereInput, l as WhereUniqueInput, Z as ZenStackPromise, e as definePlugin } from './contract-DUtKB39c.js';
+import { C as ClientConstructor } from './contract-5Wcmlo5v.js';
+export { K as AggregateArgs, L as AggregateResult, B as BatchResult, i as BooleanFilter, h as BytesFilter, c as ClientContract, d as ClientOptions, G as CountArgs, H as CountResult, u as CreateArgs, w as CreateManyAndReturnArgs, v as CreateManyArgs, D as DateTimeFilter, A as DeleteArgs, E as DeleteManyArgs, F as FindArgs, s as FindFirstArgs, r as FindManyArgs, t as FindUniqueArgs, P as GroupByArgs, Q as GroupByResult, I as IncludeInput, J as JsonArray, a as JsonObject, b as JsonValue, M as ModelResult, k as NullsOrder, N as NumberFilter, m as OmitInput, O as OrderBy, n as SelectIncludeOmit, o as SelectInput, q as SelectSubset, S as SimplifiedModelResult, j as SortOrder, g as StringFilter, p as Subset, T as ToKysely, f as TypeDefResult, U as UpdateArgs, y as UpdateManyAndReturnArgs, x as UpdateManyArgs, z as UpsertArgs, W as WhereInput, l as WhereUniqueInput, Z as ZenStackPromise, e as definePlugin } from './contract-5Wcmlo5v.js';
 export { sql } from 'kysely';
 import 'decimal.js';
 import '@zenstackhq/sdk/schema';

package/dist/index.js

@@ -6,7 +6,7 @@ var __export = (target, all) => {
 };
 
 // src/client/client-impl.ts
-import { invariant as invariant12 } from "@zenstackhq/common-helpers";
+import { invariant as invariant13 } from "@zenstackhq/common-helpers";
 import { CompiledQuery, DefaultConnectionProvider, DefaultQueryExecutor as DefaultQueryExecutor2, Kysely, Log, sql as sql9 } from "kysely";
 
 // src/client/crud/operations/aggregate.ts
@@ -5350,10 +5350,11 @@ var InputValidator = class {
   makeGroupBySchema(model) {
     const modelDef = requireModel(this.schema, model);
     const nonRelationFields = Object.keys(modelDef.fields).filter((field) => !modelDef.fields[field]?.relation);
+    const bySchema = nonRelationFields.length > 0 ? this.orArray(z.enum(nonRelationFields), true) : z.never();
     let schema = z.strictObject({
       where: this.makeWhereSchema(model, false).optional(),
       orderBy: this.orArray(this.makeOrderBySchema(model, false, true), true).optional(),
-      by: this.orArray(z.enum(nonRelationFields), true),
+      by: bySchema,
       having: this.makeHavingSchema(model).optional(),
       skip: this.makeSkipSchema().optional(),
       take: this.makeTakeSchema().optional(),
@@ -5625,12 +5626,13 @@ function performanceNow() {
 __name(performanceNow, "performanceNow");
 
 // src/client/executor/zenstack-query-executor.ts
-import { AndNode as AndNode2, DefaultQueryExecutor, DeleteQueryNode as DeleteQueryNode2, InsertQueryNode as InsertQueryNode2, ReturningNode as ReturningNode3, SelectionNode as SelectionNode4, UpdateQueryNode as UpdateQueryNode2, WhereNode as WhereNode3 } from "kysely";
+import { AndNode as AndNode2, DefaultQueryExecutor, DeleteQueryNode as DeleteQueryNode2, InsertQueryNode as InsertQueryNode2, ReturningNode as ReturningNode2, SelectionNode as SelectionNode4, UpdateQueryNode as UpdateQueryNode2, WhereNode as WhereNode3 } from "kysely";
 import { nanoid as nanoid2 } from "nanoid";
 import { match as match16 } from "ts-pattern";
 
 // src/client/executor/name-mapper.ts
-import { AliasNode as AliasNode4, ColumnNode as ColumnNode3, IdentifierNode as IdentifierNode3, OperationNodeTransformer as OperationNodeTransformer2, ReferenceNode as ReferenceNode3, ReturningNode as ReturningNode2, SelectAllNode, SelectionNode as SelectionNode3, TableNode as TableNode4 } from "kysely";
+import { invariant as invariant9 } from "@zenstackhq/common-helpers";
+import { AliasNode as AliasNode4, ColumnNode as ColumnNode3, FromNode as FromNode3, IdentifierNode as IdentifierNode3, OperationNodeTransformer as OperationNodeTransformer2, ReferenceNode as ReferenceNode3, SelectAllNode, SelectionNode as SelectionNode3, TableNode as TableNode4 } from "kysely";
 var QueryNameMapper = class extends OperationNodeTransformer2 {
   static {
     __name(this, "QueryNameMapper");
@@ -5638,7 +5640,7 @@ var QueryNameMapper = class extends OperationNodeTransformer2 {
   schema;
   modelToTableMap = /* @__PURE__ */ new Map();
   fieldToColumnMap = /* @__PURE__ */ new Map();
-  modelStack = [];
+  modelScopes = [];
   constructor(schema) {
     super(), this.schema = schema;
     for (const [modelName, modelDef] of Object.entries(schema.models)) {
@@ -5654,150 +5656,182 @@ var QueryNameMapper = class extends OperationNodeTransformer2 {
       }
     }
   }
-  get currentModel() {
-    return this.modelStack[this.modelStack.length - 1];
-  }
-  transformCreateTable(node) {
-    try {
-      this.modelStack.push(node.table.table.identifier.name);
-      return super.transformCreateTable(node);
-    } finally {
-      this.modelStack.pop();
+  // #region overrides
+  transformSelectQuery(node) {
+    if (!node.from?.froms) {
+      return super.transformSelectQuery(node);
     }
+    const scopes = this.createScopesFromFroms(node.from, true);
+    return this.withScopes(scopes, () => {
+      return {
+        ...super.transformSelectQuery(node),
+        // convert "from" to nested query as needed
+        from: this.processFrom(node.from)
+      };
+    });
   }
   transformInsertQuery(node) {
-    try {
-      if (node.into?.table.identifier.name) {
-        this.modelStack.push(node.into.table.identifier.name);
-      }
+    if (!node.into) {
       return super.transformInsertQuery(node);
-    } finally {
-      if (node.into?.table.identifier.name) {
-        this.modelStack.pop();
-      }
     }
+    return this.withScope({
+      model: node.into.table.identifier.name
+    }, () => ({
+      ...super.transformInsertQuery(node),
+      // map table name
+      into: this.processTableRef(node.into)
+    }));
   }
   transformReturning(node) {
-    return ReturningNode2.create(this.transformSelections(node.selections, node));
+    return {
+      kind: node.kind,
+      // map column names in returning selections (include returningAll)
+      selections: this.processSelections(node.selections)
+    };
   }
-  transformUpdateQuery(node) {
-    let pushed = false;
-    if (node.table && TableNode4.is(node.table)) {
-      this.modelStack.push(node.table.table.identifier.name);
-      pushed = true;
-    }
-    try {
-      return super.transformUpdateQuery(node);
-    } finally {
-      if (pushed) {
-        this.modelStack.pop();
+  transformJoin(node) {
+    const { alias, node: innerNode } = this.stripAlias(node.table);
+    if (TableNode4.is(innerNode)) {
+      const modelName = innerNode.table.identifier.name;
+      if (this.hasMappedColumns(modelName)) {
+        const select = this.createSelectAll(modelName);
+        return {
+          ...super.transformJoin(node),
+          table: this.wrapAlias(select, alias ?? modelName)
+        };
       }
     }
+    return super.transformJoin(node);
   }
-  transformDeleteQuery(node) {
-    let pushed = false;
-    if (node.from?.froms && node.from.froms.length === 1 && node.from.froms[0]) {
-      const from = node.from.froms[0];
-      if (TableNode4.is(from)) {
-        this.modelStack.push(from.table.identifier.name);
-        pushed = true;
-      } else if (AliasNode4.is(from) && TableNode4.is(from.node)) {
-        this.modelStack.push(from.node.table.identifier.name);
-        pushed = true;
-      }
+  transformReference(node) {
+    if (!ColumnNode3.is(node.column)) {
+      return super.transformReference(node);
     }
-    try {
-      return super.transformDeleteQuery(node);
-    } finally {
-      if (pushed) {
-        this.modelStack.pop();
+    const { fieldDef, modelDef, scope } = this.resolveFieldFromScopes(node.column.column.name, node.table?.table.identifier.name);
+    if (fieldDef && !scope.namesMapped) {
+      const mappedFieldName = this.mapFieldName(modelDef.name, fieldDef.name);
+      let mappedTableName = node.table?.table.identifier.name;
+      if (mappedTableName) {
+        if (scope.alias === mappedTableName) {
+        } else if (scope.model === mappedTableName) {
+          mappedTableName = this.mapTableName(scope.model);
+        }
       }
+      return ReferenceNode3.create(ColumnNode3.create(mappedFieldName), mappedTableName ? TableNode4.create(mappedTableName) : void 0);
+    } else {
+      return super.transformReference(node);
     }
   }
-  transformSelectQuery(node) {
-    if (!node.from?.froms || node.from.froms.length === 0) {
-      return super.transformSelectQuery(node);
-    }
-    if (node.from.froms.length > 1) {
-      throw new InternalError(`SelectQueryNode must have a single table in from clause`);
+  transformColumn(node) {
+    const { modelDef, fieldDef, scope } = this.resolveFieldFromScopes(node.column.name);
+    if (!fieldDef || scope.namesMapped) {
+      return super.transformColumn(node);
     }
-    let pushed = false;
-    const from = node.from.froms[0];
-    if (TableNode4.is(from)) {
-      this.modelStack.push(from.table.identifier.name);
-      pushed = true;
-    } else if (AliasNode4.is(from) && TableNode4.is(from.node)) {
-      this.modelStack.push(from.node.table.identifier.name);
-      pushed = true;
+    const mappedName = this.mapFieldName(modelDef.name, fieldDef.name);
+    return ColumnNode3.create(mappedName);
+  }
+  transformUpdateQuery(node) {
+    const { alias, node: innerTable } = this.stripAlias(node.table);
+    if (!innerTable || !TableNode4.is(innerTable)) {
+      return super.transformUpdateQuery(node);
     }
-    const selections = node.selections ? this.transformSelections(node.selections, node) : node.selections;
-    try {
+    return this.withScope({
+      model: innerTable.table.identifier.name,
+      alias
+    }, () => {
       return {
-        ...super.transformSelectQuery(node),
-        selections
+        ...super.transformUpdateQuery(node),
+        // map table name
+        table: this.wrapAlias(this.processTableRef(innerTable), alias)
       };
-    } finally {
-      if (pushed) {
-        this.modelStack.pop();
+    });
+  }
+  transformDeleteQuery(node) {
+    const scopes = this.createScopesFromFroms(node.from, false);
+    const froms = node.from.froms.map((from) => {
+      const { alias, node: innerNode } = this.stripAlias(from);
+      if (TableNode4.is(innerNode)) {
+        return this.wrapAlias(this.processTableRef(innerNode), alias);
+      } else {
+        return super.transformNode(from);
       }
-    }
+    });
+    return this.withScopes(scopes, () => {
+      return {
+        ...super.transformDeleteQuery(node),
+        from: FromNode3.create(froms)
+      };
+    });
   }
-  transformSelections(selections, contextNode) {
-    const result = [];
-    for (const selection of selections) {
-      let selectAllFromModel = void 0;
-      let isSelectAll = false;
-      if (SelectAllNode.is(selection.selection)) {
-        selectAllFromModel = this.currentModel;
-        isSelectAll = true;
-      } else if (ReferenceNode3.is(selection.selection) && SelectAllNode.is(selection.selection.column)) {
-        selectAllFromModel = selection.selection.table?.table.identifier.name ?? this.currentModel;
-        isSelectAll = true;
-      }
-      if (isSelectAll) {
-        if (!selectAllFromModel) {
-          continue;
+  // #endregion
+  // #region utils
+  resolveFieldFromScopes(name, qualifier) {
+    for (const scope of this.modelScopes.toReversed()) {
+      if (qualifier) {
+        if (scope.alias) {
+          if (qualifier !== scope.alias) {
+            continue;
+          }
         } else {
-          const scalarFields = this.getModelScalarFields(contextNode, selectAllFromModel);
-          const fromModelDef = requireModel(this.schema, selectAllFromModel);
-          const mappedTableName = this.getMappedName(fromModelDef) ?? selectAllFromModel;
-          result.push(...scalarFields.map((fieldName) => {
-            const fieldRef = ReferenceNode3.create(ColumnNode3.create(this.mapFieldName(fieldName)), TableNode4.create(mappedTableName));
-            return SelectionNode3.create(this.fieldHasMappedName(fieldName) ? AliasNode4.create(fieldRef, IdentifierNode3.create(fieldName)) : fieldRef);
-          }));
+          if (qualifier !== scope.model) {
+            continue;
+          }
         }
-      } else {
-        result.push(this.transformSelectionWithAlias(selection));
+      }
+      const modelDef = getModel(this.schema, scope.model);
+      if (!modelDef) {
+        continue;
+      }
+      if (modelDef.fields[name]) {
+        return {
+          modelDef,
+          fieldDef: modelDef.fields[name],
+          scope
+        };
       }
     }
-    return result;
+    return {
+      modelDef: void 0,
+      fieldDef: void 0,
+      scope: void 0
+    };
   }
-  transformSelectionWithAlias(node) {
-    if (ColumnNode3.is(node.selection) && this.fieldHasMappedName(node.selection.column.name)) {
-      return SelectionNode3.create(AliasNode4.create(this.transformColumn(node.selection), IdentifierNode3.create(node.selection.column.name)));
-    } else if (ReferenceNode3.is(node.selection) && this.fieldHasMappedName(node.selection.column.column.name)) {
-      return SelectionNode3.create(AliasNode4.create(this.transformReference(node.selection), IdentifierNode3.create(node.selection.column.column.name)));
-    } else {
-      return this.transformSelection(node);
+  pushScope(scope) {
+    this.modelScopes.push(scope);
+  }
+  withScope(scope, fn) {
+    this.pushScope(scope);
+    try {
+      return fn();
+    } finally {
+      this.modelScopes.pop();
     }
   }
-  fieldHasMappedName(name) {
-    if (!this.currentModel) {
-      return false;
+  withScopes(scopes, fn) {
+    scopes.forEach((s) => this.pushScope(s));
+    try {
+      return fn();
+    } finally {
+      scopes.forEach(() => this.modelScopes.pop());
     }
-    return this.fieldToColumnMap.has(`${this.currentModel}.${name}`);
   }
-  transformTable(node) {
-    const tableName = node.table.identifier.name;
-    const mappedName = this.modelToTableMap.get(tableName);
-    if (mappedName) {
-      return TableNode4.create(mappedName);
-    } else {
+  wrapAlias(node, alias) {
+    return alias ? AliasNode4.create(node, IdentifierNode3.create(alias)) : node;
+  }
+  ensureAlias(node, alias, fallbackName) {
+    if (!node) {
       return node;
     }
+    return alias ? AliasNode4.create(node, IdentifierNode3.create(alias)) : AliasNode4.create(node, IdentifierNode3.create(fallbackName));
   }
-  transformColumn(node) {
-    return ColumnNode3.create(this.mapFieldName(node.column.name));
+  processTableRef(node) {
+    if (!node) {
+      return node;
+    }
+    if (!TableNode4.is(node)) {
+      return super.transformNode(node);
+    }
+    return TableNode4.create(this.mapTableName(node.table.identifier.name));
   }
   getMappedName(def) {
     const mapAttr = def.attributes?.find((attr) => attr.name === "@@map" || attr.name === "@map");
@@ -5809,28 +5843,151 @@ var QueryNameMapper = class extends OperationNodeTransformer2 {
     }
     return void 0;
   }
-  mapFieldName(fieldName) {
-    if (!this.currentModel) {
-      return fieldName;
+  mapFieldName(model, field) {
+    const mappedName = this.fieldToColumnMap.get(`${model}.${field}`);
+    if (mappedName) {
+      return mappedName;
+    } else {
+      return field;
     }
-    const mappedName = this.fieldToColumnMap.get(`${this.currentModel}.${fieldName}`);
+  }
+  mapTableName(tableName) {
+    const mappedName = this.modelToTableMap.get(tableName);
     if (mappedName) {
       return mappedName;
     } else {
-      return fieldName;
+      return tableName;
     }
   }
-  requireCurrentModel(node) {
-    if (!this.currentModel) {
-      throw new InternalError(`Missing model context for "${node}"`);
+  stripAlias(node) {
+    if (!node) {
+      return {
+        alias: void 0,
+        node
+      };
     }
+    if (AliasNode4.is(node)) {
+      invariant9(IdentifierNode3.is(node.alias), "Expected identifier as alias");
+      return {
+        alias: node.alias.name,
+        node: node.node
+      };
+    }
+    return {
+      alias: void 0,
+      node
+    };
   }
-  getModelScalarFields(contextNode, model) {
-    this.requireCurrentModel(contextNode);
-    model = model ?? this.currentModel;
+  hasMappedColumns(modelName) {
+    return [
+      ...this.fieldToColumnMap.keys()
+    ].some((key) => key.startsWith(modelName + "."));
+  }
+  createScopesFromFroms(node, namesMapped) {
+    if (!node) {
+      return [];
+    }
+    return node.froms.map((from) => {
+      const { alias, node: innerNode } = this.stripAlias(from);
+      if (innerNode && TableNode4.is(innerNode)) {
+        return {
+          model: innerNode.table.identifier.name,
+          alias,
+          namesMapped
+        };
+      } else {
+        return void 0;
+      }
+    }).filter((s) => !!s);
+  }
+  // convert a "from" node to a nested query if there are columns with name mapping
+  processFrom(node) {
+    return {
+      ...super.transformFrom(node),
+      froms: node.froms.map((from) => {
+        const { alias, node: innerNode } = this.stripAlias(from);
+        if (!innerNode) {
+          return super.transformNode(from);
+        }
+        if (TableNode4.is(innerNode)) {
+          if (this.hasMappedColumns(innerNode.table.identifier.name)) {
+            const selectAll = this.createSelectAll(innerNode.table.identifier.name);
+            return this.ensureAlias(selectAll, alias, innerNode.table.identifier.name);
+          }
+        }
+        return this.transformNode(from);
+      })
+    };
+  }
+  // create a `SelectQueryNode` for the given model with all columns mapped
+  createSelectAll(model) {
     const modelDef = requireModel(this.schema, model);
-    const scalarFields = Object.entries(modelDef.fields).filter(([, fieldDef]) => !fieldDef.relation && !fieldDef.computed && !fieldDef.originModel).map(([fieldName]) => fieldName);
-    return scalarFields;
+    const tableName = this.mapTableName(model);
+    return {
+      kind: "SelectQueryNode",
+      from: FromNode3.create([
+        TableNode4.create(tableName)
+      ]),
+      selections: this.getModelFields(modelDef).map((fieldDef) => {
+        const columnName = this.mapFieldName(model, fieldDef.name);
+        const columnRef = ReferenceNode3.create(ColumnNode3.create(columnName), TableNode4.create(tableName));
+        if (columnName !== fieldDef.name) {
+          const aliased = AliasNode4.create(columnRef, IdentifierNode3.create(fieldDef.name));
+          return SelectionNode3.create(aliased);
+        } else {
+          return SelectionNode3.create(columnRef);
+        }
+      })
+    };
+  }
+  getModelFields(modelDef) {
+    return Object.values(modelDef.fields).filter((f) => !f.relation && !f.computed && !f.originModel);
+  }
+  processSelections(selections) {
+    const result = [];
+    selections.forEach((selection) => {
+      if (SelectAllNode.is(selection.selection)) {
+        const processed = this.processSelectAll(selection.selection);
+        if (Array.isArray(processed)) {
+          result.push(...processed.map((s) => SelectionNode3.create(s)));
+        } else {
+          result.push(SelectionNode3.create(processed));
+        }
+      } else {
+        result.push(SelectionNode3.create(this.processSelection(selection.selection)));
+      }
+    });
+    return result;
+  }
+  processSelection(node) {
+    let alias;
+    if (!AliasNode4.is(node)) {
+      alias = this.extractFieldName(node);
+    }
+    const result = super.transformNode(node);
+    return this.wrapAlias(result, alias);
+  }
+  processSelectAll(node) {
+    const scope = this.modelScopes[this.modelScopes.length - 1];
+    invariant9(scope);
+    if (!this.hasMappedColumns(scope.model)) {
+      return super.transformSelectAll(node);
+    }
+    const modelDef = requireModel(this.schema, scope.model);
+    return this.getModelFields(modelDef).map((fieldDef) => {
+      const columnName = this.mapFieldName(scope.model, fieldDef.name);
+      const columnRef = ReferenceNode3.create(ColumnNode3.create(columnName));
+      return columnName !== fieldDef.name ? this.wrapAlias(columnRef, fieldDef.name) : columnRef;
+    });
+  }
+  extractFieldName(node) {
+    if (ReferenceNode3.is(node) && ColumnNode3.is(node.column)) {
+      return node.column.column.name;
+    } else if (ColumnNode3.is(node)) {
+      return node.column.name;
+    } else {
+      return void 0;
+    }
   }
 };
 
@@ -5868,7 +6025,7 @@ var ZenStackQueryExecutor = class _ZenStackQueryExecutor extends DefaultQueryExe
       if ((InsertQueryNode2.is(queryNode) || UpdateQueryNode2.is(queryNode)) && mutationInterceptionInfo?.loadAfterMutationEntities) {
         queryNode = {
           ...queryNode,
-          returning: ReturningNode3.create([
+          returning: ReturningNode2.create([
             SelectionNode4.createSelectAll()
           ])
         };
@@ -6131,7 +6288,7 @@ __export(functions_exports, {
   search: () => search,
   startsWith: () => startsWith
 });
-import { invariant as invariant9, lowerCaseFirst, upperCaseFirst } from "@zenstackhq/common-helpers";
+import { invariant as invariant10, lowerCaseFirst, upperCaseFirst } from "@zenstackhq/common-helpers";
 import { sql as sql7, ValueNode as ValueNode4 } from "kysely";
 import { match as match17 } from "ts-pattern";
 var contains = /* @__PURE__ */ __name((eb, args) => {
@@ -6238,7 +6395,7 @@ var currentOperation = /* @__PURE__ */ __name((_eb, args, { operation }) => {
 }, "currentOperation");
 function processCasing(casing, result, model) {
   const opNode = casing.toOperationNode();
-  invariant9(ValueNode4.is(opNode) && typeof opNode.value === "string", '"casting" parameter must be a string value');
+  invariant10(ValueNode4.is(opNode) && typeof opNode.value === "string", '"casting" parameter must be a string value');
   result = match17(opNode.value).with("original", () => model).with("upper", () => result.toUpperCase()).with("lower", () => result.toLowerCase()).with("capitalize", () => upperCaseFirst(result)).with("uncapitalize", () => lowerCaseFirst(result)).otherwise(() => {
     throw new Error(`Invalid casing value: ${opNode.value}. Must be "original", "upper", "lower", "capitalize", or "uncapitalize".`);
   });
@@ -6247,7 +6404,7 @@ function processCasing(casing, result, model) {
 __name(processCasing, "processCasing");
 
 // src/client/helpers/schema-db-pusher.ts
-import { invariant as invariant10 } from "@zenstackhq/common-helpers";
+import { invariant as invariant11 } from "@zenstackhq/common-helpers";
 import { sql as sql8 } from "kysely";
 import toposort from "toposort";
 import { match as match18 } from "ts-pattern";
@@ -6343,7 +6500,7 @@ var SchemaDbPusher = class {
   }
   addUniqueConstraint(table, modelDef) {
     for (const [key, value] of Object.entries(modelDef.uniqueFields)) {
-      invariant10(typeof value === "object", "expecting an object");
+      invariant11(typeof value === "object", "expecting an object");
       if ("type" in value) {
         const fieldDef = modelDef.fields[key];
         if (fieldDef.unique) {
@@ -6411,7 +6568,7 @@ var SchemaDbPusher = class {
     return fieldDef.default && ExpressionUtils.isCall(fieldDef.default) && fieldDef.default.function === "autoincrement";
   }
   addForeignKeyConstraint(table, model, fieldName, fieldDef) {
-    invariant10(fieldDef.relation, "field must be a relation");
+    invariant11(fieldDef.relation, "field must be a relation");
     if (!fieldDef.relation.fields || !fieldDef.relation.references) {
       return table;
     }
@@ -6466,7 +6623,7 @@ function valueToPromise(thing) {
 __name(valueToPromise, "valueToPromise");
 
 // src/client/result-processor.ts
-import { invariant as invariant11 } from "@zenstackhq/common-helpers";
+import { invariant as invariant12 } from "@zenstackhq/common-helpers";
 import Decimal2 from "decimal.js";
 import { match as match19 } from "ts-pattern";
 var ResultProcessor = class {
@@ -6566,14 +6723,14 @@ var ResultProcessor = class {
     if (value instanceof Decimal2) {
       return value;
     }
-    invariant11(typeof value === "string" || typeof value === "number" || value instanceof Decimal2, `Expected string, number or Decimal, got ${typeof value}`);
+    invariant12(typeof value === "string" || typeof value === "number" || value instanceof Decimal2, `Expected string, number or Decimal, got ${typeof value}`);
     return new Decimal2(value);
   }
   transformBigInt(value) {
     if (typeof value === "bigint") {
       return value;
     }
-    invariant11(typeof value === "string" || typeof value === "number", `Expected string or number, got ${typeof value}`);
+    invariant12(typeof value === "string" || typeof value === "number", `Expected string or number, got ${typeof value}`);
     return BigInt(value);
   }
   transformBoolean(value) {
@@ -6617,7 +6774,7 @@ var ResultProcessor = class {
   }
   transformJson(value) {
     return match19(this.schema.provider.type).with("sqlite", () => {
-      invariant11(typeof value === "string", "Expected string, got " + typeof value);
+      invariant12(typeof value === "string", "Expected string, got " + typeof value);
       return JSON.parse(value);
     }).otherwise(() => value);
   }
@@ -6694,7 +6851,7 @@ var ClientImpl = class _ClientImpl {
   }
   // implementation
   async $transaction(input, options) {
-    invariant12(typeof input === "function" || Array.isArray(input) && input.every((p) => p.then && p.cb), "Invalid transaction input, expected a function or an array of ZenStackPromise");
+    invariant13(typeof input === "function" || Array.isArray(input) && input.every((p) => p.then && p.cb), "Invalid transaction input, expected a function or an array of ZenStackPromise");
     if (typeof input === "function") {
       return this.interactiveTransaction(input, options);
     } else {