rawsql-ts 0.12.0 → 0.13.0

package/dist/transformers/UpdateResultSelectConverter.js

@@ -0,0 +1,359 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.UpdateResultSelectConverter = void 0;
+const Clause_1 = require("../models/Clause");
+const SelectQuery_1 = require("../models/SelectQuery");
+const ValueComponent_1 = require("../models/ValueComponent");
+const TableSourceCollector_1 = require("./TableSourceCollector");
+const FixtureCteBuilder_1 = require("./FixtureCteBuilder");
+const SelectQueryWithClauseHelper_1 = require("../utils/SelectQueryWithClauseHelper");
+const ValueComponentRewriter_1 = require("../utils/ValueComponentRewriter");
+const TableNameUtils_1 = require("../utils/TableNameUtils");
+class UpdateResultSelectConverter {
+    /**
+     * Converts an UPDATE with RETURNING (or a bare UPDATE) into a SELECT that mirrors its output rows.
+     */
+    static toSelectQuery(updateQuery, options) {
+        var _a, _b, _c, _d;
+        const targetTableName = this.extractTargetTableName(updateQuery.updateClause);
+        const tableDefinition = this.resolveTableDefinition(targetTableName, options);
+        const targetAlias = updateQuery.updateClause.getSourceAliasName();
+        const fromClause = this.buildFromClause(updateQuery.updateClause.source, updateQuery.fromClause);
+        const whereClause = (_a = updateQuery.whereClause) !== null && _a !== void 0 ? _a : null;
+        // Decide whether RETURNING or a row count should drive the SELECT clause.
+        const selectClause = updateQuery.returningClause
+            ? this.buildReturningSelectClause(updateQuery.returningClause, updateQuery.setClause, targetAlias, tableDefinition)
+            : this.buildCountSelectClause();
+        // Assemble the skeleton SELECT that mirrors the UPDATE's source and predicates.
+        const selectQuery = new SelectQuery_1.SimpleSelectQuery({
+            withClause: (_b = updateQuery.withClause) !== null && _b !== void 0 ? _b : undefined,
+            selectClause,
+            fromClause,
+            whereClause
+        });
+        // Prepare fixture descriptors for the tables that will be touched by the SELECT.
+        const fixtureTables = (_c = options === null || options === void 0 ? void 0 : options.fixtureTables) !== null && _c !== void 0 ? _c : [];
+        const fixtureMap = this.buildFixtureTableMap(fixtureTables);
+        const missingStrategy = (_d = options === null || options === void 0 ? void 0 : options.missingFixtureStrategy) !== null && _d !== void 0 ? _d : this.DEFAULT_MISSING_FIXTURE_STRATEGY;
+        const originalWithClause = SelectQueryWithClauseHelper_1.SelectQueryWithClauseHelper.detachWithClause(selectQuery);
+        const referencedTables = this.collectPhysicalTableReferences(selectQuery, originalWithClause);
+        const cteNames = this.collectCteNamesFromWithClause(originalWithClause);
+        const targetVariants = (0, TableNameUtils_1.tableNameVariants)(targetTableName);
+        for (const variant of targetVariants) {
+            if (!cteNames.has(variant)) {
+                referencedTables.add(variant);
+            }
+        }
+        // Ensure each referenced table is covered by a fixture (or allowed to skip it).
+        this.ensureFixtureCoverage(referencedTables, fixtureMap, missingStrategy);
+        const filteredFixtures = this.filterFixtureTablesForReferences(fixtureTables, referencedTables);
+        // Turn the fixture definitions into CommonTable entries before reinjecting the WITH clause.
+        const fixtureCtes = this.buildFixtureCtes(filteredFixtures);
+        const recombinedWithClause = this.mergeWithClause(originalWithClause, fixtureCtes);
+        // Reattach the combined WITH clause so fixture CTEs precede any existing definitions.
+        SelectQueryWithClauseHelper_1.SelectQueryWithClauseHelper.setWithClause(selectQuery, recombinedWithClause);
+        return selectQuery;
+    }
+    static buildReturningSelectClause(returning, setClause, targetAlias, tableDefinition) {
+        const setExpressionMap = this.mapSetExpressions(setClause);
+        const selectItems = this.buildReturningSelectItems(returning, setExpressionMap, targetAlias, tableDefinition);
+        return new Clause_1.SelectClause(selectItems);
+    }
+    static buildReturningSelectItems(returning, setExpressions, targetAlias, tableDefinition) {
+        // Convert each RETURNING item into a select entry, expanding wildcards up front.
+        const selectItems = [];
+        for (const item of returning.items) {
+            if (this.isWildcardReturningItem(item)) {
+                selectItems.push(...this.expandReturningWildcard(tableDefinition, setExpressions, targetAlias));
+                continue;
+            }
+            selectItems.push(this.buildUpdateReturningSelectItem(item, setExpressions, targetAlias, tableDefinition));
+        }
+        return selectItems;
+    }
+    static isWildcardReturningItem(item) {
+        return (item.value instanceof ValueComponent_1.ColumnReference &&
+            item.value.column.name === '*');
+    }
+    static expandReturningWildcard(tableDefinition, setExpressions, targetAlias) {
+        // Use metadata to expand RETURNING * so each column can honor SET overrides.
+        if (!tableDefinition) {
+            throw new Error('Cannot expand RETURNING * without table definition.');
+        }
+        return tableDefinition.columns.map((column) => {
+            const expression = this.buildUpdateColumnExpression(column.name, setExpressions, targetAlias, tableDefinition);
+            return new Clause_1.SelectItem(expression, column.name);
+        });
+    }
+    static buildUpdateReturningSelectItem(item, setExpressions, targetAlias, tableDefinition) {
+        // Rewrite the item expression so column references honor SET overrides.
+        const expression = (0, ValueComponentRewriter_1.rewriteValueComponentWithColumnResolver)(item.value, (column) => this.buildUpdateColumnExpression(column, setExpressions, targetAlias, tableDefinition));
+        const alias = this.getReturningAlias(item);
+        return new Clause_1.SelectItem(expression, alias);
+    }
+    static buildUpdateColumnExpression(columnOrName, setExpressions, targetAlias, tableDefinition) {
+        const columnName = typeof columnOrName === 'string'
+            ? columnOrName
+            : this.getColumnReferenceName(columnOrName);
+        const normalized = this.normalizeIdentifier(columnName);
+        const overrideExpression = setExpressions.get(normalized);
+        // Prefer the SET expression when the column is updated, otherwise preserve the target reference.
+        if (overrideExpression) {
+            return overrideExpression;
+        }
+        this.ensureColumnExists(columnName, tableDefinition);
+        return new ValueComponent_1.ColumnReference(targetAlias, columnName);
+    }
+    static getColumnReferenceName(column) {
+        const nameComponent = column.qualifiedName.name;
+        if (nameComponent instanceof ValueComponent_1.IdentifierString) {
+            return nameComponent.name;
+        }
+        return nameComponent.value;
+    }
+    static getReturningAlias(item) {
+        var _a;
+        if ((_a = item.identifier) === null || _a === void 0 ? void 0 : _a.name) {
+            return item.identifier.name;
+        }
+        if (item.value instanceof ValueComponent_1.ColumnReference) {
+            return item.value.toString();
+        }
+        return null;
+    }
+    static buildCountSelectClause() {
+        // Count rows when the UPDATE does not expose RETURNING output.
+        const countFunction = new ValueComponent_1.FunctionCall(null, 'count', new ValueComponent_1.RawString('*'), null);
+        const selectItem = new Clause_1.SelectItem(countFunction, 'count');
+        return new Clause_1.SelectClause([selectItem]);
+    }
+    static buildFromClause(targetSource, fromClause) {
+        if (!fromClause) {
+            return new Clause_1.FromClause(targetSource, null);
+        }
+        const joins = [];
+        // Cross join any explicit FROM sources so their columns remain accessible.
+        joins.push(new Clause_1.JoinClause('cross join', fromClause.source, null, false));
+        if (fromClause.joins) {
+            joins.push(...fromClause.joins);
+        }
+        return new Clause_1.FromClause(targetSource, joins);
+    }
+    static mapSetExpressions(setClause) {
+        // Normalize each column name so lookups are case-insensitive.
+        const expressionMap = new Map();
+        for (const item of setClause.items) {
+            const columnName = this.extractColumnName(item);
+            expressionMap.set(this.normalizeIdentifier(columnName), item.value);
+        }
+        return expressionMap;
+    }
+    static ensureColumnExists(columnName, tableDefinition) {
+        // Guard against referencing columns that do not exist when metadata is available.
+        if (!tableDefinition) {
+            return;
+        }
+        const normalized = this.normalizeIdentifier(columnName);
+        const exists = tableDefinition.columns.some((column) => this.normalizeIdentifier(column.name) === normalized);
+        if (!exists) {
+            throw new Error(`Column '${columnName}' cannot be resolved for RETURNING output.`);
+        }
+    }
+    static resolveTableDefinition(tableName, options) {
+        // Prefer resolver callback results before falling back to the static registry.
+        if (options === null || options === void 0 ? void 0 : options.tableDefinitionResolver) {
+            const resolved = options.tableDefinitionResolver(tableName);
+            if (resolved !== undefined) {
+                return resolved;
+            }
+        }
+        const normalizedVariants = new Set((0, TableNameUtils_1.tableNameVariants)(tableName));
+        if (options === null || options === void 0 ? void 0 : options.tableDefinitions) {
+            const map = this.buildTableDefinitionMap(options.tableDefinitions);
+            for (const variant of normalizedVariants) {
+                const definition = map.get(variant);
+                if (definition) {
+                    return definition;
+                }
+            }
+        }
+        if (options === null || options === void 0 ? void 0 : options.fixtureTables) {
+            const fixture = options.fixtureTables.find((f) => (0, TableNameUtils_1.tableNameVariants)(f.tableName).some((variant) => normalizedVariants.has(variant)));
+            if (fixture) {
+                return this.convertFixtureToTableDefinition(fixture);
+            }
+        }
+        return undefined;
+    }
+    static convertFixtureToTableDefinition(fixture) {
+        return {
+            name: fixture.tableName,
+            columns: fixture.columns.map(col => {
+                var _a;
+                return ({
+                    name: col.name,
+                    typeName: col.typeName,
+                    required: false,
+                    defaultValue: (_a = col.defaultValue) !== null && _a !== void 0 ? _a : null
+                });
+            })
+        };
+    }
+    static buildTableDefinitionMap(registry) {
+        // Normalize registry keys so lookups ignore casing.
+        const map = new Map();
+        for (const definition of Object.values(registry)) {
+            for (const variant of (0, TableNameUtils_1.tableNameVariants)(definition.name)) {
+                map.set(variant, definition);
+            }
+        }
+        return map;
+    }
+    static extractTargetTableName(updateClause) {
+        const datasource = updateClause.source.datasource;
+        if (datasource instanceof Clause_1.TableSource) {
+            return datasource.getSourceName();
+        }
+        throw new Error('Update target must be a table source for conversion.');
+    }
+    static extractColumnName(item) {
+        const columnComponent = item.qualifiedName.name;
+        if (columnComponent instanceof ValueComponent_1.RawString) {
+            return columnComponent.value;
+        }
+        return columnComponent.name;
+    }
+    static buildFixtureCtes(fixtures) {
+        if (!fixtures || fixtures.length === 0) {
+            return [];
+        }
+        return FixtureCteBuilder_1.FixtureCteBuilder.buildFixtures(fixtures);
+    }
+    static collectPhysicalTableReferences(selectQuery, withClause) {
+        const referencedTables = this.collectReferencedTables(selectQuery);
+        const ignoredTables = this.collectCteNamesFromWithClause(withClause);
+        const tablesToShadow = new Set();
+        // Record only concrete tables that are not already defined via CTE aliases.
+        for (const table of referencedTables) {
+            if (ignoredTables.has(table)) {
+                continue;
+            }
+            tablesToShadow.add(table);
+        }
+        const cteReferencedTables = this.collectReferencedTablesFromWithClause(withClause);
+        for (const table of cteReferencedTables) {
+            if (ignoredTables.has(table)) {
+                continue;
+            }
+            tablesToShadow.add(table);
+        }
+        return tablesToShadow;
+    }
+    static filterFixtureTablesForReferences(fixtures, referencedTables) {
+        if (!fixtures.length || referencedTables.size === 0) {
+            return [];
+        }
+        const filtered = [];
+        for (const fixture of fixtures) {
+            const fixtureVariants = (0, TableNameUtils_1.tableNameVariants)(fixture.tableName);
+            if (fixtureVariants.some((variant) => referencedTables.has(variant))) {
+                filtered.push(fixture);
+            }
+        }
+        return filtered;
+    }
+    static collectReferencedTablesFromWithClause(withClause) {
+        const tables = new Set();
+        if (!(withClause === null || withClause === void 0 ? void 0 : withClause.tables)) {
+            return tables;
+        }
+        for (const cte of withClause.tables) {
+            for (const table of this.collectReferencedTables(cte.query)) {
+                tables.add(table);
+            }
+        }
+        return tables;
+    }
+    static buildFixtureTableMap(fixtures) {
+        // Normalize fixture table names to keep comparisons consistent.
+        const map = new Map();
+        for (const fixture of fixtures) {
+            for (const variant of (0, TableNameUtils_1.tableNameVariants)(fixture.tableName)) {
+                map.set(variant, fixture);
+            }
+        }
+        return map;
+    }
+    static ensureFixtureCoverage(referencedTables, fixtureMap, strategy) {
+        if (referencedTables.size === 0) {
+            return;
+        }
+        const missingTables = this.getMissingFixtureTables(referencedTables, fixtureMap);
+        if (missingTables.length === 0) {
+            return;
+        }
+        if (strategy === 'error') {
+            throw new Error(`Update SELECT refers to tables without fixture coverage: ${missingTables.join(', ')}.`);
+        }
+    }
+    static collectReferencedTables(query) {
+        // Use the collector to track every TableSource referenced by the SELECT.
+        const collector = new TableSourceCollector_1.TableSourceCollector(false);
+        const sources = collector.collect(query);
+        const normalized = new Set();
+        for (const source of sources) {
+            for (const variant of (0, TableNameUtils_1.tableNameVariants)(source.getSourceName())) {
+                normalized.add(variant);
+            }
+        }
+        return normalized;
+    }
+    static collectCteNamesFromWithClause(withClause) {
+        // Determine which table names come from CTE aliases so they can be ignored.
+        const names = new Set();
+        if (!(withClause === null || withClause === void 0 ? void 0 : withClause.tables)) {
+            return names;
+        }
+        for (const table of withClause.tables) {
+            for (const variant of (0, TableNameUtils_1.tableNameVariants)(table.getSourceAliasName())) {
+                names.add(variant);
+            }
+        }
+        return names;
+    }
+    static getMissingFixtureTables(referencedTables, fixtureMap) {
+        // Return every referenced table that lacks an overriding fixture definition.
+        const missing = [];
+        for (const table of referencedTables) {
+            const covered = (0, TableNameUtils_1.tableNameVariants)(table).some((variant) => fixtureMap.has(variant));
+            if (!covered) {
+                missing.push(table);
+            }
+        }
+        return missing;
+    }
+    static mergeWithClause(original, fixtureCtes) {
+        var _a;
+        // Combine fixture CTEs ahead of any original definitions so they shadow physical tables.
+        if (!fixtureCtes.length && !original) {
+            return null;
+        }
+        const combinedTables = [...fixtureCtes];
+        if (original === null || original === void 0 ? void 0 : original.tables) {
+            combinedTables.push(...original.tables);
+        }
+        if (!combinedTables.length) {
+            return null;
+        }
+        const merged = new Clause_1.WithClause((_a = original === null || original === void 0 ? void 0 : original.recursive) !== null && _a !== void 0 ? _a : false, combinedTables);
+        merged.globalComments = (original === null || original === void 0 ? void 0 : original.globalComments) ? [...original.globalComments] : null;
+        merged.trailingComments = (original === null || original === void 0 ? void 0 : original.trailingComments) ? [...original.trailingComments] : null;
+        return merged;
+    }
+    static normalizeIdentifier(value) {
+        return value.trim().toLowerCase();
+    }
+}
+exports.UpdateResultSelectConverter = UpdateResultSelectConverter;
+UpdateResultSelectConverter.DEFAULT_MISSING_FIXTURE_STRATEGY = 'error';
+//# sourceMappingURL=UpdateResultSelectConverter.js.map

package/dist/transformers/UpdateResultSelectConverter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"UpdateResultSelectConverter.js","sourceRoot":"","sources":["../../src/transformers/UpdateResultSelectConverter.ts"],"names":[],"mappings":";;;AAAA,6CAY0B;AAE1B,uDAAuE;AACvE,6DAMkC;AAMlC,iEAA8D;AAC9D,2DAAgF;AAChF,sFAAmF;AACnF,4EAA0F;AAC1F,4DAA4D;AAe5D,MAAa,2BAA2B;IAGpC;;OAEG;IACI,MAAM,CAAC,aAAa,CAAC,WAAwB,EAAE,OAAmC;;QACrF,MAAM,eAAe,GAAG,IAAI,CAAC,sBAAsB,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;QAC9E,MAAM,eAAe,GAAG,IAAI,CAAC,sBAAsB,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC;QAC9E,MAAM,WAAW,GAAG,WAAW,CAAC,YAAY,CAAC,kBAAkB,EAAE,CAAC;QAElE,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,WAAW,CAAC,YAAY,CAAC,MAAM,EAAE,WAAW,CAAC,UAAU,CAAC,CAAC;QACjG,MAAM,WAAW,GAAG,MAAA,WAAW,CAAC,WAAW,mCAAI,IAAI,CAAC;QAEpD,0EAA0E;QAC1E,MAAM,YAAY,GAAG,WAAW,CAAC,eAAe;YAC5C,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAC7B,WAAW,CAAC,eAAe,EAC3B,WAAW,CAAC,SAAS,EACrB,WAAW,EACX,eAAe,CAClB;YACD,CAAC,CAAC,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAEpC,gFAAgF;QAChF,MAAM,WAAW,GAAG,IAAI,+BAAiB,CAAC;YACtC,UAAU,EAAE,MAAA,WAAW,CAAC,UAAU,mCAAI,SAAS;YAC/C,YAAY;YACZ,UAAU;YACV,WAAW;SACd,CAAC,CAAC;QAEH,iFAAiF;QACjF,MAAM,aAAa,GAAG,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,aAAa,mCAAI,EAAE,CAAC;QACnD,MAAM,UAAU,GAAG,IAAI,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAC;QAC5D,MAAM,eAAe,GAAG,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,sBAAsB,mCAAI,IAAI,CAAC,gCAAgC,CAAC;QAEjG,MAAM,kBAAkB,GAAG,yDAA2B,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC;QACrF,MAAM,gBAAgB,GAAG,IAAI,CAAC,8BAA8B,CAAC,WAAW,EAAE,kBAAkB,CAAC,CAAC;QAC9F,MAAM,QAAQ,GAAG,IAAI,CAAC,6BAA6B,CAAC,kBAAkB,CAAC,CAAC;QACxE,MAAM,cAAc,GAAG,IAAA,kCAAiB,EAAC,eAAe,CAAC,CAAC;QAC1D,KAAK,MAAM,OAAO,IAAI,cAAc,EAAE,CAAC;YACnC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;gBACzB,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAClC,CAAC;QACL,CAAC;QACD,gFAAgF;QAChF,IAAI,CAAC,qBAAqB,CAAC,gBAAgB,EAAE,UAAU,EAAE,eAAe,CAAC,CAAC;QAE1E,MAAM,gBAAgB,GAAG,IAAI,CAAC,gCAAgC,CAAC,aAAa,EAAE,gBAAgB,CAAC,CAAC;QAChG,4FAA4F;QAC5F,MAAM,WAAW,GAAG,IAAI,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,CAAC;QAC5D,MAAM,oBAAoB,GAAG,IAAI,CAAC,eAAe,CAAC,kBAAkB,EAAE,WAAW,CAAC,CAAC;QACnF,sFAAsF;QACtF,yDAA2B,CAAC,aAAa,CAAC,WAAW,EAAE,oBAAoB,CAAC,CAAC;QAE7E,OAAO,WAAW,CAAC;IACvB,CAAC;IAEO,MAAM,CAAC,0BAA0B,CACrC,SAA0B,EAC1B,SAAoB,EACpB,WAA0B,EAC1B,eAAsC;QAEtC,MAAM,gBAAgB,GAAG,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAC3D,MAAM,WAAW,GAAG,IAAI,CAAC,yBAAyB,CAC9C,SAAS,EACT,gBAAgB,EAChB,WAAW,EACX,eAAe,CAClB,CAAC;QAEF,OAAO,IAAI,qBAAY,CAAC,WAAW,CAAC,CAAC;IACzC,CAAC;IAEO,MAAM,CAAC,yBAAyB,CACpC,SAA0B,EAC1B,cAA2C,EAC3C,WAA0B,EAC1B,eAAsC;QAEtC,iFAAiF;QACjF,MAAM,WAAW,GAAiB,EAAE,CAAC;QACrC,KAAK,MAAM,IAAI,IAAI,SAAS,CAAC,KAAK,EAAE,CAAC;YACjC,IAAI,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,EAAE,CAAC;gBACrC,WAAW,CAAC,IAAI,CACZ,GAAG,IAAI,CAAC,uBAAuB,CAAC,eAAe,EAAE,cAAc,EAAE,WAAW,CAAC,CAChF,CAAC;gBACF,SAAS;YACb,CAAC;YACD,WAAW,CAAC,IAAI,CACZ,IAAI,CAAC,8BAA8B,CAAC,IAAI,EAAE,cAAc,EAAE,WAAW,EAAE,eAAe,CAAC,CAC1F,CAAC;QACN,CAAC;QACD,OAAO,WAAW,CAAC;IACvB,CAAC;IAEO,MAAM,CAAC,uBAAuB,CAAC,IAAgB;QACnD,OAAO,CACH,IAAI,CAAC,KAAK,YAAY,gCAAe;YACrC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,GAAG,CACjC,CAAC;IACN,CAAC;IAEO,MAAM,CAAC,uBAAuB,CAClC,eAAiD,EACjD,cAA2C,EAC3C,WAA0B;QAE1B,6EAA6E;QAC7E,IAAI,CAAC,eAAe,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;QAC3E,CAAC;QACD,OAAO,eAAe,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;YAC1C,MAAM,UAAU,GAAG,IAAI,CAAC,2BAA2B,CAC/C,MAAM,CAAC,IAAI,EACX,cAAc,EACd,WAAW,EACX,eAAe,CAClB,CAAC;YACF,OAAO,IAAI,mBAAU,CAAC,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;IACP,CAAC;IAEO,MAAM,CAAC,8BAA8B,CACzC,IAAgB,EAChB,cAA2C,EAC3C,WAA0B,EAC1B,eAAsC;QAEtC,wEAAwE;QACxE,MAAM,UAAU,GAAG,IAAA,gEAAuC,EAAC,IAAI,CAAC,KAAK,EAAE,CAAC,MAAM,EAAE,EAAE,CAC9E,IAAI,CAAC,2BAA2B,CAAC,MAAM,EAAE,cAAc,EAAE,WAAW,EAAE,eAAe,CAAC,CACzF,CAAC;QACF,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAC3C,OAAO,IAAI,mBAAU,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;IAC7C,CAAC;IAEO,MAAM,CAAC,2BAA2B,CACtC,YAAsC,EACtC,cAA2C,EAC3C,WAA0B,EAC1B,eAAsC;QAEtC,MAAM,UAAU,GACZ,OAAO,YAAY,KAAK,QAAQ;YAC5B,CAAC,CAAC,YAAY;YACd,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,YAAY,CAAC,CAAC;QACpD,MAAM,UAAU,GAAG,IAAI,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC;QACxD,MAAM,kBAAkB,GAAG,cAAc,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC1D,iGAAiG;QACjG,IAAI,kBAAkB,EAAE,CAAC;YACrB,OAAO,kBAAkB,CAAC;QAC9B,CAAC;QAED,IAAI,CAAC,kBAAkB,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC;QACrD,OAAO,IAAI,gCAAe,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;IACxD,CAAC;IAEO,MAAM,CAAC,sBAAsB,CAAC,MAAuB;QACzD,MAAM,aAAa,GAAG,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC;QAChD,IAAI,aAAa,YAAY,iCAAgB,EAAE,CAAC;YAC5C,OAAO,aAAa,CAAC,IAAI,CAAC;QAC9B,CAAC;QACD,OAAO,aAAa,CAAC,KAAK,CAAC;IAC/B,CAAC;IAEO,MAAM,CAAC,iBAAiB,CAAC,IAAgB;;QAC7C,IAAI,MAAA,IAAI,CAAC,UAAU,0CAAE,IAAI,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;QAChC,CAAC;QACD,IAAI,IAAI,CAAC,KAAK,YAAY,gCAAe,EAAE,CAAC;YACxC,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;QACjC,CAAC;QACD,OAAO,IAAI,CAAC;IAChB,CAAC;IAEO,MAAM,CAAC,sBAAsB;QACjC,+DAA+D;QAC/D,MAAM,aAAa,GAAG,IAAI,6BAAY,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,0BAAS,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;QAChF,MAAM,UAAU,GAAG,IAAI,mBAAU,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;QAC1D,OAAO,IAAI,qBAAY,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;IAC1C,CAAC;IAEO,MAAM,CAAC,eAAe,CAAC,YAA8B,EAAE,UAA6B;QACxF,IAAI,CAAC,UAAU,EAAE,CAAC;YACd,OAAO,IAAI,mBAAU,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;QAC9C,CAAC;QAED,MAAM,KAAK,GAAiB,EAAE,CAAC;QAC/B,2EAA2E;QAC3E,KAAK,CAAC,IAAI,CAAC,IAAI,mBAAU,CAAC,YAAY,EAAE,UAAU,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;QACzE,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;YACnB,KAAK,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;QACpC,CAAC;QAED,OAAO,IAAI,mBAAU,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;IAC/C,CAAC;IAEO,MAAM,CAAC,iBAAiB,CAAC,SAAoB;QACjD,8DAA8D;QAC9D,MAAM,aAAa,GAAG,IAAI,GAAG,EAA0B,CAAC;QACxD,KAAK,MAAM,IAAI,IAAI,SAAS,CAAC,KAAK,EAAE,CAAC;YACjC,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAChD,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,mBAAmB,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QACxE,CAAC;QACD,OAAO,aAAa,CAAC;IACzB,CAAC;IAEO,MAAM,CAAC,kBAAkB,CAAC,UAAkB,EAAE,eAAsC;QACxF,kFAAkF;QAClF,IAAI,CAAC,eAAe,EAAE,CAAC;YACnB,OAAO;QACX,CAAC;QACD,MAAM,UAAU,GAAG,IAAI,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC;QACxD,MAAM,MAAM,GAAG,eAAe,CAAC,OAAO,CAAC,IAAI,CACvC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,UAAU,CACnE,CAAC;QACF,IAAI,CAAC,MAAM,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,WAAW,UAAU,4CAA4C,CAAC,CAAC;QACvF,CAAC;IACL,CAAC;IAEO,MAAM,CAAC,sBAAsB,CACjC,SAAiB,EACjB,OAAmC;QAEnC,+EAA+E;QAC/E,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,uBAAuB,EAAE,CAAC;YACnC,MAAM,QAAQ,GAAG,OAAO,CAAC,uBAAuB,CAAC,SAAS,CAAC,CAAC;YAC5D,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;gBACzB,OAAO,QAAQ,CAAC;YACpB,CAAC;QACL,CAAC;QAED,MAAM,kBAAkB,GAAG,IAAI,GAAG,CAAC,IAAA,kCAAiB,EAAC,SAAS,CAAC,CAAC,CAAC;QAEjE,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,gBAAgB,EAAE,CAAC;YAC5B,MAAM,GAAG,GAAG,IAAI,CAAC,uBAAuB,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;YACnE,KAAK,MAAM,OAAO,IAAI,kBAAkB,EAAE,CAAC;gBACvC,MAAM,UAAU,GAAG,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACpC,IAAI,UAAU,EAAE,CAAC;oBACb,OAAO,UAAU,CAAC;gBACtB,CAAC;YACL,CAAC;QACL,CAAC;QAED,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,aAAa,EAAE,CAAC;YACzB,MAAM,OAAO,GAAG,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAC7C,IAAA,kCAAiB,EAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CACpF,CAAC;YACF,IAAI,OAAO,EAAE,CAAC;gBACV,OAAO,IAAI,CAAC,+BAA+B,CAAC,OAAO,CAAC,CAAC;YACzD,CAAC;QACL,CAAC;QAED,OAAO,SAAS,CAAC;IACrB,CAAC;IAEO,MAAM,CAAC,+BAA+B,CAAC,OAA+B;QAC1E,OAAO;YACH,IAAI,EAAE,OAAO,CAAC,SAAS;YACvB,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;;gBAAC,OAAA,CAAC;oBACjC,IAAI,EAAE,GAAG,CAAC,IAAI;oBACd,QAAQ,EAAE,GAAG,CAAC,QAAQ;oBACtB,QAAQ,EAAE,KAAK;oBACf,YAAY,EAAE,MAAA,GAAG,CAAC,YAAY,mCAAI,IAAI;iBACzC,CAAC,CAAA;aAAA,CAAC;SACN,CAAC;IACN,CAAC;IAEO,MAAM,CAAC,uBAAuB,CAClC,QAAiC;QAEjC,oDAAoD;QACpD,MAAM,GAAG,GAAG,IAAI,GAAG,EAAgC,CAAC;QACpD,KAAK,MAAM,UAAU,IAAI,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC/C,KAAK,MAAM,OAAO,IAAI,IAAA,kCAAiB,EAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;gBACvD,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;YACjC,CAAC;QACL,CAAC;QACD,OAAO,GAAG,CAAC;IACf,CAAC;IAEO,MAAM,CAAC,sBAAsB,CAAC,YAAyC;QAC3E,MAAM,UAAU,GAAG,YAAY,CAAC,MAAM,CAAC,UAAU,CAAC;QAClD,IAAI,UAAU,YAAY,oBAAW,EAAE,CAAC;YACpC,OAAO,UAAU,CAAC,aAAa,EAAE,CAAC;QACtC,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;IAC5E,CAAC;IAEO,MAAM,CAAC,iBAAiB,CAAC,IAAmB;QAChD,MAAM,eAAe,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;QAChD,IAAI,eAAe,YAAY,0BAAS,EAAE,CAAC;YACvC,OAAO,eAAe,CAAC,KAAK,CAAC;QACjC,CAAC;QACD,OAAO,eAAe,CAAC,IAAI,CAAC;IAChC,CAAC;IAEO,MAAM,CAAC,gBAAgB,CAAC,QAAkC;QAC9D,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACrC,OAAO,EAAE,CAAC;QACd,CAAC;QACD,OAAO,qCAAiB,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;IACrD,CAAC;IAEO,MAAM,CAAC,8BAA8B,CACzC,WAAwB,EACxB,UAA6B;QAE7B,MAAM,gBAAgB,GAAG,IAAI,CAAC,uBAAuB,CAAC,WAAW,CAAC,CAAC;QACnE,MAAM,aAAa,GAAG,IAAI,CAAC,6BAA6B,CAAC,UAAU,CAAC,CAAC;QAErE,MAAM,cAAc,GAAG,IAAI,GAAG,EAAU,CAAC;QACzC,4EAA4E;QAC5E,KAAK,MAAM,KAAK,IAAI,gBAAgB,EAAE,CAAC;YACnC,IAAI,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC3B,SAAS;YACb,CAAC;YACD,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC9B,CAAC;QAED,MAAM,mBAAmB,GAAG,IAAI,CAAC,qCAAqC,CAAC,UAAU,CAAC,CAAC;QACnF,KAAK,MAAM,KAAK,IAAI,mBAAmB,EAAE,CAAC;YACtC,IAAI,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC3B,SAAS;YACb,CAAC;YACD,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC9B,CAAC;QAED,OAAO,cAAc,CAAC;IAC1B,CAAC;IAEO,MAAM,CAAC,gCAAgC,CAC3C,QAAkC,EAClC,gBAA6B;QAE7B,IAAI,CAAC,QAAQ,CAAC,MAAM,IAAI,gBAAgB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YAClD,OAAO,EAAE,CAAC;QACd,CAAC;QAED,MAAM,QAAQ,GAA6B,EAAE,CAAC;QAC9C,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC7B,MAAM,eAAe,GAAG,IAAA,kCAAiB,EAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAC7D,IAAI,eAAe,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;gBACnE,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC3B,CAAC;QACL,CAAC;QAED,OAAO,QAAQ,CAAC;IACpB,CAAC;IAEO,MAAM,CAAC,qCAAqC,CAAC,UAA6B;QAC9E,MAAM,MAAM,GAAG,IAAI,GAAG,EAAU,CAAC;QACjC,IAAI,CAAC,CAAA,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAE,MAAM,CAAA,EAAE,CAAC;YACtB,OAAO,MAAM,CAAC;QAClB,CAAC;QAED,KAAK,MAAM,GAAG,IAAI,UAAU,CAAC,MAAM,EAAE,CAAC;YAClC,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC1D,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACtB,CAAC;QACL,CAAC;QAED,OAAO,MAAM,CAAC;IAClB,CAAC;IAEO,MAAM,CAAC,oBAAoB,CAAC,QAAkC;QAClE,gEAAgE;QAChE,MAAM,GAAG,GAAG,IAAI,GAAG,EAAkC,CAAC;QACtD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC7B,KAAK,MAAM,OAAO,IAAI,IAAA,kCAAiB,EAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;gBACzD,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC9B,CAAC;QACL,CAAC;QACD,OAAO,GAAG,CAAC;IACf,CAAC;IAEO,MAAM,CAAC,qBAAqB,CAChC,gBAA6B,EAC7B,UAA+C,EAC/C,QAAgC;QAEhC,IAAI,gBAAgB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YAC9B,OAAO;QACX,CAAC;QAED,MAAM,aAAa,GAAG,IAAI,CAAC,uBAAuB,CAAC,gBAAgB,EAAE,UAAU,CAAC,CAAC;QACjF,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,OAAO;QACX,CAAC;QAED,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CACX,4DAA4D,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAC1F,CAAC;QACN,CAAC;IACL,CAAC;IAEO,MAAM,CAAC,uBAAuB,CAAC,KAAkB;QACrD,yEAAyE;QACzE,MAAM,SAAS,GAAG,IAAI,2CAAoB,CAAC,KAAK,CAAC,CAAC;QAClD,MAAM,OAAO,GAAG,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACzC,MAAM,UAAU,GAAG,IAAI,GAAG,EAAU,CAAC;QACrC,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC3B,KAAK,MAAM,OAAO,IAAI,IAAA,kCAAiB,EAAC,MAAM,CAAC,aAAa,EAAE,CAAC,EAAE,CAAC;gBAC9D,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC5B,CAAC;QACL,CAAC;QACD,OAAO,UAAU,CAAC;IACtB,CAAC;IAEO,MAAM,CAAC,6BAA6B,CAAC,UAA8B;QACvE,4EAA4E;QAC5E,MAAM,KAAK,GAAG,IAAI,GAAG,EAAU,CAAC;QAChC,IAAI,CAAC,CAAA,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAE,MAAM,CAAA,EAAE,CAAC;YACtB,OAAO,KAAK,CAAC;QACjB,CAAC;QACD,KAAK,MAAM,KAAK,IAAI,UAAU,CAAC,MAAM,EAAE,CAAC;YACpC,KAAK,MAAM,OAAO,IAAI,IAAA,kCAAiB,EAAC,KAAK,CAAC,kBAAkB,EAAE,CAAC,EAAE,CAAC;gBAClE,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;QACL,CAAC;QACD,OAAO,KAAK,CAAC;IACjB,CAAC;IAEO,MAAM,CAAC,uBAAuB,CAClC,gBAA6B,EAC7B,UAA+C;QAE/C,6EAA6E;QAC7E,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,KAAK,MAAM,KAAK,IAAI,gBAAgB,EAAE,CAAC;YACnC,MAAM,OAAO,GAAG,IAAA,kCAAiB,EAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;YACpF,IAAI,CAAC,OAAO,EAAE,CAAC;gBACX,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACxB,CAAC;QACL,CAAC;QACD,OAAO,OAAO,CAAC;IACnB,CAAC;IAEO,MAAM,CAAC,eAAe,CAAC,QAA2B,EAAE,WAA0B;;QAClF,yFAAyF;QACzF,IAAI,CAAC,WAAW,CAAC,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnC,OAAO,IAAI,CAAC;QAChB,CAAC;QACD,MAAM,cAAc,GAAG,CAAC,GAAG,WAAW,CAAC,CAAC;QACxC,IAAI,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,CAAE,MAAM,EAAE,CAAC;YACnB,cAAc,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC;QAC5C,CAAC;QACD,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC;QAChB,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,mBAAU,CAAC,MAAA,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,CAAE,SAAS,mCAAI,KAAK,EAAE,cAAc,CAAC,CAAC;QAC5E,MAAM,CAAC,cAAc,GAAG,CAAA,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,CAAE,cAAc,EAAC,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACvF,MAAM,CAAC,gBAAgB,GAAG,CAAA,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,CAAE,gBAAgB,EAAC,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QAC7F,OAAO,MAAM,CAAC;IAClB,CAAC;IAEO,MAAM,CAAC,mBAAmB,CAAC,KAAa;QAC5C,OAAO,KAAK,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACtC,CAAC;;AA/cL,kEAgdC;AA/c2B,4DAAgC,GAA2B,OAAO,CAAC"}

package/dist/transformers/UpstreamSelectQueryFinder.js

@@ -0,0 +1,184 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.UpstreamSelectQueryFinder = void 0;
+const SelectQuery_1 = require("../models/SelectQuery");
+const Clause_1 = require("../models/Clause");
+const SelectableColumnCollector_1 = require("./SelectableColumnCollector");
+const CTECollector_1 = require("./CTECollector");
+/**
+ * UpstreamSelectQueryFinder searches upstream queries for the specified columns.
+ * If a query (including its upstream CTEs or subqueries) contains all columns,
+ * it returns the highest such SelectQuery. Otherwise, it searches downstream.
+ *
+ * For BinarySelectQuery (UNION/INTERSECT/EXCEPT), this finder processes each branch
+ * independently, as SelectableColumnCollector is designed for SimpleSelectQuery only.
+ * This approach ensures accurate column detection within individual SELECT branches
+ * while maintaining compatibility with compound query structures.
+ */
+class UpstreamSelectQueryFinder {
+    constructor(tableColumnResolver, options) {
+        this.options = options || {};
+        this.tableColumnResolver = tableColumnResolver;
+        // Pass the tableColumnResolver instead of options to fix type mismatch.
+        this.columnCollector = new SelectableColumnCollector_1.SelectableColumnCollector(this.tableColumnResolver, false, // includeWildCard
+        SelectableColumnCollector_1.DuplicateDetectionMode.FullName, // Use FullName to preserve JOIN table columns
+        { upstream: true } // Enable upstream collection for qualified name resolution
+        );
+    }
+    /**
+     * Finds the highest SelectQuery containing all specified columns.
+     * @param query The root SelectQuery to search.
+     * @param columnNames A column name or array of column names to check for.
+     * @returns An array of SelectQuery objects, or an empty array if not found.
+     */
+    find(query, columnNames) {
+        // Normalize columnNames to array
+        const namesArray = typeof columnNames === 'string' ? [columnNames] : columnNames;
+        // Use CTECollector to collect CTEs from the root query only once and reuse
+        const cteCollector = new CTECollector_1.CTECollector();
+        const ctes = cteCollector.collect(query);
+        const cteMap = new Map();
+        for (const cte of ctes) {
+            cteMap.set(cte.getSourceAliasName(), cte);
+        }
+        return this.findUpstream(query, namesArray, cteMap);
+    }
+    handleTableSource(src, columnNames, cteMap) {
+        // Handles the logic for TableSource in findUpstream
+        const cte = cteMap.get(src.table.name);
+        if (cte) {
+            // Remove the current CTE name from the map to prevent infinite recursion
+            const nextCteMap = new Map(cteMap);
+            nextCteMap.delete(src.table.name);
+            const result = this.findUpstream(cte.query, columnNames, nextCteMap);
+            if (result.length === 0) {
+                return null;
+            }
+            return result;
+        }
+        return null;
+    }
+    handleSubQuerySource(src, columnNames, cteMap) {
+        // Handles the logic for SubQuerySource in findUpstream
+        const result = this.findUpstream(src.query, columnNames, cteMap);
+        if (result.length === 0) {
+            return null;
+        }
+        return result;
+    }
+    /**
+     * Processes all source branches in a FROM clause and checks if all upstream queries contain the specified columns.
+     * Returns a flat array of SelectQuery if all branches are valid, otherwise null.
+     */
+    processFromClauseBranches(fromClause, columnNames, cteMap) {
+        const sources = fromClause.getSources();
+        if (sources.length === 0)
+            return null;
+        let allBranchResults = [];
+        let allBranchesOk = true;
+        let validBranchCount = 0; // Count only filterable branches
+        for (const sourceExpr of sources) {
+            const src = sourceExpr.datasource;
+            let branchResult = null;
+            if (src instanceof Clause_1.TableSource) {
+                branchResult = this.handleTableSource(src, columnNames, cteMap);
+                validBranchCount++;
+            }
+            else if (src instanceof Clause_1.SubQuerySource) {
+                branchResult = this.handleSubQuerySource(src, columnNames, cteMap);
+                validBranchCount++;
+            }
+            else if (src instanceof SelectQuery_1.ValuesQuery) {
+                // Skip ValuesQuery: not filterable, do not count as a valid branch
+                continue;
+            }
+            else {
+                allBranchesOk = false;
+                break;
+            }
+            // If the branch result is null, 
+            // it means it didn't find the required columns in this branch
+            if (branchResult === null) {
+                allBranchesOk = false;
+                break;
+            }
+            allBranchResults.push(branchResult);
+        }
+        // Check if all valid (filterable) branches are valid and contain the required columns
+        if (allBranchesOk && allBranchResults.length === validBranchCount) {
+            return allBranchResults.flat();
+        }
+        return null;
+    }
+    findUpstream(query, columnNames, cteMap) {
+        if (query instanceof SelectQuery_1.SimpleSelectQuery) {
+            // First, try to find upstream queries from FROM clause
+            const fromClause = query.fromClause;
+            if (fromClause) {
+                const branchResult = this.processFromClauseBranches(fromClause, columnNames, cteMap);
+                if (branchResult && branchResult.length > 0) {
+                    return branchResult;
+                }
+            }
+            // If no upstream queries found, check if current query contains all columns
+            const columns = this.columnCollector.collect(query).map(col => col.name);
+            // Collect columns defined in CTEs as well
+            const cteColumns = this.collectCTEColumns(query, cteMap);
+            const allColumns = [...columns, ...cteColumns];
+            const normalize = (s) => this.options.ignoreCaseAndUnderscore ? s.toLowerCase().replace(/_/g, '') : s;
+            // Normalize both the columns and the required names for comparison.
+            const hasAll = columnNames.every(name => allColumns.some(col => normalize(col) === normalize(name)));
+            if (hasAll) {
+                return [query];
+            }
+            return [];
+        }
+        else if (query instanceof SelectQuery_1.BinarySelectQuery) {
+            // Process BinarySelectQuery by decomposing into individual branches.
+            // SelectableColumnCollector is designed for SimpleSelectQuery only,
+            // so we handle UNION/INTERSECT/EXCEPT by processing left and right branches separately.
+            const left = this.findUpstream(query.left, columnNames, cteMap);
+            const right = this.findUpstream(query.right, columnNames, cteMap);
+            return [...left, ...right];
+        }
+        return [];
+    }
+    /**
+     * Collects columns defined in CTEs
+     */
+    collectCTEColumns(query, cteMap) {
+        const cteColumns = [];
+        // If WITH clause exists, collect columns defined in CTEs
+        if (query.withClause) {
+            for (const cte of query.withClause.tables) {
+                // Collect columns from CTE query
+                const columns = this.collectColumnsFromSelectQuery(cte.query);
+                cteColumns.push(...columns);
+            }
+        }
+        return cteColumns;
+    }
+    /**
+     * Recursively collects columns from SelectQuery
+     */
+    collectColumnsFromSelectQuery(query) {
+        if (query instanceof SelectQuery_1.SimpleSelectQuery) {
+            try {
+                return this.columnCollector.collect(query).map(col => col.name);
+            }
+            catch (error) {
+                // Return empty array if SelectableColumnCollector fails
+                console.warn('Failed to collect columns from SimpleSelectQuery:', error);
+                return [];
+            }
+        }
+        else if (query instanceof SelectQuery_1.BinarySelectQuery) {
+            // For BinarySelectQuery (UNION etc.), get column names from the left query
+            // In UNION statements, left and right must have matching column count/types, so left side is sufficient
+            return this.collectColumnsFromSelectQuery(query.left);
+        }
+        return [];
+    }
+}
+exports.UpstreamSelectQueryFinder = UpstreamSelectQueryFinder;
+//# sourceMappingURL=UpstreamSelectQueryFinder.js.map

package/dist/transformers/UpstreamSelectQueryFinder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"UpstreamSelectQueryFinder.js","sourceRoot":"","sources":["../../src/transformers/UpstreamSelectQueryFinder.ts"],"names":[],"mappings":";;;AAAA,uDAAuG;AACvG,6CAAwF;AACxF,2EAAgG;AAChG,iDAA8C;AAE9C;;;;;;;;;GASG;AACH,MAAa,yBAAyB;IAKlC,YAAY,mBAAqD,EAAE,OAA+C;QAC9G,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,EAAE,CAAC;QAC7B,IAAI,CAAC,mBAAmB,GAAG,mBAAmB,CAAC;QAC/C,wEAAwE;QACxE,IAAI,CAAC,eAAe,GAAG,IAAI,qDAAyB,CAChD,IAAI,CAAC,mBAAmB,EACxB,KAAK,EAAE,kBAAkB;QACzB,kDAAsB,CAAC,QAAQ,EAAE,8CAA8C;QAC/E,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,2DAA2D;SACjF,CAAC;IACN,CAAC;IAED;;;;;OAKG;IACI,IAAI,CAAC,KAAkB,EAAE,WAA8B;QAC1D,iCAAiC;QACjC,MAAM,UAAU,GAAG,OAAO,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC;QACjF,2EAA2E;QAC3E,MAAM,YAAY,GAAG,IAAI,2BAAY,EAAE,CAAC;QACxC,MAAM,IAAI,GAAG,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACzC,MAAM,MAAM,GAA6B,IAAI,GAAG,EAAE,CAAC;QACnD,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACrB,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,kBAAkB,EAAE,EAAE,GAAG,CAAC,CAAC;QAC9C,CAAC;QACD,OAAO,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;IACxD,CAAC;IAEO,iBAAiB,CAAC,GAAgB,EAAE,WAAqB,EAAE,MAAgC;QAC/F,oDAAoD;QACpD,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACvC,IAAI,GAAG,EAAE,CAAC;YACN,yEAAyE;YACzE,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC;YACnC,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,KAAK,EAAE,WAAW,EAAE,UAAU,CAAC,CAAC;YACrE,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACtB,OAAO,IAAI,CAAC;YAChB,CAAC;YACD,OAAO,MAAM,CAAC;QAClB,CAAC;QACD,OAAO,IAAI,CAAC;IAChB,CAAC;IAEO,oBAAoB,CAAC,GAAmB,EAAE,WAAqB,EAAE,MAAgC;QACrG,uDAAuD;QACvD,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,KAAK,EAAE,WAAW,EAAE,MAAM,CAAC,CAAC;QACjE,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtB,OAAO,IAAI,CAAC;QAChB,CAAC;QACD,OAAO,MAAM,CAAC;IAClB,CAAC;IAED;;;OAGG;IACK,yBAAyB,CAC7B,UAAe,EACf,WAAqB,EACrB,MAAgC;QAEhC,MAAM,OAAO,GAAG,UAAU,CAAC,UAAU,EAAE,CAAC;QACxC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QAEtC,IAAI,gBAAgB,GAA0B,EAAE,CAAC;QACjD,IAAI,aAAa,GAAG,IAAI,CAAC;QACzB,IAAI,gBAAgB,GAAG,CAAC,CAAC,CAAC,iCAAiC;QAE3D,KAAK,MAAM,UAAU,IAAI,OAAO,EAAE,CAAC;YAC/B,MAAM,GAAG,GAAG,UAAU,CAAC,UAAU,CAAC;YAClC,IAAI,YAAY,GAA+B,IAAI,CAAC;YACpD,IAAI,GAAG,YAAY,oBAAW,EAAE,CAAC;gBAC7B,YAAY,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,EAAE,WAAW,EAAE,MAAM,CAAC,CAAC;gBAChE,gBAAgB,EAAE,CAAC;YACvB,CAAC;iBAAM,IAAI,GAAG,YAAY,uBAAc,EAAE,CAAC;gBACvC,YAAY,GAAG,IAAI,CAAC,oBAAoB,CAAC,GAAG,EAAE,WAAW,EAAE,MAAM,CAAC,CAAC;gBACnE,gBAAgB,EAAE,CAAC;YACvB,CAAC;iBAAM,IAAI,GAAG,YAAY,yBAAW,EAAE,CAAC;gBACpC,mEAAmE;gBACnE,SAAS;YACb,CAAC;iBAAM,CAAC;gBACJ,aAAa,GAAG,KAAK,CAAC;gBACtB,MAAM;YACV,CAAC;YAED,iCAAiC;YACjC,8DAA8D;YAC9D,IAAI,YAAY,KAAK,IAAI,EAAE,CAAC;gBACxB,aAAa,GAAG,KAAK,CAAC;gBACtB,MAAM;YACV,CAAC;YACD,gBAAgB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACxC,CAAC;QAED,sFAAsF;QACtF,IAAI,aAAa,IAAI,gBAAgB,CAAC,MAAM,KAAK,gBAAgB,EAAE,CAAC;YAChE,OAAO,gBAAgB,CAAC,IAAI,EAAE,CAAC;QACnC,CAAC;QACD,OAAO,IAAI,CAAC;IAChB,CAAC;IAEO,YAAY,CAAC,KAAkB,EAAE,WAAqB,EAAE,MAAgC;QAC5F,IAAI,KAAK,YAAY,+BAAiB,EAAE,CAAC;YACrC,uDAAuD;YACvD,MAAM,UAAU,GAAG,KAAK,CAAC,UAAU,CAAC;YACpC,IAAI,UAAU,EAAE,CAAC;gBACb,MAAM,YAAY,GAAG,IAAI,CAAC,yBAAyB,CAAC,UAAU,EAAE,WAAW,EAAE,MAAM,CAAC,CAAC;gBACrF,IAAI,YAAY,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC1C,OAAO,YAAY,CAAC;gBACxB,CAAC;YACL,CAAC;YAED,4EAA4E;YAC5E,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACzE,0CAA0C;YAC1C,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YACzD,MAAM,UAAU,GAAG,CAAC,GAAG,OAAO,EAAE,GAAG,UAAU,CAAC,CAAC;YAE/C,MAAM,SAAS,GAAG,CAAC,CAAS,EAAE,EAAE,CAC5B,IAAI,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACjF,oEAAoE;YACpE,MAAM,MAAM,GAAG,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAErG,IAAI,MAAM,EAAE,CAAC;gBACT,OAAO,CAAC,KAAK,CAAC,CAAC;YACnB,CAAC;YAED,OAAO,EAAE,CAAC;QACd,CAAC;aAAM,IAAI,KAAK,YAAY,+BAAiB,EAAE,CAAC;YAC5C,qEAAqE;YACrE,oEAAoE;YACpE,wFAAwF;YACxF,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,EAAE,WAAW,EAAE,MAAM,CAAC,CAAC;YAChE,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,KAAK,EAAE,WAAW,EAAE,MAAM,CAAC,CAAC;YAClE,OAAO,CAAC,GAAG,IAAI,EAAE,GAAG,KAAK,CAAC,CAAC;QAC/B,CAAC;QACD,OAAO,EAAE,CAAC;IACd,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,KAAwB,EAAE,MAAgC;QAChF,MAAM,UAAU,GAAa,EAAE,CAAC;QAEhC,yDAAyD;QACzD,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;YACnB,KAAK,MAAM,GAAG,IAAI,KAAK,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;gBACxC,iCAAiC;gBACjC,MAAM,OAAO,GAAG,IAAI,CAAC,6BAA6B,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBAC9D,UAAU,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,CAAC;YAChC,CAAC;QACL,CAAC;QAED,OAAO,UAAU,CAAC;IACtB,CAAC;IAED;;OAEG;IACK,6BAA6B,CAAC,KAAkB;QACpD,IAAI,KAAK,YAAY,+BAAiB,EAAE,CAAC;YACrC,IAAI,CAAC;gBACD,OAAO,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACpE,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,wDAAwD;gBACxD,OAAO,CAAC,IAAI,CAAC,mDAAmD,EAAE,KAAK,CAAC,CAAC;gBACzE,OAAO,EAAE,CAAC;YACd,CAAC;QACL,CAAC;aAAM,IAAI,KAAK,YAAY,+BAAiB,EAAE,CAAC;YAC5C,2EAA2E;YAC3E,wGAAwG;YACxG,OAAO,IAAI,CAAC,6BAA6B,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC1D,CAAC;QACD,OAAO,EAAE,CAAC;IACd,CAAC;CACJ;AAzLD,8DAyLC"}