pqb 0.30.7 → 0.31.1

package/dist/index.js

@@ -3676,15 +3676,16 @@ class AsMethods {
 }
 
 function queryFrom(self, arg) {
-  var _a;
+  var _a, _b;
   const data = self.q;
   if (typeof arg === "string") {
     data.as || (data.as = arg);
+    data.shape = (_a = data.withShapes) == null ? void 0 : _a[arg];
   } else if (orchidCore.isExpression(arg)) {
     data.as || (data.as = "t");
   } else if (Array.isArray(arg)) {
     const { shape } = data;
-    const parsers = (_a = data.parsers) != null ? _a : data.parsers = {};
+    const parsers = (_b = data.parsers) != null ? _b : data.parsers = {};
     for (const item of arg) {
       if (typeof item === "string") {
         const withShape = data.withShapes[item];
@@ -3714,7 +3715,7 @@ function queryFromSql(self, args) {
   data.from = sqlQueryArgsToExpression(args);
   return self;
 }
-class From {
+class FromMethods {
   /**
    * Set the `FROM` value, by default the table name is used.
    *
@@ -4071,20 +4072,21 @@ const havingToSql = (ctx, query, quotedAs) => {
   ).join(" AND ");
 };
 
-const pushWithSql = (ctx, withData) => {
-  if (!withData.length)
+const pushWithSql = (ctx, items) => {
+  if (!items.length)
     return;
   ctx.sql.push(
     "WITH",
-    withData.map((withItem) => {
-      const [name, options, query] = withItem;
+    items.map((item) => {
+      var _a;
       let inner;
-      if (orchidCore.isExpression(query)) {
-        inner = query.toSQL(ctx, `"${name}"`);
+      if (item.q) {
+        inner = makeSQL(item.q, ctx).text;
       } else {
-        inner = makeSQL(query, ctx).text;
+        inner = item.s.toSQL(ctx, `"${item.n}"`);
       }
-      return `${options.recursive ? "RECURSIVE " : ""}"${name}"${options.columns ? `(${options.columns.map((x) => `"${x}"`).join(", ")})` : ""} AS ${options.materialized ? "MATERIALIZED " : options.notMaterialized ? "NOT MATERIALIZED " : ""}(${inner})`;
+      const o = (_a = item.o) != null ? _a : orchidCore.emptyObject;
+      return `${o.recursive ? "RECURSIVE " : ""}"${item.n}"${o.columns ? `(${o.columns.map((x) => `"${x}"`).join(", ")})` : ""} AS ${o.materialized ? "MATERIALIZED " : o.notMaterialized ? "NOT MATERIALIZED " : ""}(${inner})`;
     }).join(", ")
   );
 };
@@ -4403,12 +4405,6 @@ const mergeColumnsSql = (quotedColumns2) => {
 };
 const encodeRow = (ctx, q, QueryClass, row, runtimeDefaults, quotedAs) => {
   const arr = row.map((value) => {
-    if (typeof value === "function") {
-      value = resolveSubQueryCallback(
-        q,
-        value
-      );
-    }
     if (value && typeof value === "object") {
       if (value instanceof orchidCore.Expression) {
         return value.toSQL(ctx, quotedAs);
@@ -4711,6 +4707,7 @@ const toSQL = (table, options) => {
   return !(options == null ? void 0 : options.clearCache) && table.q[toSQLCacheKey] || (table.q[toSQLCacheKey] = makeSQL(table, options));
 };
 const makeSQL = (table, options) => {
+  var _a;
   const query = table.q;
   const sql = [];
   const values = (options == null ? void 0 : options.values) || [];
@@ -4724,24 +4721,21 @@ const makeSQL = (table, options) => {
     pushWithSql(ctx, query.with);
   }
   if (query.type) {
+    const tableName = (_a = table.table) != null ? _a : query.as;
+    if (!tableName)
+      throw new Error(`Table is missing for ${query.type}`);
     if (query.type === "truncate") {
-      if (!table.table)
-        throw new Error("Table is missing for truncate");
-      pushTruncateSql(ctx, table.table, query);
+      pushTruncateSql(ctx, tableName, query);
       return { text: sql.join(" "), values };
     }
     if (query.type === "columnInfo") {
-      if (!table.table)
-        throw new Error("Table is missing for truncate");
       pushColumnInfoSql(ctx, table, query);
       return { text: sql.join(" "), values };
     }
-    if (!table.table)
-      throw new Error(`Table is missing for ${query.type}`);
-    const quotedAs2 = `"${query.as || table.table}"`;
+    const quotedAs2 = `"${query.as || tableName}"`;
     if (query.type === "insert") {
       return {
-        hookSelect: pushInsertSql(ctx, table, query, `"${table.table}"`),
+        hookSelect: pushInsertSql(ctx, table, query, `"${tableName}"`),
         text: sql.join(" "),
         values
       };
@@ -4766,54 +4760,50 @@ const makeSQL = (table, options) => {
     }
   }
   const quotedAs = (query.as || table.table) && `"${query.as || table.table}"`;
-  sql.push("SELECT");
-  if (query.distinct) {
-    pushDistinctSql(ctx, table, query.distinct, quotedAs);
-  }
-  pushSelectSql(ctx, table, query, quotedAs);
-  if (table.table || query.from) {
-    pushFromAndAs(ctx, table, query, quotedAs);
-  }
-  if (query.join) {
-    pushJoinSql(
-      ctx,
-      table,
-      query,
-      quotedAs
-    );
-  }
-  if (query.and || query.or) {
-    pushWhereStatementSql(ctx, table, query, quotedAs);
-  }
-  if (query.group) {
-    const group = query.group.map(
-      (item) => orchidCore.isExpression(item) ? item.toSQL(ctx, quotedAs) : columnToSql(ctx, table.q, table.q.shape, item, quotedAs)
-    );
-    sql.push(`GROUP BY ${group.join(", ")}`);
-  }
-  if (query.having)
-    pushHavingSql(ctx, query, quotedAs);
-  if (query.window) {
-    const window = [];
-    query.window.forEach((item) => {
-      for (const key in item) {
-        window.push(
-          `"${key}" AS ${windowToSql(ctx, query, item[key], quotedAs)}`
-        );
-      }
-    });
-    sql.push(`WINDOW ${window.join(", ")}`);
-  }
   if (query.union) {
-    for (const item of query.union) {
-      let itemSql;
-      if (orchidCore.isExpression(item.arg)) {
-        itemSql = item.arg.toSQL(ctx, quotedAs);
-      } else {
-        const argSql = makeSQL(item.arg, { values });
-        itemSql = argSql.text;
-      }
-      sql.push(`${item.kind} ${item.wrap ? `(${itemSql})` : itemSql}`);
+    sql.push(`(${makeSQL(query.union.b, { values }).text})`);
+    for (const u of query.union.u) {
+      const itemSql = orchidCore.isExpression(u.a) ? u.a.toSQL(ctx, quotedAs) : makeSQL(u.a, { values }).text;
+      sql.push(`${u.k} (${itemSql})`);
+    }
+  } else {
+    sql.push("SELECT");
+    if (query.distinct) {
+      pushDistinctSql(ctx, table, query.distinct, quotedAs);
+    }
+    pushSelectSql(ctx, table, query, quotedAs);
+    if (table.table || query.from) {
+      pushFromAndAs(ctx, table, query, quotedAs);
+    }
+    if (query.join) {
+      pushJoinSql(
+        ctx,
+        table,
+        query,
+        quotedAs
+      );
+    }
+    if (query.and || query.or) {
+      pushWhereStatementSql(ctx, table, query, quotedAs);
+    }
+    if (query.group) {
+      const group = query.group.map(
+        (item) => orchidCore.isExpression(item) ? item.toSQL(ctx, quotedAs) : columnToSql(ctx, table.q, table.q.shape, item, quotedAs)
+      );
+      sql.push(`GROUP BY ${group.join(", ")}`);
+    }
+    if (query.having)
+      pushHavingSql(ctx, query, quotedAs);
+    if (query.window) {
+      const window = [];
+      query.window.forEach((item) => {
+        for (const key in item) {
+          window.push(
+            `"${key}" AS ${windowToSql(ctx, query, item[key], quotedAs)}`
+          );
+        }
+      });
+      sql.push(`WINDOW ${window.join(", ")}`);
     }
   }
   if (query.order) {
@@ -4897,7 +4887,7 @@ const cloneQuery = (q) => {
     if (q.window)
       q.window = q.window.slice(0);
     if (q.union)
-      q.union = q.union.slice(0);
+      q.union = { b: q.union.b, u: q.union.u.slice(0) };
     if (q.order)
       q.order = q.order.slice(0);
   } else if (q.type === "insert") {
@@ -6157,9 +6147,17 @@ const processCreateItem = (q, item, rowIndex, ctx, encoders) => {
         item,
         rowIndex
       );
-    } else if (!ctx.columns.has(key) && (shape[key] && !shape[key].data.computed || shape === anyShape)) {
-      ctx.columns.set(key, ctx.columns.size);
-      encoders[key] = (_c = shape[key]) == null ? void 0 : _c.encodeFn;
+    } else {
+      if (typeof item[key] === "function") {
+        item[key] = resolveSubQueryCallback(
+          q,
+          item[key]
+        );
+      }
+      if (!ctx.columns.has(key) && (shape[key] && !shape[key].data.computed || shape === anyShape)) {
+        ctx.columns.set(key, ctx.columns.size);
+        encoders[key] = (_c = shape[key]) == null ? void 0 : _c.encodeFn;
+      }
     }
   }
 };
@@ -6365,6 +6363,22 @@ class Create {
    * });
    * ```
    *
+   * `create` and `insert` can be used in {@link WithMethods.with} expressions:
+   *
+   * ```ts
+   * db.$queryBuilder
+   *   // create a record in one table
+   *   .with('a', db.table.select('id').create(data))
+   *   // create a record in other table using the first table record id
+   *   .with('b', (q) =>
+   *     db.otherTable.select('id').create({
+   *       ...otherData,
+   *       aId: () => q.from('a').get('id'),
+   *     }),
+   *   )
+   *   .from('b');
+   * ```
+   *
    * @param data - data for the record, may have values, raw SQL, queries, relation operations.
    */
   create(data) {
@@ -6873,6 +6887,19 @@ class Delete {
    * // delete all users who have corresponding profile records:
    * db.table.join(Profile, 'profile.userId', 'user.id').all().delete();
    * ```
+   *
+   * `delete` can be used in {@link WithMethods.with} expressions:
+   *
+   * ```ts
+   * db.$queryBuilder
+   *   // delete a record in one table
+   *   .with('a', db.table.find(1).select('id').delete())
+   *   // delete a record in other table using the first table record id
+   *   .with('b', (q) =>
+   *     db.otherTable.select('id').whereIn('aId', q.from('a').pluck('id')).delete(),
+   *   )
+   *   .from('b');
+   * ```
    */
   // eslint-disable-next-line @typescript-eslint/no-unused-vars
   delete(..._args) {
@@ -8295,6 +8322,11 @@ class MergeQueryMethods {
             a[key] = a[key] ? [...a[key], ...value] : value;
           } else if (mergableObjects[key]) {
             a[key] = a[key] ? __spreadValues$5(__spreadValues$5({}, a[key]), value) : value;
+          } else if (key === "union") {
+            a[key] = a[key] ? {
+              b: a[key].b,
+              u: [...a[key].u, ...value.u]
+            } : value;
           } else {
             a[key] = value;
           }
@@ -8308,196 +8340,225 @@ class MergeQueryMethods {
   }
 }
 
-var __defProp$4 = Object.defineProperty;
-var __defProps$1 = Object.defineProperties;
-var __getOwnPropDescs$1 = Object.getOwnPropertyDescriptors;
-var __getOwnPropSymbols$4 = Object.getOwnPropertySymbols;
-var __hasOwnProp$4 = Object.prototype.hasOwnProperty;
-var __propIsEnum$4 = Object.prototype.propertyIsEnumerable;
-var __defNormalProp$4 = (obj, key, value) => key in obj ? __defProp$4(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __spreadValues$4 = (a, b) => {
-  for (var prop in b || (b = {}))
-    if (__hasOwnProp$4.call(b, prop))
-      __defNormalProp$4(a, prop, b[prop]);
-  if (__getOwnPropSymbols$4)
-    for (var prop of __getOwnPropSymbols$4(b)) {
-      if (__propIsEnum$4.call(b, prop))
-        __defNormalProp$4(a, prop, b[prop]);
-    }
-  return a;
+const _queryUnion = (base, args, k) => {
+  const q = base.baseQuery.clone();
+  const u = args.map(
+    (a) => ({
+      a: typeof a === "function" ? a(q) : a,
+      k
+    })
+  );
+  const union = q.q.union = base.q.union;
+  if (union) {
+    union.u.push(...u);
+  } else {
+    q.q.union = {
+      b: base,
+      u
+    };
+  }
+  return q;
 };
-var __spreadProps$1 = (a, b) => __defProps$1(a, __getOwnPropDescs$1(b));
-class With {
+class Union {
   /**
-   * Add Common Table Expression (CTE) to the query.
+   * Creates a union query, takes one or more queries or SQL expressions.
    *
    * ```ts
-   * import { columnTypes } from 'orchid-orm';
-   * import { NumberColumn } from './number';
-   *
-   * // .with optionally accepts such options:
-   * type WithOptions = {
-   *   // list of columns returned by this WITH statement
-   *   // by default all columns from provided column shape will be included
-   *   // true is for default behavior
-   *   columns?: string[] | boolean;
-   *
-   *   // Adds RECURSIVE keyword:
-   *   recursive?: true;
-   *
-   *   // Adds MATERIALIZED keyword:
-   *   materialized?: true;
-   *
-   *   // Adds NOT MATERIALIZED keyword:
-   *   notMaterialized?: true;
-   * };
-   *
-   * // accepts columns shape and a raw expression:
-   * db.table.with(
-   *   'alias',
-   *   {
-   *     id: columnTypes.integer(),
-   *     name: columnTypes.text(3, 100),
-   *   },
-   *   sql`SELECT id, name FROM "someTable"`,
-   * );
-   *
-   * // accepts query:
-   * db.table.with('alias', db.table.all());
-   *
-   * // accepts a callback for a query builder:
-   * db.table.with('alias', (qb) =>
-   *   qb.select({ one: sql`1`.type((t) => t.integer()) }),
-   * );
-   *
-   * // All mentioned forms can accept options as a second argument:
-   * db.table.with(
-   *   'alias',
-   *   {
-   *     recursive: true,
-   *     materialized: true,
-   *   },
-   *   rawOrQueryOrCallback,
-   * );
+   * // The first query of the union
+   * db.one
+   *   .select('id', 'name')
+   *   // add two more queries to the union
+   *   .union(
+   *     db.two.select('id', 'name'),
+   *     (q = q.sql`SELECT id, name FROM "thirdTable"`),
+   *   )
+   *   // sub-sequent `union` is equivalent to passing multiple queries into a single `union`
+   *   .union(db.three.select('id', 'name'));
    * ```
    *
-   * Defined `WITH` table can be used in `.from` or `.join` with all the type safeness:
+   * `order`, `limit`, `offset` are special, it matters if you place them **before** or **after** the `union`, it also have a meaning to place them before and after.
    *
    * ```ts
-   * db.table.with('alias', db.table.all()).from('alias').select('alias.id');
-   *
-   * db.table
-   *   .with('alias', db.table.all())
-   *   .join('alias', 'alias.id', 'user.id')
-   *   .select('alias.id');
+   * // order, limit, offset are applied ONLY to 'one'
+   * db.one
+   *   .order('x')
+   *   .limit(1)
+   *   .offset(1)
+   *   // 'two' also has order, limit, and offset
+   *   .unionAll(db.two.order('y').limit(2).offset(2))
+   *   // sets order, limit, offset for all records
+   *   .order('z')
+   *   .limit(3)
+   *   .offset(3);
    * ```
    *
-   * @param args - first argument is an alias for this CTE, other arguments can be column shape, query object, or raw SQL.
-   */
-  with(...args) {
-    const q = this.clone();
-    let options = args.length === 3 && !orchidCore.isExpression(args[2]) || args.length === 4 ? args[1] : void 0;
-    const last = args[args.length - 1];
-    const query = typeof last === "function" ? last(q.queryBuilder) : last;
-    const shape = args.length === 4 ? args[2] : orchidCore.isExpression(query) ? args[1] : query.q.shape;
-    if ((options == null ? void 0 : options.columns) === true) {
-      options = __spreadProps$1(__spreadValues$4({}, options), {
-        columns: Object.keys(shape)
-      });
-    }
-    pushQueryValue(q, "with", [args[0], options || orchidCore.emptyObject, query]);
-    return setQueryObjectValue(q, "withShapes", args[0], shape);
-  }
-}
-
-class Union {
-  /**
-   * Creates a union query, taking an array or a list of callbacks, builders, or raw statements to build the union statement, with optional boolean `wrap`.
-   * If the `wrap` parameter is true, the queries will be individually wrapped in parentheses.
+   * Equivalent SQL:
    *
-   * ```ts
-   * SomeTable.select('id', 'name').union(
-   *   [
-   *     OtherTable.select('id', 'name'),
-   *     sql`SELECT id, name FROM "thirdTable"`,
-   *   ],
-   *   true, // optional wrap parameter
-   * );
+   * ```sql
+   * -- both union parts have their own order, limit, offset
+   * ( SELECT * FROM one ORDER x ASC LIMIT 1 OFFSET 1 )
+   * UNION ALL
+   * ( SELECT * FROM two ORDER y ASC LIMIT 2 OFFSET 2 )
+   * -- order, limit, offset of the whole query
+   * ORDER BY z ASC LIMIT 3 OFFSET 3
    * ```
    *
-   * @param args - array of queries or raw SQLs
-   * @param wrap - provide `true` if you want the queries to be wrapped into parentheses
+   * All the listed methods have the same signature, they are only different by SQL keyword:
+   *
+   * - `union` - union of all queries, performs deduplication
+   * - `unionAll` - `union` that allows duplicated rows
+   * - `intersect` - get only rows that are present in all queries
+   * - `intersectAll` - `intersect` that allows duplicated rows
+   * - `except` - get only rows that are in the first query but not in the second
+   * - `exceptAll` - `except` that allows duplicated rows
+   *
+   * @param args - array of queries or SQL expressions
    */
-  union(args, wrap) {
-    return pushQueryArray(
+  union(...args) {
+    return _queryUnion(
       this.clone(),
-      "union",
-      args.map((arg) => ({ arg, kind: "UNION", wrap }))
+      args,
+      "UNION"
     );
   }
   /**
-   * Same as `union`, but allows duplicated rows.
+   * Same as {@link union}, but allows duplicated rows.
    *
-   * @param args - array of queries or raw SQLs
-   * @param wrap - provide `true` if you want the queries to be wrapped into parentheses
+   * @param args - array of queries or SQL expressions
    */
-  unionAll(args, wrap) {
-    return pushQueryArray(
+  unionAll(...args) {
+    return _queryUnion(
       this.clone(),
-      "union",
-      args.map((arg) => ({ arg, kind: "UNION ALL", wrap }))
+      args,
+      "UNION ALL"
     );
   }
   /**
-   * Same as `union`, but uses a `INTERSECT` SQL keyword instead
+   * Same as {@link union}, but uses a `INTERSECT` SQL keyword instead
    *
-   * @param args - array of queries or raw SQLs
-   * @param wrap - provide `true` if you want the queries to be wrapped into parentheses
+   * @param args - array of queries or SQL expressions
    */
-  intersect(args, wrap) {
-    return pushQueryArray(
+  intersect(...args) {
+    return _queryUnion(
       this.clone(),
-      "union",
-      args.map((arg) => ({ arg, kind: "INTERSECT", wrap }))
+      args,
+      "INTERSECT"
     );
   }
   /**
-   * Same as `intersect`, but allows duplicated rows.
+   * Same as {@link intersect}, but allows duplicated rows.
    *
-   * @param args - array of queries or raw SQLs
-   * @param wrap - provide `true` if you want the queries to be wrapped into parentheses
+   * @param args - array of queries or SQL expressions
    */
-  intersectAll(args, wrap) {
-    return pushQueryArray(
+  intersectAll(...args) {
+    return _queryUnion(
       this.clone(),
-      "union",
-      args.map((arg) => ({ arg, kind: "INTERSECT ALL", wrap }))
+      args,
+      "INTERSECT ALL"
     );
   }
   /**
-   * Same as `union`, but uses an `EXCEPT` SQL keyword instead
+   * Same as {@link union}, but uses an `EXCEPT` SQL keyword instead
    *
-   * @param args - array of queries or raw SQLs
-   * @param wrap - provide `true` if you want the queries to be wrapped into parentheses
+   * @param args - array of queries or SQL expressions
    */
-  except(args, wrap) {
-    return pushQueryArray(
+  except(...args) {
+    return _queryUnion(
       this.clone(),
-      "union",
-      args.map((arg) => ({ arg, kind: "EXCEPT", wrap }))
+      args,
+      "EXCEPT"
     );
   }
   /**
-   * Same as `except`, but allows duplicated rows.
+   * Same as {@link except}, but allows duplicated rows.
    *
-   * @param args - array of queries or raw SQLs
-   * @param wrap - provide `true` if you want the queries to be wrapped into parentheses
+   * @param args - array of queries or SQL expressions
    */
-  exceptAll(args, wrap) {
-    return pushQueryArray(
+  exceptAll(...args) {
+    return _queryUnion(
       this.clone(),
-      "union",
-      args.map((arg) => ({ arg, kind: "EXCEPT ALL", wrap }))
+      args,
+      "EXCEPT ALL"
+    );
+  }
+}
+
+var __defProp$4 = Object.defineProperty;
+var __defProps$1 = Object.defineProperties;
+var __getOwnPropDescs$1 = Object.getOwnPropertyDescriptors;
+var __getOwnPropSymbols$4 = Object.getOwnPropertySymbols;
+var __hasOwnProp$4 = Object.prototype.hasOwnProperty;
+var __propIsEnum$4 = Object.prototype.propertyIsEnumerable;
+var __defNormalProp$4 = (obj, key, value) => key in obj ? __defProp$4(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues$4 = (a, b) => {
+  for (var prop in b || (b = {}))
+    if (__hasOwnProp$4.call(b, prop))
+      __defNormalProp$4(a, prop, b[prop]);
+  if (__getOwnPropSymbols$4)
+    for (var prop of __getOwnPropSymbols$4(b)) {
+      if (__propIsEnum$4.call(b, prop))
+        __defNormalProp$4(a, prop, b[prop]);
+    }
+  return a;
+};
+var __spreadProps$1 = (a, b) => __defProps$1(a, __getOwnPropDescs$1(b));
+class WithMethods {
+  with(name, second, third) {
+    const q = this.clone();
+    let [options, queryArg] = third ? [second, third] : [void 0, second];
+    let query;
+    if (typeof queryArg === "function") {
+      const arg = q.queryBuilder.clone();
+      arg.q.withShapes = q.q.withShapes;
+      query = queryArg(arg);
+    } else {
+      query = queryArg;
+    }
+    if ((options == null ? void 0 : options.columns) === true) {
+      options = __spreadProps$1(__spreadValues$4({}, options), {
+        columns: Object.keys(query.shape)
+      });
+    }
+    pushQueryValue(q, "with", { n: name, o: options, q: query });
+    const shape = getShapeFromSelect(query, true);
+    return setQueryObjectValue(q, "withShapes", name, shape);
+  }
+  withRecursive(name, ...args) {
+    var _a, _b, _c;
+    const q = this.clone();
+    let [options, baseFn, recursiveFn] = args.length === 2 ? [{}, args[0], args[1]] : args;
+    const arg = q.queryBuilder.clone();
+    arg.q.withShapes = q.q.withShapes;
+    let query = typeof baseFn === "function" ? baseFn(arg) : baseFn;
+    const shape = ((_b = (_a = arg.q).withShapes) != null ? _b : _a.withShapes = {})[name] = getShapeFromSelect(
+      query,
+      true
+    );
+    const recursive = recursiveFn(arg);
+    query = _queryUnion(query, [recursive], (_c = options.union) != null ? _c : "UNION ALL");
+    options.recursive = true;
+    if (options.columns === true) {
+      options = __spreadProps$1(__spreadValues$4({}, options), {
+        columns: Object.keys(shape)
+      });
+    }
+    pushQueryValue(q, "with", { n: name, o: options, q: query });
+    return setQueryObjectValue(q, "withShapes", name, shape);
+  }
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  withSql(name, ...args) {
+    const q = this.clone();
+    const [options, shape, sql] = args.length === 2 ? [void 0, args[0], args[1]] : args;
+    pushQueryValue(q, "with", {
+      n: name,
+      o: options,
+      s: sql(q)
+    });
+    return setQueryObjectValue(
+      q,
+      "withShapes",
+      name,
+      shape(this.columnTypes)
     );
   }
 }
@@ -9331,11 +9392,10 @@ const _queryUpdate = (query, arg) => {
               "q",
               "withShapes"
             );
-            pushQueryValue(query, "with", [
-              as,
-              orchidCore.emptyObject,
-              value
-            ]);
+            pushQueryValue(query, "with", {
+              n: as,
+              q: value
+            });
             set[key] = new RawSQL(`(SELECT * FROM "${as}")`);
           }
         } else {
@@ -9502,7 +9562,24 @@ class Update {
    * })
    * ```
    *
-   * It is not supported because query inside `WITH` cannot reference the table in `UPDATE`.
+   * `update` can be used in {@link WithMethods.with} expressions:
+   *
+   * ```ts
+   * db.$queryBuilder
+   *   // update record in one table
+   *   .with('a', db.table.find(1).select('id').update(data))
+   *   // update record in other table using the first table record id
+   *   .with('b', (q) =>
+   *     db.otherTable
+   *       .find(1)
+   *       .select('id')
+   *       .update({
+   *         ...otherData,
+   *         aId: () => q.from('a').get('id'),
+   *       }),
+   *   )
+   *   .from('b');
+   * ```
    *
    * ### null, undefined, unknown columns
    *
@@ -11296,10 +11373,10 @@ orchidCore.applyMixins(QueryMethods, [
   AsMethods,
   AggregateMethods,
   Select,
-  From,
+  FromMethods,
   Join,
   OnMethods,
-  With,
+  WithMethods,
   Union,
   JsonModifiers,
   JsonMethods,
@@ -11933,7 +12010,7 @@ exports.EnumColumn = EnumColumn;
 exports.ExpressionMethods = ExpressionMethods;
 exports.FnExpression = FnExpression;
 exports.For = For;
-exports.From = From;
+exports.FromMethods = FromMethods;
 exports.Having = Having;
 exports.InetColumn = InetColumn;
 exports.IntegerBaseColumn = IntegerBaseColumn;
@@ -12000,7 +12077,7 @@ exports.Update = Update;
 exports.VarCharColumn = VarCharColumn;
 exports.VirtualColumn = VirtualColumn;
 exports.Where = Where;
-exports.With = With;
+exports.WithMethods = WithMethods;
 exports.XMLColumn = XMLColumn;
 exports._initQueryBuilder = _initQueryBuilder;
 exports._queryAfterSaveCommit = _queryAfterSaveCommit;
@@ -12048,6 +12125,7 @@ exports._queryRows = _queryRows;
 exports._querySelect = _querySelect;
 exports._queryTake = _queryTake;
 exports._queryTakeOptional = _queryTakeOptional;
+exports._queryUnion = _queryUnion;
 exports._queryUpdate = _queryUpdate;
 exports._queryUpdateOrThrow = _queryUpdateOrThrow;
 exports._queryUpdateRaw = _queryUpdateRaw;