@atscript/db 0.1.146 → 0.1.147

package/dist/relation-loader-D9XuXaMv.mjs

@@ -1,462 +0,0 @@
-import { C as findRemoteFK, S as findFKForRelation, w as resolveRelationTargetTable } from "./nested-writer-CnOOAehr.mjs";
-//#region \0rolldown/runtime.js
-var __defProp = Object.defineProperty;
-var __exportAll = (all, no_symbols) => {
-	let target = {};
-	for (var name in all) __defProp(target, name, {
-		get: all[name],
-		enumerable: true
-	});
-	if (!no_symbols) __defProp(target, Symbol.toStringTag, { value: "Module" });
-	return target;
-};
-//#endregion
-//#region src/rel/relation-loader.ts
-var relation_loader_exports = /* @__PURE__ */ __exportAll({
-	ensureSelectIncludesFields: () => ensureSelectIncludesFields,
-	loadRelationsImpl: () => loadRelationsImpl
-});
-/**
-* Loads related data for `$with` relations and attaches them to result rows.
-*/
-async function loadRelationsImpl(rows, withRelations, host) {
-	if (rows.length === 0 || withRelations.length === 0) return;
-	if (host.adapter.supportsNativeRelations()) return host.adapter.loadRelations(rows, withRelations, host._meta.relations, host._meta.foreignKeys, host._tableResolver);
-	if (!host._tableResolver) return;
-	const tasks = [];
-	for (const withRel of withRelations) {
-		const relName = withRel.name;
-		if (relName.includes(".")) continue;
-		const relation = host._meta.relations.get(relName);
-		if (!relation) throw new Error(`Unknown relation "${relName}" in $with. Available relations: ${[...host._meta.relations.keys()].join(", ") || "(none)"}`);
-		const targetType = relation.targetType();
-		if (!targetType) continue;
-		const targetTable = host._tableResolver(targetType);
-		if (!targetTable) {
-			host.logger.warn(`Could not resolve table for relation "${relName}" — skipping`);
-			continue;
-		}
-		const filter = withRel.filter && Object.keys(withRel.filter).length > 0 ? withRel.filter : void 0;
-		const flatRel = withRel;
-		const controls = { ...withRel.controls || {} };
-		if (flatRel.$sort && !controls.$sort) controls.$sort = flatRel.$sort;
-		if (flatRel.$limit !== null && flatRel.$limit !== void 0 && (controls.$limit === null || controls.$limit === void 0)) controls.$limit = flatRel.$limit;
-		if (flatRel.$skip !== null && flatRel.$skip !== void 0 && (controls.$skip === null || controls.$skip === void 0)) controls.$skip = flatRel.$skip;
-		if (flatRel.$select && !controls.$select) controls.$select = flatRel.$select;
-		if (flatRel.$with && !controls.$with) controls.$with = flatRel.$with;
-		const relQuery = {
-			filter,
-			controls
-		};
-		if (relation.direction === "to") tasks.push(loadToRelation(rows, {
-			relName,
-			relation,
-			targetTable,
-			relQuery
-		}, host));
-		else if (relation.direction === "via") tasks.push(loadViaRelation(rows, {
-			relName,
-			relation,
-			targetTable,
-			relQuery
-		}, host));
-		else tasks.push(loadFromRelation(rows, {
-			relName,
-			relation,
-			targetTable,
-			relQuery
-		}, host));
-	}
-	await Promise.all(tasks);
-}
-/**
-* Loads a `@db.rel.to` relation (FK is on this table).
-*/
-async function loadToRelation(rows, opts, host) {
-	const { relName, relation, targetTable, relQuery } = opts;
-	const fkEntry = findFKForRelation(relation, host._meta.foreignKeys);
-	if (!fkEntry) return;
-	const { localFields, targetFields } = fkEntry;
-	if (localFields.length === 1) {
-		const localField = localFields[0];
-		const targetField = targetFields[0];
-		const fkValues = collectUniqueValues(rows, localField);
-		if (fkValues.length === 0) {
-			for (const row of rows) row[relName] = null;
-			return;
-		}
-		const inFilter = { [targetField]: { $in: fkValues } };
-		const targetFilter = relQuery.filter ? { $and: [inFilter, relQuery.filter] } : inFilter;
-		const controls = ensureSelectIncludesFields(relQuery.controls, targetFields);
-		assignSingle({
-			rows,
-			related: await targetTable.findMany({
-				filter: targetFilter,
-				controls
-			}),
-			localField,
-			remoteField: targetField,
-			relName
-		});
-	} else {
-		const related = await queryCompositeFK(rows, {
-			localFields,
-			targetFields,
-			targetTable,
-			relQuery
-		});
-		const index = /* @__PURE__ */ new Map();
-		for (const item of related) index.set(compositeKey(targetFields, item), item);
-		for (const row of rows) row[relName] = index.get(compositeKey(localFields, row)) ?? null;
-	}
-}
-/**
-* Loads a `@db.rel.from` relation (FK is on the target table).
-*/
-async function loadFromRelation(rows, opts, host) {
-	const { relName, relation, targetTable, relQuery } = opts;
-	const remoteFK = findRemoteFK(targetTable, host.tableName, relation.alias);
-	if (!remoteFK) {
-		host.logger.warn(`Could not find FK on target table for relation "${relName}"`);
-		return;
-	}
-	const localFields = remoteFK.targetFields;
-	const remoteFields = remoteFK.fields;
-	if (localFields.length === 1) {
-		const localField = localFields[0];
-		const remoteField = remoteFields[0];
-		const pkValues = collectUniqueValues(rows, localField);
-		if (pkValues.length === 0) return;
-		const inFilter = { [remoteField]: { $in: pkValues } };
-		const targetFilter = relQuery.filter ? { $and: [inFilter, relQuery.filter] } : inFilter;
-		const controls = ensureSelectIncludesFields(relQuery.controls, remoteFields);
-		const related = await targetTable.findMany({
-			filter: targetFilter,
-			controls
-		});
-		if (relation.isArray) assignGrouped({
-			rows,
-			related,
-			localField,
-			remoteField,
-			relName
-		});
-		else assignSingle({
-			rows,
-			related,
-			localField,
-			remoteField,
-			relName
-		});
-	} else {
-		const related = await queryCompositeFK(rows, {
-			localFields,
-			targetFields: remoteFields,
-			targetTable,
-			relQuery
-		});
-		if (relation.isArray) {
-			const groups = /* @__PURE__ */ new Map();
-			for (const item of related) {
-				const key = compositeKey(remoteFields, item);
-				let group = groups.get(key);
-				if (!group) {
-					group = [];
-					groups.set(key, group);
-				}
-				group.push(item);
-			}
-			for (const row of rows) row[relName] = groups.get(compositeKey(localFields, row)) ?? [];
-		} else {
-			const index = /* @__PURE__ */ new Map();
-			for (const item of related) {
-				const key = compositeKey(remoteFields, item);
-				if (!index.has(key)) index.set(key, item);
-			}
-			for (const row of rows) row[relName] = index.get(compositeKey(localFields, row)) ?? null;
-		}
-	}
-}
-/**
-* Loads a `@db.rel.via` relation (M:N through a junction table).
-*/
-async function loadViaRelation(rows, opts, host) {
-	const { relName, relation, targetTable, relQuery } = opts;
-	if (!relation.viaType || !host._tableResolver) return;
-	const junctionType = relation.viaType();
-	if (!junctionType) return;
-	const junctionTable = host._tableResolver(junctionType);
-	if (!junctionTable) {
-		host.logger.warn(`Could not resolve junction table for via relation "${relName}"`);
-		return;
-	}
-	const fkToThis = findRemoteFK(junctionTable, host.tableName);
-	if (!fkToThis) {
-		host.logger.warn(`Could not find FK on junction table pointing to "${host.tableName}" for via relation "${relName}"`);
-		return;
-	}
-	const targetTableName = resolveRelationTargetTable(relation);
-	const fkToTarget = findRemoteFK(junctionTable, targetTableName);
-	if (!fkToTarget) {
-		host.logger.warn(`Could not find FK on junction table pointing to target "${targetTableName}" for via relation "${relName}"`);
-		return;
-	}
-	const localPKFields = fkToThis.targetFields;
-	const junctionLocalFields = fkToThis.fields;
-	const targetPKFields = fkToTarget.targetFields;
-	const junctionTargetFields = fkToTarget.fields;
-	if (localPKFields.length === 1) await loadViaSingleKey(rows, {
-		relName,
-		relation,
-		targetTable,
-		relQuery,
-		localPKFields,
-		junctionLocalFields,
-		targetPKFields,
-		junctionTargetFields,
-		junctionTable
-	});
-	else await loadViaCompositeKey(rows, {
-		relName,
-		relation,
-		targetTable,
-		relQuery,
-		localPKFields,
-		junctionLocalFields,
-		targetPKFields,
-		junctionTargetFields,
-		junctionTable
-	});
-}
-async function loadViaSingleKey(rows, opts) {
-	const { relName, relation, targetTable, relQuery, localPKFields, junctionLocalFields, targetPKFields, junctionTargetFields, junctionTable } = opts;
-	const localField = localPKFields[0];
-	const junctionLocalField = junctionLocalFields[0];
-	const junctionTargetField = junctionTargetFields[0];
-	const targetPKField = targetPKFields[0];
-	const pkValues = collectUniqueValues(rows, localField);
-	if (pkValues.length === 0) {
-		for (const row of rows) row[relName] = [];
-		return;
-	}
-	const junctionFilter = { [junctionLocalField]: { $in: pkValues } };
-	const junctionRows = await junctionTable.findMany({
-		filter: junctionFilter,
-		controls: { $select: [junctionLocalField, junctionTargetField] }
-	});
-	if (junctionRows.length === 0) {
-		for (const row of rows) row[relName] = relation.isArray ? [] : null;
-		return;
-	}
-	const targetFKValues = collectUniqueValues(junctionRows, junctionTargetField);
-	const inFilter = { [targetPKField]: { $in: targetFKValues } };
-	const targetFilter = relQuery.filter ? { $and: [inFilter, relQuery.filter] } : inFilter;
-	const controls = ensureSelectIncludesFields(relQuery.controls, targetPKFields);
-	const targetRows = await targetTable.findMany({
-		filter: targetFilter,
-		controls
-	});
-	const targetIndex = /* @__PURE__ */ new Map();
-	for (const item of targetRows) targetIndex.set(String(item[targetPKField]), item);
-	const groups = /* @__PURE__ */ new Map();
-	for (const jRow of junctionRows) {
-		const localKey = String(jRow[junctionLocalField]);
-		const targetKey = String(jRow[junctionTargetField]);
-		const target = targetIndex.get(targetKey);
-		if (!target) continue;
-		let group = groups.get(localKey);
-		if (!group) {
-			group = [];
-			groups.set(localKey, group);
-		}
-		group.push(target);
-	}
-	for (const row of rows) {
-		const key = String(row[localField]);
-		row[relName] = relation.isArray ? groups.get(key) ?? [] : groups.get(key)?.[0] ?? null;
-	}
-}
-async function loadViaCompositeKey(rows, opts) {
-	const { relName, relation, targetTable, relQuery, localPKFields, junctionLocalFields, targetPKFields, junctionTargetFields, junctionTable } = opts;
-	const orFilters = [];
-	for (const row of rows) {
-		const condition = {};
-		let valid = true;
-		for (let i = 0; i < localPKFields.length; i++) {
-			const val = row[localPKFields[i]];
-			if (val === null || val === void 0) {
-				valid = false;
-				break;
-			}
-			condition[junctionLocalFields[i]] = val;
-		}
-		if (valid) orFilters.push(condition);
-	}
-	if (orFilters.length === 0) {
-		for (const row of rows) row[relName] = relation.isArray ? [] : null;
-		return;
-	}
-	const junctionFilter = orFilters.length === 1 ? orFilters[0] : { $or: orFilters };
-	const junctionRows = await junctionTable.findMany({
-		filter: junctionFilter,
-		controls: { $select: [...junctionLocalFields, ...junctionTargetFields] }
-	});
-	if (junctionRows.length === 0) {
-		for (const row of rows) row[relName] = relation.isArray ? [] : null;
-		return;
-	}
-	const targetOrFilters = [];
-	const seenTargets = /* @__PURE__ */ new Set();
-	for (const jRow of junctionRows) {
-		const key = compositeKey(junctionTargetFields, jRow);
-		if (seenTargets.has(key)) continue;
-		seenTargets.add(key);
-		const condition = {};
-		for (let i = 0; i < junctionTargetFields.length; i++) condition[targetPKFields[i]] = jRow[junctionTargetFields[i]];
-		targetOrFilters.push(condition);
-	}
-	const targetBaseFilter = targetOrFilters.length === 1 ? targetOrFilters[0] : { $or: targetOrFilters };
-	const finalFilter = relQuery.filter ? { $and: [targetBaseFilter, relQuery.filter] } : targetBaseFilter;
-	const controls = ensureSelectIncludesFields(relQuery.controls, targetPKFields);
-	const targetRows = await targetTable.findMany({
-		filter: finalFilter,
-		controls
-	});
-	const targetIndex = /* @__PURE__ */ new Map();
-	for (const item of targetRows) targetIndex.set(compositeKey(targetPKFields, item), item);
-	const groups = /* @__PURE__ */ new Map();
-	for (const jRow of junctionRows) {
-		const localKey = compositeKey(junctionLocalFields, jRow);
-		const targetKey = compositeKey(junctionTargetFields, jRow);
-		const target = targetIndex.get(targetKey);
-		if (!target) continue;
-		let group = groups.get(localKey);
-		if (!group) {
-			group = [];
-			groups.set(localKey, group);
-		}
-		group.push(target);
-	}
-	for (const row of rows) {
-		const key = compositeKey(localPKFields, row);
-		row[relName] = relation.isArray ? groups.get(key) ?? [] : groups.get(key)?.[0] ?? null;
-	}
-}
-/**
-* Ensure the given join fields survive the user's $select on a $with relation
-* read. Without this, the JS-side join would key off undefined and silently
-* return empty relations.
-*
-* - Array $select: append missing FK fields.
-* - Include-mode object $select (any value is 1/true, or empty object): set
-*   missing FK fields to 1.
-* - Exclude-mode object $select: leave alone — FK survives by default. If the
-*   FK is explicitly excluded, throw rather than silently override user intent.
-*/
-function ensureSelectIncludesFields(controls, fields) {
-	if (!controls) return controls;
-	const sel = controls.$select;
-	if (sel === void 0 || sel === null) return controls;
-	if (Array.isArray(sel)) {
-		const augmented = [...sel];
-		for (const f of fields) if (!augmented.includes(f)) augmented.push(f);
-		return {
-			...controls,
-			$select: augmented
-		};
-	}
-	if (typeof sel === "object") {
-		const selObj = sel;
-		const isInclude = Object.keys(selObj).length === 0 || Object.values(selObj).some((v) => v === 1 || v === true);
-		const augmented = { ...selObj };
-		let mutated = false;
-		for (const f of fields) {
-			const v = augmented[f];
-			if (v === 0 || v === false) throw new Error(`Cannot exclude join column "${f}" from $select on $with relation`);
-			if (v === void 0 && isInclude) {
-				augmented[f] = 1;
-				mutated = true;
-			}
-		}
-		return mutated ? {
-			...controls,
-			$select: augmented
-		} : controls;
-	}
-	return controls;
-}
-function compositeKey(fields, obj) {
-	let key = "";
-	for (let i = 0; i < fields.length; i++) {
-		if (i > 0) key += "\0\0";
-		const v = obj[fields[i]];
-		key += v === null || v === void 0 ? "\0" : String(v);
-	}
-	return key;
-}
-/** Collects unique non-null values for a field across rows. */
-function collectUniqueValues(rows, field) {
-	const set = /* @__PURE__ */ new Set();
-	for (const row of rows) {
-		const v = row[field];
-		if (v !== null && v !== void 0) set.add(v);
-	}
-	return [...set];
-}
-/** Assigns related items grouped by FK value (one-to-many). */
-function assignGrouped(opts) {
-	const { rows, related, localField, remoteField, relName } = opts;
-	const groups = /* @__PURE__ */ new Map();
-	for (const item of related) {
-		const key = item[remoteField];
-		let group = groups.get(key);
-		if (!group) {
-			group = [];
-			groups.set(key, group);
-		}
-		group.push(item);
-	}
-	for (const row of rows) row[relName] = groups.get(row[localField]) ?? [];
-}
-/** Assigns related items by FK value (many-to-one / one-to-one). */
-function assignSingle(opts) {
-	const { rows, related, localField, remoteField, relName } = opts;
-	const index = /* @__PURE__ */ new Map();
-	for (const item of related) {
-		const key = item[remoteField];
-		if (!index.has(key)) index.set(key, item);
-	}
-	for (const row of rows) row[relName] = index.get(row[localField]) ?? null;
-}
-/** Batch query for composite FK. */
-function queryCompositeFK(rows, opts) {
-	const { localFields, targetFields, targetTable, relQuery } = opts;
-	const seen = /* @__PURE__ */ new Set();
-	const orFilters = [];
-	for (const row of rows) {
-		const key = compositeKey(localFields, row);
-		if (seen.has(key)) continue;
-		seen.add(key);
-		const condition = {};
-		let valid = true;
-		for (let i = 0; i < localFields.length; i++) {
-			const val = row[localFields[i]];
-			if (val === null || val === void 0) {
-				valid = false;
-				break;
-			}
-			condition[targetFields[i]] = val;
-		}
-		if (valid) orFilters.push(condition);
-	}
-	if (orFilters.length === 0) return Promise.resolve([]);
-	const baseFilter = orFilters.length === 1 ? orFilters[0] : { $or: orFilters };
-	const targetFilter = relQuery.filter ? { $and: [baseFilter, relQuery.filter] } : baseFilter;
-	const controls = ensureSelectIncludesFields(relQuery.controls, targetFields);
-	return targetTable.findMany({
-		filter: targetFilter,
-		controls
-	});
-}
-//#endregion
-export { relation_loader_exports as n, loadRelationsImpl as t };