orchid-orm 1.72.5 → 1.72.6

package/dist/index.mjs

@@ -1,4 +1,4 @@
-import { DynamicRawSQL, QueryHooks, RawSql, VirtualColumn, _appendQuery, _clone, _createDbSqlMethod, _hookSelectColumns, _initQueryBuilder, _orCreate, _prependWith, _queryCreate, _queryCreateMany, _queryCreateManyFrom, _queryDefaults, _queryDelete, _queryFindBy, _queryFindByOptional, _queryHookAfterCreate, _queryHookAfterUpdate, _queryInsert, _queryInsertMany, _queryJoinOn, _queryRows, _querySelect, _queryTake, _queryTakeOptional, _queryUpdate, _queryUpdateOrThrow, _queryUpsert, _queryWhere, _queryWhereExists, _queryWhereIn, applyMixins, cloneQueryBaseUnscoped, defaultSchemaConfig, emptyArray, emptyObject, getCallerFilePath, getClonedQueryData, getColumnTypes, getFreeAlias, getPrimaryKeys, getQueryAs, getShapeFromSelect, getStackTrace, internalSchemaConfig, isExpression, isQueryReturnsAll, makeColumnTypes, noop, objectHasValues, parseTableData, pick, prepareSubQueryForSql, pushQueryOnForOuter, setQueryObjectValueImmutable, toArray, toSnakeCase } from "pqb/internal";
+import { DynamicRawSQL, QueryHooks, RawSql, VirtualColumn, _appendQuery, _appendQueryOnUpsertCreate, _clone, _createDbSqlMethod, _hookSelectColumns, _initQueryBuilder, _orCreate, _prependWith, _queryCreate, _queryCreateMany, _queryCreateManyFrom, _queryDefaults, _queryDelete, _queryFindBy, _queryFindByOptional, _queryHookAfterCreate, _queryHookAfterUpdate, _queryInsert, _queryInsertMany, _queryJoinOn, _queryRows, _querySelect, _queryTake, _queryTakeOptional, _queryUpdate, _queryUpdateOrThrow, _queryUpsert, _queryWhere, _queryWhereExists, _queryWhereIn, applyMixins, cloneQueryBaseUnscoped, defaultSchemaConfig, emptyArray, emptyObject, getCallerFilePath, getClonedQueryData, getColumnTypes, getFreeAlias, getPrimaryKeys, getQueryAs, getShapeFromSelect, getStackTrace, internalSchemaConfig, isExpression, isQueryReturnsAll, makeColumnTypes, noop, objectHasValues, parseTableData, pick, prepareSubQueryForSql, pushQueryOnForOuter, setQueryObjectValueImmutable, toArray, toSnakeCase } from "pqb/internal";
 import { Db, NotFoundError, OrchidOrmInternalError } from "pqb";
 import { AsyncLocalStorage } from "node:async_hooks";
 export * from "pqb";
@@ -137,6 +137,25 @@ function createBaseTable({ schemaConfig = defaultSchemaConfig, columnTypes: colu
 	base.prototype.autoForeignKeys = autoForeignKeys === true ? {} : autoForeignKeys || void 0;
 	return base;
 }
+const throwIfQueryReturnsAllForNestedUpdate = (q, params) => {
+	if (!(params.set || params.create || params.upsert) || !isQueryReturnsAll(q)) return;
+	let key;
+	if (params.set) key = "set";
+	else if (params.create) key = "create";
+	else key = "upsert";
+	throw new Error(`\`${key}\` option is not allowed in a batch update`);
+};
+const getNestedUpdateUpsertCreate = (upsert) => {
+	if (typeof upsert.create === "function") return upsert.create();
+	return upsert.create || emptyObject;
+};
+const makeNestedUpdateUpsertData = (upsert, setIds) => ({
+	update: upsert.update,
+	create: {
+		...getNestedUpdateUpsertCreate(upsert),
+		...setIds
+	}
+});
 const getThroughRelation = (table, through) => {
 	return table.relations[through];
 };
@@ -224,8 +243,27 @@ const getColumnKeyFromDbName = (query, name) => {
 	for (const k in query.shape) if (query.shape[k].data.name === name) return k;
 	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 selectCteColumnsSql$1 = (cteAs, columns) => `(SELECT ${columns.map((c) => `"${cteAs}"."${c}"`).join(", ")} FROM "${cteAs}")`;
+const selectCteColumnSql$1 = (cteAs, column) => `(SELECT "${cteAs}"."${column}" FROM "${cteAs}")`;
+const makeNestedUpdateRelationIds = (q, relQuery, primaryKeys, foreignKeys) => {
+	const selectIdsSql = new RawSql("");
+	const setIds = {};
+	for (const foreignKey of foreignKeys) setIds[foreignKey] = new RawSql("");
+	let appendedAs;
+	_hookSelectColumns(q, primaryKeys, (aliasedPrimaryKeys) => {
+		selectIdsSql._sql = selectCteColumnsSql$1(appendedAs, aliasedPrimaryKeys);
+		foreignKeys.forEach((foreignKey, i) => {
+			setIds[foreignKey]._sql = selectCteColumnSql$1(appendedAs, aliasedPrimaryKeys[i]);
+		});
+	});
+	return {
+		existingRelQuery: _queryWhereIn(_clone(relQuery), true, foreignKeys, selectIdsSql),
+		setIds,
+		setAppendedAs(as) {
+			appendedAs = as;
+		}
+	};
+};
 const selectCteColumnFromManySql = (cteAs, column, rowIndex, count) => {
 	let sql = `(SELECT "${cteAs}"."${column}" FROM "${cteAs}"`;
 	if (count > 1) {
@@ -406,7 +444,7 @@ const nestedUpdate$2 = ({ query, primaryKeys, foreignKeys, len }) => {
 		} else if (params.update) {
 			let appendedAs;
 			_hookSelectColumns(self, foreignKeys, (aliasedForeignKeys) => {
-				selectIdsSql._sql = selectCteColumnsSql(appendedAs, aliasedForeignKeys);
+				selectIdsSql._sql = selectCteColumnsSql$1(appendedAs, aliasedForeignKeys);
 			});
 			const selectIdsSql = new RawSql("");
 			const updateQuery = _queryUpdate(_queryWhereIn(query.clone(), true, primaryKeys, selectIdsSql), params.update);
@@ -449,7 +487,7 @@ const relWithSelectIds = (self, rel, primaryKeys, foreignKeys) => {
 	const selectIdsQuery = makeSelectIdsQuery(self, foreignKeys);
 	const selectIdsSql = new RawSql("");
 	_prependWith(self, (as) => {
-		selectIdsSql._sql = selectCteColumnsSql(as, foreignKeys);
+		selectIdsSql._sql = selectCteColumnsSql$1(as, foreignKeys);
 	}, selectIdsQuery);
 	return {
 		selectIdsSql,
@@ -470,12 +508,12 @@ const setForeignKeysFromCte = (record, primaryKeys, foreignKeys, mustExist) => {
 		foreignKeys.forEach(mustExist ? (foreignKey, i) => {
 			record[foreignKey]._sql = selectCteColumnMustExistSql(i, as, primaryKeys[i]);
 		} : (foreignKey, i) => {
-			record[foreignKey]._sql = selectCteColumnSql(as, primaryKeys[i]);
+			record[foreignKey]._sql = selectCteColumnSql$1(as, primaryKeys[i]);
 		});
 	};
 };
 const selectCteColumnMustExistSql = (i, cteAs, column) => {
-	const selectColumn = selectCteColumnSql(cteAs, column);
+	const selectColumn = selectCteColumnSql$1(cteAs, column);
 	return i === 0 ? `CASE WHEN (SELECT count(*) FROM "${cteAs}") = 0 AND (SELECT 'not-found')::int = 0 THEN NULL ELSE ${selectColumn} END` : selectColumn;
 };
 var HasOneVirtualColumn = class extends VirtualColumn {
@@ -541,43 +579,19 @@ var HasOneVirtualColumn = class extends VirtualColumn {
 	update(self, set) {
 		const querySelf = self;
 		const params = set[this.key];
-		if ((params.set || params.create || params.upsert) && isQueryReturnsAll(querySelf)) {
-			const key = params.set ? "set" : params.create ? "create" : "upsert";
-			throw new Error(`\`${key}\` option is not allowed in a batch update`);
-		}
+		throwIfQueryReturnsAllForNestedUpdate(querySelf, params);
 		const { primaryKeys, foreignKeys, query: relQuery } = this.state;
 		if (params.create || params.update || params.upsert || params.disconnect || params.set || params.delete) {
-			let appendedAs;
-			_hookSelectColumns(querySelf, primaryKeys, (aliasedPrimaryKeys) => {
-				selectIdsSql._sql = selectCteColumnsSql(appendedAs, aliasedPrimaryKeys);
-				if (params.create || params.set || params.upsert) foreignKeys.forEach((foreignKey, i) => {
-					setIds[foreignKey]._sql = selectCteColumnSql(appendedAs, aliasedPrimaryKeys[i]);
-				});
-			});
-			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);
+			const ids = makeNestedUpdateRelationIds(querySelf, relQuery, primaryKeys, foreignKeys);
+			const nullifyOrDeleteQuery = params.update ? _queryUpdate(ids.existingRelQuery, params.update) : params.upsert ? _queryUpsert(ids.existingRelQuery, makeNestedUpdateUpsertData(params.upsert, ids.setIds)) : params.delete ? _queryDelete(ids.existingRelQuery) : _queryUpdate(ids.existingRelQuery, this.setNulls);
 			nullifyOrDeleteQuery.q.returnType = "void";
-			_appendQuery(querySelf, nullifyOrDeleteQuery, (as) => appendedAs = as);
+			_appendQuery(querySelf, nullifyOrDeleteQuery, ids.setAppendedAs);
 			if (params.create) _appendQuery(querySelf, _queryInsert(_clone(relQuery), {
 				...params.create,
-				...setIds
+				...ids.setIds
 			}), noop);
 			else if (params.set) {
-				const setQuery = _queryUpdate(_queryWhere(_clone(relQuery), [params.set]), setIds);
+				const setQuery = _queryUpdate(_queryWhere(_clone(relQuery), [params.set]), ids.setIds);
 				setQuery.q.returnType = "void";
 				_appendQuery(querySelf, setQuery, noop);
 			}
@@ -691,6 +705,8 @@ var HasManyVirtualColumn = class extends VirtualColumn {
 		this.state = state;
 		this.nestedInsert = nestedInsert$1(state);
 		this.nestedUpdate = nestedUpdate$1(state);
+		this.setNulls = {};
+		for (const foreignKey of state.foreignKeys) this.setNulls[foreignKey] = null;
 	}
 	create(self, ctx, items, rowIndexes, count) {
 		const querySelf = self;
@@ -761,14 +777,18 @@ var HasManyVirtualColumn = class extends VirtualColumn {
 			});
 		} else hasRelationHandleCreate(querySelf, ctx, items, rowIndexes, this.key, this.state.primaryKeys, this.nestedInsert);
 	}
-	update(q, set) {
-		const query = q;
+	update(self, set) {
+		const querySelf = self;
 		const params = set[this.key];
-		if ((params.set || params.create) && isQueryReturnsAll(query)) {
-			const key = params.set ? "set" : "create";
-			throw new Error(`\`${key}\` option is not allowed in a batch update`);
+		throwIfQueryReturnsAllForNestedUpdate(querySelf, params);
+		hasRelationHandleUpdate(querySelf, set, this.key, this.state.primaryKeys, this.nestedUpdate);
+		if (params.upsert) {
+			const { primaryKeys, foreignKeys, query: relQuery } = this.state;
+			const ids = makeNestedUpdateRelationIds(querySelf, relQuery, primaryKeys, foreignKeys);
+			const appendedQuery = _queryUpsert(ids.existingRelQuery, makeNestedUpdateUpsertData(params.upsert, ids.setIds));
+			appendedQuery.q.returnType = "void";
+			_appendQuery(querySelf, appendedQuery, ids.setAppendedAs);
 		}
-		hasRelationHandleUpdate(query, set, this.key, this.state.primaryKeys, this.nestedUpdate);
 	}
 };
 const makeHasManyMethod = (tableConfig, table, relation, relationName, query) => {
@@ -908,10 +928,10 @@ const nestedInsert$1 = ({ query, primaryKeys, foreignKeys }) => {
 		}
 	});
 };
-const nestedUpdate$1 = ({ query, primaryKeys, foreignKeys }) => {
+const nestedUpdate$1 = ({ query: relQuery, primaryKeys, foreignKeys }) => {
 	const len = primaryKeys.length;
 	return (async (_, data, params) => {
-		const t = query.clone();
+		const t = relQuery.clone();
 		if (params.create) {
 			const obj = {};
 			for (let i = 0; i < len; i++) obj[foreignKeys[i]] = data[0][primaryKeys[i]];
@@ -921,12 +941,12 @@ const nestedUpdate$1 = ({ query, primaryKeys, foreignKeys }) => {
 			})));
 		}
 		if (params.add) {
-			if (data.length > 1) throw new OrchidOrmInternalError(query, "`connect` is not available when updating multiple records, it is only applicable for a single record update");
+			if (data.length > 1) throw new OrchidOrmInternalError(relQuery, "`connect` is not available when updating multiple records, it is only applicable for a single record update");
 			const obj = {};
 			for (let i = 0; i < len; i++) obj[foreignKeys[i]] = data[0][primaryKeys[i]];
 			const relatedWheres = toArray(params.add);
 			const count = await _queryUpdate(t.where({ OR: relatedWheres }), obj);
-			if (count < relatedWheres.length) throw new OrchidOrmInternalError(query, `Expected to find at least ${relatedWheres.length} record(s) based on \`add\` conditions, but found ${count}`);
+			if (count < relatedWheres.length) throw new OrchidOrmInternalError(relQuery, `Expected to find at least ${relatedWheres.length} record(s) based on \`add\` conditions, but found ${count}`);
 		}
 		if (params.disconnect || params.set) {
 			const obj = {};
@@ -961,9 +981,48 @@ var HasAndBelongsToManyVirtualColumn = class extends VirtualColumn {
 		hasRelationHandleCreate(q, ctx, items, rowIndexes, this.key, this.state.primaryKeys, this.nestedInsert);
 	}
 	update(q, set) {
-		hasRelationHandleUpdate(q, set, this.key, this.state.primaryKeys, this.nestedUpdate);
+		const querySelf = q;
+		const params = set[this.key];
+		hasRelationHandleUpdate(querySelf, set, this.key, this.state.primaryKeys, this.nestedUpdate);
+		if (params.upsert) {
+			throwIfQueryReturnsAllForNestedUpdate(querySelf, { upsert: params.upsert });
+			nestedUpdateUpsert(querySelf, this.state, params.upsert);
+		}
 	}
 };
+const selectCteColumnsSql = (cteAs, columns) => `(SELECT ${columns.map((c) => `"${cteAs}"."${c}"`).join(", ")} FROM "${cteAs}")`;
+const selectCteColumnSql = (cteAs, column) => `(SELECT "${cteAs}"."${column}" FROM "${cteAs}")`;
+const nestedUpdateUpsert = (querySelf, state, upsert) => {
+	const parentIdsSql = new RawSql("");
+	const parentValues = state.foreignKeys.map(() => new RawSql(""));
+	const relatedIdsSql = new RawSql("");
+	const relatedValues = state.throughPrimaryKeys.map(() => new RawSql(""));
+	let parentAs;
+	_hookSelectColumns(querySelf, state.primaryKeys, (aliasedPrimaryKeys) => {
+		parentIdsSql._sql = selectCteColumnsSql(parentAs, aliasedPrimaryKeys);
+		for (let i = 0; i < state.foreignKeys.length; i++) parentValues[i]._sql = selectCteColumnSql(parentAs, aliasedPrimaryKeys[i]);
+	});
+	const upsertQuery = _querySelect(_queryUpsert(_queryWhereExists(_queryWhere(_clone(state.relatedTableQuery), [upsert.findBy]), state.joinTableQuery, [(q) => {
+		for (let i = 0; i < state.throughPrimaryKeys.length; i++) _queryJoinOn(q, [state.throughForeignKeysFull[i], state.throughPrimaryKeysFull[i]]);
+		return _queryWhereIn(q, true, state.foreignKeysFull, parentIdsSql);
+	}]), {
+		update: upsert.update,
+		create: upsert.create || {}
+	}), state.throughPrimaryKeys);
+	const selectAs = {};
+	for (let i = 0; i < state.foreignKeys.length; i++) selectAs[state.foreignKeys[i]] = parentValues[i];
+	for (let i = 0; i < state.throughForeignKeys.length; i++) selectAs[state.throughForeignKeys[i]] = relatedValues[i];
+	const joinRows = _querySelect(_clone(state.queryBuilder), [selectAs]);
+	_queryWhere(joinRows, [relatedIdsSql]);
+	const joinQuery = state.joinTableQuery.insertForEachFrom(joinRows);
+	joinQuery.q.returnType = "void";
+	_appendQuery(querySelf, _appendQueryOnUpsertCreate(upsertQuery, joinQuery, (as) => {
+		relatedIdsSql._sql = `EXISTS (SELECT 1 FROM "${as}")`;
+		for (let i = 0; i < relatedValues.length; i++) relatedValues[i]._sql = selectCteColumnSql(as, state.throughPrimaryKeys[i]);
+	}), (as) => {
+		parentAs = as;
+	});
+};
 const removeColumnName = (column) => {
 	if (!column.data.name) return column;
 	const cloned = Object.create(column);
@@ -1021,6 +1080,7 @@ const makeHasAndBelongsToManyMethod = (tableConfig, table, qb, relation, relatio
 	addAutoForeignKey(tableConfig, subQuery, table, primaryKeys, foreignKeys, relation.options, originalForeignKeys);
 	addAutoForeignKey(tableConfig, subQuery, query, throughPrimaryKeys, throughForeignKeys, relation.options.through, originalThroughForeignKeys);
 	const state = {
+		queryBuilder: qb,
 		relatedTableQuery: query,
 		joinTableQuery: subQuery,
 		primaryKeys,