rawsql-ts 0.1.1-beta.1 → 0.1.1-beta.3

package/dist/esm/transformers/TableSourceCollector.js

@@ -0,0 +1,380 @@
+import { CommonTable, ForClause, FromClause, GroupByClause, HavingClause, JoinClause, JoinOnClause, JoinUsingClause, LimitClause, OrderByClause, OrderByItem, ParenSource, SelectClause, SelectItem, SourceExpression, SubQuerySource, TableSource, WhereClause, WindowFrameClause, WithClause } from "../models/Clause";
+import { BinarySelectQuery, SimpleSelectQuery, ValuesQuery } from "../models/SelectQuery";
+import { ArrayExpression, BetweenExpression, BinaryExpression, CaseExpression, CaseKeyValuePair, CastExpression, FunctionCall, InlineQuery, ParenExpression, SwitchCaseArgument, TupleExpression, UnaryExpression } from "../models/ValueComponent";
+import { CTECollector } from "./CTECollector";
+/**
+ * A visitor that collects all table source names from a SQL query structure.
+ *
+ * When selectableOnly is true (default behavior):
+ * - Includes only table sources from FROM and JOIN clauses
+ * - Excludes inline queries, subqueries, and CTEs
+ *
+ * When selectableOnly is false:
+ * - Scans all parts of the query including WITH clauses, subqueries, etc.
+ * - Collects all table sources from the entire query
+ * - Excludes tables that are managed by CTEs
+ *
+ * For UNION-like queries, it scans both the left and right parts.
+ */
+export class TableSourceCollector {
+    constructor(selectableOnly = true) {
+        this.tableSources = [];
+        this.visitedNodes = new Set();
+        this.tableNameMap = new Map();
+        this.cteNames = new Set();
+        this.isRootVisit = true;
+        this.selectableOnly = selectableOnly;
+        this.handlers = new Map();
+        // Setup handlers for query components
+        this.handlers.set(SimpleSelectQuery.kind, (expr) => this.visitSimpleSelectQuery(expr));
+        this.handlers.set(BinarySelectQuery.kind, (expr) => this.visitBinarySelectQuery(expr));
+        this.handlers.set(ValuesQuery.kind, (expr) => this.visitValuesQuery(expr));
+        // WITH clause and common tables
+        this.handlers.set(WithClause.kind, (expr) => this.visitWithClause(expr));
+        this.handlers.set(CommonTable.kind, (expr) => this.visitCommonTable(expr));
+        // Handlers for FROM and JOIN components
+        this.handlers.set(FromClause.kind, (expr) => this.visitFromClause(expr));
+        this.handlers.set(JoinClause.kind, (expr) => this.visitJoinClause(expr));
+        this.handlers.set(JoinOnClause.kind, (expr) => this.visitJoinOnClause(expr));
+        this.handlers.set(JoinUsingClause.kind, (expr) => this.visitJoinUsingClause(expr));
+        // Source components
+        this.handlers.set(SourceExpression.kind, (expr) => this.visitSourceExpression(expr));
+        this.handlers.set(TableSource.kind, (expr) => this.visitTableSource(expr));
+        this.handlers.set(ParenSource.kind, (expr) => this.visitParenSource(expr));
+        this.handlers.set(SubQuerySource.kind, (expr) => this.visitSubQuerySource(expr));
+        this.handlers.set(InlineQuery.kind, (expr) => this.visitInlineQuery(expr));
+        // Only register these handlers when not in selectableOnly mode
+        if (!selectableOnly) {
+            // Additional clause handlers for full scanning
+            this.handlers.set(WhereClause.kind, (expr) => this.visitWhereClause(expr));
+            this.handlers.set(GroupByClause.kind, (expr) => this.visitGroupByClause(expr));
+            this.handlers.set(HavingClause.kind, (expr) => this.visitHavingClause(expr));
+            this.handlers.set(OrderByClause.kind, (expr) => this.visitOrderByClause(expr));
+            this.handlers.set(WindowFrameClause.kind, (expr) => this.visitWindowFrameClause(expr));
+            this.handlers.set(LimitClause.kind, (expr) => this.visitLimitClause(expr));
+            this.handlers.set(ForClause.kind, (expr) => this.visitForClause(expr));
+            this.handlers.set(OrderByItem.kind, (expr) => this.visitOrderByItem(expr));
+            this.handlers.set(SelectClause.kind, (expr) => this.visitSelectClause(expr));
+            this.handlers.set(SelectItem.kind, (expr) => this.visitSelectItem(expr));
+            // Value components that might contain table references
+            this.handlers.set(ParenExpression.kind, (expr) => this.visitParenExpression(expr));
+            this.handlers.set(BinaryExpression.kind, (expr) => this.visitBinaryExpression(expr));
+            this.handlers.set(UnaryExpression.kind, (expr) => this.visitUnaryExpression(expr));
+            this.handlers.set(CaseExpression.kind, (expr) => this.visitCaseExpression(expr));
+            this.handlers.set(CaseKeyValuePair.kind, (expr) => this.visitCaseKeyValuePair(expr));
+            this.handlers.set(SwitchCaseArgument.kind, (expr) => this.visitSwitchCaseArgument(expr));
+            this.handlers.set(BetweenExpression.kind, (expr) => this.visitBetweenExpression(expr));
+            this.handlers.set(FunctionCall.kind, (expr) => this.visitFunctionCall(expr));
+            this.handlers.set(ArrayExpression.kind, (expr) => this.visitArrayExpression(expr));
+            this.handlers.set(TupleExpression.kind, (expr) => this.visitTupleExpression(expr));
+            this.handlers.set(CastExpression.kind, (expr) => this.visitCastExpression(expr));
+        }
+    }
+    /**
+     * Gets all collected table sources
+     */
+    getTableSources() {
+        return this.tableSources;
+    }
+    /**
+     * Reset the collection of table sources
+     */
+    reset() {
+        this.tableSources = [];
+        this.tableNameMap.clear();
+        this.visitedNodes.clear();
+        this.cteNames.clear();
+    }
+    /**
+     * Gets a unique identifier for a table source
+     */
+    getTableIdentifier(source) {
+        let identifier = source.table.name;
+        if (source.namespaces && source.namespaces.length > 0) {
+            identifier = source.namespaces.map(ns => ns.name).join('.') + '.' + identifier;
+        }
+        return identifier;
+    }
+    collect(query) {
+        // Visit the SQL component to collect table sources
+        this.visit(query);
+        return this.getTableSources();
+    }
+    /**
+     * Main entry point for the visitor pattern.
+     * Implements the shallow visit pattern to distinguish between root and recursive visits.
+     */
+    visit(arg) {
+        // If not a root visit, just visit the node and return
+        if (!this.isRootVisit) {
+            this.visitNode(arg);
+            return;
+        }
+        // If this is a root visit, we need to reset the state
+        this.reset();
+        this.isRootVisit = false;
+        try {
+            // When in full scan mode, collect CTEs first to exclude them from table sources
+            if (!this.selectableOnly) {
+                this.collectCTEs(arg);
+            }
+            this.visitNode(arg);
+        }
+        finally {
+            // Regardless of success or failure, reset the root visit flag
+            this.isRootVisit = true;
+        }
+    }
+    /**
+     * Internal visit method used for all nodes.
+     * This separates the visit flag management from the actual node visitation logic.
+     */
+    visitNode(arg) {
+        // Skip if we've already visited this node to prevent infinite recursion
+        if (this.visitedNodes.has(arg)) {
+            return;
+        }
+        // Mark as visited
+        this.visitedNodes.add(arg);
+        const handler = this.handlers.get(arg.getKind());
+        if (handler) {
+            handler(arg);
+            return;
+        }
+        // If no handler found, that's ok - we only care about specific components
+    }
+    /**
+     * Collects all CTE names to exclude them from real table sources
+     */
+    collectCTEs(query) {
+        // Use CommonTableCollector to get all CTEs
+        const cteCollector = new CTECollector();
+        cteCollector.visit(query);
+        const commonTables = cteCollector.getCommonTables();
+        // Add CTE names to the set
+        for (const cte of commonTables) {
+            this.cteNames.add(cte.aliasExpression.table.name);
+        }
+    }
+    visitSimpleSelectQuery(query) {
+        // Process the FROM and JOIN clauses
+        if (query.fromClause) {
+            query.fromClause.accept(this);
+        }
+        // If in full scan mode, visit all other clauses too
+        if (!this.selectableOnly) {
+            if (query.WithClause) {
+                query.WithClause.accept(this);
+            }
+            if (query.whereClause) {
+                query.whereClause.accept(this);
+            }
+            if (query.groupByClause) {
+                query.groupByClause.accept(this);
+            }
+            if (query.havingClause) {
+                query.havingClause.accept(this);
+            }
+            if (query.orderByClause) {
+                query.orderByClause.accept(this);
+            }
+            if (query.windowFrameClause) {
+                query.windowFrameClause.accept(this);
+            }
+            if (query.rowLimitClause) {
+                query.rowLimitClause.accept(this);
+            }
+            if (query.forClause) {
+                query.forClause.accept(this);
+            }
+            query.selectClause.accept(this);
+        }
+    }
+    visitBinarySelectQuery(query) {
+        // For UNION-like queries, visit both sides
+        query.left.accept(this);
+        query.right.accept(this);
+    }
+    visitValuesQuery(query) {
+        if (!this.selectableOnly) {
+            // VALUES queries might contain subqueries in tuple expressions
+            for (const tuple of query.tuples) {
+                tuple.accept(this);
+            }
+        }
+    }
+    visitWithClause(withClause) {
+        if (!this.selectableOnly) {
+            // Visit each CommonTable
+            for (const table of withClause.tables) {
+                table.accept(this);
+            }
+        }
+    }
+    visitCommonTable(commonTable) {
+        if (!this.selectableOnly) {
+            // Process the query within the common table
+            commonTable.query.accept(this);
+        }
+    }
+    visitFromClause(fromClause) {
+        // Check the main source in FROM clause
+        fromClause.source.accept(this);
+        // Check all JOIN clauses
+        if (fromClause.joins) {
+            for (const join of fromClause.joins) {
+                join.accept(this);
+            }
+        }
+    }
+    visitSourceExpression(source) {
+        // Process the actual data source, ignoring aliases
+        source.datasource.accept(this);
+    }
+    visitTableSource(source) {
+        // Get the table identifier for uniqueness check
+        const identifier = this.getTableIdentifier(source);
+        // Check if this is a table managed by a CTE
+        if (!this.tableNameMap.has(identifier) && !this.isCTETable(source.table.name)) {
+            this.tableNameMap.set(identifier, true);
+            this.tableSources.push(source);
+        }
+    }
+    /**
+     * Checks if a table name is a CTE name
+     */
+    isCTETable(tableName) {
+        return this.cteNames.has(tableName);
+    }
+    visitParenSource(source) {
+        // For parenthesized sources, visit the inner source
+        source.source.accept(this);
+    }
+    visitSubQuerySource(subQuery) {
+        if (!this.selectableOnly) {
+            // In full scan mode, we also check subqueries
+            subQuery.query.accept(this);
+        }
+        // In selectableOnly mode, we don't collect sources from subqueries
+    }
+    visitInlineQuery(inlineQuery) {
+        if (!this.selectableOnly) {
+            // In full scan mode, visit inline queries too
+            inlineQuery.selectQuery.accept(this);
+        }
+    }
+    visitJoinClause(joinClause) {
+        // Visit the source being joined
+        joinClause.source.accept(this);
+        // If full scanning, also visit the join condition
+        if (!this.selectableOnly && joinClause.condition) {
+            joinClause.condition.accept(this);
+        }
+    }
+    visitJoinOnClause(joinOn) {
+        if (!this.selectableOnly) {
+            // In full scan mode, check ON condition for table references
+            joinOn.condition.accept(this);
+        }
+    }
+    visitJoinUsingClause(joinUsing) {
+        if (!this.selectableOnly) {
+            // In full scan mode, check USING condition for table references
+            joinUsing.condition.accept(this);
+        }
+    }
+    // Additional visitor methods only used in full scan mode
+    visitWhereClause(whereClause) {
+        whereClause.condition.accept(this);
+    }
+    visitGroupByClause(clause) {
+        for (const item of clause.grouping) {
+            item.accept(this);
+        }
+    }
+    visitHavingClause(clause) {
+        clause.condition.accept(this);
+    }
+    visitOrderByClause(clause) {
+        for (const item of clause.order) {
+            item.accept(this);
+        }
+    }
+    visitWindowFrameClause(clause) {
+        clause.expression.accept(this);
+    }
+    visitLimitClause(clause) {
+        clause.limit.accept(this);
+        if (clause.offset) {
+            clause.offset.accept(this);
+        }
+    }
+    visitForClause(clause) {
+        // FOR clause doesn't contain table sources
+    }
+    visitOrderByItem(item) {
+        item.value.accept(this);
+    }
+    visitSelectClause(clause) {
+        for (const item of clause.items) {
+            item.accept(this);
+        }
+    }
+    visitSelectItem(item) {
+        item.value.accept(this);
+    }
+    visitParenExpression(expr) {
+        expr.expression.accept(this);
+    }
+    visitBinaryExpression(expr) {
+        expr.left.accept(this);
+        expr.right.accept(this);
+    }
+    visitUnaryExpression(expr) {
+        expr.expression.accept(this);
+    }
+    visitCaseExpression(expr) {
+        if (expr.condition) {
+            expr.condition.accept(this);
+        }
+        expr.switchCase.accept(this);
+    }
+    visitSwitchCaseArgument(switchCase) {
+        for (const caseItem of switchCase.cases) {
+            caseItem.accept(this);
+        }
+        if (switchCase.elseValue) {
+            switchCase.elseValue.accept(this);
+        }
+    }
+    visitCaseKeyValuePair(pair) {
+        pair.key.accept(this);
+        pair.value.accept(this);
+    }
+    visitBetweenExpression(expr) {
+        expr.expression.accept(this);
+        expr.lower.accept(this);
+        expr.upper.accept(this);
+    }
+    visitFunctionCall(func) {
+        if (func.argument) {
+            func.argument.accept(this);
+        }
+        if (func.over) {
+            func.over.accept(this);
+        }
+    }
+    visitArrayExpression(expr) {
+        expr.expression.accept(this);
+    }
+    visitTupleExpression(expr) {
+        for (const value of expr.values) {
+            value.accept(this);
+        }
+    }
+    visitCastExpression(expr) {
+        expr.input.accept(this);
+        expr.castType.accept(this);
+    }
+}
+//# sourceMappingURL=TableSourceCollector.js.map

package/dist/esm/transformers/TableSourceCollector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"TableSourceCollector.js","sourceRoot":"","sources":["../../../src/transformers/TableSourceCollector.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,UAAU,EAAE,aAAa,EAAE,YAAY,EAAE,UAAU,EAAE,YAAY,EAAE,eAAe,EAAE,WAAW,EAAE,aAAa,EAAE,WAAW,EAAE,WAAW,EAAqB,YAAY,EAAE,UAAU,EAAE,gBAAgB,EAAE,cAAc,EAAE,WAAW,EAAE,WAAW,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC5U,OAAO,EAAE,iBAAiB,EAAe,iBAAiB,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAEvG,OAAO,EACH,eAAe,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,cAAc,EAAE,gBAAgB,EACtF,cAAc,EAAmB,YAAY,EAAE,WAAW,EAAE,eAAe,EACtD,kBAAkB,EAAE,eAAe,EAAE,eAAe,EAK5E,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAE9C;;;;;;;;;;;;;GAaG;AACH,MAAM,OAAO,oBAAoB;IAS7B,YAAY,iBAA0B,IAAI;QAPlC,iBAAY,GAAkB,EAAE,CAAC;QACjC,iBAAY,GAAsB,IAAI,GAAG,EAAE,CAAC;QAC5C,iBAAY,GAAyB,IAAI,GAAG,EAAmB,CAAC;QAEhE,aAAQ,GAAgB,IAAI,GAAG,EAAU,CAAC;QAC1C,gBAAW,GAAY,IAAI,CAAC;QAGhC,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,QAAQ,GAAG,IAAI,GAAG,EAA8B,CAAC;QAEtD,sCAAsC;QACtC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,IAAyB,CAAC,CAAC,CAAC;QAC5G,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,IAAyB,CAAC,CAAC,CAAC;QAC5G,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAmB,CAAC,CAAC,CAAC;QAE1F,gCAAgC;QAChC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,IAAkB,CAAC,CAAC,CAAC;QACvF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAmB,CAAC,CAAC,CAAC;QAE1F,wCAAwC;QACxC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,IAAkB,CAAC,CAAC,CAAC;QACvF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,IAAkB,CAAC,CAAC,CAAC;QACvF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAoB,CAAC,CAAC,CAAC;QAC7F,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAuB,CAAC,CAAC,CAAC;QAEtG,oBAAoB;QACpB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,gBAAgB,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,IAAwB,CAAC,CAAC,CAAC;QACzG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAmB,CAAC,CAAC,CAAC;QAC1F,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAmB,CAAC,CAAC,CAAC;QAC1F,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,mBAAmB,CAAC,IAAsB,CAAC,CAAC,CAAC;QACnG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAmB,CAAC,CAAC,CAAC;QAE1F,+DAA+D;QAC/D,IAAI,CAAC,cAAc,EAAE,CAAC;YAClB,+CAA+C;YAC/C,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAmB,CAAC,CAAC,CAAC;YAC1F,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,kBAAkB,CAAC,IAAqB,CAAC,CAAC,CAAC;YAChG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAoB,CAAC,CAAC,CAAC;YAC7F,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,kBAAkB,CAAC,IAAqB,CAAC,CAAC,CAAC;YAChG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,IAAyB,CAAC,CAAC,CAAC;YAC5G,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAmB,CAAC,CAAC,CAAC;YAC1F,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,IAAiB,CAAC,CAAC,CAAC;YACpF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAmB,CAAC,CAAC,CAAC;YAC1F,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAoB,CAAC,CAAC,CAAC;YAC7F,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,IAAkB,CAAC,CAAC,CAAC;YAEvF,uDAAuD;YACvD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAuB,CAAC,CAAC,CAAC;YACtG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,gBAAgB,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,IAAwB,CAAC,CAAC,CAAC;YACzG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAuB,CAAC,CAAC,CAAC;YACtG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,mBAAmB,CAAC,IAAsB,CAAC,CAAC,CAAC;YACnG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,gBAAgB,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,IAAwB,CAAC,CAAC,CAAC;YACzG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,kBAAkB,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,uBAAuB,CAAC,IAA0B,CAAC,CAAC,CAAC;YAC/G,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,IAAyB,CAAC,CAAC,CAAC;YAC5G,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAoB,CAAC,CAAC,CAAC;YAC7F,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAuB,CAAC,CAAC,CAAC;YACtG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAuB,CAAC,CAAC,CAAC;YACtG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,mBAAmB,CAAC,IAAsB,CAAC,CAAC,CAAC;QACvG,CAAC;IACL,CAAC;IAED;;OAEG;IACI,eAAe;QAClB,OAAO,IAAI,CAAC,YAAY,CAAC;IAC7B,CAAC;IAED;;OAEG;IACK,KAAK;QACT,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC;QACvB,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;QAC1B,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;QAC1B,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC;IAED;;OAEG;IACK,kBAAkB,CAAC,MAAmB;QAC1C,IAAI,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC;QACnC,IAAI,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpD,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,GAAG,UAAU,CAAC;QACnF,CAAC;QACD,OAAO,UAAU,CAAC;IACtB,CAAC;IAEM,OAAO,CAAC,KAAmB;QAC9B,mDAAmD;QACnD,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAClB,OAAO,IAAI,CAAC,eAAe,EAAE,CAAC;IAClC,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,GAAiB;QAC1B,sDAAsD;QACtD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACpB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YACpB,OAAO;QACX,CAAC;QAED,sDAAsD;QACtD,IAAI,CAAC,KAAK,EAAE,CAAC;QACb,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;QAEzB,IAAI,CAAC;YACD,gFAAgF;YAChF,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;gBACvB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;YAC1B,CAAC;YACD,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QACxB,CAAC;gBAAS,CAAC;YACP,8DAA8D;YAC9D,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAC5B,CAAC;IACL,CAAC;IAED;;;OAGG;IACK,SAAS,CAAC,GAAiB;QAC/B,wEAAwE;QACxE,IAAI,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YAC7B,OAAO;QACX,CAAC;QAED,kBAAkB;QAClB,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAE3B,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;QACjD,IAAI,OAAO,EAAE,CAAC;YACV,OAAO,CAAC,GAAG,CAAC,CAAC;YACb,OAAO;QACX,CAAC;QAED,0EAA0E;IAC9E,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,KAAmB;QACnC,2CAA2C;QAC3C,MAAM,YAAY,GAAG,IAAI,YAAY,EAAE,CAAC;QACxC,YAAY,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC1B,MAAM,YAAY,GAAG,YAAY,CAAC,eAAe,EAAE,CAAC;QAEpD,2BAA2B;QAC3B,KAAK,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;YAC7B,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACtD,CAAC;IACL,CAAC;IAEO,sBAAsB,CAAC,KAAwB;QACnD,oCAAoC;QACpC,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;YACnB,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAClC,CAAC;QAED,oDAAoD;QACpD,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACvB,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;gBACnB,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAClC,CAAC;YAED,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;gBACpB,KAAK,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACnC,CAAC;YAED,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;gBACtB,KAAK,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACrC,CAAC;YAED,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC;gBACrB,KAAK,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACpC,CAAC;YAED,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;gBACtB,KAAK,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACrC,CAAC;YAED,IAAI,KAAK,CAAC,iBAAiB,EAAE,CAAC;gBAC1B,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACzC,CAAC;YAED,IAAI,KAAK,CAAC,cAAc,EAAE,CAAC;gBACvB,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACtC,CAAC;YAED,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;gBAClB,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACjC,CAAC;YAED,KAAK,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACpC,CAAC;IACL,CAAC;IAEO,sBAAsB,CAAC,KAAwB;QACnD,2CAA2C;QAC3C,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACxB,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC7B,CAAC;IAEO,gBAAgB,CAAC,KAAkB;QACvC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACvB,+DAA+D;YAC/D,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;gBAC/B,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACvB,CAAC;QACL,CAAC;IACL,CAAC;IAEO,eAAe,CAAC,UAAsB;QAC1C,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACvB,yBAAyB;YACzB,KAAK,MAAM,KAAK,IAAI,UAAU,CAAC,MAAM,EAAE,CAAC;gBACpC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACvB,CAAC;QACL,CAAC;IACL,CAAC;IAEO,gBAAgB,CAAC,WAAwB;QAC7C,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACvB,4CAA4C;YAC5C,WAAW,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACnC,CAAC;IACL,CAAC;IAEO,eAAe,CAAC,UAAsB;QAC1C,uCAAuC;QACvC,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAE/B,yBAAyB;QACzB,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;YACnB,KAAK,MAAM,IAAI,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;gBAClC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACtB,CAAC;QACL,CAAC;IACL,CAAC;IAEO,qBAAqB,CAAC,MAAwB;QAClD,mDAAmD;QACnD,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAEO,gBAAgB,CAAC,MAAmB;QACxC,gDAAgD;QAChD,MAAM,UAAU,GAAG,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAEnD,4CAA4C;QAC5C,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5E,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;YACxC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,UAAU,CAAC,SAAiB;QAChC,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACxC,CAAC;IAEO,gBAAgB,CAAC,MAAmB;QACxC,oDAAoD;QACpD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC/B,CAAC;IAEO,mBAAmB,CAAC,QAAwB;QAChD,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACvB,8CAA8C;YAC9C,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAChC,CAAC;QACD,mEAAmE;IACvE,CAAC;IAEO,gBAAgB,CAAC,WAAwB;QAC7C,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACvB,8CAA8C;YAC9C,WAAW,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACzC,CAAC;IACL,CAAC;IAEO,eAAe,CAAC,UAAsB;QAC1C,gCAAgC;QAChC,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAE/B,kDAAkD;QAClD,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,UAAU,CAAC,SAAS,EAAE,CAAC;YAC/C,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACtC,CAAC;IACL,CAAC;IAEO,iBAAiB,CAAC,MAAoB;QAC1C,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACvB,6DAA6D;YAC7D,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAClC,CAAC;IACL,CAAC;IAEO,oBAAoB,CAAC,SAA0B;QACnD,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACvB,gEAAgE;YAChE,SAAS,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACrC,CAAC;IACL,CAAC;IAED,yDAAyD;IAEjD,gBAAgB,CAAC,WAAwB;QAC7C,WAAW,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACvC,CAAC;IAEO,kBAAkB,CAAC,MAAqB;QAC5C,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YACjC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACtB,CAAC;IACL,CAAC;IAEO,iBAAiB,CAAC,MAAoB;QAC1C,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAClC,CAAC;IAEO,kBAAkB,CAAC,MAAqB;QAC5C,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YAC9B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACtB,CAAC;IACL,CAAC;IAEO,sBAAsB,CAAC,MAAyB;QACpD,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAEO,gBAAgB,CAAC,MAAmB;QACxC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC1B,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC/B,CAAC;IACL,CAAC;IAEO,cAAc,CAAC,MAAiB;QACpC,2CAA2C;IAC/C,CAAC;IAEO,gBAAgB,CAAC,IAAiB;QACtC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;IAEO,iBAAiB,CAAC,MAAoB;QAC1C,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YAC9B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACtB,CAAC;IACL,CAAC;IAEO,eAAe,CAAC,IAAgB;QACpC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;IAEO,oBAAoB,CAAC,IAAqB;QAC9C,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC;IAEO,qBAAqB,CAAC,IAAsB;QAChD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACvB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;IAEO,oBAAoB,CAAC,IAAqB;QAC9C,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC;IAEO,mBAAmB,CAAC,IAAoB;QAC5C,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACjB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAChC,CAAC;QACD,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC;IAEO,uBAAuB,CAAC,UAA8B;QAC1D,KAAK,MAAM,QAAQ,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;YACtC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC1B,CAAC;QAED,IAAI,UAAU,CAAC,SAAS,EAAE,CAAC;YACvB,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACtC,CAAC;IACL,CAAC;IAEO,qBAAqB,CAAC,IAAsB;QAChD,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACtB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;IAEO,sBAAsB,CAAC,IAAuB;QAClD,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC7B,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACxB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;IAEO,iBAAiB,CAAC,IAAkB;QACxC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAChB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC/B,CAAC;QAED,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACZ,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC;IACL,CAAC;IAEO,oBAAoB,CAAC,IAAqB;QAC9C,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC;IAEO,oBAAoB,CAAC,IAAqB;QAC9C,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAC9B,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACvB,CAAC;IACL,CAAC;IAEO,mBAAmB,CAAC,IAAoB;QAC5C,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACxB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC/B,CAAC;CACJ"}

package/dist/esm/transformers/UpstreamSelectQueryFinder.js

@@ -0,0 +1,125 @@
+import { SimpleSelectQuery, BinarySelectQuery, ValuesQuery } from "../models/SelectQuery";
+import { SubQuerySource, TableSource } from "../models/Clause";
+import { SelectableColumnCollector } from "./SelectableColumnCollector";
+import { CTECollector } from "./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 UNION queries, it checks each branch independently.
+ */
+export class UpstreamSelectQueryFinder {
+    constructor(tableColumnResolver) {
+        this.tableColumnResolver = tableColumnResolver;
+        this.columnCollector = new SelectableColumnCollector(this.tableColumnResolver);
+    }
+    /**
+     * Finds the highest SelectQuery containing all specified columns.
+     * @param query The root SelectQuery to search.
+     * @param columnNames Array of column names to check for.
+     * @returns An array of SelectQuery objects, or an empty array if not found.
+     */
+    find(query, columnNames) {
+        // Use CTECollector to collect CTEs from the root query only once and reuse
+        const cteCollector = new CTECollector();
+        const ctes = cteCollector.collect(query);
+        const cteMap = new Map();
+        for (const cte of ctes) {
+            cteMap.set(cte.getAliasSourceName(), cte);
+        }
+        return this.findUpstream(query, columnNames, 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 TableSource) {
+                branchResult = this.handleTableSource(src, columnNames, cteMap);
+                validBranchCount++;
+            }
+            else if (src instanceof SubQuerySource) {
+                branchResult = this.handleSubQuerySource(src, columnNames, cteMap);
+                validBranchCount++;
+            }
+            else if (src instanceof 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 SimpleSelectQuery) {
+            // Check upstream sources first: prioritize searching upstream branches for the required columns.
+            const fromClause = query.fromClause;
+            if (fromClause) {
+                const branchResult = this.processFromClauseBranches(fromClause, columnNames, cteMap);
+                if (branchResult) {
+                    return branchResult;
+                }
+            }
+            // If not found in all upstream branches, check this query itself
+            const columns = this.columnCollector.collect(query).map(col => col.name);
+            const hasAll = columnNames.every(name => columns.includes(name));
+            if (hasAll) {
+                return [query];
+            }
+            return [];
+        }
+        else if (query instanceof BinarySelectQuery) {
+            const left = this.findUpstream(query.left, columnNames, cteMap);
+            const right = this.findUpstream(query.right, columnNames, cteMap);
+            return [...left, ...right];
+        }
+        return [];
+    }
+}
+//# sourceMappingURL=UpstreamSelectQueryFinder.js.map

package/dist/esm/transformers/UpstreamSelectQueryFinder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"UpstreamSelectQueryFinder.js","sourceRoot":"","sources":["../../../src/transformers/UpstreamSelectQueryFinder.ts"],"names":[],"mappings":"AAAA,OAAO,EAAe,iBAAiB,EAAE,iBAAiB,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACvG,OAAO,EAAe,cAAc,EAAE,WAAW,EAAc,MAAM,kBAAkB,CAAC;AACxF,OAAO,EAAE,yBAAyB,EAAE,MAAM,6BAA6B,CAAC;AACxE,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAE9C;;;;;GAKG;AACH,MAAM,OAAO,yBAAyB;IAIlC,YAAY,mBAAqD;QAC7D,IAAI,CAAC,mBAAmB,GAAG,mBAAmB,CAAC;QAC/C,IAAI,CAAC,eAAe,GAAG,IAAI,yBAAyB,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IACnF,CAAC;IAED;;;;;OAKG;IACI,IAAI,CAAC,KAAkB,EAAE,WAAqB;QACjD,2EAA2E;QAC3E,MAAM,YAAY,GAAG,IAAI,YAAY,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,WAAW,EAAE,MAAM,CAAC,CAAC;IACzD,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,MAAiD;QACtH,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,GAAoB,EAAE,CAAC;QAC3C,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,GAAyB,IAAI,CAAC;YAC9C,IAAI,GAAG,YAAY,WAAW,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,cAAc,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,WAAW,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,iBAAiB,EAAE,CAAC;YACrC,iGAAiG;YACjG,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,EAAE,CAAC;oBACf,OAAO,YAAY,CAAC;gBACxB,CAAC;YACL,CAAC;YACD,iEAAiE;YACjE,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACzE,MAAM,MAAM,GAAG,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;YACjE,IAAI,MAAM,EAAE,CAAC;gBACT,OAAO,CAAC,KAAK,CAAC,CAAC;YACnB,CAAC;YACD,OAAO,EAAE,CAAC;QACd,CAAC;aAAM,IAAI,KAAK,YAAY,iBAAiB,EAAE,CAAC;YAC5C,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;CACJ"}

package/dist/esm/types/index.d.ts

@@ -0,0 +1,14 @@
+export * from './parsers/SelectQueryParser';
+export * from './models/BinarySelectQuery';
+export * from './models/SelectQuery';
+export * from './models/ValueComponent';
+export * from './models/ValuesQuery';
+export * from './transformers/CTECollector';
+export * from './transformers/CTENormalizer';
+export * from './transformers/Formatter';
+export * from './transformers/QueryConverter';
+export * from './transformers/SelectValueCollector';
+export * from './transformers/SelectableColumnCollector';
+export * from './transformers/TableColumnResolver';
+export * from './transformers/TableSourceCollector';
+export * from './transformers/UpstreamSelectQueryFinder';

package/dist/esm/types/models/BinarySelectQuery.d.ts

@@ -0,0 +1,91 @@
+import { SourceExpression } from "./Clause";
+import type { SelectQuery } from "./SelectQuery";
+import { SqlComponent } from "./SqlComponent";
+import { RawString } from "./ValueComponent";
+/**
+ * Represents a binary SELECT query (e.g., UNION, INTERSECT, EXCEPT).
+ */
+export declare class BinarySelectQuery extends SqlComponent {
+    static kind: symbol;
+    left: SelectQuery;
+    operator: RawString;
+    right: SelectQuery;
+    constructor(left: SelectQuery, operator: string, right: SelectQuery);
+    /**
+     * Appends another query to this binary query using UNION as the operator.
+     * This creates a new BinarySelectQuery where the left side is this binary query
+     * and the right side is the provided query.
+     *
+     * @param query The query to append with UNION
+     * @returns A new BinarySelectQuery representing "(this) UNION query"
+     */
+    union(query: SelectQuery): BinarySelectQuery;
+    /**
+     * Appends another query to this binary query using UNION ALL as the operator.
+     * This creates a new BinarySelectQuery where the left side is this binary query
+     * and the right side is the provided query.
+     *
+     * @param query The query to append with UNION ALL
+     * @returns A new BinarySelectQuery representing "(this) UNION ALL query"
+     */
+    unionAll(query: SelectQuery): BinarySelectQuery;
+    /**
+     * Appends another query to this binary query using INTERSECT as the operator.
+     * This creates a new BinarySelectQuery where the left side is this binary query
+     * and the right side is the provided query.
+     *
+     * @param query The query to append with INTERSECT
+     * @returns A new BinarySelectQuery representing "(this) INTERSECT query"
+     */
+    intersect(query: SelectQuery): BinarySelectQuery;
+    /**
+     * Appends another query to this binary query using INTERSECT ALL as the operator.
+     * This creates a new BinarySelectQuery where the left side is this binary query
+     * and the right side is the provided query.
+     *
+     * @param query The query to append with INTERSECT ALL
+     * @returns A new BinarySelectQuery representing "(this) INTERSECT ALL query"
+     */
+    intersectAll(query: SelectQuery): BinarySelectQuery;
+    /**
+     * Appends another query to this binary query using EXCEPT as the operator.
+     * This creates a new BinarySelectQuery where the left side is this binary query
+     * and the right side is the provided query.
+     *
+     * @param query The query to append with EXCEPT
+     * @returns A new BinarySelectQuery representing "(this) EXCEPT query"
+     */
+    except(query: SelectQuery): BinarySelectQuery;
+    /**
+     * Appends another query to this binary query using EXCEPT ALL as the operator.
+     * This creates a new BinarySelectQuery where the left side is this binary query
+     * and the right side is the provided query.
+     *
+     * @param query The query to append with EXCEPT ALL
+     * @returns A new BinarySelectQuery representing "(this) EXCEPT ALL query"
+     */
+    exceptAll(query: SelectQuery): BinarySelectQuery;
+    /**
+     * Appends another query to this binary query using the specified operator.
+     * This creates a new BinarySelectQuery where the left side is this binary query
+     * and the right side is the provided query.
+     *
+     * @param operator SQL operator to use (e.g. 'union', 'union all', 'intersect', 'except')
+     * @param query The query to append with the specified operator
+     * @returns A new BinarySelectQuery representing "(this) [operator] query"
+     */
+    appendSelectQuery(operator: string, query: SelectQuery): BinarySelectQuery;
+    /**
+     * Appends another query to this binary query using UNION as the operator, accepting a raw SQL string.
+     * This method parses the SQL string and appends the resulting query using UNION.
+     * @param sql The SQL string to parse and union
+     * @returns A new BinarySelectQuery representing "(this) UNION (parsed query)"
+     */
+    unionRaw(sql: string): BinarySelectQuery;
+    unionAllRaw(sql: string): BinarySelectQuery;
+    intersectRaw(sql: string): BinarySelectQuery;
+    intersectAllRaw(sql: string): BinarySelectQuery;
+    exceptRaw(sql: string): BinarySelectQuery;
+    exceptAllRaw(sql: string): BinarySelectQuery;
+    toSource(alias?: string): SourceExpression;
+}