joist-codegen 2.3.0-next.9 → 2.3.0-next.91

package/build/inheritance.js

@@ -1,265 +1,255 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.applyInheritanceUpdates = applyInheritanceUpdates;
-exports.getStiEntities = getStiEntities;
-const EntityDbMetadata_1 = require("./EntityDbMetadata");
-const utils_1 = require("./utils");
+import { fail } from "./utils.js";
+import "./config.js";
+import { canonicalizeOtherEntities, makeEntity } from "./EntityDbMetadata.js";
+//#region src/inheritance.ts
+/** Preserves physical storage metadata, then applies domain inheritance specialization. */
 function applyInheritanceUpdates(config, db) {
-    const { entities, entitiesByName } = db;
-    setClassTableInheritance(entities, entitiesByName);
-    expandSingleTableInheritance(config, entitiesByName, entities);
-    rewriteSingleTableForeignKeys(config, entities);
-    setupSubTypeSpecialization(config, entities);
-    // Now that all relations (incl. specialized ones) exist, collapse the duplicate `otherEntity` objects
-    (0, EntityDbMetadata_1.canonicalizeOtherEntities)(db);
+	const { entities, entitiesByName } = db;
+	for (const meta of entities) {
+		meta.physicalMetadata = {
+			...meta,
+			name: meta.name,
+			primitives: meta.primitives.map((field) => ({ ...field })),
+			enums: meta.enums.map((field) => ({ ...field })),
+			pgEnums: meta.pgEnums.map((field) => ({ ...field })),
+			manyToOnes: meta.manyToOnes.map((field) => ({ ...field })),
+			oneToManys: meta.oneToManys.map((field) => ({ ...field })),
+			largeOneToManys: meta.largeOneToManys.map((field) => ({ ...field })),
+			oneToOnes: meta.oneToOnes.map((field) => ({ ...field })),
+			manyToManys: meta.manyToManys.map((field) => ({ ...field })),
+			largeManyToManys: meta.largeManyToManys.map((field) => ({ ...field })),
+			polymorphics: meta.polymorphics.map((field) => ({
+				...field,
+				components: field.components.map((component) => ({ ...component }))
+			})),
+			nonDeferredFks: meta.nonDeferredFks,
+			nonDeferredManyToManyFks: meta.nonDeferredManyToManyFks
+		};
+		for (const timestamp of [
+			"createdAt",
+			"updatedAt",
+			"deletedAt"
+		]) meta.physicalMetadata[timestamp] = meta.physicalMetadata.primitives.find((field) => field.fieldName === meta[timestamp]?.fieldName);
+	}
+	setClassTableInheritance(entities, entitiesByName);
+	expandSingleTableInheritance(config, entitiesByName, entities);
+	rewriteSingleTableForeignKeys(config, entities);
+	setupSubTypeSpecialization(config, entities);
+	canonicalizeOtherEntities(db);
 }
 /**
- * Looks for subtypes specializing a FK of their base type, and give them a specialized m2o.
- *
- * I.e. `SmallPublishers.group: SmallPublisherGroup`.
- *
- * This means the `publishers.group_id` FK will actually be in the `EntityDbMetadata` twice,
- * once for the base and once for the child, which is unusual, but gets the types generated
- * that we want, and then we fixup/dedupe them at runtime in `configure.ts` when creating
- * `meta.allFields`.
- */
+* Looks for subtypes specializing a FK of their base type, and give them a specialized m2o.
+*
+* I.e. `SmallPublishers.group: SmallPublisherGroup`.
+*
+* This means the `publishers.group_id` FK will actually be in the `EntityDbMetadata` twice,
+* once for the base and once for the child, which is unusual, but gets the types generated
+* that we want, and then we fixup/dedupe them at runtime in `configure.ts` when creating
+* `meta.allFields`.
+*/
 function setupSubTypeSpecialization(config, entities) {
-    for (const entity of entities) {
-        if (entity.baseType) {
-            const entityConfig = config.entities[entity.name];
-            const baseConfig = config.entities[entity.baseType.name];
-            // Look through the base's m2os & o2ms, looking for a config specialization
-            for (const rel of [...entity.baseType.manyToOnes, ...entity.baseType.oneToManys]) {
-                const { notNull } = entityConfig?.relations?.[rel.fieldName] ?? {};
-                const subType = entityConfig?.relations?.[rel.fieldName]?.subType ??
-                    baseConfig?.relations?.[rel.fieldName]?.subType ??
-                    // Probe the otherFieldName for a `subType: "self"` on self-referential relations in STI tables
-                    baseConfig?.relations?.[rel.otherFieldName]?.subType ??
-                    // if we are only marking the relation as required and not specifying a subtype, then use the original type
-                    (notNull ? rel.otherEntity.name : undefined);
-                if (!subType) {
-                    // continue...
-                }
-                else if (subType === "self" && rel.kind === "m2o") {
-                    // Specialize `TaskNew.copiedFrom: TaskNew` & `TaskOld.copiedFrom: TaskOld`
-                    entity.manyToOnes.push({ ...rel, ...(notNull ? { notNull } : {}), otherEntity: (0, EntityDbMetadata_1.makeEntity)(entity.name) });
-                }
-                else if (subType === "self" && rel.kind === "o2m") {
-                    // Specialize `TaskNew.copiedTo: TaskNew[]` & `TaskOld.copiedTo: TaskOld[]`
-                    entity.oneToManys.push({ ...rel, ...(notNull ? { notNull } : {}), otherEntity: (0, EntityDbMetadata_1.makeEntity)(entity.name) });
-                }
-                else if (rel.kind === "m2o") {
-                    // Specialize `SmallPublisher.group: SmallPublisherGroup`
-                    entity.manyToOnes.push({ ...rel, ...(notNull ? { notNull } : {}), otherEntity: (0, EntityDbMetadata_1.makeEntity)(subType) });
-                }
-                else if (rel.kind === "o2m") {
-                    // Specialize `SmallPublisherGroup.publishers: SmallPublisher`
-                    entity.oneToManys.push({ ...rel, ...(notNull ? { notNull } : {}), otherEntity: (0, EntityDbMetadata_1.makeEntity)(subType) });
-                }
-            }
-            // Look through the base's primitives looking for a config specialization
-            for (const field of entity.baseType?.primitives ?? []) {
-                // the only specialization a primitive can have is nullability
-                const { notNull } = entityConfig?.fields?.[field.fieldName] ?? {};
-                if (notNull) {
-                    entity.primitives.push({ ...field, notNull });
-                }
-            }
-        }
-    }
+	for (const entity of entities) if (entity.baseType) {
+		const entityConfig = config.entities[entity.name];
+		const baseConfig = config.entities[entity.baseType.name];
+		for (const rel of [...entity.baseType.manyToOnes, ...entity.baseType.oneToManys]) {
+			const { notNull } = entityConfig?.relations?.[rel.fieldName] ?? {};
+			const subType = entityConfig?.relations?.[rel.fieldName]?.subType ?? baseConfig?.relations?.[rel.fieldName]?.subType ?? baseConfig?.relations?.[rel.otherFieldName]?.subType ?? (notNull ? rel.otherEntity.name : void 0);
+			if (!subType) {} else if (subType === "self" && rel.kind === "m2o") entity.manyToOnes.push({
+				...rel,
+				...notNull ? { notNull } : {},
+				otherEntity: makeEntity(entity.name)
+			});
+			else if (subType === "self" && rel.kind === "o2m") entity.oneToManys.push({
+				...rel,
+				...notNull ? { notNull } : {},
+				otherEntity: makeEntity(entity.name)
+			});
+			else if (rel.kind === "m2o") entity.manyToOnes.push({
+				...rel,
+				...notNull ? { notNull } : {},
+				otherEntity: makeEntity(subType)
+			});
+			else if (rel.kind === "o2m") entity.oneToManys.push({
+				...rel,
+				...notNull ? { notNull } : {},
+				otherEntity: makeEntity(subType)
+			});
+		}
+		for (const field of entity.baseType?.primitives ?? []) {
+			const { notNull } = entityConfig?.fields?.[field.fieldName] ?? {};
+			if (notNull) entity.primitives.push({
+				...field,
+				notNull
+			});
+		}
+	}
 }
 /**
- * Ensure CTI base types have their inheritanceType set.
- *
- * (We automatically set `inheritanceType` for STI tables when we see their config
- * setup, see `expandSingleTableInheritance`.)
- */
+* Ensure CTI base types have their inheritanceType set.
+*
+* (We automatically set `inheritanceType` for STI tables when we see their config
+* setup, see `expandSingleTableInheritance`.)
+*/
 function setClassTableInheritance(entities, entitiesByName) {
-    const ctiBaseNames = [];
-    for (const entity of entities) {
-        if (entity.baseClassName) {
-            ctiBaseNames.push(entity.baseClassName);
-            const base = entitiesByName[entity.baseClassName];
-            entity.baseType = base;
-            base.subTypes.push(entity);
-        }
-    }
-    for (const entity of entities) {
-        if (ctiBaseNames.includes(entity.name))
-            entity.inheritanceType = "cti";
-    }
+	const ctiBaseNames = [];
+	for (const entity of entities) if (entity.baseClassName) {
+		ctiBaseNames.push(entity.baseClassName);
+		const base = entitiesByName[entity.baseClassName];
+		entity.baseType = base;
+		base.subTypes.push(entity);
+	}
+	for (const entity of entities) if (ctiBaseNames.includes(entity.name)) entity.inheritanceType = "cti";
 }
 /** Expands STI tables into multiple entities, so they get separate `SubTypeCodegen.ts` & `SubType.ts` files. */
 function expandSingleTableInheritance(config, entitiesByName, entities) {
-    for (const entity of entities) {
-        const [fieldName, stiField] = Object.entries(config.entities[entity.name]?.fields || {}).find(([, f]) => !!f.stiDiscriminator) ?? [];
-        if (fieldName && stiField && stiField.stiDiscriminator) {
-            entity.inheritanceType = "sti";
-            // Ensure we have an enum field so that we can bake the STI discriminators into the metadata.ts file
-            const enumField = entity.enums.find((e) => e.fieldName === fieldName) ??
-                (0, utils_1.fail)(`No enum column found for ${entity.name}.${fieldName}, which is required to use singleTableInheritance`);
-            entity.stiDiscriminatorField = enumField.fieldName;
-            // Find the available discriminators/subtypes, i.e. NEW => NewTask, OLD => OldTask, etc.
-            const subTypes = Object.entries(stiField.stiDiscriminator);
-            const allFields = [
-                ...Object.entries(config.entities[entity.name]?.fields ?? {}),
-                ...Object.entries(config.entities[entity.name]?.relations ?? {}),
-            ];
-            // Make sure there aren't any typos in `stiType`s
-            const availableCodes = subTypes.map(([code]) => code);
-            const availableNames = subTypes.map(([, name]) => name);
-            const available = [...availableCodes, ...availableNames];
-            allFields.filter(([name, f]) => {
-                if (f.stiType && !available.includes(f.stiType)) {
-                    (0, utils_1.fail)(`${name}.stiType '${f.stiType}' is invalid, expected one of ${available.join(", ")}`);
-                }
-            });
-            // Now split each subType out into its out entity
-            for (const [enumCode, subTypeName] of subTypes) {
-                // Find all the base entity's fields that belong to us
-                const subTypeFields = allFields.filter(([, f]) => f.stiType === subTypeName || f.stiType === enumCode);
-                const subTypeFieldNames = subTypeFields.map(([name]) => name);
-                // Make fields as required
-                function maybeRequired(field) {
-                    const config = subTypeFields.find(([name]) => name === field.fieldName)?.[1];
-                    if (config.notNull)
-                        field.notNull = true;
-                    return field;
-                }
-                // Synthesize an entity for this STI sub-entity
-                const subEntity = {
-                    name: subTypeName,
-                    entity: (0, EntityDbMetadata_1.makeEntity)(subTypeName),
-                    tableName: entity.tableName,
-                    primaryKey: entity.primaryKey,
-                    primitives: entity.primitives.filter((f) => subTypeFieldNames.includes(f.fieldName)).map(maybeRequired),
-                    enums: entity.enums.filter((f) => subTypeFieldNames.includes(f.fieldName)).map(maybeRequired),
-                    pgEnums: entity.pgEnums.filter((f) => subTypeFieldNames.includes(f.fieldName)).map(maybeRequired),
-                    manyToOnes: entity.manyToOnes.filter((f) => subTypeFieldNames.includes(f.fieldName)).map(maybeRequired),
-                    oneToManys: entity.oneToManys.filter((f) => subTypeFieldNames.includes(f.fieldName)),
-                    largeOneToManys: entity.largeOneToManys.filter((f) => subTypeFieldNames.includes(f.fieldName)),
-                    oneToOnes: entity.oneToOnes.filter((f) => subTypeFieldNames.includes(f.fieldName)),
-                    manyToManys: entity.manyToManys.filter((f) => subTypeFieldNames.includes(f.fieldName)),
-                    largeManyToManys: entity.largeManyToManys.filter((f) => subTypeFieldNames.includes(f.fieldName)),
-                    manyToManyEnums: entity.manyToManyEnums.filter((f) => subTypeFieldNames.includes(f.fieldName)),
-                    polymorphics: entity.polymorphics.filter((f) => subTypeFieldNames.includes(f.fieldName)),
-                    tagName: entity.tagName,
-                    createdAt: undefined,
-                    updatedAt: undefined,
-                    deletedAt: undefined,
-                    baseClassName: entity.name,
-                    baseType: entity,
-                    subTypes: [],
-                    inheritanceType: "sti",
-                    stiDiscriminatorValue: (enumField.enumRows.find((r) => r.code === enumCode) ??
-                        (0, utils_1.fail)(`No enum row found for ${entity.name}.${fieldName}.${enumCode}`)).id,
-                    abstract: false,
-                    nonDeferredFkOrder: entity.nonDeferredFkOrder,
-                    get nonDeferredFks() {
-                        return [
-                            ...this.manyToOnes.filter((r) => !r.isDeferredAndDeferrable),
-                            ...this.polymorphics.flatMap((p) => p.components).filter((c) => !c.isDeferredAndDeferrable),
-                        ];
-                    },
-                    get nonDeferredManyToManyFks() {
-                        return this.manyToManys.filter((r) => !r.isDeferredAndDeferrable);
-                    },
-                };
-                // Now strip all the subclass fields from the base class
-                entity.primitives = entity.primitives.filter((f) => !subTypeFieldNames.includes(f.fieldName));
-                entity.enums = entity.enums.filter((f) => !subTypeFieldNames.includes(f.fieldName));
-                entity.pgEnums = entity.pgEnums.filter((f) => !subTypeFieldNames.includes(f.fieldName));
-                entity.manyToOnes = entity.manyToOnes.filter((f) => !subTypeFieldNames.includes(f.fieldName));
-                entity.oneToManys = entity.oneToManys.filter((f) => !subTypeFieldNames.includes(f.fieldName));
-                entity.largeOneToManys = entity.largeOneToManys.filter((f) => !subTypeFieldNames.includes(f.fieldName));
-                entity.oneToOnes = entity.oneToOnes.filter((f) => !subTypeFieldNames.includes(f.fieldName));
-                entity.manyToManys = entity.manyToManys.filter((f) => !subTypeFieldNames.includes(f.fieldName));
-                entity.largeManyToManys = entity.largeManyToManys.filter((f) => !subTypeFieldNames.includes(f.fieldName));
-                entity.manyToManyEnums = entity.manyToManyEnums.filter((f) => !subTypeFieldNames.includes(f.fieldName));
-                entity.polymorphics = entity.polymorphics.filter((f) => !subTypeFieldNames.includes(f.fieldName));
-                entity.subTypes.push(subEntity);
-                entities.push(subEntity);
-                entitiesByName[subEntity.name] = subEntity;
-            }
-        }
-    }
+	for (const entity of entities) {
+		const [fieldName, stiField] = Object.entries(config.entities[entity.name]?.fields || {}).find(([, f]) => !!f.stiDiscriminator) ?? [];
+		if (fieldName && stiField && stiField.stiDiscriminator) {
+			entity.inheritanceType = "sti";
+			const enumField = entity.enums.find((e) => e.fieldName === fieldName) ?? fail(`No enum column found for ${entity.name}.${fieldName}, which is required to use singleTableInheritance`);
+			entity.stiDiscriminatorField = enumField.fieldName;
+			const subTypes = Object.entries(stiField.stiDiscriminator);
+			const allFields = [...Object.entries(config.entities[entity.name]?.fields ?? {}), ...Object.entries(config.entities[entity.name]?.relations ?? {})];
+			const availableCodes = subTypes.map(([code]) => code);
+			const availableNames = subTypes.map(([, name]) => name);
+			const available = [...availableCodes, ...availableNames];
+			for (const [name, f] of allFields) for (const stiType of stiTypesOf(f)) if (!available.includes(stiType)) fail(`${name}.stiType '${stiType}' is invalid, expected one of ${available.join(", ")}`);
+			const requiredFieldNames = new Set([
+				...entity.primitives,
+				...entity.enums,
+				...entity.pgEnums,
+				...entity.manyToOnes
+			].filter((f) => f.notNull && f.columnDefault === null && !("columnGenerated" in f && f.columnGenerated)).map((f) => f.fieldName));
+			for (const [name, f] of allFields) {
+				const stiTypes = stiTypesOf(f);
+				const claimsEverySubtype = subTypes.every(([code, name]) => stiTypes.some((t) => t === code || t === name));
+				if (stiTypes.length > 0 && !claimsEverySubtype && requiredFieldNames.has(name)) fail(`${entity.name}.${name} is notNull with no database default, so it cannot be pushed down to ${stiTypes.map((t) => `'${t}'`).join(", ")} -- the other subtypes would have no value to insert. Either give the column a default, make it nullable (stiType will still make it required on those subtypes), or leave the field on ${entity.name}.`);
+			}
+			const claimedFieldNames = new Set(allFields.filter(([, f]) => stiTypesOf(f).length > 0).map(([name]) => name));
+			for (const [enumCode, subTypeName] of subTypes) {
+				const subTypeFields = allFields.filter(([, f]) => stiTypesOf(f).some((t) => t === subTypeName || t === enumCode));
+				const subTypeFieldNames = subTypeFields.map(([name]) => name);
+				function maybeRequired(field) {
+					if ((subTypeFields.find(([name]) => name === field.fieldName)?.[1]).notNull) field.notNull = true;
+					return field;
+				}
+				const subEntity = {
+					name: subTypeName,
+					entity: makeEntity(subTypeName),
+					tableName: entity.tableName,
+					physicalMetadata: entity.physicalMetadata,
+					primaryKey: entity.primaryKey,
+					primitives: entity.primitives.filter((f) => subTypeFieldNames.includes(f.fieldName)).map(maybeRequired),
+					enums: entity.enums.filter((f) => subTypeFieldNames.includes(f.fieldName)).map(maybeRequired),
+					pgEnums: entity.pgEnums.filter((f) => subTypeFieldNames.includes(f.fieldName)).map(maybeRequired),
+					manyToOnes: entity.manyToOnes.filter((f) => subTypeFieldNames.includes(f.fieldName)).map(maybeRequired),
+					oneToManys: entity.oneToManys.filter((f) => subTypeFieldNames.includes(f.fieldName)),
+					largeOneToManys: entity.largeOneToManys.filter((f) => subTypeFieldNames.includes(f.fieldName)),
+					oneToOnes: entity.oneToOnes.filter((f) => subTypeFieldNames.includes(f.fieldName)),
+					manyToManys: entity.manyToManys.filter((f) => subTypeFieldNames.includes(f.fieldName)),
+					largeManyToManys: entity.largeManyToManys.filter((f) => subTypeFieldNames.includes(f.fieldName)),
+					manyToManyEnums: entity.manyToManyEnums.filter((f) => subTypeFieldNames.includes(f.fieldName)),
+					polymorphics: entity.polymorphics.filter((f) => subTypeFieldNames.includes(f.fieldName)),
+					tagName: entity.tagName,
+					createdAt: void 0,
+					updatedAt: void 0,
+					deletedAt: void 0,
+					baseClassName: entity.name,
+					baseType: entity,
+					subTypes: [],
+					inheritanceType: "sti",
+					stiDiscriminatorValue: (enumField.enumRows.find((r) => r.code === enumCode) ?? fail(`No enum row found for ${entity.name}.${fieldName}.${enumCode}`)).id,
+					abstract: false,
+					nonDeferredFkOrder: entity.nonDeferredFkOrder,
+					get nonDeferredFks() {
+						return [...this.manyToOnes.filter((r) => !r.isDeferredAndDeferrable), ...this.polymorphics.flatMap((p) => p.components).filter((c) => !c.isDeferredAndDeferrable)];
+					},
+					get nonDeferredManyToManyFks() {
+						return this.manyToManys.filter((r) => !r.isDeferredAndDeferrable);
+					}
+				};
+				entity.subTypes.push(subEntity);
+				entities.push(subEntity);
+				entitiesByName[subEntity.name] = subEntity;
+			}
+			const keep = (f) => !claimedFieldNames.has(f.fieldName);
+			entity.primitives = entity.primitives.filter(keep);
+			entity.enums = entity.enums.filter(keep);
+			entity.pgEnums = entity.pgEnums.filter(keep);
+			entity.manyToOnes = entity.manyToOnes.filter(keep);
+			entity.oneToManys = entity.oneToManys.filter(keep);
+			entity.largeOneToManys = entity.largeOneToManys.filter(keep);
+			entity.oneToOnes = entity.oneToOnes.filter(keep);
+			entity.manyToManys = entity.manyToManys.filter(keep);
+			entity.largeManyToManys = entity.largeManyToManys.filter(keep);
+			entity.manyToManyEnums = entity.manyToManyEnums.filter(keep);
+			entity.polymorphics = entity.polymorphics.filter(keep);
+		}
+	}
 }
 let stiEntities;
 function getStiEntities(entities) {
-    if (stiEntities)
-        return stiEntities;
-    stiEntities = new Map();
-    for (const entity of entities) {
-        if (entity.inheritanceType === "sti" && entity.stiDiscriminatorField) {
-            const base = entity;
-            const subTypes = entities.filter((s) => s.baseClassName === entity.name && s !== entity);
-            stiEntities.set(entity.name, { base, subTypes });
-            // Allow looking up by subType name
-            for (const subType of subTypes) {
-                stiEntities.set(subType.name, { base, subTypes: [] });
-            }
-        }
-    }
-    return stiEntities;
+	if (stiEntities) return stiEntities;
+	stiEntities = /* @__PURE__ */ new Map();
+	for (const entity of entities) if (entity.inheritanceType === "sti" && entity.stiDiscriminatorField) {
+		const base = entity;
+		const subTypes = entities.filter((s) => s.baseClassName === entity.name && s !== entity);
+		stiEntities.set(entity.name, {
+			base,
+			subTypes
+		});
+		for (const subType of subTypes) stiEntities.set(subType.name, {
+			base,
+			subTypes: []
+		});
+	}
+	return stiEntities;
 }
 /** Finds FKs pointing to the base table and, if configured, rewrites them to point to the sub-tables. */
 function rewriteSingleTableForeignKeys(config, entities) {
-    // See if we even have any STI tables
-    const stiEntities = getStiEntities(entities);
-    if (stiEntities.size === 0)
-        return;
-    // Scan for other entities/relations that point to the STI table
-    for (const entity of entities) {
-        // m2os -- Look for `entity.task_id` FKs pointing at `Task` and, if configured, rewrite them to point at `TaskOld`
-        for (const m2o of entity.manyToOnes) {
-            const target = stiEntities.get(m2o.otherEntity.name);
-            const base = target?.base.entity.name;
-            // See if the user has pushed `Task.entities` down to a subtype
-            const stiType = base && config.entities[base]?.relations?.[m2o.otherFieldName]?.stiType;
-            if (target && stiType) {
-                const { subTypes } = target;
-                m2o.otherEntity = (subTypes.find((s) => s.name === stiType) ??
-                    (0, utils_1.fail)(`Could not find STI type '${stiType}' in ${subTypes.map((s) => s.name)}`)).entity;
-            }
-        }
-        // polys -- Look for `entity.parent_task_id` FKs pointing at `Task` and, if configured, rewrite them to point at `TaskOld`
-        for (const poly of entity.polymorphics) {
-            for (const comp of poly.components) {
-                const target = stiEntities.get(comp.otherEntity.name);
-                const base = target?.base.entity.name;
-                // See if the user has pushed `Task.entities` down to a subtype
-                const stiType = base && config.entities[base]?.relations?.[comp.otherFieldName]?.stiType;
-                if (target && stiType) {
-                    const { subTypes } = target;
-                    comp.otherEntity = (subTypes.find((s) => s.name === stiType) ??
-                        (0, utils_1.fail)(`Could not find STI type '${stiType}' in ${subTypes.map((s) => s.name)}`)).entity;
-                }
-            }
-        }
-        // o2ms -- Look for `entity.tasks` collections loading `task.entity_id`, but entity has been pushed down to `TaskOld`
-        for (const o2m of entity.oneToManys) {
-            const target = stiEntities.get(o2m.otherEntity.name);
-            if (target && target.base.inheritanceType === "sti") {
-                // Ensure the incoming FK is not in the base type, and find the 1st subtype (eventually N subtypes?)
-                const otherField = target.subTypes.find((st) => !target.base.manyToOnes.some((m) => m.fieldName === o2m.otherFieldName) &&
-                    st.manyToOnes.some((m) => m.fieldName === o2m.otherFieldName));
-                if (otherField) {
-                    o2m.otherEntity = otherField.entity;
-                }
-            }
-        }
-        // m2ms -- Look for `entity.tasks` collections loading m2m rows, but `entity` has been pushed down to `TaskOld`
-        for (const m2m of entity.manyToManys) {
-            const target = stiEntities.get(m2m.otherEntity.name);
-            if (target && target.base.inheritanceType === "sti") {
-                // Ensure the incoming FK is not in the base type, and find the 1st subtype (eventually N subtypes?)
-                const otherField = target.subTypes.find((st) => !target.base.manyToManys.some((m) => m.fieldName === m2m.otherFieldName) &&
-                    st.manyToManys.some((m) => m.fieldName === m2m.otherFieldName));
-                if (otherField) {
-                    m2m.otherEntity = otherField.entity;
-                }
-            }
-        }
-    }
+	const stiEntities = getStiEntities(entities);
+	if (stiEntities.size === 0) return;
+	for (const entity of entities) {
+		for (const m2o of entity.manyToOnes) {
+			const target = stiEntities.get(m2o.otherEntity.name);
+			const base = target?.base.entity.name;
+			const [stiType, ...others] = stiTypesOf(base ? config.entities[base]?.relations?.[m2o.otherFieldName] : void 0);
+			if (target && stiType && others.length === 0) {
+				const { subTypes } = target;
+				m2o.otherEntity = (subTypes.find((s) => s.name === stiType) ?? fail(`Could not find STI type '${stiType}' in ${subTypes.map((s) => s.name)}`)).entity;
+			}
+		}
+		for (const poly of entity.polymorphics) for (const comp of poly.components) {
+			const target = stiEntities.get(comp.otherEntity.name);
+			const base = target?.base.entity.name;
+			const [stiType, ...others] = stiTypesOf(base ? config.entities[base]?.relations?.[comp.otherFieldName] : void 0);
+			if (target && stiType && others.length === 0) {
+				const { subTypes } = target;
+				comp.otherEntity = (subTypes.find((s) => s.name === stiType) ?? fail(`Could not find STI type '${stiType}' in ${subTypes.map((s) => s.name)}`)).entity;
+			}
+		}
+		for (const o2m of entity.oneToManys) {
+			const target = stiEntities.get(o2m.otherEntity.name);
+			if (target && target.base.inheritanceType === "sti") {
+				const otherField = target.subTypes.find((st) => !target.base.manyToOnes.some((m) => m.fieldName === o2m.otherFieldName) && st.manyToOnes.some((m) => m.fieldName === o2m.otherFieldName));
+				if (otherField) o2m.otherEntity = otherField.entity;
+			}
+		}
+		for (const m2m of entity.manyToManys) {
+			const target = stiEntities.get(m2m.otherEntity.name);
+			if (target && target.base.inheritanceType === "sti") {
+				const otherField = target.subTypes.find((st) => !target.base.manyToManys.some((m) => m.fieldName === m2m.otherFieldName) && st.manyToManys.some((m) => m.fieldName === m2m.otherFieldName));
+				if (otherField) m2m.otherEntity = otherField.entity;
+			}
+		}
+	}
 }
+/** Normalizes a field/relation's `stiType`, which may name one subtype or several, to an array. */
+function stiTypesOf(config) {
+	const stiType = config?.stiType;
+	return stiType === void 0 ? [] : typeof stiType === "string" ? [stiType] : stiType;
+}
+//#endregion
+export { applyInheritanceUpdates, getStiEntities };
+
 //# sourceMappingURL=inheritance.js.map

package/build/inheritance.js.map

@@ -1 +1 @@
-{"version":3,"file":"inheritance.js","sourceRoot":"","sources":["../src/inheritance.ts"],"names":[],"mappings":";;AAYA,0DAQC;AAoMD,wCAeC;AAvOD,yDAQ4B;AAE5B,mCAA+B;AAE/B,SAAgB,uBAAuB,CAAC,MAAc,EAAE,EAAc;IACpE,MAAM,EAAE,QAAQ,EAAE,cAAc,EAAE,GAAG,EAAE,CAAC;IACxC,wBAAwB,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;IACnD,4BAA4B,CAAC,MAAM,EAAE,cAAc,EAAE,QAAQ,CAAC,CAAC;IAC/D,6BAA6B,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IAChD,0BAA0B,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IAC7C,sGAAsG;IACtG,IAAA,4CAAyB,EAAC,EAAE,CAAC,CAAC;AAChC,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,0BAA0B,CAAC,MAAc,EAAE,QAA4B;IAC9E,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;QAC9B,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YACpB,MAAM,YAAY,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,UAAU,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YACzD,2EAA2E;YAC3E,KAAK,MAAM,GAAG,IAAI,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,GAAG,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;gBACjF,MAAM,EAAE,OAAO,EAAE,GAAG,YAAY,EAAE,SAAS,EAAE,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;gBACnE,MAAM,OAAO,GACX,YAAY,EAAE,SAAS,EAAE,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,OAAO;oBACjD,UAAU,EAAE,SAAS,EAAE,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,OAAO;oBAC/C,+FAA+F;oBAC/F,UAAU,EAAE,SAAS,EAAE,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,OAAO;oBACpD,2GAA2G;oBAC3G,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;gBAE/C,IAAI,CAAC,OAAO,EAAE,CAAC;oBACb,cAAc;gBAChB,CAAC;qBAAM,IAAI,OAAO,KAAK,MAAM,IAAI,GAAG,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;oBACpD,2EAA2E;oBAC3E,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,GAAG,GAAG,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,WAAW,EAAE,IAAA,6BAAU,EAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC5G,CAAC;qBAAM,IAAI,OAAO,KAAK,MAAM,IAAI,GAAG,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;oBACpD,2EAA2E;oBAC3E,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,GAAG,GAAG,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,WAAW,EAAE,IAAA,6BAAU,EAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC5G,CAAC;qBAAM,IAAI,GAAG,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;oBAC9B,yDAAyD;oBACzD,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,GAAG,GAAG,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,WAAW,EAAE,IAAA,6BAAU,EAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBACxG,CAAC;qBAAM,IAAI,GAAG,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;oBAC9B,8DAA8D;oBAC9D,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,GAAG,GAAG,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,WAAW,EAAE,IAAA,6BAAU,EAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBACxG,CAAC;YACH,CAAC;YAED,yEAAyE;YACzE,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,QAAQ,EAAE,UAAU,IAAI,EAAE,EAAE,CAAC;gBACtD,8DAA8D;gBAC9D,MAAM,EAAE,OAAO,EAAE,GAAG,YAAY,EAAE,MAAM,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;gBAClE,IAAI,OAAO,EAAE,CAAC;oBACZ,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,GAAG,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;gBAChD,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC;AAED;;;;;GAKG;AACH,SAAS,wBAAwB,CAC/B,QAA4B,EAC5B,cAAgD;IAEhD,MAAM,YAAY,GAAa,EAAE,CAAC;IAClC,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;QAC9B,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;YACzB,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YACxC,MAAM,IAAI,GAAG,cAAc,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YAClD,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC;YACvB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;IACD,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;QAC9B,IAAI,YAAY,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC;YAAE,MAAM,CAAC,eAAe,GAAG,KAAK,CAAC;IACzE,CAAC;AACH,CAAC;AAED,gHAAgH;AAChH,SAAS,4BAA4B,CACnC,MAAc,EACd,cAAgD,EAChD,QAA4B;IAE5B,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;QAC9B,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,GACzB,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,MAAM,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,IAAI,EAAE,CAAC;QACzG,IAAI,SAAS,IAAI,QAAQ,IAAI,QAAQ,CAAC,gBAAgB,EAAE,CAAC;YACvD,MAAM,CAAC,eAAe,GAAG,KAAK,CAAC;YAE/B,oGAAoG;YACpG,MAAM,SAAS,GACb,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,SAAS,CAAC;gBACnD,IAAA,YAAI,EAAC,4BAA4B,MAAM,CAAC,IAAI,IAAI,SAAS,mDAAmD,CAAC,CAAC;YAChH,MAAM,CAAC,qBAAqB,GAAG,SAAS,CAAC,SAAS,CAAC;YAEnD,wFAAwF;YACxF,MAAM,QAAQ,GAAG,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;YAE3D,MAAM,SAAS,GAAG;gBAChB,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,MAAM,IAAI,EAAE,CAAC;gBAC7D,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,IAAI,EAAE,CAAC;aACjE,CAAC;YAEF,iDAAiD;YACjD,MAAM,cAAc,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;YACtD,MAAM,cAAc,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;YACxD,MAAM,SAAS,GAAG,CAAC,GAAG,cAAc,EAAE,GAAG,cAAc,CAAC,CAAC;YACzD,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE;gBAC7B,IAAI,CAAC,CAAC,OAAO,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;oBAChD,IAAA,YAAI,EAAC,GAAG,IAAI,aAAa,CAAC,CAAC,OAAO,iCAAiC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC7F,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,iDAAiD;YACjD,KAAK,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,IAAI,QAAQ,EAAE,CAAC;gBAC/C,sDAAsD;gBACtD,MAAM,aAAa,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,KAAK,WAAW,IAAI,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC;gBACvG,MAAM,iBAAiB,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;gBAE9D,0BAA0B;gBAC1B,SAAS,aAAa,CAAoD,KAAQ;oBAChF,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,IAAI,KAAK,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAE,CAAC;oBAC9E,IAAI,MAAM,CAAC,OAAO;wBAAE,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC;oBACzC,OAAO,KAAK,CAAC;gBACf,CAAC;gBAED,+CAA+C;gBAC/C,MAAM,SAAS,GAAqB;oBAClC,IAAI,EAAE,WAAW;oBACjB,MAAM,EAAE,IAAA,6BAAU,EAAC,WAAW,CAAC;oBAC/B,SAAS,EAAE,MAAM,CAAC,SAAS;oBAC3B,UAAU,EAAE,MAAM,CAAC,UAAU;oBAC7B,UAAU,EAAE,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,aAAa,CAAC;oBACvG,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,aAAa,CAAC;oBAC7F,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,aAAa,CAAC;oBACjG,UAAU,EAAE,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,aAAa,CAAC;oBACvG,UAAU,EAAE,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;oBACpF,eAAe,EAAE,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;oBAC9F,SAAS,EAAE,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;oBAClF,WAAW,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;oBACtF,gBAAgB,EAAE,MAAM,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;oBAChG,eAAe,EAAE,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;oBAC9F,YAAY,EAAE,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;oBACxF,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,SAAS,EAAE,SAAS;oBACpB,SAAS,EAAE,SAAS;oBACpB,SAAS,EAAE,SAAS;oBACpB,aAAa,EAAE,MAAM,CAAC,IAAI;oBAC1B,QAAQ,EAAE,MAAM;oBAChB,QAAQ,EAAE,EAAE;oBACZ,eAAe,EAAE,KAAK;oBACtB,qBAAqB,EAAE,CACrB,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;wBACnD,IAAA,YAAI,EAAC,yBAAyB,MAAM,CAAC,IAAI,IAAI,SAAS,IAAI,QAAQ,EAAE,CAAC,CACtE,CAAC,EAAE;oBACJ,QAAQ,EAAE,KAAK;oBACf,kBAAkB,EAAE,MAAM,CAAC,kBAAkB;oBAC7C,IAAI,cAAc;wBAChB,OAAO;4BACL,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,uBAAuB,CAAC;4BAC5D,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,uBAAuB,CAAC;yBAC5F,CAAC;oBACJ,CAAC;oBACD,IAAI,wBAAwB;wBAC1B,OAAO,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,uBAAuB,CAAC,CAAC;oBACpE,CAAC;iBACF,CAAC;gBAEF,wDAAwD;gBACxD,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;gBAC9F,MAAM,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;gBACpF,MAAM,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;gBACxF,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;gBAC9F,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;gBAC9F,MAAM,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;gBACxG,MAAM,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;gBAC5F,MAAM,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;gBAChG,MAAM,CAAC,gBAAgB,GAAG,MAAM,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;gBAC1G,MAAM,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;gBACxG,MAAM,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;gBAClG,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAEhC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACzB,cAAc,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC;YAC7C,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC;AAGD,IAAI,WAAyB,CAAC;AAE9B,SAAgB,cAAc,CAAC,QAA4B;IACzD,IAAI,WAAW;QAAE,OAAO,WAAW,CAAC;IACpC,WAAW,GAAG,IAAI,GAAG,EAAE,CAAC;IACxB,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;QAC9B,IAAI,MAAM,CAAC,eAAe,KAAK,KAAK,IAAI,MAAM,CAAC,qBAAqB,EAAE,CAAC;YACrE,MAAM,IAAI,GAAG,MAAM,CAAC;YACpB,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,KAAK,MAAM,CAAC,CAAC;YACzF,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;YACjD,mCAAmC;YACnC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC/B,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;YACxD,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,yGAAyG;AACzG,SAAS,6BAA6B,CAAC,MAAc,EAAE,QAA4B;IACjF,qCAAqC;IACrC,MAAM,WAAW,GAAG,cAAc,CAAC,QAAQ,CAAC,CAAC;IAC7C,IAAI,WAAW,CAAC,IAAI,KAAK,CAAC;QAAE,OAAO;IACnC,gEAAgE;IAChE,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;QAC9B,kHAAkH;QAClH,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;YACpC,MAAM,MAAM,GAAG,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACrD,MAAM,IAAI,GAAG,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;YACtC,+DAA+D;YAC/D,MAAM,OAAO,GAAG,IAAI,IAAI,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,OAAO,CAAC;YACxF,IAAI,MAAM,IAAI,OAAO,EAAE,CAAC;gBACtB,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC;gBAC5B,GAAG,CAAC,WAAW,GAAG,CAChB,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC;oBACxC,IAAA,YAAI,EAAC,4BAA4B,OAAO,QAAQ,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAC/E,CAAC,MAAM,CAAC;YACX,CAAC;QACH,CAAC;QACD,0HAA0H;QAC1H,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;YACvC,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBACnC,MAAM,MAAM,GAAG,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;gBACtD,MAAM,IAAI,GAAG,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;gBACtC,+DAA+D;gBAC/D,MAAM,OAAO,GAAG,IAAI,IAAI,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,EAAE,OAAO,CAAC;gBACzF,IAAI,MAAM,IAAI,OAAO,EAAE,CAAC;oBACtB,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC;oBAC5B,IAAI,CAAC,WAAW,GAAG,CACjB,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC;wBACxC,IAAA,YAAI,EAAC,4BAA4B,OAAO,QAAQ,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAC/E,CAAC,MAAM,CAAC;gBACX,CAAC;YACH,CAAC;QACH,CAAC;QACD,qHAAqH;QACrH,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;YACpC,MAAM,MAAM,GAAG,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACrD,IAAI,MAAM,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,KAAK,KAAK,EAAE,CAAC;gBACpD,oGAAoG;gBACpG,MAAM,UAAU,GAAG,MAAM,CAAC,QAAQ,CAAC,IAAI,CACrC,CAAC,EAAE,EAAE,EAAE,CACL,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,GAAG,CAAC,cAAc,CAAC;oBACvE,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,GAAG,CAAC,cAAc,CAAC,CAChE,CAAC;gBACF,IAAI,UAAU,EAAE,CAAC;oBACf,GAAG,CAAC,WAAW,GAAG,UAAU,CAAC,MAAM,CAAC;gBACtC,CAAC;YACH,CAAC;QACH,CAAC;QACD,+GAA+G;QAC/G,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;YACrC,MAAM,MAAM,GAAG,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACrD,IAAI,MAAM,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,KAAK,KAAK,EAAE,CAAC;gBACpD,oGAAoG;gBACpG,MAAM,UAAU,GAAG,MAAM,CAAC,QAAQ,CAAC,IAAI,CACrC,CAAC,EAAE,EAAE,EAAE,CACL,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,GAAG,CAAC,cAAc,CAAC;oBACxE,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,GAAG,CAAC,cAAc,CAAC,CACjE,CAAC;gBACF,IAAI,UAAU,EAAE,CAAC;oBACf,GAAG,CAAC,WAAW,GAAG,UAAU,CAAC,MAAM,CAAC;gBACtC,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC"}
+{"version":3,"file":"inheritance.js","names":[],"sources":["../src/inheritance.ts"],"sourcesContent":["import { type Config } from \"./config.ts\";\nimport {\n  type DbMetadata,\n  type EntityDbMetadata,\n  type ManyToManyField,\n  type ManyToOneField,\n  type PolymorphicFieldComponent,\n  canonicalizeOtherEntities,\n  makeEntity,\n} from \"./EntityDbMetadata.ts\";\nimport { fail } from \"./utils.ts\";\n\n/** Preserves physical storage metadata, then applies domain inheritance specialization. */\nexport function applyInheritanceUpdates(config: Config, db: DbMetadata): void {\n  const { entities, entitiesByName } = db;\n  // Preserve storage domains before STI moves fields or rewrites relationship targets.\n  for (const meta of entities) {\n    meta.physicalMetadata = {\n      ...meta,\n      name: meta.name,\n      primitives: meta.primitives.map((field) => ({ ...field })),\n      enums: meta.enums.map((field) => ({ ...field })),\n      pgEnums: meta.pgEnums.map((field) => ({ ...field })),\n      manyToOnes: meta.manyToOnes.map((field) => ({ ...field })),\n      oneToManys: meta.oneToManys.map((field) => ({ ...field })),\n      largeOneToManys: meta.largeOneToManys.map((field) => ({ ...field })),\n      oneToOnes: meta.oneToOnes.map((field) => ({ ...field })),\n      manyToManys: meta.manyToManys.map((field) => ({ ...field })),\n      largeManyToManys: meta.largeManyToManys.map((field) => ({ ...field })),\n      polymorphics: meta.polymorphics.map((field) => ({\n        ...field,\n        components: field.components.map((component) => ({ ...component })),\n      })),\n      nonDeferredFks: meta.nonDeferredFks,\n      nonDeferredManyToManyFks: meta.nonDeferredManyToManyFks,\n    };\n    for (const timestamp of [\"createdAt\", \"updatedAt\", \"deletedAt\"] as const) {\n      meta.physicalMetadata[timestamp] = meta.physicalMetadata.primitives.find(\n        (field) => field.fieldName === meta[timestamp]?.fieldName,\n      );\n    }\n  }\n  setClassTableInheritance(entities, entitiesByName);\n  expandSingleTableInheritance(config, entitiesByName, entities);\n  rewriteSingleTableForeignKeys(config, entities);\n  setupSubTypeSpecialization(config, entities);\n  // Now that all relations (incl. specialized ones) exist, collapse the duplicate `otherEntity` objects\n  canonicalizeOtherEntities(db);\n}\n\n/**\n * Looks for subtypes specializing a FK of their base type, and give them a specialized m2o.\n *\n * I.e. `SmallPublishers.group: SmallPublisherGroup`.\n *\n * This means the `publishers.group_id` FK will actually be in the `EntityDbMetadata` twice,\n * once for the base and once for the child, which is unusual, but gets the types generated\n * that we want, and then we fixup/dedupe them at runtime in `configure.ts` when creating\n * `meta.allFields`.\n */\nfunction setupSubTypeSpecialization(config: Config, entities: EntityDbMetadata[]): void {\n  for (const entity of entities) {\n    if (entity.baseType) {\n      const entityConfig = config.entities[entity.name];\n      const baseConfig = config.entities[entity.baseType.name];\n      // Look through the base's m2os & o2ms, looking for a config specialization\n      for (const rel of [...entity.baseType.manyToOnes, ...entity.baseType.oneToManys]) {\n        const { notNull } = entityConfig?.relations?.[rel.fieldName] ?? {};\n        const subType =\n          entityConfig?.relations?.[rel.fieldName]?.subType ??\n          baseConfig?.relations?.[rel.fieldName]?.subType ??\n          // Probe the otherFieldName for a `subType: \"self\"` on self-referential relations in STI tables\n          baseConfig?.relations?.[rel.otherFieldName]?.subType ??\n          // if we are only marking the relation as required and not specifying a subtype, then use the original type\n          (notNull ? rel.otherEntity.name : undefined);\n\n        if (!subType) {\n          // continue...\n        } else if (subType === \"self\" && rel.kind === \"m2o\") {\n          // Specialize `TaskNew.copiedFrom: TaskNew` & `TaskOld.copiedFrom: TaskOld`\n          entity.manyToOnes.push({ ...rel, ...(notNull ? { notNull } : {}), otherEntity: makeEntity(entity.name) });\n        } else if (subType === \"self\" && rel.kind === \"o2m\") {\n          // Specialize `TaskNew.copiedTo: TaskNew[]` & `TaskOld.copiedTo: TaskOld[]`\n          entity.oneToManys.push({ ...rel, ...(notNull ? { notNull } : {}), otherEntity: makeEntity(entity.name) });\n        } else if (rel.kind === \"m2o\") {\n          // Specialize `SmallPublisher.group: SmallPublisherGroup`\n          entity.manyToOnes.push({ ...rel, ...(notNull ? { notNull } : {}), otherEntity: makeEntity(subType) });\n        } else if (rel.kind === \"o2m\") {\n          // Specialize `SmallPublisherGroup.publishers: SmallPublisher`\n          entity.oneToManys.push({ ...rel, ...(notNull ? { notNull } : {}), otherEntity: makeEntity(subType) });\n        }\n      }\n\n      // Look through the base's primitives looking for a config specialization\n      for (const field of entity.baseType?.primitives ?? []) {\n        // the only specialization a primitive can have is nullability\n        const { notNull } = entityConfig?.fields?.[field.fieldName] ?? {};\n        if (notNull) {\n          entity.primitives.push({ ...field, notNull });\n        }\n      }\n    }\n  }\n}\n\n/**\n * Ensure CTI base types have their inheritanceType set.\n *\n * (We automatically set `inheritanceType` for STI tables when we see their config\n * setup, see `expandSingleTableInheritance`.)\n */\nfunction setClassTableInheritance(\n  entities: EntityDbMetadata[],\n  entitiesByName: Record<string, EntityDbMetadata>,\n): void {\n  const ctiBaseNames: string[] = [];\n  for (const entity of entities) {\n    if (entity.baseClassName) {\n      ctiBaseNames.push(entity.baseClassName);\n      const base = entitiesByName[entity.baseClassName];\n      entity.baseType = base;\n      base.subTypes.push(entity);\n    }\n  }\n  for (const entity of entities) {\n    if (ctiBaseNames.includes(entity.name)) entity.inheritanceType = \"cti\";\n  }\n}\n\n/** Expands STI tables into multiple entities, so they get separate `SubTypeCodegen.ts` & `SubType.ts` files. */\nfunction expandSingleTableInheritance(\n  config: Config,\n  entitiesByName: Record<string, EntityDbMetadata>,\n  entities: EntityDbMetadata[],\n): void {\n  for (const entity of entities) {\n    const [fieldName, stiField] =\n      Object.entries(config.entities[entity.name]?.fields || {}).find(([, f]) => !!f.stiDiscriminator) ?? [];\n    if (fieldName && stiField && stiField.stiDiscriminator) {\n      entity.inheritanceType = \"sti\";\n\n      // Ensure we have an enum field so that we can bake the STI discriminators into the metadata.ts file\n      const enumField =\n        entity.enums.find((e) => e.fieldName === fieldName) ??\n        fail(`No enum column found for ${entity.name}.${fieldName}, which is required to use singleTableInheritance`);\n      entity.stiDiscriminatorField = enumField.fieldName;\n\n      // Find the available discriminators/subtypes, i.e. NEW => NewTask, OLD => OldTask, etc.\n      const subTypes = Object.entries(stiField.stiDiscriminator);\n\n      const allFields = [\n        ...Object.entries(config.entities[entity.name]?.fields ?? {}),\n        ...Object.entries(config.entities[entity.name]?.relations ?? {}),\n      ];\n\n      // Make sure there aren't any typos in `stiType`s\n      const availableCodes = subTypes.map(([code]) => code);\n      const availableNames = subTypes.map(([, name]) => name);\n      const available = [...availableCodes, ...availableNames];\n      for (const [name, f] of allFields) {\n        for (const stiType of stiTypesOf(f)) {\n          if (!available.includes(stiType)) {\n            fail(`${name}.stiType '${stiType}' is invalid, expected one of ${available.join(\", \")}`);\n          }\n        }\n      }\n\n      // A `notNull` column with no database default cannot be pushed down to a subtype: the other\n      // subtypes omit it from their INSERT (see `addInserts`), and the database has nothing to fall back\n      // on. A `notNull` column *with* a default is fine -- the other subtypes simply get the default.\n      const requiredFieldNames = new Set(\n        [...entity.primitives, ...entity.enums, ...entity.pgEnums, ...entity.manyToOnes]\n          .filter((f) => f.notNull && f.columnDefault === null && !(\"columnGenerated\" in f && f.columnGenerated))\n          .map((f) => f.fieldName),\n      );\n      for (const [name, f] of allFields) {\n        const stiTypes = stiTypesOf(f);\n        // Claiming every subtype leaves no one without a value, so it's equivalent to staying on the base\n        const claimsEverySubtype = subTypes.every(([code, name]) => stiTypes.some((t) => t === code || t === name));\n        if (stiTypes.length > 0 && !claimsEverySubtype && requiredFieldNames.has(name)) {\n          fail(\n            `${entity.name}.${name} is notNull with no database default, so it cannot be pushed down to` +\n              ` ${stiTypes.map((t) => `'${t}'`).join(\", \")} -- the other subtypes would have no value to` +\n              ` insert. Either give the column a default, make it nullable (stiType will still make it` +\n              ` required on those subtypes), or leave the field on ${entity.name}.`,\n          );\n        }\n      }\n\n      // Names claimed by any subtype; stripped from the base after the loop, not during it, so that a\n      // field claimed by two subtypes is still on the base when the second iteration looks for it.\n      const claimedFieldNames = new Set(allFields.filter(([, f]) => stiTypesOf(f).length > 0).map(([name]) => name));\n\n      // Now split each subType out into its out entity\n      for (const [enumCode, subTypeName] of subTypes) {\n        // Find all the base entity's fields that belong to us\n        const subTypeFields = allFields.filter(([, f]) =>\n          stiTypesOf(f).some((t) => t === subTypeName || t === enumCode),\n        );\n        const subTypeFieldNames = subTypeFields.map(([name]) => name);\n\n        // Make fields as required\n        function maybeRequired<T extends { notNull: boolean; fieldName: string }>(field: T): T {\n          const config = subTypeFields.find(([name]) => name === field.fieldName)?.[1]!;\n          if (config.notNull) field.notNull = true;\n          return field;\n        }\n\n        // Synthesize an entity for this STI sub-entity\n        const subEntity: EntityDbMetadata = {\n          name: subTypeName,\n          entity: makeEntity(subTypeName),\n          tableName: entity.tableName,\n          physicalMetadata: entity.physicalMetadata,\n          primaryKey: entity.primaryKey,\n          primitives: entity.primitives.filter((f) => subTypeFieldNames.includes(f.fieldName)).map(maybeRequired),\n          enums: entity.enums.filter((f) => subTypeFieldNames.includes(f.fieldName)).map(maybeRequired),\n          pgEnums: entity.pgEnums.filter((f) => subTypeFieldNames.includes(f.fieldName)).map(maybeRequired),\n          manyToOnes: entity.manyToOnes.filter((f) => subTypeFieldNames.includes(f.fieldName)).map(maybeRequired),\n          oneToManys: entity.oneToManys.filter((f) => subTypeFieldNames.includes(f.fieldName)),\n          largeOneToManys: entity.largeOneToManys.filter((f) => subTypeFieldNames.includes(f.fieldName)),\n          oneToOnes: entity.oneToOnes.filter((f) => subTypeFieldNames.includes(f.fieldName)),\n          manyToManys: entity.manyToManys.filter((f) => subTypeFieldNames.includes(f.fieldName)),\n          largeManyToManys: entity.largeManyToManys.filter((f) => subTypeFieldNames.includes(f.fieldName)),\n          manyToManyEnums: entity.manyToManyEnums.filter((f) => subTypeFieldNames.includes(f.fieldName)),\n          polymorphics: entity.polymorphics.filter((f) => subTypeFieldNames.includes(f.fieldName)),\n          tagName: entity.tagName,\n          createdAt: undefined,\n          updatedAt: undefined,\n          deletedAt: undefined,\n          baseClassName: entity.name,\n          baseType: entity,\n          subTypes: [],\n          inheritanceType: \"sti\",\n          stiDiscriminatorValue: (\n            enumField.enumRows.find((r) => r.code === enumCode) ??\n            fail(`No enum row found for ${entity.name}.${fieldName}.${enumCode}`)\n          ).id,\n          abstract: false,\n          nonDeferredFkOrder: entity.nonDeferredFkOrder,\n          get nonDeferredFks(): Array<ManyToOneField | PolymorphicFieldComponent> {\n            return [\n              ...this.manyToOnes.filter((r) => !r.isDeferredAndDeferrable),\n              ...this.polymorphics.flatMap((p) => p.components).filter((c) => !c.isDeferredAndDeferrable),\n            ];\n          },\n          get nonDeferredManyToManyFks(): Array<ManyToManyField> {\n            return this.manyToManys.filter((r) => !r.isDeferredAndDeferrable);\n          },\n        };\n\n        entity.subTypes.push(subEntity);\n\n        entities.push(subEntity);\n        entitiesByName[subEntity.name] = subEntity;\n      }\n\n      // Now strip the subclass fields from the base class\n      const keep = <T extends { fieldName: string }>(f: T) => !claimedFieldNames.has(f.fieldName);\n      entity.primitives = entity.primitives.filter(keep);\n      entity.enums = entity.enums.filter(keep);\n      entity.pgEnums = entity.pgEnums.filter(keep);\n      entity.manyToOnes = entity.manyToOnes.filter(keep);\n      entity.oneToManys = entity.oneToManys.filter(keep);\n      entity.largeOneToManys = entity.largeOneToManys.filter(keep);\n      entity.oneToOnes = entity.oneToOnes.filter(keep);\n      entity.manyToManys = entity.manyToManys.filter(keep);\n      entity.largeManyToManys = entity.largeManyToManys.filter(keep);\n      entity.manyToManyEnums = entity.manyToManyEnums.filter(keep);\n      entity.polymorphics = entity.polymorphics.filter(keep);\n    }\n  }\n}\n\ntype StiEntityMap = Map<string, { base: EntityDbMetadata; subTypes: EntityDbMetadata[] }>;\nlet stiEntities: StiEntityMap;\n\nexport function getStiEntities(entities: EntityDbMetadata[]): StiEntityMap {\n  if (stiEntities) return stiEntities;\n  stiEntities = new Map();\n  for (const entity of entities) {\n    if (entity.inheritanceType === \"sti\" && entity.stiDiscriminatorField) {\n      const base = entity;\n      const subTypes = entities.filter((s) => s.baseClassName === entity.name && s !== entity);\n      stiEntities.set(entity.name, { base, subTypes });\n      // Allow looking up by subType name\n      for (const subType of subTypes) {\n        stiEntities.set(subType.name, { base, subTypes: [] });\n      }\n    }\n  }\n  return stiEntities;\n}\n\n/** Finds FKs pointing to the base table and, if configured, rewrites them to point to the sub-tables. */\nfunction rewriteSingleTableForeignKeys(config: Config, entities: EntityDbMetadata[]): void {\n  // See if we even have any STI tables\n  const stiEntities = getStiEntities(entities);\n  if (stiEntities.size === 0) return;\n  // Scan for other entities/relations that point to the STI table\n  for (const entity of entities) {\n    // m2os -- Look for `entity.task_id` FKs pointing at `Task` and, if configured, rewrite them to point at `TaskOld`\n    for (const m2o of entity.manyToOnes) {\n      const target = stiEntities.get(m2o.otherEntity.name);\n      const base = target?.base.entity.name;\n      // See if the user has pushed `Task.entities` down to a subtype\n      // Only a lone subtype can retype the FK; naming several means the FK can point at any of them,\n      // which is just the base type, so leave it alone.\n      const [stiType, ...others] = stiTypesOf(\n        base ? config.entities[base]?.relations?.[m2o.otherFieldName] : undefined,\n      );\n      if (target && stiType && others.length === 0) {\n        const { subTypes } = target;\n        m2o.otherEntity = (\n          subTypes.find((s) => s.name === stiType) ??\n          fail(`Could not find STI type '${stiType}' in ${subTypes.map((s) => s.name)}`)\n        ).entity;\n      }\n    }\n    // polys -- Look for `entity.parent_task_id` FKs pointing at `Task` and, if configured, rewrite them to point at `TaskOld`\n    for (const poly of entity.polymorphics) {\n      for (const comp of poly.components) {\n        const target = stiEntities.get(comp.otherEntity.name);\n        const base = target?.base.entity.name;\n        // See if the user has pushed `Task.entities` down to a subtype\n        const [stiType, ...others] = stiTypesOf(\n          base ? config.entities[base]?.relations?.[comp.otherFieldName] : undefined,\n        );\n        if (target && stiType && others.length === 0) {\n          const { subTypes } = target;\n          comp.otherEntity = (\n            subTypes.find((s) => s.name === stiType) ??\n            fail(`Could not find STI type '${stiType}' in ${subTypes.map((s) => s.name)}`)\n          ).entity;\n        }\n      }\n    }\n    // o2ms -- Look for `entity.tasks` collections loading `task.entity_id`, but entity has been pushed down to `TaskOld`\n    for (const o2m of entity.oneToManys) {\n      const target = stiEntities.get(o2m.otherEntity.name);\n      if (target && target.base.inheritanceType === \"sti\") {\n        // Ensure the incoming FK is not in the base type, and find the 1st subtype (eventually N subtypes?)\n        const otherField = target.subTypes.find(\n          (st) =>\n            !target.base.manyToOnes.some((m) => m.fieldName === o2m.otherFieldName) &&\n            st.manyToOnes.some((m) => m.fieldName === o2m.otherFieldName),\n        );\n        if (otherField) {\n          o2m.otherEntity = otherField.entity;\n        }\n      }\n    }\n    // m2ms -- Look for `entity.tasks` collections loading m2m rows, but `entity` has been pushed down to `TaskOld`\n    for (const m2m of entity.manyToManys) {\n      const target = stiEntities.get(m2m.otherEntity.name);\n      if (target && target.base.inheritanceType === \"sti\") {\n        // Ensure the incoming FK is not in the base type, and find the 1st subtype (eventually N subtypes?)\n        const otherField = target.subTypes.find(\n          (st) =>\n            !target.base.manyToManys.some((m) => m.fieldName === m2m.otherFieldName) &&\n            st.manyToManys.some((m) => m.fieldName === m2m.otherFieldName),\n        );\n        if (otherField) {\n          m2m.otherEntity = otherField.entity;\n        }\n      }\n    }\n  }\n}\n\n/** Normalizes a field/relation's `stiType`, which may name one subtype or several, to an array. */\nfunction stiTypesOf(config: { stiType?: string | string[] } | undefined): string[] {\n  const stiType = config?.stiType;\n  return stiType === undefined ? [] : typeof stiType === \"string\" ? [stiType] : stiType;\n}\n"],"mappings":";;;;;AAaA,SAAgB,wBAAwB,QAAgB,IAAsB;CAC5E,MAAM,EAAE,UAAU,mBAAmB;CAErC,KAAK,MAAM,QAAQ,UAAU;EAC3B,KAAK,mBAAmB;GACtB,GAAG;GACH,MAAM,KAAK;GACX,YAAY,KAAK,WAAW,KAAK,WAAW,EAAE,GAAG,MAAM,EAAE;GACzD,OAAO,KAAK,MAAM,KAAK,WAAW,EAAE,GAAG,MAAM,EAAE;GAC/C,SAAS,KAAK,QAAQ,KAAK,WAAW,EAAE,GAAG,MAAM,EAAE;GACnD,YAAY,KAAK,WAAW,KAAK,WAAW,EAAE,GAAG,MAAM,EAAE;GACzD,YAAY,KAAK,WAAW,KAAK,WAAW,EAAE,GAAG,MAAM,EAAE;GACzD,iBAAiB,KAAK,gBAAgB,KAAK,WAAW,EAAE,GAAG,MAAM,EAAE;GACnE,WAAW,KAAK,UAAU,KAAK,WAAW,EAAE,GAAG,MAAM,EAAE;GACvD,aAAa,KAAK,YAAY,KAAK,WAAW,EAAE,GAAG,MAAM,EAAE;GAC3D,kBAAkB,KAAK,iBAAiB,KAAK,WAAW,EAAE,GAAG,MAAM,EAAE;GACrE,cAAc,KAAK,aAAa,KAAK,WAAW;IAC9C,GAAG;IACH,YAAY,MAAM,WAAW,KAAK,eAAe,EAAE,GAAG,UAAU,EAAE;GACpE,EAAE;GACF,gBAAgB,KAAK;GACrB,0BAA0B,KAAK;EACjC;EACA,KAAK,MAAM,aAAa;GAAC;GAAa;GAAa;EAAW,GAC5D,KAAK,iBAAiB,aAAa,KAAK,iBAAiB,WAAW,MACjE,UAAU,MAAM,cAAc,KAAK,UAAU,EAAE,SAClD;CAEJ;CACA,yBAAyB,UAAU,cAAc;CACjD,6BAA6B,QAAQ,gBAAgB,QAAQ;CAC7D,8BAA8B,QAAQ,QAAQ;CAC9C,2BAA2B,QAAQ,QAAQ;CAE3C,0BAA0B,EAAE;AAC9B;;;;;;;;;;;AAYA,SAAS,2BAA2B,QAAgB,UAAoC;CACtF,KAAK,MAAM,UAAU,UACnB,IAAI,OAAO,UAAU;EACnB,MAAM,eAAe,OAAO,SAAS,OAAO;EAC5C,MAAM,aAAa,OAAO,SAAS,OAAO,SAAS;EAEnD,KAAK,MAAM,OAAO,CAAC,GAAG,OAAO,SAAS,YAAY,GAAG,OAAO,SAAS,UAAU,GAAG;GAChF,MAAM,EAAE,YAAY,cAAc,YAAY,IAAI,cAAc,CAAC;GACjE,MAAM,UACJ,cAAc,YAAY,IAAI,UAAU,EAAE,WAC1C,YAAY,YAAY,IAAI,UAAU,EAAE,WAExC,YAAY,YAAY,IAAI,eAAe,EAAE,YAE5C,UAAU,IAAI,YAAY,OAAO,KAAA;GAEpC,IAAI,CAAC,SAAS,CAEd,OAAO,IAAI,YAAY,UAAU,IAAI,SAAS,OAE5C,OAAO,WAAW,KAAK;IAAE,GAAG;IAAK,GAAI,UAAU,EAAE,QAAQ,IAAI,CAAC;IAAI,aAAa,WAAW,OAAO,IAAI;GAAE,CAAC;QACnG,IAAI,YAAY,UAAU,IAAI,SAAS,OAE5C,OAAO,WAAW,KAAK;IAAE,GAAG;IAAK,GAAI,UAAU,EAAE,QAAQ,IAAI,CAAC;IAAI,aAAa,WAAW,OAAO,IAAI;GAAE,CAAC;QACnG,IAAI,IAAI,SAAS,OAEtB,OAAO,WAAW,KAAK;IAAE,GAAG;IAAK,GAAI,UAAU,EAAE,QAAQ,IAAI,CAAC;IAAI,aAAa,WAAW,OAAO;GAAE,CAAC;QAC/F,IAAI,IAAI,SAAS,OAEtB,OAAO,WAAW,KAAK;IAAE,GAAG;IAAK,GAAI,UAAU,EAAE,QAAQ,IAAI,CAAC;IAAI,aAAa,WAAW,OAAO;GAAE,CAAC;EAExG;EAGA,KAAK,MAAM,SAAS,OAAO,UAAU,cAAc,CAAC,GAAG;GAErD,MAAM,EAAE,YAAY,cAAc,SAAS,MAAM,cAAc,CAAC;GAChE,IAAI,SACF,OAAO,WAAW,KAAK;IAAE,GAAG;IAAO;GAAQ,CAAC;EAEhD;CACF;AAEJ;;;;;;;AAQA,SAAS,yBACP,UACA,gBACM;CACN,MAAM,eAAyB,CAAC;CAChC,KAAK,MAAM,UAAU,UACnB,IAAI,OAAO,eAAe;EACxB,aAAa,KAAK,OAAO,aAAa;EACtC,MAAM,OAAO,eAAe,OAAO;EACnC,OAAO,WAAW;EAClB,KAAK,SAAS,KAAK,MAAM;CAC3B;CAEF,KAAK,MAAM,UAAU,UACnB,IAAI,aAAa,SAAS,OAAO,IAAI,GAAG,OAAO,kBAAkB;AAErE;;AAGA,SAAS,6BACP,QACA,gBACA,UACM;CACN,KAAK,MAAM,UAAU,UAAU;EAC7B,MAAM,CAAC,WAAW,YAChB,OAAO,QAAQ,OAAO,SAAS,OAAO,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,OAAO,CAAC,CAAC,EAAE,gBAAgB,KAAK,CAAC;EACvG,IAAI,aAAa,YAAY,SAAS,kBAAkB;GACtD,OAAO,kBAAkB;GAGzB,MAAM,YACJ,OAAO,MAAM,MAAM,MAAM,EAAE,cAAc,SAAS,KAClD,KAAK,4BAA4B,OAAO,KAAK,GAAG,UAAU,kDAAkD;GAC9G,OAAO,wBAAwB,UAAU;GAGzC,MAAM,WAAW,OAAO,QAAQ,SAAS,gBAAgB;GAEzD,MAAM,YAAY,CAChB,GAAG,OAAO,QAAQ,OAAO,SAAS,OAAO,KAAK,EAAE,UAAU,CAAC,CAAC,GAC5D,GAAG,OAAO,QAAQ,OAAO,SAAS,OAAO,KAAK,EAAE,aAAa,CAAC,CAAC,CACjE;GAGA,MAAM,iBAAiB,SAAS,KAAK,CAAC,UAAU,IAAI;GACpD,MAAM,iBAAiB,SAAS,KAAK,GAAG,UAAU,IAAI;GACtD,MAAM,YAAY,CAAC,GAAG,gBAAgB,GAAG,cAAc;GACvD,KAAK,MAAM,CAAC,MAAM,MAAM,WACtB,KAAK,MAAM,WAAW,WAAW,CAAC,GAChC,IAAI,CAAC,UAAU,SAAS,OAAO,GAC7B,KAAK,GAAG,KAAK,YAAY,QAAQ,gCAAgC,UAAU,KAAK,IAAI,GAAG;GAQ7F,MAAM,qBAAqB,IAAI,IAC7B;IAAC,GAAG,OAAO;IAAY,GAAG,OAAO;IAAO,GAAG,OAAO;IAAS,GAAG,OAAO;GAAU,CAAC,CAC7E,QAAQ,MAAM,EAAE,WAAW,EAAE,kBAAkB,QAAQ,EAAE,qBAAqB,KAAK,EAAE,gBAAgB,CAAC,CACtG,KAAK,MAAM,EAAE,SAAS,CAC3B;GACA,KAAK,MAAM,CAAC,MAAM,MAAM,WAAW;IACjC,MAAM,WAAW,WAAW,CAAC;IAE7B,MAAM,qBAAqB,SAAS,OAAO,CAAC,MAAM,UAAU,SAAS,MAAM,MAAM,MAAM,QAAQ,MAAM,IAAI,CAAC;IAC1G,IAAI,SAAS,SAAS,KAAK,CAAC,sBAAsB,mBAAmB,IAAI,IAAI,GAC3E,KACE,GAAG,OAAO,KAAK,GAAG,KAAK,uEACjB,SAAS,KAAK,MAAM,IAAI,EAAE,EAAE,CAAC,CAAC,KAAK,IAAI,EAAE,0LAEU,OAAO,KAAK,EACvE;GAEJ;GAIA,MAAM,oBAAoB,IAAI,IAAI,UAAU,QAAQ,GAAG,OAAO,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,UAAU,IAAI,CAAC;GAG7G,KAAK,MAAM,CAAC,UAAU,gBAAgB,UAAU;IAE9C,MAAM,gBAAgB,UAAU,QAAQ,GAAG,OACzC,WAAW,CAAC,CAAC,CAAC,MAAM,MAAM,MAAM,eAAe,MAAM,QAAQ,CAC/D;IACA,MAAM,oBAAoB,cAAc,KAAK,CAAC,UAAU,IAAI;IAG5D,SAAS,cAAiE,OAAa;KAErF,KADe,cAAc,MAAM,CAAC,UAAU,SAAS,MAAM,SAAS,CAAC,GAAG,GAAA,CAC/D,SAAS,MAAM,UAAU;KACpC,OAAO;IACT;IAGA,MAAM,YAA8B;KAClC,MAAM;KACN,QAAQ,WAAW,WAAW;KAC9B,WAAW,OAAO;KAClB,kBAAkB,OAAO;KACzB,YAAY,OAAO;KACnB,YAAY,OAAO,WAAW,QAAQ,MAAM,kBAAkB,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,aAAa;KACtG,OAAO,OAAO,MAAM,QAAQ,MAAM,kBAAkB,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,aAAa;KAC5F,SAAS,OAAO,QAAQ,QAAQ,MAAM,kBAAkB,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,aAAa;KAChG,YAAY,OAAO,WAAW,QAAQ,MAAM,kBAAkB,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,aAAa;KACtG,YAAY,OAAO,WAAW,QAAQ,MAAM,kBAAkB,SAAS,EAAE,SAAS,CAAC;KACnF,iBAAiB,OAAO,gBAAgB,QAAQ,MAAM,kBAAkB,SAAS,EAAE,SAAS,CAAC;KAC7F,WAAW,OAAO,UAAU,QAAQ,MAAM,kBAAkB,SAAS,EAAE,SAAS,CAAC;KACjF,aAAa,OAAO,YAAY,QAAQ,MAAM,kBAAkB,SAAS,EAAE,SAAS,CAAC;KACrF,kBAAkB,OAAO,iBAAiB,QAAQ,MAAM,kBAAkB,SAAS,EAAE,SAAS,CAAC;KAC/F,iBAAiB,OAAO,gBAAgB,QAAQ,MAAM,kBAAkB,SAAS,EAAE,SAAS,CAAC;KAC7F,cAAc,OAAO,aAAa,QAAQ,MAAM,kBAAkB,SAAS,EAAE,SAAS,CAAC;KACvF,SAAS,OAAO;KAChB,WAAW,KAAA;KACX,WAAW,KAAA;KACX,WAAW,KAAA;KACX,eAAe,OAAO;KACtB,UAAU;KACV,UAAU,CAAC;KACX,iBAAiB;KACjB,wBACE,UAAU,SAAS,MAAM,MAAM,EAAE,SAAS,QAAQ,KAClD,KAAK,yBAAyB,OAAO,KAAK,GAAG,UAAU,GAAG,UAAU,EAAA,CACpE;KACF,UAAU;KACV,oBAAoB,OAAO;KAC3B,IAAI,iBAAoE;MACtE,OAAO,CACL,GAAG,KAAK,WAAW,QAAQ,MAAM,CAAC,EAAE,uBAAuB,GAC3D,GAAG,KAAK,aAAa,SAAS,MAAM,EAAE,UAAU,CAAC,CAAC,QAAQ,MAAM,CAAC,EAAE,uBAAuB,CAC5F;KACF;KACA,IAAI,2BAAmD;MACrD,OAAO,KAAK,YAAY,QAAQ,MAAM,CAAC,EAAE,uBAAuB;KAClE;IACF;IAEA,OAAO,SAAS,KAAK,SAAS;IAE9B,SAAS,KAAK,SAAS;IACvB,eAAe,UAAU,QAAQ;GACnC;GAGA,MAAM,QAAyC,MAAS,CAAC,kBAAkB,IAAI,EAAE,SAAS;GAC1F,OAAO,aAAa,OAAO,WAAW,OAAO,IAAI;GACjD,OAAO,QAAQ,OAAO,MAAM,OAAO,IAAI;GACvC,OAAO,UAAU,OAAO,QAAQ,OAAO,IAAI;GAC3C,OAAO,aAAa,OAAO,WAAW,OAAO,IAAI;GACjD,OAAO,aAAa,OAAO,WAAW,OAAO,IAAI;GACjD,OAAO,kBAAkB,OAAO,gBAAgB,OAAO,IAAI;GAC3D,OAAO,YAAY,OAAO,UAAU,OAAO,IAAI;GAC/C,OAAO,cAAc,OAAO,YAAY,OAAO,IAAI;GACnD,OAAO,mBAAmB,OAAO,iBAAiB,OAAO,IAAI;GAC7D,OAAO,kBAAkB,OAAO,gBAAgB,OAAO,IAAI;GAC3D,OAAO,eAAe,OAAO,aAAa,OAAO,IAAI;EACvD;CACF;AACF;AAGA,IAAI;AAEJ,SAAgB,eAAe,UAA4C;CACzE,IAAI,aAAa,OAAO;CACxB,8BAAc,IAAI,IAAI;CACtB,KAAK,MAAM,UAAU,UACnB,IAAI,OAAO,oBAAoB,SAAS,OAAO,uBAAuB;EACpE,MAAM,OAAO;EACb,MAAM,WAAW,SAAS,QAAQ,MAAM,EAAE,kBAAkB,OAAO,QAAQ,MAAM,MAAM;EACvF,YAAY,IAAI,OAAO,MAAM;GAAE;GAAM;EAAS,CAAC;EAE/C,KAAK,MAAM,WAAW,UACpB,YAAY,IAAI,QAAQ,MAAM;GAAE;GAAM,UAAU,CAAC;EAAE,CAAC;CAExD;CAEF,OAAO;AACT;;AAGA,SAAS,8BAA8B,QAAgB,UAAoC;CAEzF,MAAM,cAAc,eAAe,QAAQ;CAC3C,IAAI,YAAY,SAAS,GAAG;CAE5B,KAAK,MAAM,UAAU,UAAU;EAE7B,KAAK,MAAM,OAAO,OAAO,YAAY;GACnC,MAAM,SAAS,YAAY,IAAI,IAAI,YAAY,IAAI;GACnD,MAAM,OAAO,QAAQ,KAAK,OAAO;GAIjC,MAAM,CAAC,SAAS,GAAG,UAAU,WAC3B,OAAO,OAAO,SAAS,KAAK,EAAE,YAAY,IAAI,kBAAkB,KAAA,CAClE;GACA,IAAI,UAAU,WAAW,OAAO,WAAW,GAAG;IAC5C,MAAM,EAAE,aAAa;IACrB,IAAI,eACF,SAAS,MAAM,MAAM,EAAE,SAAS,OAAO,KACvC,KAAK,4BAA4B,QAAQ,OAAO,SAAS,KAAK,MAAM,EAAE,IAAI,GAAG,EAAA,CAC7E;GACJ;EACF;EAEA,KAAK,MAAM,QAAQ,OAAO,cACxB,KAAK,MAAM,QAAQ,KAAK,YAAY;GAClC,MAAM,SAAS,YAAY,IAAI,KAAK,YAAY,IAAI;GACpD,MAAM,OAAO,QAAQ,KAAK,OAAO;GAEjC,MAAM,CAAC,SAAS,GAAG,UAAU,WAC3B,OAAO,OAAO,SAAS,KAAK,EAAE,YAAY,KAAK,kBAAkB,KAAA,CACnE;GACA,IAAI,UAAU,WAAW,OAAO,WAAW,GAAG;IAC5C,MAAM,EAAE,aAAa;IACrB,KAAK,eACH,SAAS,MAAM,MAAM,EAAE,SAAS,OAAO,KACvC,KAAK,4BAA4B,QAAQ,OAAO,SAAS,KAAK,MAAM,EAAE,IAAI,GAAG,EAAA,CAC7E;GACJ;EACF;EAGF,KAAK,MAAM,OAAO,OAAO,YAAY;GACnC,MAAM,SAAS,YAAY,IAAI,IAAI,YAAY,IAAI;GACnD,IAAI,UAAU,OAAO,KAAK,oBAAoB,OAAO;IAEnD,MAAM,aAAa,OAAO,SAAS,MAChC,OACC,CAAC,OAAO,KAAK,WAAW,MAAM,MAAM,EAAE,cAAc,IAAI,cAAc,KACtE,GAAG,WAAW,MAAM,MAAM,EAAE,cAAc,IAAI,cAAc,CAChE;IACA,IAAI,YACF,IAAI,cAAc,WAAW;GAEjC;EACF;EAEA,KAAK,MAAM,OAAO,OAAO,aAAa;GACpC,MAAM,SAAS,YAAY,IAAI,IAAI,YAAY,IAAI;GACnD,IAAI,UAAU,OAAO,KAAK,oBAAoB,OAAO;IAEnD,MAAM,aAAa,OAAO,SAAS,MAChC,OACC,CAAC,OAAO,KAAK,YAAY,MAAM,MAAM,EAAE,cAAc,IAAI,cAAc,KACvE,GAAG,YAAY,MAAM,MAAM,EAAE,cAAc,IAAI,cAAc,CACjE;IACA,IAAI,YACF,IAAI,cAAc,WAAW;GAEjC;EACF;CACF;AACF;;AAGA,SAAS,WAAW,QAA+D;CACjF,MAAM,UAAU,QAAQ;CACxB,OAAO,YAAY,KAAA,IAAY,CAAC,IAAI,OAAO,YAAY,WAAW,CAAC,OAAO,IAAI;AAChF"}