rawsql-ts 0.1.0-beta.4 → 0.1.0-beta.6

package/dist/transformers/TableSourceCollector.js

@@ -0,0 +1,384 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TableSourceCollector = void 0;
+const Clause_1 = require("../models/Clause");
+const SelectQuery_1 = require("../models/SelectQuery");
+const ValueComponent_1 = require("../models/ValueComponent");
+const CTECollector_1 = require("./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.
+ */
+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(SelectQuery_1.SimpleSelectQuery.kind, (expr) => this.visitSimpleSelectQuery(expr));
+        this.handlers.set(SelectQuery_1.BinarySelectQuery.kind, (expr) => this.visitBinarySelectQuery(expr));
+        this.handlers.set(SelectQuery_1.ValuesQuery.kind, (expr) => this.visitValuesQuery(expr));
+        // WITH clause and common tables
+        this.handlers.set(Clause_1.WithClause.kind, (expr) => this.visitWithClause(expr));
+        this.handlers.set(Clause_1.CommonTable.kind, (expr) => this.visitCommonTable(expr));
+        // Handlers for FROM and JOIN components
+        this.handlers.set(Clause_1.FromClause.kind, (expr) => this.visitFromClause(expr));
+        this.handlers.set(Clause_1.JoinClause.kind, (expr) => this.visitJoinClause(expr));
+        this.handlers.set(Clause_1.JoinOnClause.kind, (expr) => this.visitJoinOnClause(expr));
+        this.handlers.set(Clause_1.JoinUsingClause.kind, (expr) => this.visitJoinUsingClause(expr));
+        // Source components
+        this.handlers.set(Clause_1.SourceExpression.kind, (expr) => this.visitSourceExpression(expr));
+        this.handlers.set(Clause_1.TableSource.kind, (expr) => this.visitTableSource(expr));
+        this.handlers.set(Clause_1.ParenSource.kind, (expr) => this.visitParenSource(expr));
+        this.handlers.set(Clause_1.SubQuerySource.kind, (expr) => this.visitSubQuerySource(expr));
+        this.handlers.set(ValueComponent_1.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(Clause_1.WhereClause.kind, (expr) => this.visitWhereClause(expr));
+            this.handlers.set(Clause_1.GroupByClause.kind, (expr) => this.visitGroupByClause(expr));
+            this.handlers.set(Clause_1.HavingClause.kind, (expr) => this.visitHavingClause(expr));
+            this.handlers.set(Clause_1.OrderByClause.kind, (expr) => this.visitOrderByClause(expr));
+            this.handlers.set(Clause_1.WindowFrameClause.kind, (expr) => this.visitWindowFrameClause(expr));
+            this.handlers.set(Clause_1.LimitClause.kind, (expr) => this.visitLimitClause(expr));
+            this.handlers.set(Clause_1.ForClause.kind, (expr) => this.visitForClause(expr));
+            this.handlers.set(Clause_1.OrderByItem.kind, (expr) => this.visitOrderByItem(expr));
+            this.handlers.set(Clause_1.SelectClause.kind, (expr) => this.visitSelectClause(expr));
+            this.handlers.set(Clause_1.SelectItem.kind, (expr) => this.visitSelectItem(expr));
+            // Value components that might contain table references
+            this.handlers.set(ValueComponent_1.ParenExpression.kind, (expr) => this.visitParenExpression(expr));
+            this.handlers.set(ValueComponent_1.BinaryExpression.kind, (expr) => this.visitBinaryExpression(expr));
+            this.handlers.set(ValueComponent_1.UnaryExpression.kind, (expr) => this.visitUnaryExpression(expr));
+            this.handlers.set(ValueComponent_1.CaseExpression.kind, (expr) => this.visitCaseExpression(expr));
+            this.handlers.set(ValueComponent_1.CaseKeyValuePair.kind, (expr) => this.visitCaseKeyValuePair(expr));
+            this.handlers.set(ValueComponent_1.SwitchCaseArgument.kind, (expr) => this.visitSwitchCaseArgument(expr));
+            this.handlers.set(ValueComponent_1.BetweenExpression.kind, (expr) => this.visitBetweenExpression(expr));
+            this.handlers.set(ValueComponent_1.FunctionCall.kind, (expr) => this.visitFunctionCall(expr));
+            this.handlers.set(ValueComponent_1.ArrayExpression.kind, (expr) => this.visitArrayExpression(expr));
+            this.handlers.set(ValueComponent_1.TupleExpression.kind, (expr) => this.visitTupleExpression(expr));
+            this.handlers.set(ValueComponent_1.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_1.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);
+    }
+}
+exports.TableSourceCollector = TableSourceCollector;
+//# sourceMappingURL=TableSourceCollector.js.map

package/dist/transformers/TableSourceCollector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"TableSourceCollector.js","sourceRoot":"","sources":["../../src/transformers/TableSourceCollector.ts"],"names":[],"mappings":";;;AAAA,6CAA4U;AAC5U,uDAAuG;AAEvG,6DAQkC;AAClC,iDAA8C;AAE9C;;;;;;;;;;;;;GAaG;AACH,MAAa,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,+BAAiB,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,IAAyB,CAAC,CAAC,CAAC;QAC5G,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,+BAAiB,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,IAAyB,CAAC,CAAC,CAAC;QAC5G,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,yBAAW,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,mBAAU,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,IAAkB,CAAC,CAAC,CAAC;QACvF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,oBAAW,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,mBAAU,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,IAAkB,CAAC,CAAC,CAAC;QACvF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,mBAAU,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,IAAkB,CAAC,CAAC,CAAC;QACvF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,qBAAY,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAoB,CAAC,CAAC,CAAC;QAC7F,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,wBAAe,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,yBAAgB,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,IAAwB,CAAC,CAAC,CAAC;QACzG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,oBAAW,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAmB,CAAC,CAAC,CAAC;QAC1F,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,oBAAW,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAmB,CAAC,CAAC,CAAC;QAC1F,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,uBAAc,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,mBAAmB,CAAC,IAAsB,CAAC,CAAC,CAAC;QACnG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,4BAAW,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,oBAAW,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAmB,CAAC,CAAC,CAAC;YAC1F,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,sBAAa,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,kBAAkB,CAAC,IAAqB,CAAC,CAAC,CAAC;YAChG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,qBAAY,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAoB,CAAC,CAAC,CAAC;YAC7F,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,sBAAa,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,kBAAkB,CAAC,IAAqB,CAAC,CAAC,CAAC;YAChG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,0BAAiB,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,IAAyB,CAAC,CAAC,CAAC;YAC5G,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,oBAAW,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAmB,CAAC,CAAC,CAAC;YAC1F,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,kBAAS,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,IAAiB,CAAC,CAAC,CAAC;YACpF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,oBAAW,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAmB,CAAC,CAAC,CAAC;YAC1F,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,qBAAY,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAoB,CAAC,CAAC,CAAC;YAC7F,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,mBAAU,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,gCAAe,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAuB,CAAC,CAAC,CAAC;YACtG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,iCAAgB,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,IAAwB,CAAC,CAAC,CAAC;YACzG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,gCAAe,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAuB,CAAC,CAAC,CAAC;YACtG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,+BAAc,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,mBAAmB,CAAC,IAAsB,CAAC,CAAC,CAAC;YACnG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,iCAAgB,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,IAAwB,CAAC,CAAC,CAAC;YACzG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,mCAAkB,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,uBAAuB,CAAC,IAA0B,CAAC,CAAC,CAAC;YAC/G,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,kCAAiB,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,IAAyB,CAAC,CAAC,CAAC;YAC5G,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,6BAAY,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAoB,CAAC,CAAC,CAAC;YAC7F,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,gCAAe,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAuB,CAAC,CAAC,CAAC;YACtG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,gCAAe,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAuB,CAAC,CAAC,CAAC;YACtG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,+BAAc,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,2BAAY,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;AAhbD,oDAgbC"}

package/dist/transformers/UpstreamSelectQueryFinder.js

@@ -0,0 +1,129 @@
+"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 UNION queries, it checks each branch independently.
+ */
+class UpstreamSelectQueryFinder {
+    constructor(tableColumnResolver) {
+        this.tableColumnResolver = tableColumnResolver;
+        this.columnCollector = new SelectableColumnCollector_1.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_1.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 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) {
+            // 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 SelectQuery_1.BinarySelectQuery) {
+            const left = this.findUpstream(query.left, columnNames, cteMap);
+            const right = this.findUpstream(query.right, columnNames, cteMap);
+            return [...left, ...right];
+        }
+        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,2EAAwE;AACxE,iDAA8C;AAE9C;;;;;GAKG;AACH,MAAa,yBAAyB;IAIlC,YAAY,mBAAqD;QAC7D,IAAI,CAAC,mBAAmB,GAAG,mBAAmB,CAAC;QAC/C,IAAI,CAAC,eAAe,GAAG,IAAI,qDAAyB,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IACnF,CAAC;IAED;;;;;OAKG;IACI,IAAI,CAAC,KAAkB,EAAE,WAAqB;QACjD,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,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,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,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,+BAAiB,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;AA5HD,8DA4HC"}

package/dist/utils/charLookupTable.js

@@ -0,0 +1,73 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CharLookupTable = void 0;
+/**
+ * Fast character classification utilities for SQL tokenization
+ */
+class CharLookupTable {
+    static isWhitespace(char) {
+        if (char.length !== 1)
+            return false;
+        const code = char.charCodeAt(0);
+        // Check for space(32), tab(9), line feed(10), carriage return(13)
+        return code === 32 || code === 9 || code === 10 || code === 13;
+    }
+    static isDigit(char) {
+        if (char.length !== 1)
+            return false;
+        const code = char.charCodeAt(0);
+        // Check if within '0'(48) to '9'(57) range
+        return code >= 48 && code <= 57;
+    }
+    static isHexChar(char) {
+        if (char.length !== 1)
+            return false;
+        const code = char.charCodeAt(0);
+        // Check if '0'(48) to '9'(57) or 'a'(97) to 'f'(102) or 'A'(65) to 'F'(70)
+        return (code >= 48 && code <= 57) ||
+            (code >= 97 && code <= 102) ||
+            (code >= 65 && code <= 70);
+    }
+    static isOperatorSymbol(char) {
+        if (char.length !== 1)
+            return false;
+        const code = char.charCodeAt(0);
+        // Check for specific operator character codes
+        // '+'=43, '-'=45, '*'=42, '/'=47, '%'=37, '~'=126, '@'=64, '#'=35, '^'=94, 
+        // '&'=38, ':'=58, '!'=33, '<'=60, '>'=62, '='=61, '|'=124
+        return code === 43 || code === 45 || code === 42 || code === 47 ||
+            code === 37 || code === 126 || code === 64 || code === 35 ||
+            code === 94 || code === 38 || code === 58 || code === 33 ||
+            code === 60 || code === 62 || code === 61 || code === 124;
+    }
+    static isDelimiter(char) {
+        if (char.length !== 1)
+            return false;
+        const code = char.charCodeAt(0);
+        // First check delimiters: '.'=46, ','=44, '('=40, ')'=41, '['=91, ']'=93, '{'=123, '}'=125, ';'=59
+        if (code === 46 || code === 44 || code === 40 || code === 41 ||
+            code === 91 || code === 93 || code === 123 || code === 125 || code === 59) {
+            return true;
+        }
+        // Then check for whitespace: ' '=32, '\t'=9, '\n'=10, '\r'=13
+        if (code === 32 || code === 9 || code === 10 || code === 13) {
+            return true;
+        }
+        // Finally check for operator symbols
+        // '+'=43, '-'=45, '*'=42, '/'=47, '%'=37, '~'=126, '@'=64, '#'=35, '^'=94, 
+        // '&'=38, ':'=58, '!'=33, '<'=60, '>'=62, '='=61, '|'=124
+        return code === 43 || code === 45 || code === 42 || code === 47 ||
+            code === 37 || code === 126 || code === 64 || code === 35 ||
+            code === 94 || code === 38 || code === 58 || code === 33 ||
+            code === 60 || code === 62 || code === 61 || code === 124;
+    }
+    static isNamedParameterPrefix(char) {
+        if (char.length !== 1)
+            return false;
+        const code = char.charCodeAt(0);
+        // Check for parameter prefix characters: '@'=64, ':'=58, '$'=36
+        return code === 64 || code === 58 || code === 36;
+    }
+}
+exports.CharLookupTable = CharLookupTable;
+//# sourceMappingURL=charLookupTable.js.map

package/dist/utils/charLookupTable.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"charLookupTable.js","sourceRoot":"","sources":["../../src/utils/charLookupTable.ts"],"names":[],"mappings":";;;AAAA;;GAEG;AACH,MAAa,eAAe;IACjB,MAAM,CAAC,YAAY,CAAC,IAAY;QACnC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC;QACpC,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QAChC,kEAAkE;QAClE,OAAO,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,CAAC,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE,CAAC;IACnE,CAAC;IAEM,MAAM,CAAC,OAAO,CAAC,IAAY;QAC9B,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC;QACpC,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QAChC,2CAA2C;QAC3C,OAAO,IAAI,IAAI,EAAE,IAAI,IAAI,IAAI,EAAE,CAAC;IACpC,CAAC;IAEM,MAAM,CAAC,SAAS,CAAC,IAAY;QAChC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC;QACpC,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QAChC,2EAA2E;QAC3E,OAAO,CAAC,IAAI,IAAI,EAAE,IAAI,IAAI,IAAI,EAAE,CAAC;YAC7B,CAAC,IAAI,IAAI,EAAE,IAAI,IAAI,IAAI,GAAG,CAAC;YAC3B,CAAC,IAAI,IAAI,EAAE,IAAI,IAAI,IAAI,EAAE,CAAC,CAAC;IACnC,CAAC;IAEM,MAAM,CAAC,gBAAgB,CAAC,IAAY;QACvC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC;QACpC,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QAEhC,8CAA8C;QAC9C,4EAA4E;QAC5E,0DAA0D;QAC1D,OAAO,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE;YAC3D,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE;YACzD,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE;YACxD,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,GAAG,CAAC;IAClE,CAAC;IAEM,MAAM,CAAC,WAAW,CAAC,IAAY;QAClC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC;QACpC,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QAEhC,mGAAmG;QACnG,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE;YACxD,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,EAAE,EAAE,CAAC;YAC5E,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,8DAA8D;QAC9D,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,CAAC,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE,EAAE,CAAC;YAC1D,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,qCAAqC;QACrC,4EAA4E;QAC5E,0DAA0D;QAC1D,OAAO,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE;YAC3D,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE;YACzD,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE;YACxD,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,GAAG,CAAC;IAClE,CAAC;IAEM,MAAM,CAAC,sBAAsB,CAAC,IAAY;QAC7C,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC;QACpC,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QAEhC,gEAAgE;QAChE,OAAO,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE,CAAC;IACrD,CAAC;CACJ;AApED,0CAoEC"}