rawsql-ts 0.28.0 → 0.30.0

package/dist/transformers/StaticPredicatePlacementOptimizer.js

@@ -5,70 +5,11 @@ const Clause_1 = require("../models/Clause");
 const SelectQuery_1 = require("../models/SelectQuery");
 const ValueComponent_1 = require("../models/ValueComponent");
 const SelectQueryParser_1 = require("../parsers/SelectQueryParser");
-const ValueParser_1 = require("../parsers/ValueParser");
 const SqlComponentFormatter_1 = require("./SqlComponentFormatter");
 const SelectOutputCollector_1 = require("./SelectOutputCollector");
+const TopLevelAndConditionDeduper_1 = require("./TopLevelAndConditionDeduper");
+const PredicateExpressionUtils_1 = require("./PredicateExpressionUtils");
 const VOLATILE_OR_UNSUPPORTED_FUNCTION_REASON = "Predicate contains a function call; volatile and expression predicates are not moved in the safe-only implementation.";
-const normalizeIdentifier = (value) => value.trim().toLowerCase();
-const unwrapParens = (expression) => {
-    let candidate = expression;
-    while (candidate instanceof ValueComponent_1.ParenExpression) {
-        candidate = candidate.expression;
-    }
-    return candidate;
-};
-const isBinaryOperator = (expression, operator) => {
-    const candidate = unwrapParens(expression);
-    return candidate instanceof ValueComponent_1.BinaryExpression
-        && candidate.operator.value.trim().toLowerCase() === operator;
-};
-const collectTopLevelAndTerms = (expression) => {
-    const candidate = unwrapParens(expression);
-    if (!isBinaryOperator(candidate, "and")) {
-        return [expression];
-    }
-    return [
-        ...collectTopLevelAndTerms(candidate.left),
-        ...collectTopLevelAndTerms(candidate.right)
-    ];
-};
-const rebuildWhereWithoutTerms = (query, termsToRemove) => {
-    if (!query.whereClause || termsToRemove.size === 0) {
-        return;
-    }
-    const remaining = collectTopLevelAndTerms(query.whereClause.condition)
-        .filter(term => !termsToRemove.has(term));
-    if (remaining.length === 0) {
-        query.whereClause = null;
-        return;
-    }
-    let rebuilt = remaining[0];
-    for (let index = 1; index < remaining.length; index += 1) {
-        rebuilt = new ValueComponent_1.BinaryExpression(rebuilt, "and", remaining[index]);
-    }
-    query.whereClause = new Clause_1.WhereClause(rebuilt);
-};
-const cloneValueComponent = (expression, options) => {
-    return ValueParser_1.ValueParser.parse((0, SqlComponentFormatter_1.formatSqlComponent)(expression, options));
-};
-const cloneColumnReference = (reference) => {
-    var _a, _b;
-    const namespaces = (_b = (_a = reference.namespaces) === null || _a === void 0 ? void 0 : _a.map(namespace => namespace.name)) !== null && _b !== void 0 ? _b : null;
-    return new ValueComponent_1.ColumnReference(namespaces, reference.column.name);
-};
-const columnReferenceText = (reference) => {
-    const namespace = reference.getNamespace();
-    return namespace ? `${namespace}.${reference.column.name}` : reference.column.name;
-};
-const sameColumnReference = (left, right) => {
-    return normalizeIdentifier(left.column.name) === normalizeIdentifier(right.column.name)
-        && normalizeIdentifier(left.getNamespace()) === normalizeIdentifier(right.getNamespace());
-};
-const appendUnique = (items, value) => {
-    if (!items.includes(value)) {
-        items.push(value);
-    }
-};
 class StaticPredicatePlacementOptimizer {
     plan(input, options = {}) {
         var _a, _b, _c;
@@ -106,71 +47,8 @@ class StaticPredicatePlacementOptimizer {
         const query = parsed.query;
         const applied = [];
         const skipped = [];
-        const movedTerms = [];
-        for (const term of query.whereClause ? collectTopLevelAndTerms(query.whereClause.condition) : []) {
-            const candidate = this.analyzeCandidate(query, term, options);
-            if (!candidate) {
-                continue;
-            }
-            if ("code" in candidate) {
-                skipped.push(this.makeSkipped(term, candidate, options));
-                continue;
-            }
-            const target = this.resolveTarget(query, candidate);
-            if ("code" in target) {
-                skipped.push(this.makeSkipped(term, target, options));
-                continue;
-            }
-            let appliedReason = "";
-            if (target.kind === "simple") {
-                const targetColumns = this.resolveTargetColumns(query, target.query, candidate.references);
-                if ("code" in targetColumns) {
-                    skipped.push(this.makeSkipped(term, targetColumns, options));
-                    continue;
-                }
-                const placement = this.resolveTargetPlacement(target.query, targetColumns);
-                if ("code" in placement) {
-                    skipped.push(this.makeSkipped(term, placement, options));
-                    continue;
-                }
-                const movedPredicate = this.rebasePredicate(candidate.expression, targetColumns, options);
-                target.query.appendWhere(movedPredicate);
-                appliedReason = placement.reason;
-            }
-            else {
-                const placements = [];
-                let skip = null;
-                for (const branch of target.branches) {
-                    const placement = this.resolveTargetPlacement(branch.query, branch.targetColumns);
-                    if ("code" in placement) {
-                        skip = placement;
-                        break;
-                    }
-                    placements.push(placement);
-                }
-                if (skip) {
-                    skipped.push(this.makeSkipped(term, skip, options));
-                    continue;
-                }
-                for (const branch of target.branches) {
-                    const movedPredicate = this.rebasePredicate(candidate.expression, branch.targetColumns, options);
-                    branch.query.appendWhere(movedPredicate);
-                }
-                appliedReason = placements.some(item => /group by/i.test(item.reason))
-                    ? "Predicate is distributed to every UNION branch by output column position; grouped branches only receive GROUP BY-key predicates."
-                    : "Predicate is distributed to every UNION branch by output column position before unsafe query boundaries.";
-            }
-            movedTerms.push(term);
-            applied.push({
-                kind: "move_static_predicate",
-                predicateSql: candidate.predicateSql,
-                fromScopeId: "scope:root",
-                toScopeId: target.scopeId,
-                reason: appliedReason,
-                columnReferences: candidate.references.map(columnReferenceText)
-            });
-        }
-        rebuildWhereWithoutTerms(query, new Set(movedTerms));
+        this.placePredicatesInScopes(query, options, applied, skipped);
+        // API output shape review: keep result.sql/result.query compatibility while expanding scope-aware placement.
         const sql = applied.length > 0 || (0, SqlComponentFormatter_1.hasSqlComponentFormatOverride)(options)
             ? (0, SqlComponentFormatter_1.formatSqlComponent)(query, options)
             : parsed.sql;
@@ -185,6 +63,121 @@ class StaticPredicatePlacementOptimizer {
             formatterGeneratedSource: parsed.formatterGeneratedSource
         });
     }
+    placePredicatesInScopes(contextRoot, options, applied, skipped) {
+        var _a;
+        const queue = [{ query: contextRoot, scopeId: "scope:root" }];
+        let processed = 0;
+        while (queue.length > 0) {
+            processed += 1;
+            if (processed > 1000) {
+                skipped.push({
+                    predicateSql: "",
+                    scopeId: "scope:root",
+                    code: "RECURSIVE_PLACEMENT_LIMIT",
+                    reason: "Static predicate recursive placement stopped after reaching the safety iteration limit."
+                });
+                return;
+            }
+            const work = queue.shift();
+            const movedTerms = [];
+            const reportSkipped = work.pendingTerms === undefined;
+            const terms = (_a = work.pendingTerms) !== null && _a !== void 0 ? _a : (work.query.whereClause
+                ? (0, PredicateExpressionUtils_1.collectTopLevelAndTerms)(work.query.whereClause.condition)
+                : []);
+            for (const term of terms) {
+                const candidate = this.analyzeCandidate(work.query, term, options);
+                if (!candidate) {
+                    continue;
+                }
+                if ("code" in candidate) {
+                    if (reportSkipped) {
+                        skipped.push(this.makeSkipped(term, work.scopeId, candidate, options));
+                    }
+                    continue;
+                }
+                const target = this.resolveTarget(contextRoot, work.query, candidate);
+                if ("code" in target) {
+                    if (reportSkipped) {
+                        skipped.push(this.makeSkipped(term, work.scopeId, target, options));
+                    }
+                    continue;
+                }
+                let appliedReason = "";
+                if (target.kind === "join_on") {
+                    this.appendJoinOnPredicate(target.join, candidate.expression, options);
+                    appliedReason = target.reason;
+                }
+                else if (target.kind === "simple") {
+                    const targetColumns = this.resolveTargetColumns(contextRoot, target.query, candidate.references);
+                    if ("code" in targetColumns) {
+                        if (reportSkipped) {
+                            skipped.push(this.makeSkipped(term, work.scopeId, targetColumns, options));
+                        }
+                        continue;
+                    }
+                    const placement = this.resolveTargetPlacement(contextRoot, target.query, targetColumns);
+                    if ("code" in placement) {
+                        if (reportSkipped) {
+                            skipped.push(this.makeSkipped(term, work.scopeId, placement, options));
+                        }
+                        continue;
+                    }
+                    const movedPredicate = this.rebasePredicate(candidate.expression, targetColumns, options);
+                    target.query.appendWhere(movedPredicate);
+                    (0, TopLevelAndConditionDeduper_1.dedupeWhereTopLevelAndConditions)(target.query, options);
+                    appliedReason = work.scopeId === "scope:root"
+                        ? placement.reason
+                        : `${placement.reason} Predicate was safely rebased from a previously moved predicate.`;
+                    const retainedTerms = target.query.whereClause
+                        ? (0, PredicateExpressionUtils_1.collectTopLevelAndTerms)(target.query.whereClause.condition)
+                        : [];
+                    if (retainedTerms.includes(movedPredicate)) {
+                        queue.push({
+                            query: target.query,
+                            scopeId: target.scopeId,
+                            pendingTerms: [movedPredicate]
+                        });
+                    }
+                }
+                else {
+                    const placements = [];
+                    let skip = null;
+                    for (const branch of target.branches) {
+                        const placement = this.resolveTargetPlacement(contextRoot, branch.query, branch.targetColumns);
+                        if ("code" in placement) {
+                            skip = placement;
+                            break;
+                        }
+                        placements.push(placement);
+                    }
+                    if (skip) {
+                        if (reportSkipped) {
+                            skipped.push(this.makeSkipped(term, work.scopeId, skip, options));
+                        }
+                        continue;
+                    }
+                    for (const branch of target.branches) {
+                        const movedPredicate = this.rebasePredicate(candidate.expression, branch.targetColumns, options);
+                        branch.query.appendWhere(movedPredicate);
+                        (0, TopLevelAndConditionDeduper_1.dedupeWhereTopLevelAndConditions)(branch.query, options);
+                    }
+                    appliedReason = placements.some(item => /group by/i.test(item.reason))
+                        ? "Predicate is distributed to every UNION branch by output column position; grouped branches only receive GROUP BY-key predicates."
+                        : "Predicate is distributed to every UNION branch by output column position before unsafe query boundaries.";
+                }
+                movedTerms.push(term);
+                applied.push({
+                    kind: "move_static_predicate",
+                    predicateSql: candidate.predicateSql,
+                    fromScopeId: work.scopeId,
+                    toScopeId: target.scopeId,
+                    reason: appliedReason,
+                    columnReferences: candidate.references.map(PredicateExpressionUtils_1.columnReferenceText)
+                });
+            }
+            (0, PredicateExpressionUtils_1.rebuildWhereWithoutTerms)(work.query, new Set(movedTerms));
+        }
+    }
     optimize(input, options = {}) {
         var _a;
         return this.plan(input, { ...options, dryRun: (_a = options.dryRun) !== null && _a !== void 0 ? _a : false });
@@ -257,12 +250,12 @@ class StaticPredicatePlacementOptimizer {
         };
     }
     isExistsPredicate(expression) {
-        const candidate = unwrapParens(expression);
+        const candidate = (0, PredicateExpressionUtils_1.unwrapParens)(expression);
         if (!(candidate instanceof ValueComponent_1.UnaryExpression)) {
             return false;
         }
         const operator = candidate.operator.value.trim().toLowerCase();
-        return operator === "exists" && unwrapParens(candidate.expression) instanceof ValueComponent_1.InlineQuery;
+        return operator === "exists" && (0, PredicateExpressionUtils_1.unwrapParens)(candidate.expression) instanceof ValueComponent_1.InlineQuery;
     }
     findUnsupportedExpression(expression, allowExistsSubquery) {
         let found = null;
@@ -311,7 +304,7 @@ class StaticPredicatePlacementOptimizer {
             if (found) {
                 return;
             }
-            const candidate = unwrapParens(value);
+            const candidate = (0, PredicateExpressionUtils_1.unwrapParens)(value);
             if (candidate instanceof ValueComponent_1.BinaryExpression) {
                 visit(candidate.left);
                 visit(candidate.right);
@@ -380,12 +373,13 @@ class StaticPredicatePlacementOptimizer {
         visit(expression);
         return found;
     }
-    resolveTarget(root, candidate) {
-        const boundary = this.findRootQueryBoundary(root);
+    resolveTarget(contextRoot, currentQuery, candidate) {
+        var _a;
+        const boundary = this.findRootQueryBoundary(currentQuery);
         if (boundary) {
             return boundary;
         }
-        if (!root.fromClause) {
+        if (!currentQuery.fromClause) {
             return {
                 code: "NO_FROM_CLAUSE",
                 reason: "Predicate has no FROM source that can receive it safely."
@@ -393,7 +387,7 @@ class StaticPredicatePlacementOptimizer {
         }
         const bindings = [];
         for (const reference of candidate.references) {
-            const binding = this.resolveSourceBinding(root, root, reference);
+            const binding = this.resolveSourceBinding(contextRoot, currentQuery, reference);
             if ("code" in binding) {
                 return binding;
             }
@@ -408,12 +402,25 @@ class StaticPredicatePlacementOptimizer {
             };
         }
         const binding = bindings[0];
-        const nullableSide = this.findNullableSideBoundary(root.fromClause, binding);
+        if ((_a = binding.join) === null || _a === void 0 ? void 0 : _a.lateral) {
+            return {
+                code: "LATERAL_JOIN_BOUNDARY",
+                reason: "Predicate crosses a LATERAL JOIN boundary; moving it may change semantics."
+            };
+        }
+        const nullableSide = this.findNullableSideBoundary(currentQuery.fromClause, binding);
         if (nullableSide) {
             return nullableSide;
         }
-        const upstream = this.resolveUpstreamQuery(root, binding, candidate.references);
+        const upstream = this.resolveUpstreamQuery(contextRoot, binding, candidate.references);
         if ("code" in upstream) {
+            if (upstream.code === "NO_SAFE_UPSTREAM_QUERY") {
+                const joinOnTarget = this.resolveBaseTableJoinOnTarget(contextRoot, currentQuery, binding);
+                if (!("code" in joinOnTarget)) {
+                    return joinOnTarget;
+                }
+                return joinOnTarget;
+            }
             return upstream;
         }
         return upstream;
@@ -433,7 +440,7 @@ class StaticPredicatePlacementOptimizer {
         }
         return null;
     }
-    resolveTargetPlacement(query, targetColumns) {
+    resolveTargetPlacement(contextRoot, query, targetColumns) {
         const hasOrdinaryDistinct = this.hasOrdinaryDistinct(query);
         if (this.hasDistinctOnBoundary(query)) {
             return {
@@ -453,11 +460,11 @@ class StaticPredicatePlacementOptimizer {
                 reason: "Predicate crosses LIMIT/OFFSET/FETCH boundary; moving it may change row selection semantics."
             };
         }
-        if (query.fromClause && this.hasOuterJoin(query.fromClause)) {
-            return {
-                code: "OUTER_JOIN_BOUNDARY",
-                reason: "Target query contains an OUTER JOIN boundary that is not moved across in the safe-only implementation."
-            };
+        if (query.fromClause) {
+            const nullableSide = this.findNullableSideBoundaryForTargetColumns(contextRoot, query, targetColumns);
+            if (nullableSide) {
+                return nullableSide;
+            }
         }
         if (query.groupByClause) {
             const allReferencesAreGroupKeys = targetColumns.every(item => this.isGroupKeyColumn(query, item.targetColumn));
@@ -486,6 +493,22 @@ class StaticPredicatePlacementOptimizer {
             reason: "All outer references resolve to direct upstream outputs before unsafe query boundaries."
         };
     }
+    findNullableSideBoundaryForTargetColumns(contextRoot, query, targetColumns) {
+        if (!query.fromClause) {
+            return null;
+        }
+        for (const item of targetColumns) {
+            const binding = this.resolveSourceBinding(contextRoot, query, item.targetColumn);
+            if ("code" in binding) {
+                return binding;
+            }
+            const nullableSide = this.findNullableSideBoundary(query.fromClause, binding);
+            if (nullableSide) {
+                return nullableSide;
+            }
+        }
+        return null;
+    }
     resolveSourceBinding(contextRoot, query, column) {
         const fromClause = query.fromClause;
         if (!fromClause) {
@@ -495,10 +518,10 @@ class StaticPredicatePlacementOptimizer {
             };
         }
         const bindings = this.getSourceBindings(fromClause);
-        const namespace = normalizeIdentifier(column.getNamespace());
+        const namespace = column.getNamespace();
         const columnName = column.column.name;
         if (namespace) {
-            const matches = bindings.filter(binding => normalizeIdentifier(binding.alias) === namespace);
+            const matches = bindings.filter(binding => (0, PredicateExpressionUtils_1.identifiersEqual)(binding.alias, namespace));
             if (matches.length !== 1) {
                 return {
                     code: "AMBIGUOUS_COLUMN_SOURCE",
@@ -506,16 +529,19 @@ class StaticPredicatePlacementOptimizer {
                 };
             }
             const matchCount = this.getOutputColumnMatchCount(contextRoot, matches[0], columnName);
+            if (matchCount === 0 && this.isBaseTableBinding(contextRoot, matches[0])) {
+                return matches[0];
+            }
             if (matchCount === 0) {
                 return {
                     code: "COLUMN_NOT_AVAILABLE_UPSTREAM",
-                    reason: `Column '${columnReferenceText(column)}' is not a direct output of the referenced source.`
+                    reason: `Column '${(0, PredicateExpressionUtils_1.columnReferenceText)(column)}' is not a direct output of the referenced source.`
                 };
             }
             if (matchCount > 1) {
                 return {
                     code: "AMBIGUOUS_COLUMN_REFERENCE",
-                    reason: `Column '${columnReferenceText(column)}' resolves to multiple outputs in the referenced source.`
+                    reason: `Column '${(0, PredicateExpressionUtils_1.columnReferenceText)(column)}' resolves to multiple outputs in the referenced source.`
                 };
             }
             return matches[0];
@@ -634,7 +660,7 @@ class StaticPredicatePlacementOptimizer {
     resolveTargetColumns(root, query, references) {
         const resolved = [];
         for (const reference of references) {
-            const matches = this.collectSelectOutputs(root, query).filter(item => normalizeIdentifier(item.name) === normalizeIdentifier(reference.column.name));
+            const matches = this.collectDirectOutputMatches(root, query, reference.column.name);
             if (matches.length !== 1) {
                 return {
                     code: "AMBIGUOUS_TARGET_COLUMN",
@@ -661,6 +687,57 @@ class StaticPredicatePlacementOptimizer {
         }
         return resolved;
     }
+    resolveBaseTableJoinOnTarget(contextRoot, currentQuery, binding) {
+        var _a, _b, _c;
+        if (!currentQuery.fromClause || !this.isBaseTableBinding(contextRoot, binding)) {
+            return {
+                code: "NO_SAFE_UPSTREAM_QUERY",
+                reason: "The referenced source is a base table, so there is no upstream query block to move into."
+            };
+        }
+        if (this.hasOuterJoin(currentQuery.fromClause)) {
+            return {
+                code: "OUTER_JOIN_BOUNDARY",
+                reason: "Predicate crosses an OUTER JOIN boundary; moving it into JOIN ON may change semantics."
+            };
+        }
+        const join = binding.isPrimary
+            ? (_b = (_a = currentQuery.fromClause.joins) === null || _a === void 0 ? void 0 : _a[0]) !== null && _b !== void 0 ? _b : null
+            : binding.join;
+        if (join === null || join === void 0 ? void 0 : join.lateral) {
+            return {
+                code: "LATERAL_JOIN_BOUNDARY",
+                reason: "Predicate crosses a LATERAL JOIN boundary; moving it into JOIN ON may change semantics."
+            };
+        }
+        if (!join || !this.isInnerJoin(join)) {
+            return {
+                code: "NO_SAFE_JOIN_ON_TARGET",
+                reason: "Base-table predicates are moved only into INNER JOIN ON clauses in the safe-only implementation."
+            };
+        }
+        if (!(join.condition instanceof Clause_1.JoinOnClause)) {
+            return {
+                code: "NO_SAFE_JOIN_ON_TARGET",
+                reason: "Base-table predicates are moved only into existing JOIN ON clauses, not USING or conditionless joins."
+            };
+        }
+        return {
+            kind: "join_on",
+            join,
+            scopeId: `join_on:${(_c = join.source.getAliasName()) !== null && _c !== void 0 ? _c : "unknown"}`,
+            reason: binding.isPrimary
+                ? "Predicate references the primary source of an INNER JOIN; it is moved into the first JOIN ON clause."
+                : "Predicate references the joined source of an INNER JOIN; it is moved into that JOIN ON clause."
+        };
+    }
+    appendJoinOnPredicate(join, expression, options) {
+        if (!(join.condition instanceof Clause_1.JoinOnClause)) {
+            return;
+        }
+        join.condition.condition = new ValueComponent_1.BinaryExpression(join.condition.condition, "and", (0, PredicateExpressionUtils_1.cloneValueComponent)(expression, options));
+        join.condition.condition = (0, TopLevelAndConditionDeduper_1.dedupeTopLevelAndConditions)(join.condition.condition, options);
+    }
     resolveDeepestBranchTarget(contextRoot, query, targetColumns, visited = new Set()) {
         if (visited.has(query) || targetColumns.length === 0) {
             return { query, targetColumns: [...targetColumns] };
@@ -688,14 +765,17 @@ class StaticPredicatePlacementOptimizer {
             return { query, targetColumns: [...targetColumns] };
         }
         const upstream = this.resolveUpstreamQuery(contextRoot, binding, targetColumns.map(item => item.targetColumn));
-        if ("code" in upstream || upstream.kind === "union") {
+        if ("code" in upstream) {
+            return { query, targetColumns: [...targetColumns] };
+        }
+        if (upstream.kind !== "simple") {
             return { query, targetColumns: [...targetColumns] };
         }
         const upstreamColumns = this.resolveTargetColumns(contextRoot, upstream.query, targetColumns.map(item => item.targetColumn));
         if ("code" in upstreamColumns) {
             return { query, targetColumns: [...targetColumns] };
         }
-        const placement = this.resolveTargetPlacement(upstream.query, upstreamColumns);
+        const placement = this.resolveTargetPlacement(contextRoot, upstream.query, upstreamColumns);
         if ("code" in placement) {
             return { query, targetColumns: [...targetColumns] };
         }
@@ -713,14 +793,14 @@ class StaticPredicatePlacementOptimizer {
             };
         }
         const bindings = this.getSourceBindings(query.fromClause);
-        const namespace = normalizeIdentifier(column.getNamespace());
+        const namespace = column.getNamespace();
         if (namespace) {
-            const matches = bindings.filter(binding => normalizeIdentifier(binding.alias) === namespace);
+            const matches = bindings.filter(binding => (0, PredicateExpressionUtils_1.identifiersEqual)(binding.alias, namespace));
             return matches.length === 1
                 ? null
                 : {
                     code: "AMBIGUOUS_TARGET_COLUMN",
-                    reason: `Target column '${columnReferenceText(column)}' is not uniquely resolvable in the destination query.`
+                    reason: `Target column '${(0, PredicateExpressionUtils_1.columnReferenceText)(column)}' is not uniquely resolvable in the destination query.`
                 };
         }
         if (bindings.length === 1) {
@@ -737,16 +817,16 @@ class StaticPredicatePlacementOptimizer {
             return false;
         }
         return groupBy.grouping.some(grouping => {
-            const candidate = unwrapParens(grouping);
+            const candidate = (0, PredicateExpressionUtils_1.unwrapParens)(grouping);
             return candidate instanceof ValueComponent_1.ColumnReference
                 && this.sameResolvableColumnInQuery(query, candidate, column);
         });
     }
     sameResolvableColumnInQuery(query, left, right) {
-        if (sameColumnReference(left, right)) {
+        if ((0, PredicateExpressionUtils_1.sameColumnReference)(left, right)) {
             return true;
         }
-        if (normalizeIdentifier(left.column.name) !== normalizeIdentifier(right.column.name)) {
+        if (!(0, PredicateExpressionUtils_1.identifiersEqual)(left.column.name, right.column.name)) {
             return false;
         }
         const leftSource = this.resolveColumnSourceAlias(query, left);
@@ -758,15 +838,15 @@ class StaticPredicatePlacementOptimizer {
             return null;
         }
         const bindings = this.getSourceBindings(query.fromClause);
-        const namespace = normalizeIdentifier(column.getNamespace());
+        const namespace = column.getNamespace();
         if (namespace) {
-            const matches = bindings.filter(binding => normalizeIdentifier(binding.alias) === namespace);
-            return matches.length === 1 ? normalizeIdentifier(matches[0].alias) : null;
+            const matches = bindings.filter(binding => (0, PredicateExpressionUtils_1.identifiersEqual)(binding.alias, namespace));
+            return matches.length === 1 ? matches[0].alias : null;
         }
-        return bindings.length === 1 ? normalizeIdentifier(bindings[0].alias) : null;
+        return bindings.length === 1 ? bindings[0].alias : null;
     }
     rebasePredicate(expression, targetColumns, options) {
-        const cloned = cloneValueComponent(expression, options);
+        const cloned = (0, PredicateExpressionUtils_1.cloneValueComponent)(expression, options);
         const visitSelect = (query, inheritedLocalAliases) => {
             var _a, _b, _c, _d;
             if (!(query instanceof SelectQuery_1.SimpleSelectQuery)) {
@@ -799,15 +879,15 @@ class StaticPredicatePlacementOptimizer {
             }
         };
         const visit = (value, localAliases) => {
-            const candidate = unwrapParens(value);
+            const candidate = (0, PredicateExpressionUtils_1.unwrapParens)(value);
             if (candidate instanceof ValueComponent_1.ColumnReference) {
-                const namespace = normalizeIdentifier(candidate.getNamespace());
+                const namespace = (0, PredicateExpressionUtils_1.normalizeIdentifier)(candidate.getNamespace());
                 if (namespace && localAliases.has(namespace)) {
                     return;
                 }
-                const target = targetColumns.find(item => sameColumnReference(candidate, item.sourceColumn));
+                const target = targetColumns.find(item => (0, PredicateExpressionUtils_1.sameColumnReference)(candidate, item.sourceColumn));
                 if (target) {
-                    candidate.qualifiedName = cloneColumnReference(target.targetColumn).qualifiedName;
+                    candidate.qualifiedName = (0, PredicateExpressionUtils_1.cloneColumnReference)(target.targetColumn).qualifiedName;
                 }
                 return;
             }
@@ -900,6 +980,14 @@ class StaticPredicatePlacementOptimizer {
         }
         return bindings;
     }
+    isBaseTableBinding(root, binding) {
+        const source = binding.source.datasource;
+        return source instanceof Clause_1.TableSource && !this.findCte(root, source.table.name);
+    }
+    isInnerJoin(join) {
+        const joinType = join.joinType.value.trim().toLowerCase();
+        return joinType === "join" || joinType === "inner join";
+    }
     findNullableSideBoundary(fromClause, binding) {
         var _a;
         const joins = (_a = fromClause.joins) !== null && _a !== void 0 ? _a : [];
@@ -967,12 +1055,12 @@ class StaticPredicatePlacementOptimizer {
             if ("code" in branches) {
                 return 0;
             }
-            return this.collectSelectOutputs(root, branches[0]).filter(item => normalizeIdentifier(item.name) === normalizeIdentifier(columnName)).length;
+            return this.collectDirectOutputMatches(root, branches[0], columnName).length;
         }
         if (!(target instanceof SelectQuery_1.SimpleSelectQuery)) {
             return 0;
         }
-        return this.collectSelectOutputs(root, target).filter(item => normalizeIdentifier(item.name) === normalizeIdentifier(columnName)).length;
+        return this.collectDirectOutputMatches(root, target, columnName).length;
     }
     collectUnionBranches(query) {
         const branches = [];
@@ -1003,7 +1091,7 @@ class StaticPredicatePlacementOptimizer {
     resolveUnionOutputIndex(root, firstBranch, outputColumnName) {
         const matches = [];
         this.collectSelectOutputs(root, firstBranch).forEach((item, index) => {
-            if (normalizeIdentifier(item.name) === normalizeIdentifier(outputColumnName)) {
+            if ((0, PredicateExpressionUtils_1.identifiersEqual)(item.name, outputColumnName)) {
                 matches.push(index);
             }
         });
@@ -1025,6 +1113,15 @@ class StaticPredicatePlacementOptimizer {
                 reason: `UNION branch does not expose column '${sourceColumn.column.name}' at output position ${outputIndex + 1}.`
             };
         }
+        else if ((0, PredicateExpressionUtils_1.identifiersEqual)(output.name, sourceColumn.column.name)) {
+            const matches = this.collectDirectOutputMatches(root, query, sourceColumn.column.name);
+            if (matches.length > 1) {
+                return {
+                    code: "AMBIGUOUS_TARGET_COLUMN",
+                    reason: `UNION branch exposes multiple '${sourceColumn.column.name}' columns.`
+                };
+            }
+        }
         if (!(output.value instanceof ValueComponent_1.ColumnReference)) {
             return {
                 code: "EXPRESSION_OUTPUT_UNSUPPORTED",
@@ -1049,6 +1146,96 @@ class StaticPredicatePlacementOptimizer {
         const collector = new SelectOutputCollector_1.SelectOutputCollector(null, commonTables.length > 0 ? commonTables : null);
         return collector.collect(query);
     }
+    collectDirectOutputMatches(root, query, columnName) {
+        const collectedMatches = this.collectSelectOutputs(root, query).filter(item => (0, PredicateExpressionUtils_1.identifiersEqual)(item.name, columnName));
+        if (collectedMatches.length > 0) {
+            return this.hasExplicitOutputMatch(query, columnName)
+                ? [...collectedMatches, ...this.inferWildcardOutputMatches(root, query, columnName, true)]
+                : collectedMatches;
+        }
+        return this.inferWildcardOutputMatches(root, query, columnName, false);
+    }
+    hasExplicitOutputMatch(query, columnName) {
+        return query.selectClause.items.some(item => {
+            if (item.identifier) {
+                return (0, PredicateExpressionUtils_1.identifiersEqual)(item.identifier.name, columnName);
+            }
+            return item.value instanceof ValueComponent_1.ColumnReference
+                && item.value.column.name !== "*"
+                && (0, PredicateExpressionUtils_1.identifiersEqual)(item.value.column.name, columnName);
+        });
+    }
+    inferWildcardOutputMatches(root, query, columnName, includeUnprovenPotential) {
+        const matches = [];
+        query.selectClause.items.forEach((item, outputIndex) => {
+            if (item.identifier || !(item.value instanceof ValueComponent_1.ColumnReference) || item.value.column.name !== "*") {
+                return;
+            }
+            const targetColumn = this.inferWildcardTargetColumn(root, query, item.value, columnName, includeUnprovenPotential);
+            if (!targetColumn) {
+                return;
+            }
+            if (targetColumn === "ambiguous") {
+                matches.push(this.createInferredOutputColumn(columnName, new ValueComponent_1.ColumnReference(null, columnName), outputIndex), this.createInferredOutputColumn(columnName, new ValueComponent_1.ColumnReference(null, columnName), outputIndex));
+                return;
+            }
+            matches.push(this.createInferredOutputColumn(columnName, targetColumn, outputIndex));
+        });
+        return matches;
+    }
+    inferWildcardTargetColumn(root, query, wildcard, columnName, includeUnprovenPotential) {
+        if (!query.fromClause) {
+            return null;
+        }
+        const bindings = this.getSourceBindings(query.fromClause);
+        if (wildcard.namespaces === null) {
+            if (bindings.length !== 1) {
+                return "ambiguous";
+            }
+            return this.resolveWildcardColumnFromBinding(root, query, bindings[0], columnName, includeUnprovenPotential);
+        }
+        const namespace = wildcard.getNamespace();
+        const matches = bindings.filter(binding => (0, PredicateExpressionUtils_1.identifiersEqual)(binding.alias, namespace));
+        if (matches.length === 0) {
+            return null;
+        }
+        return matches.length === 1
+            ? this.resolveWildcardColumnFromBinding(root, query, matches[0], columnName, includeUnprovenPotential)
+            : "ambiguous";
+    }
+    resolveWildcardColumnFromBinding(root, query, binding, columnName, includeUnprovenPotential) {
+        const matches = this.collectSourceOutputMatches(root, query, binding.source, columnName);
+        if (matches.length > 1) {
+            return "ambiguous";
+        }
+        if (matches.length === 1) {
+            const value = matches[0].value;
+            return value instanceof ValueComponent_1.ColumnReference ? value : null;
+        }
+        return includeUnprovenPotential ? this.createColumnForBinding(binding, columnName) : null;
+    }
+    collectSourceOutputMatches(root, query, source, columnName) {
+        var _a, _b, _c, _d;
+        const commonTables = [
+            ...((_b = (_a = query.withClause) === null || _a === void 0 ? void 0 : _a.tables) !== null && _b !== void 0 ? _b : []),
+            ...((_d = (_c = root.withClause) === null || _c === void 0 ? void 0 : _c.tables) !== null && _d !== void 0 ? _d : [])
+        ];
+        const collector = new SelectOutputCollector_1.SelectOutputCollector(null, commonTables.length > 0 ? commonTables : null);
+        return collector.collect(source).filter(item => (0, PredicateExpressionUtils_1.identifiersEqual)(item.name, columnName));
+    }
+    createColumnForBinding(binding, columnName) {
+        return new ValueComponent_1.ColumnReference(binding.alias ? [binding.alias] : null, columnName);
+    }
+    createInferredOutputColumn(name, value, outputIndex) {
+        return {
+            name,
+            value,
+            outputIndex,
+            sourceAlias: value.getNamespace() || null,
+            sourceName: null,
+            sourceColumnName: name
+        };
+    }
     resolveSourceQueryForColumns(root, source) {
         var _a;
         if (source.datasource instanceof Clause_1.SubQuerySource) {
@@ -1064,13 +1251,13 @@ class StaticPredicatePlacementOptimizer {
     }
     findCte(root, name) {
         var _a, _b;
-        const normalized = normalizeIdentifier(name);
+        const normalized = (0, PredicateExpressionUtils_1.normalizeIdentifier)(name);
         const matches = ((_b = (_a = root.withClause) === null || _a === void 0 ? void 0 : _a.tables) !== null && _b !== void 0 ? _b : [])
-            .filter(table => normalizeIdentifier(table.getSourceAliasName()) === normalized);
+            .filter(table => (0, PredicateExpressionUtils_1.normalizeIdentifier)(table.getSourceAliasName()) === normalized);
         return matches.length === 1 ? matches[0] : null;
     }
     countTableSourceReferences(query, tableName) {
-        const normalized = normalizeIdentifier(tableName);
+        const normalized = (0, PredicateExpressionUtils_1.normalizeIdentifier)(tableName);
         let count = 0;
         const visitSelect = (select) => {
             var _a, _b;
@@ -1085,7 +1272,7 @@ class StaticPredicatePlacementOptimizer {
             if (select.fromClause) {
                 for (const binding of this.getSourceBindings(select.fromClause)) {
                     const source = binding.source.datasource;
-                    if (source instanceof Clause_1.TableSource && normalizeIdentifier(source.table.name) === normalized) {
+                    if (source instanceof Clause_1.TableSource && (0, PredicateExpressionUtils_1.normalizeIdentifier)(source.table.name) === normalized) {
                         count += 1;
                     }
                     if (source instanceof Clause_1.SubQuerySource) {
@@ -1108,7 +1295,7 @@ class StaticPredicatePlacementOptimizer {
         for (const source of (_b = (_a = query.fromClause) === null || _a === void 0 ? void 0 : _a.getSources()) !== null && _b !== void 0 ? _b : []) {
             const alias = source.getAliasName();
             if (alias) {
-                aliases.add(normalizeIdentifier(alias));
+                aliases.add((0, PredicateExpressionUtils_1.normalizeIdentifier)(alias));
             }
         }
         return aliases;
@@ -1148,14 +1335,14 @@ class StaticPredicatePlacementOptimizer {
             }
         };
         const collect = (reference) => {
-            if (!references.some(existing => sameColumnReference(existing, reference))) {
+            if (!references.some(existing => (0, PredicateExpressionUtils_1.sameColumnReference)(existing, reference))) {
                 references.push(reference);
             }
         };
         const visit = (value, localAliases) => {
-            const candidate = unwrapParens(value);
+            const candidate = (0, PredicateExpressionUtils_1.unwrapParens)(value);
             if (candidate instanceof ValueComponent_1.ColumnReference) {
-                const namespace = normalizeIdentifier(candidate.getNamespace());
+                const namespace = (0, PredicateExpressionUtils_1.normalizeIdentifier)(candidate.getNamespace());
                 if (namespace) {
                     if (rootAliases.has(namespace) && !localAliases.has(namespace)) {
                         collect(candidate);
@@ -1248,7 +1435,7 @@ class StaticPredicatePlacementOptimizer {
             if (found) {
                 return;
             }
-            const candidate = unwrapParens(value);
+            const candidate = (0, PredicateExpressionUtils_1.unwrapParens)(value);
             if (candidate instanceof ValueComponent_1.FunctionCall) {
                 if (candidate.over) {
                     found = true;
@@ -1330,9 +1517,9 @@ class StaticPredicatePlacementOptimizer {
             }
         };
         const visit = (value) => {
-            const candidate = unwrapParens(value);
+            const candidate = (0, PredicateExpressionUtils_1.unwrapParens)(value);
             if (candidate instanceof ValueComponent_1.ParameterExpression) {
-                appendUnique(names, candidate.name.value);
+                (0, PredicateExpressionUtils_1.appendUnique)(names, candidate.name.value);
                 return;
             }
             if (candidate instanceof ValueComponent_1.BinaryExpression) {
@@ -1407,10 +1594,10 @@ class StaticPredicatePlacementOptimizer {
         visit(expression);
         return names;
     }
-    makeSkipped(expression, draft, options) {
+    makeSkipped(expression, scopeId, draft, options) {
         return {
             predicateSql: (0, SqlComponentFormatter_1.formatSqlComponent)(expression, options),
-            scopeId: "scope:root",
+            scopeId,
             ...draft
         };
     }