orchid-orm 1.59.4 → 1.60.0

package/dist/index.mjs

@@ -1,4 +1,4 @@
-import { makeColumnTypes, getStackTrace, applyMixins, QueryHooks, defaultSchemaConfig, raw, getCallerFilePath, snakeCaseKey, getColumnTypes, parseTableData, toSnakeCase, emptyObject, emptyArray, _queryHookAfterCreate, _queryHookAfterUpdate, getQueryAs, setQueryObjectValueImmutable, pushQueryOnForOuter, isExpression, cloneQueryBaseUnscoped, DynamicRawSQL, RawSql, getShapeFromSelect, _queryWhere, _queryDefaults, getPrimaryKeys, VirtualColumn, _with, _queryCreate, _orCreate, getFreeAlias, isQueryReturnsAll, _queryHookBeforeUpdate, _queryFindBy, _queryRows, _queryUpdate, _queryDelete, prepareSubQueryForSql, _queryUpdateOrThrow, _queryCreateMany, OrchidOrmInternalError, toArray, objectHasValues, _queryJoinOn, _queryCreateManyFrom, NotFoundError, _queryFindByOptional, _querySelect, pick, _queryWhereExists, _queryTake, _queryTakeOptional, _initQueryBuilder, Db, getClonedQueryData } from 'pqb';
+import { makeColumnTypes, getStackTrace, applyMixins, QueryHooks, defaultSchemaConfig, raw, getCallerFilePath, snakeCaseKey, getColumnTypes, parseTableData, toSnakeCase, emptyObject, emptyArray, _queryHookAfterCreate, _queryHookAfterUpdate, getQueryAs, setQueryObjectValueImmutable, pushQueryOnForOuter, isExpression, cloneQueryBaseUnscoped, DynamicRawSQL, RawSql, getShapeFromSelect, _queryWhere, _queryDefaults, getPrimaryKeys, VirtualColumn, _prependWith, _queryCreate, _orCreate, getFreeAlias, _querySelect, _queryUpdate, _queryWhereIn, _appendQuery, isQueryReturnsAll, _queryUpsert, _queryDelete, noop, _queryFindBy, prepareSubQueryForSql, _clone, _queryUpdateOrThrow, _queryInsert, _queryCreateMany, OrchidOrmInternalError, toArray, objectHasValues, _queryJoinOn, _queryCreateManyFrom, NotFoundError, _queryFindByOptional, _queryRows, pick, _queryWhereExists, _queryTake, _queryTakeOptional, _initQueryBuilder, Db, getClonedQueryData } from 'pqb';
 export * from 'pqb';
 import { AsyncLocalStorage } from 'node:async_hooks';
 
@@ -190,7 +190,7 @@ const hasRelationHandleUpdate = (q, set, key, primaryKeys, nestedUpdate) => {
     );
   });
 };
-const selectIfNotSelected = (q, columns) => {
+const _selectIfNotSelected = (q, columns) => {
   const select = q.q.select || [];
   if (!select.includes("*")) {
     for (const column of columns) {
@@ -286,6 +286,8 @@ const getColumnKeyFromDbName = (query, name) => {
   }
   return name;
 };
+const selectCteColumnsSql = (cteAs, columns) => `(SELECT ${columns.map((c) => `"${cteAs}"."${c}"`).join(", ")} FROM "${cteAs}")`;
+const selectCteColumnSql = (cteAs, column) => `(SELECT "${cteAs}"."${column}" FROM "${cteAs}")`;
 
 const joinQueryChainHOF = (relPKeys, reverseJoin, joinQuery) => (joiningQuery, baseQuery) => {
   const jq = joiningQuery;
@@ -401,7 +403,7 @@ class BelongsToVirtualColumn extends VirtualColumn {
     super(schema);
     this.key = key;
     this.state = state;
-    this.nestedUpdate = nestedUpdate$3(this.state);
+    this.nestedUpdate = nestedUpdate$2(this.state);
   }
   create(q, ctx, item) {
     const {
@@ -415,18 +417,11 @@ class BelongsToVirtualColumn extends VirtualColumn {
     }
     const value = item[key];
     if ("create" in value || "connectOrCreate" in value) {
-      foreignKeys.forEach((foreignKey) => item[foreignKey] = new RawSql(""));
+      const asFn = setForeignKeysFromCte(item, primaryKeys, foreignKeys);
       const selectPKeys = query.select(...primaryKeys);
-      _with(
+      _prependWith(
         q,
-        (as) => {
-          foreignKeys.forEach((foreignKey, i) => {
-            item[foreignKey]._sql = selectCteColumnSql(
-              as,
-              primaryKeys[i]
-            );
-          });
-        },
+        asFn,
         "create" in value ? _queryCreate(selectPKeys, value.create) : _orCreate(
           _queryWhere(selectPKeys, [
             value.connectOrCreate.where
@@ -437,7 +432,7 @@ class BelongsToVirtualColumn extends VirtualColumn {
       return;
     } else if ("connect" in value) {
       const as = getFreeAlias(q.q.withShapes, "q");
-      _with(q, as, query.select(...primaryKeys).findBy(value.connect));
+      _prependWith(q, as, query.select(...primaryKeys).findBy(value.connect));
       foreignKeys.map((foreignKey, i) => {
         item[foreignKey] = new RawSql(
           selectCteColumnMustExistSql(i, as, primaryKeys[i])
@@ -446,10 +441,10 @@ class BelongsToVirtualColumn extends VirtualColumn {
       return;
     }
   }
-  update(q, ctx, set) {
+  update(q, set) {
     q.q.wrapInTransaction = true;
     const data = set[this.key];
-    this.nestedUpdate(q, set, data, ctx);
+    this.nestedUpdate(q, set, data);
   }
 }
 const makeBelongsToMethod = (tableConfig, table, relation, relationName, query) => {
@@ -520,139 +515,138 @@ const makeBelongsToMethod = (tableConfig, table, relation, relationName, query)
     reverseJoin
   };
 };
-const nestedUpdate$3 = ({ query, primaryKeys, foreignKeys, len }) => {
-  return (q, update, params, state) => {
-    if (params.upsert && isQueryReturnsAll(q)) {
-      throw new Error("`upsert` option is not allowed in a batch update");
-    }
-    let idsForDelete;
-    _queryHookBeforeUpdate(q, async ({ query: q2 }) => {
-      if (params.disconnect) {
-        for (const key of foreignKeys) {
-          update[key] = null;
-        }
-      } else if (params.set) {
-        let loadPrimaryKeys;
-        let loadForeignKeys;
-        for (let i = 0; i < len; i++) {
-          const primaryKey = primaryKeys[i];
-          if (primaryKey in params.set) {
-            update[foreignKeys[i]] = params.set[primaryKey];
-          } else {
-            (loadPrimaryKeys ?? (loadPrimaryKeys = [])).push(primaryKey);
-            (loadForeignKeys ?? (loadForeignKeys = [])).push(foreignKeys[i]);
-          }
-        }
-        if (loadPrimaryKeys) {
-          for (let i = 0, len2 = loadPrimaryKeys.length; i < len2; i++) {
-            update[loadForeignKeys[i]] = new RawSql("");
-          }
-          _with(
-            q2,
-            (as) => {
-              for (let i = 0, len2 = loadPrimaryKeys.length; i < len2; i++) {
-                update[loadForeignKeys[i]]._sql = selectCteColumnMustExistSql(i, as, loadPrimaryKeys[i]);
-              }
-            },
-            _queryFindBy(query.select(...loadPrimaryKeys), params.set)
-          );
+const nestedUpdate$2 = ({ query, primaryKeys, foreignKeys, len }) => {
+  return (self, update, params) => {
+    if (params.create) {
+      const createQuery = _querySelect(
+        _queryCreate(query.clone(), params.create),
+        primaryKeys
+      );
+      const asFn = setForeignKeysFromCte(update, primaryKeys, foreignKeys);
+      _prependWith(self, asFn, createQuery);
+    } else if (params.update) {
+      _selectIfNotSelected(self, foreignKeys);
+      const selectIdsSql = new RawSql("");
+      const updateQuery = _queryUpdate(
+        _queryWhereIn(query.clone(), true, primaryKeys, selectIdsSql),
+        params.update
+      );
+      updateQuery.q.returnType = "value";
+      _appendQuery(self, updateQuery, (as) => {
+        selectIdsSql._sql = selectCteColumnsSql(as, foreignKeys);
+      });
+    } else if (params.upsert) {
+      if (isQueryReturnsAll(self)) {
+        throw new Error("`upsert` option is not allowed in a batch update");
+      }
+      const { relQuery } = relWithSelectIds(
+        self,
+        query,
+        primaryKeys,
+        foreignKeys
+      );
+      const upsertQuery = _querySelect(
+        _queryUpsert(
+          relQuery,
+          params.upsert
+        ),
+        primaryKeys
+      );
+      const asFn = setForeignKeysFromCte(update, primaryKeys, foreignKeys);
+      _prependWith(self, asFn, upsertQuery);
+    } else if (params.delete) {
+      _selectIfNotSelected(self, foreignKeys);
+      disconnect(update, foreignKeys);
+      const { selectIdsSql, relQuery } = relWithSelectIds(
+        self,
+        query,
+        primaryKeys,
+        foreignKeys
+      );
+      self.q.and = self.q.or = void 0;
+      _queryWhereIn(self, true, foreignKeys, selectIdsSql);
+      const deleteQuery = _queryDelete(relQuery);
+      deleteQuery.q.returnType = "value";
+      _appendQuery(self, deleteQuery, noop);
+    } else if (params.disconnect) {
+      disconnect(update, foreignKeys);
+    } else if (params.set) {
+      let loadPrimaryKeys;
+      let loadForeignKeys;
+      for (let i = 0; i < len; i++) {
+        const primaryKey = primaryKeys[i];
+        if (primaryKey in params.set) {
+          update[foreignKeys[i]] = params.set[primaryKey];
+        } else {
+          (loadPrimaryKeys ?? (loadPrimaryKeys = [])).push(primaryKey);
+          (loadForeignKeys ?? (loadForeignKeys = [])).push(foreignKeys[i]);
         }
-      } else if (params.create) {
-        const q3 = query.clone();
-        q3.q.select = primaryKeys;
-        const record = await _queryCreate(
-          q3,
-          params.create
+      }
+      if (loadPrimaryKeys) {
+        const asFn = setForeignKeysFromCte(
+          update,
+          loadPrimaryKeys,
+          loadForeignKeys,
+          true
+        );
+        _prependWith(
+          self,
+          asFn,
+          _queryFindBy(query.select(...loadPrimaryKeys), params.set)
         );
-        for (let i = 0; i < len; i++) {
-          update[foreignKeys[i]] = record[primaryKeys[i]];
-        }
-      } else if (params.delete) {
-        const selectQuery = q2.clone();
-        selectQuery.q.type = void 0;
-        selectQuery.q.distinct = emptyArray;
-        selectIfNotSelected(selectQuery, foreignKeys);
-        idsForDelete = await _queryRows(selectQuery);
-        for (const foreignKey of foreignKeys) {
-          update[foreignKey] = null;
-        }
       }
-    });
-    const { upsert } = params;
-    if (upsert || params.update || params.delete) {
-      selectIfNotSelected(q, foreignKeys);
-    }
-    if (upsert) {
-      (state.queries ?? (state.queries = [])).push(async (queryResult) => {
-        const row = queryResult.rows[0];
-        let obj = {};
-        for (let i = 0; i < len; i++) {
-          const id = row[foreignKeys[i]];
-          if (id === null) {
-            obj = void 0;
-            break;
-          }
-          obj[primaryKeys[i]] = id;
-        }
-        const count = obj ? await _queryUpdate(
-          query.findBy(obj),
-          upsert.update
-        ) : 0;
-        if (!count) {
-          const data = typeof upsert.create === "function" ? upsert.create() : upsert.create;
-          const result = await _queryCreate(
-            query.select(...primaryKeys),
-            data
-          );
-          const collectData = {};
-          state.collect = {
-            data: collectData
-          };
-          for (let i = 0; i < len; i++) {
-            collectData[foreignKeys[i]] = result[primaryKeys[i]];
-          }
-        }
-      });
-    } else if (params.delete || params.update) {
-      _queryHookAfterUpdate(
-        q,
-        params.update ? foreignKeys : emptyArray,
-        async (data) => {
-          let ids;
-          if (params.delete) {
-            ids = idsForDelete;
-          } else {
-            ids = [];
-            for (const item of data) {
-              let row;
-              for (const foreignKey of foreignKeys) {
-                const id = item[foreignKey];
-                if (id === null) {
-                  row = void 0;
-                  break;
-                } else {
-                  (row ?? (row = [])).push(id);
-                }
-              }
-              if (row) ids.push(row);
-            }
-          }
-          if (!ids?.length) return;
-          const t = query.whereIn(
-            primaryKeys,
-            ids
-          );
-          if (params.delete) {
-            await _queryDelete(t);
-          } else {
-            await _queryUpdate(t, params.update);
-          }
-        }
-      );
     }
   };
 };
-const selectCteColumnSql = (cteAs, column) => `(SELECT "${cteAs}"."${column}" FROM "${cteAs}")`;
+const disconnect = (update, foreignKeys) => {
+  for (const foreignKey of foreignKeys) {
+    update[foreignKey] = null;
+  }
+};
+const relWithSelectIds = (self, rel, primaryKeys, foreignKeys) => {
+  const selectIdsQuery = makeSelectIdsQuery(self, foreignKeys);
+  const selectIdsSql = new RawSql("");
+  _prependWith(
+    self,
+    (as) => {
+      selectIdsSql._sql = selectCteColumnsSql(as, foreignKeys);
+    },
+    selectIdsQuery
+  );
+  return {
+    selectIdsSql,
+    relQuery: _queryWhereIn(rel.clone(), true, primaryKeys, selectIdsSql)
+  };
+};
+const makeSelectIdsQuery = (self, foreignKeys) => {
+  const selectIdsQuery = self.baseQuery.clone();
+  selectIdsQuery.q.distinct = emptyArray;
+  selectIdsQuery.q.select = foreignKeys;
+  selectIdsQuery.q.and = self.q.and;
+  selectIdsQuery.q.or = self.q.or;
+  return selectIdsQuery;
+};
+const setForeignKeysFromCte = (record, primaryKeys, foreignKeys, mustExist) => {
+  for (const key of foreignKeys) {
+    record[key] = new RawSql("");
+  }
+  return (as) => {
+    foreignKeys.forEach(
+      mustExist ? (foreignKey, i) => {
+        record[foreignKey]._sql = selectCteColumnMustExistSql(
+          i,
+          as,
+          primaryKeys[i]
+        );
+      } : (foreignKey, i) => {
+        record[foreignKey]._sql = selectCteColumnSql(
+          as,
+          primaryKeys[i]
+        );
+      }
+    );
+  };
+};
 const selectCteColumnMustExistSql = (i, cteAs, column) => {
   const selectColumn = selectCteColumnSql(cteAs, column);
   return i === 0 ? `CASE WHEN (SELECT count(*) FROM "${cteAs}") = 0 AND (SELECT 'not-found')::int = 0 THEN NULL ELSE ${selectColumn} END` : selectColumn;
@@ -664,32 +658,107 @@ class HasOneVirtualColumn extends VirtualColumn {
     this.key = key;
     this.state = state;
     this.nestedInsert = nestedInsert$2(state);
-    this.nestedUpdate = nestedUpdate$2(state);
+    this.setNulls = {};
+    for (const foreignKey of state.foreignKeys) {
+      this.setNulls[foreignKey] = null;
+    }
   }
-  create(q, ctx, item, rowIndex) {
-    hasRelationHandleCreate(
-      q,
-      ctx,
-      item,
-      rowIndex,
-      this.key,
-      this.state.primaryKeys,
-      this.nestedInsert
-    );
+  create(self, ctx, item, rowIndex, one) {
+    if (one) {
+      const value = item[this.key];
+      if (!value.create && !value.connect && !value.connectOrCreate) {
+        return;
+      }
+      const { query: rel, primaryKeys, foreignKeys } = this.state;
+      _selectIfNotSelected(self, primaryKeys);
+      const data = value.create ? { ...value.create } : {};
+      foreignKeys.forEach((key) => {
+        data[key] = new RawSql("");
+      });
+      const query = value.create ? _queryCreate(_clone(rel), data) : value.connect ? _queryUpdateOrThrow(
+        rel.where(value.connect),
+        data
+      ) : value.connectOrCreate ? _queryUpsert(rel.where(value.connectOrCreate.where), {
+        update: data,
+        create: {
+          ...value.connectOrCreate.create,
+          ...data
+        }
+      }) : void 0;
+      _appendQuery(self, query, (as) => {
+        foreignKeys.forEach((key, i) => {
+          data[key]._sql = selectCteColumnSql(as, primaryKeys[i]);
+        });
+      });
+    } else {
+      hasRelationHandleCreate(
+        self,
+        ctx,
+        item,
+        rowIndex,
+        this.key,
+        this.state.primaryKeys,
+        this.nestedInsert
+      );
+    }
   }
-  update(q, _, set) {
+  update(self, set) {
     const params = set[this.key];
-    if ((params.set || params.create || params.upsert) && isQueryReturnsAll(q)) {
+    if ((params.set || params.create || params.upsert) && isQueryReturnsAll(self)) {
       const key = params.set ? "set" : params.create ? "create" : "upsert";
       throw new Error(`\`${key}\` option is not allowed in a batch update`);
     }
-    hasRelationHandleUpdate(
-      q,
-      set,
-      this.key,
-      this.state.primaryKeys,
-      this.nestedUpdate
-    );
+    const { primaryKeys, foreignKeys, query: relQuery } = this.state;
+    if (params.create || params.update || params.upsert || params.disconnect || params.set || params.delete) {
+      _selectIfNotSelected(self, primaryKeys);
+      const selectIdsSql = new RawSql("");
+      const existingRelQuery = _queryWhereIn(
+        _clone(relQuery),
+        true,
+        foreignKeys,
+        selectIdsSql
+      );
+      let setIds = void 0;
+      if (params.create || params.set || params.upsert) {
+        setIds = {};
+        foreignKeys.forEach((foreignKey) => {
+          setIds[foreignKey] = new RawSql("");
+        });
+      }
+      const nullifyOrDeleteQuery = params.update ? _queryUpdate(existingRelQuery, params.update) : params.upsert ? _queryUpsert(existingRelQuery, {
+        update: params.upsert.update,
+        create: {
+          ...typeof params.upsert.create === "function" ? params.upsert.create() : params.upsert.create,
+          ...setIds
+        }
+      }) : params.delete ? _queryDelete(existingRelQuery) : _queryUpdate(existingRelQuery, this.setNulls);
+      nullifyOrDeleteQuery.q.returnType = "void";
+      _appendQuery(self, nullifyOrDeleteQuery, (as) => {
+        selectIdsSql._sql = selectCteColumnsSql(as, primaryKeys);
+        if (params.create || params.set || params.upsert) {
+          foreignKeys.forEach((foreignKey, i) => {
+            setIds[foreignKey]._sql = selectCteColumnSql(
+              as,
+              primaryKeys[i]
+            );
+          });
+        }
+      });
+      if (params.create) {
+        const createQuery = _queryInsert(_clone(relQuery), {
+          ...params.create,
+          ...setIds
+        });
+        _appendQuery(self, createQuery, noop);
+      } else if (params.set) {
+        const setQuery = _queryUpdate(
+          _queryWhere(_clone(relQuery), [params.set]),
+          setIds
+        );
+        setQuery.q.returnType = "void";
+        _appendQuery(self, setQuery, noop);
+      }
+    }
   }
 }
 const makeHasOneMethod = (tableConfig, table, relation, relationName, query) => {
@@ -817,9 +886,9 @@ const nestedInsert$2 = ({ query, primaryKeys, foreignKeys }) => {
       for (let i = 0, len = items.length; i < len; i++) {
         const [selfData, item] = items[i];
         const data2 = {};
-        for (let i2 = 0; i2 < len; i2++) {
-          data2[foreignKeys[i2]] = selfData[primaryKeys[i2]];
-        }
+        primaryKeys.forEach((primaryKey, i2) => {
+          data2[foreignKeys[i2]] = selfData[primaryKey];
+        });
         items[i] = "connect" in item ? _queryUpdateOrThrow(
           t.where(item.connect),
           data2
@@ -860,78 +929,6 @@ const nestedInsert$2 = ({ query, primaryKeys, foreignKeys }) => {
     }
   };
 };
-const nestedUpdate$2 = ({ query, primaryKeys, foreignKeys }) => {
-  const len = primaryKeys.length;
-  const setNulls = {};
-  for (const foreignKey of foreignKeys) {
-    setNulls[foreignKey] = null;
-  }
-  return async (_, data, params) => {
-    const t = query.clone();
-    const ids = data.map(
-      (item) => primaryKeys.map((primaryKey) => item[primaryKey])
-    );
-    const currentRelationsQuery = t.whereIn(
-      foreignKeys,
-      ids
-    );
-    if (params.create || params.disconnect || params.set) {
-      let queryToDisconnect = currentRelationsQuery;
-      if (params.set) {
-        queryToDisconnect = queryToDisconnect.whereNot(params.set);
-      }
-      await _queryUpdate(queryToDisconnect, setNulls);
-      const record = data[0];
-      if (params.create) {
-        const obj = { ...params.create };
-        for (let i = 0; i < len; i++) {
-          obj[foreignKeys[i]] = record[primaryKeys[i]];
-        }
-        await t.insert(obj);
-      }
-      if (params.set) {
-        const obj = {};
-        for (let i = 0; i < len; i++) {
-          obj[foreignKeys[i]] = record[primaryKeys[i]];
-        }
-        await _queryUpdate(
-          _queryWhere(t, [params.set]),
-          obj
-        );
-      }
-    } else if (params.update) {
-      await _queryUpdate(currentRelationsQuery, params.update);
-    } else if (params.delete) {
-      const q = _queryDelete(currentRelationsQuery);
-      q.q.returnType = "value";
-      await q;
-    } else if (params.upsert) {
-      const { update, create } = params.upsert;
-      currentRelationsQuery.q.select = foreignKeys;
-      const updatedIds = await _queryUpdate(
-        _queryRows(currentRelationsQuery),
-        update
-      );
-      if (updatedIds.length < ids.length) {
-        const data2 = typeof create === "function" ? create() : create;
-        await t.createMany(
-          ids.reduce((rows, ids2) => {
-            if (!updatedIds.some(
-              (updated) => updated.every((value, i) => value === ids2[i])
-            )) {
-              const obj = { ...data2 };
-              for (let i = 0; i < len; i++) {
-                obj[foreignKeys[i]] = ids2[i];
-              }
-              rows.push(obj);
-            }
-            return rows;
-          }, [])
-        );
-      }
-    }
-  };
-};
 
 class HasManyVirtualColumn extends VirtualColumn {
   constructor(schema, key, state) {
@@ -952,7 +949,7 @@ class HasManyVirtualColumn extends VirtualColumn {
       this.nestedInsert
     );
   }
-  update(q, _, set) {
+  update(q, set) {
     const params = set[this.key];
     if ((params.set || params.create) && isQueryReturnsAll(q)) {
       const key = params.set ? "set" : "create";
@@ -1295,7 +1292,7 @@ class HasAndBelongsToManyVirtualColumn extends VirtualColumn {
       this.nestedInsert
     );
   }
-  update(q, _, set) {
+  update(q, set) {
     hasRelationHandleUpdate(
       q,
       set,