cure_engine/rules/

mod.rs

//! 引擎内置通用规则：只依赖 `Lang` 抽象，任何语言 crate 均可复用。
//!
//! 每条规则文档化其语义前提；所有提案都必须严格降低成本（runner 校验）。



use crate::analysis::{prefix_effects_readable, subtree_contains};
use crate::effect::Effect;
use crate::kind::{BinOp, LitRef, NodeKind, UnOp};
use crate::pattern::{matches, Pat};
use crate::rule::{Edit, RewriteCtx, Rule};
use crate::lang::{Lang, ReassocOutcome, StmtKill};
use crate::walk::Walk;

// ---------------------------------------------------------------------------
// 括号消除：Paren(x) → x（纯分组节点，删除永远安全）
// ---------------------------------------------------------------------------

pub struct ParenRemoval;

impl<L: Lang> Rule<L> for ParenRemoval {
    fn name(&self) -> &'static str {
        "paren_removal"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Paren]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk } = ctx;
        if lang.kind(id) != NodeKind::Paren {
            return None;
        }
        let inner = *lang.children(id).first()?;
        // 赋值目标位的括号可能语义显著（JS 具名求值）
        if lang.assign_target_paren_required(inner) {
            if let Some((p, i)) = walk.parents.get(&id) {
                if lang.kind(*p) == NodeKind::Assign && *i == 0 {
                    return None;
                }
            }
        }
        // 语法必需括号不删（Lang 钩子）：JS 的 Paren(箭头/Raw/自定义
        // 形态) 在运算数位由产生式强制——删了打印机要再加，形成
        // 「删括号→打印补→再解析成 Paren→再删」的 AST 振荡（真实
        // 语料二轮仍 ~900 编辑的根源）。Java 优先级模型完备 → 恒 false。
        if lang.grammar_paren_required(inner) {
            return None;
        }
        Some(Edit::Replace {
            target: id,
            with: inner,
        })
    }
}

// ---------------------------------------------------------------------------
// 常量条件：if (true) {A} [else B] → A；if (false) {A} else B → 删除
// ---------------------------------------------------------------------------

/// 语句自身终结其所在**语句列表**的后续（return/throw/yield 终结方法/块，
/// break/continue 终结所在块——JLS 14.21 语义）。
/// javac 对 `if (true) 终结语句` 后跟语句有「条件编译豁免」——后续按可达
/// 处理；剥掉 if 壳后豁免消失，同列表后续变硬不可达（javac 拒绝）。
fn self_is_terminal<L: Lang>(lang: &L, n: L::Id) -> bool {
    matches!(
        lang.kind(n),
        NodeKind::Return | NodeKind::Throw | NodeKind::Break | NodeKind::Continue | NodeKind::Yield
    )
}

/// 子树是否含 switch 表达式产出（yield）。折叠丢弃含 yield 的分支会
/// 缩小产出集 → switch 表达式静态类型改变（var 推断随之破坏）。
fn contains_yield<L: Lang>(lang: &L, n: L::Id) -> bool {
    subtree_contains(lang, n, |x| lang.kind(x) == NodeKind::Yield)
}

pub struct ConstCondition;

impl<L: Lang> Rule<L> for ConstCondition {
    fn name(&self) -> &'static str {
        "const_condition"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::If]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk } = ctx;
        if lang.kind(id) != NodeKind::If {
            return None;
        }
        let ch = lang.children(id);
        let cond = *ch.first()?;
        match lang.literal(cond) {
            Some(LitRef::Bool(true)) => {
                // then 分支必然执行：拼进父块（避免嵌套块）；非块父级则整体替换。
                // then 必然终结（return/break 等）→ 后续兄弟不可达，须一并
                // 删除——否则产出 `return 1; return 2;` 不可编译
                //（InputArrayTrailingComma 抓获；unreachable_after_terminal
                // 是 --dead-code 门控，默认模式依赖本规则自洽）
                let then = ch.get(1).copied()?;
                // 可提升声明（JS 的 if (c) function f(){}——Annex B 条件
                // 初始化）不可搬出分支（Test262 annexB 族抓获）
                if lang.contains_hoisted_decl(then) {
                    return None;
                }
                // R15 P0-1（o4 形态）抓获：被丢弃的 else 含 yield → switch
                // 表达式产出集缩小 → 静态类型变化（var 推断破坏）→ 拒绝
                if let Some(&els) = ch.get(2) {
                    if contains_yield(&*lang, els) || lang.contains_hoisted_decl(els) {
                        return None;
                    }
                }
                let insert: Vec<L::Id> = if lang.kind(then) == NodeKind::Block {
                    lang.children(then).to_vec()
                } else {
                    vec![then]
                };
                let terminates = insert.iter().any(|&s| self_is_terminal(lang, s));
                if let (Some(parent), Some(index)) = (walk.parent(id), walk.index(id)) {
                    if lang.kind(parent) == NodeKind::Block {
                        let siblings = lang.children(parent).to_vec();
                        let remove = if terminates && index + 1 < siblings.len() {
                            siblings.len() - index
                        } else {
                            1
                        };
                        // R15 P0-1（o3 同类）抓获：删除的尾随兄弟含 yield →
                        // 产出集缩小 → 静态类型变化 → 拒绝
                        if remove > 1
                            && siblings[index + 1..]
                                .iter()
                                .any(|&s| contains_yield(&*lang, s))
                        {
                            return None;
                        }
                        return Some(Edit::Splice {
                            node: parent,
                            index,
                            remove,
                            insert,
                        });
                    }
                    // R15 P0-3（UnreachableVar）抓获：case 组（标签后的
                    // 语句序列）中整体替换时，then 终结且 if 后还有尾随
                    // 语句 → 折叠使尾随不可达（JLS 14.21 条件编译豁免
                    // 消失）。case 组中声明的作用域覆盖整个 switch 块
                    //（删 `int i=1` 会断 `default: i=2` 引用）——无法安全
                    // 删除，唯一安全动作是拒绝折叠。其余非块父级（if/while
                    // 等控制体的唯一语句孩子、else 分支、do-while 条件）
                    // 不存在尾随语句——照常替换。
                    if lang.kind(parent) == NodeKind::Case && terminates {
                        let siblings = lang.children(parent).to_vec();
                        if index + 1 < siblings.len() {
                            return None;
                        }
                    }
                }
                Some(Edit::Replace {
                    target: id,
                    with: then,
                })
            }
            Some(LitRef::Bool(false)) => {
                // else 分支必然执行：保留之；无 else（或空 else）才可整删
                // R15 P0-1（o3 形态）抓获：then 分支整体被丢弃（换 else 或
                // 删除）——含 yield 则 switch 表达式产出集缩小 → 静态类型
                // 变化（var 推断破坏）→ 拒绝
                let then = *ch.get(1)?;
                if contains_yield(&*lang, then) {
                    return None;
                }
                // 选中的 else 含可提升声明 → 搬出分支改变初始化时机
                //（Test262 annexB if-decl-else-decl 家族抓获）
                if let Some(&els) = ch.get(2) {
                    if lang.contains_hoisted_decl(els) {
                        return None;
                    }
                }
                let els = ch.get(2).copied();
                if let (Some(parent), Some(index)) = (walk.parent(id), walk.index(id)) {
                    if lang.kind(parent) == NodeKind::Block {
                        let insert: Vec<L::Id> = match els {
                            Some(e) => {
                                if lang.kind(e) == NodeKind::Block {
                                    lang.children(e).to_vec()
                                } else {
                                    vec![e]
                                }
                            }
                            None => Vec::new(),
                        };
                        return Some(if insert.is_empty() {
                            Edit::Delete { node: id }
                        } else {
                            Edit::Splice {
                                node: parent,
                                index,
                                remove: 1,
                                insert,
                            }
                        });
                    }
                }
                match els {
                    Some(e) => Some(Edit::Replace {
                        target: id,
                        with: e,
                    }),
                    None => Some(Edit::Delete { node: id }),
                }
            }
            _ => None,
        }
    }
}

// ---------------------------------------------------------------------------
// 布尔返回：if (c) return true else return false → return c（反极性 → return !c）
// 前提：c 为原始 boolean。
// ---------------------------------------------------------------------------

// 裸分支：if (c) return true; else return false;
static P_IF_RET_TF: Pat = Pat::Kind(
    NodeKind::If,
    &[Pat::Bind("cond"), Pat::Kind(NodeKind::Return, &[Pat::LitBool(true)]), Pat::Kind(NodeKind::Return, &[Pat::LitBool(false)])],
);
static P_IF_RET_FT: Pat = Pat::Kind(
    NodeKind::If,
    &[Pat::Bind("cond"), Pat::Kind(NodeKind::Return, &[Pat::LitBool(false)]), Pat::Kind(NodeKind::Return, &[Pat::LitBool(true)])],
);
// 块包裹分支（反编译器/规范形态）：if (c) { return true; } else { return false; }
static P_IF_BRET_TF: Pat = Pat::Kind(
    NodeKind::If,
    &[
        Pat::Bind("cond"),
        Pat::Kind(NodeKind::Block, &[Pat::Kind(NodeKind::Return, &[Pat::LitBool(true)])]),
        Pat::Kind(NodeKind::Block, &[Pat::Kind(NodeKind::Return, &[Pat::LitBool(false)])]),
    ],
);
static P_IF_BRET_FT: Pat = Pat::Kind(
    NodeKind::If,
    &[
        Pat::Bind("cond"),
        Pat::Kind(NodeKind::Block, &[Pat::Kind(NodeKind::Return, &[Pat::LitBool(false)])]),
        Pat::Kind(NodeKind::Block, &[Pat::Kind(NodeKind::Return, &[Pat::LitBool(true)])]),
    ],
);

pub struct BooleanReturn;

impl<L: Lang> Rule<L> for BooleanReturn {
    fn name(&self) -> &'static str {
        "boolean_return"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::If]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk: _ } = ctx;
        if lang.kind(id) != NodeKind::If {
            return None;
        }
        // 四种形态：裸/块分支 × 正/反极性。块分支整体被裸 return 替换。
        let negated = |cond: L::Id, lang: &mut L| {
            let nc = lang.build_unary(UnOp::Not, cond);
            lang.build_return(Some(nc))
        };
        for (pat, invert) in [
            (&P_IF_RET_TF, false),
            (&P_IF_BRET_TF, false),
            (&P_IF_RET_FT, true),
            (&P_IF_BRET_FT, true),
        ] {
            if let Some(b) = matches(&*lang, id, pat) {
                let cond = lookup(&b, "cond")?;
                if !lang.is_bool(cond) {
                    return None;
                }
                let ret = if invert {
                    negated(cond, lang)
                } else {
                    lang.build_return(Some(cond))
                };
                return Some(Edit::Replace {
                    target: id,
                    with: ret,
                });
            }
        }
        None
    }
}

fn lookup<Id: Copy>(binds: &[(&'static str, Id)], name: &str) -> Option<Id> {
    binds.iter().find(|(n, _)| *n == name).map(|(_, id)| *id)
}

// ---------------------------------------------------------------------------
// 空分支：if (c) {} else {B} → if (!c) {B}；if (c) {A} else {} → if (c) {A}；
//         if (c) {} → 删除（仅当 c 无副作用）。
// ---------------------------------------------------------------------------

pub struct IfElseEmpty;

impl<L: Lang> Rule<L> for IfElseEmpty {
    fn name(&self) -> &'static str {
        "if_else_empty"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::If]
    }
    fn structural(&self) -> bool {
        // `if (c) {} else {B}` → `if (!c) {B}` 在代价模型上可能恰好
        // 持平（空 Block 1 + cond 1 vs Unary 3）——这是可读性结构
        // 改写，允许零代价应用（否则 debug 断言 panic）。
        true
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk } = ctx;
        if lang.kind(id) != NodeKind::If {
            return None;
        }
        let ch = lang.children(id);
        let cond = *ch.first()?;
        let is_empty_block = |n: L::Id| -> bool {
            lang.kind(n) == NodeKind::Block && lang.children(n).is_empty()
        };
        match ch.len() {
            3 => {
                let (then, els) = (ch[1], ch[2]);
                if is_empty_block(then) {
                    if is_empty_block(els) {
                        // 两支全空：塌缩成 if(c){}，交给 len==2 分支处理
                        return Some(Edit::Splice {
                            node: id,
                            index: 0,
                            remove: 3,
                            insert: vec![cond, then],
                        });
                    }
                    // then 空、else 有内容：取反条件交换分支（条件两侧都恰好求值一次，永远安全）
                    let nc = lang.build_unary(UnOp::Not, cond);
                    return Some(Edit::Splice {
                        node: id,
                        index: 0,
                        remove: 3,
                        insert: vec![nc, els],
                    });
                }
                if is_empty_block(els) {
                    return Some(Edit::Splice {
                        node: id,
                        index: 0,
                        remove: 3,
                        insert: vec![cond, then],
                    });
                }
                None
            }
            2 => {
                let then = ch[1];
                // 【位置守卫】Delete 只在语句位置（父为 Block）合法：
                // if 处于分支位置（if(c){ if(c2){} }——else-if 链化后的
                // 常见形态）时，抽走分支会给父 If 留下"只有条件没有体"的
                // 残骸（1-child If，打印机越界 panic——真实大语料抓获）。
                // 分支位置的空体 if 由父 If 的规则吸收（塌缩后自身再走
                // len==2/len==3 变换）。
                let parent_is_block = walk
                    .parent(id)
                    .map(|p| lang.kind(p) == NodeKind::Block)
                    .unwrap_or(false);
                if is_empty_block(then)
                    && lang.effect(cond) <= Effect::MayRead
                    && parent_is_block
                {
                    // 删除整个 if 会丢掉 cond 的求值，仅当 cond 无副作用时安全
                    return Some(Edit::Delete { node: id });
                }
                None
            }
            _ => None,
        }
    }
}

// ---------------------------------------------------------------------------
// 布尔比较：x == true → x；x == false → !x；x != true → !x；x != false → x（两个操作数序）
// 前提：x 为原始 boolean（装箱比较是引用比较，is_bool 必须排除）。
// ---------------------------------------------------------------------------

pub struct BoolCompare;

impl<L: Lang> Rule<L> for BoolCompare {
    fn name(&self) -> &'static str {
        "bool_compare"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Binary]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk } = ctx;
        if lang.kind(id) != NodeKind::Binary {
            return None;
        }
        let op = lang.bin_op(id)?;
        if !matches!(op, BinOp::Eq | BinOp::Ne) {
            return None;
        }
        let ch = lang.children(id);
        let (l, r) = (*ch.first()?, *ch.get(1)?);
        let lit_bool = |n: L::Id| -> Option<bool> {
            match lang.literal(n) {
                Some(LitRef::Bool(b)) => Some(b),
                _ => None,
            }
        };
        // 统一成 (x, lit)
        let (x, lit) = match (lit_bool(l), lit_bool(r)) {
            (None, Some(b)) => (l, b),
            (Some(b), None) => (r, b),
            _ => return None,
        };
        if !lang.is_bool(x) {
            // 条件位上下文松弛：if/while 的条件表达式必为 boolean
            //（IteratorWildcard 抓获：`x == false` 的 x 是方法调用——
            // 返回类型无类型信息即被拒）。仅当本节点恰是 If/While 的
            // 条件孩子时放行
            let in_cond = walk.parent(id).is_some_and(|p| {
                matches!(lang.kind(p), NodeKind::If | NodeKind::While) && lang.children(p).first() == Some(&id)
            });
            if !in_cond {
                return None;
            }
        }
        // (op, lit) → 替换结果
        let neg = match (op, lit) {
            (BinOp::Eq, true) | (BinOp::Ne, false) => false,
            (BinOp::Eq, false) | (BinOp::Ne, true) => true,
            _ => return None,
        };
        let with = if neg {
            lang.build_unary(UnOp::Not, x)
        } else {
            x
        };
        Some(Edit::Replace {
            target: id,
            with,
        })
    }
}

// ---------------------------------------------------------------------------
// 双重否定：!!x → x
// 安全性：`!x` 要能通过类型检查，x 必为 boolean，无需类型查询。
// ---------------------------------------------------------------------------

pub struct DoubleNot;

impl<L: Lang> Rule<L> for DoubleNot {
    fn name(&self) -> &'static str {
        "double_not"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Unary]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk: _ } = ctx;
        if lang.kind(id) != NodeKind::Unary || lang.un_op(id) != Some(UnOp::Not) {
            return None;
        }
        let inner = *lang.children(id).first()?;
        if lang.kind(inner) != NodeKind::Unary || lang.un_op(inner) != Some(UnOp::Not) {
            return None;
        }
        let x = *lang.children(inner).first()?;
        // 语言裁决（Lang 钩子）：Java 的 !!x 蕴含 x:boolean（类型系统
        // 保证，恒等式无条件成立——!!it.hasNext() 形态依赖）；JS 的
        // !x 是布尔胁迫（!!0 === false ≠ 0）——仅语法级布尔放行
        if !lang.double_not_sound(x) {
            return None;
        }
        Some(Edit::Replace {
            target: id,
            with: x,
        })
    }
}

// ---------------------------------------------------------------------------
// 常量折叠：!true → false / !false → true
// （局部传播把常量内联进 !x 后，由此规则收敛到自然形态）
// ---------------------------------------------------------------------------

pub struct BoolNotFold;

impl<L: Lang> Rule<L> for BoolNotFold {
    fn name(&self) -> &'static str {
        "bool_not_fold"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Unary]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk: _ } = ctx;
        if lang.kind(id) != NodeKind::Unary || lang.un_op(id) != Some(UnOp::Not) {
            return None;
        }
        let operand = *lang.children(id).first()?;
        match lang.literal(operand) {
            Some(LitRef::Bool(b)) => {
                let folded = lang.build_bool(!b);
                Some(Edit::Replace {
                    target: id,
                    with: folded,
                })
            }
            _ => None,
        }
    }
}

// ---------------------------------------------------------------------------
// 自赋值：x = x → 删除（仅限语言层局部变量）
// ---------------------------------------------------------------------------

pub struct SelfAssign;

impl<L: Lang> Rule<L> for SelfAssign {
    fn name(&self) -> &'static str {
        "self_assign"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Assign]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk } = ctx;
        if lang.kind(id) != NodeKind::Assign {
            return None;
        }
        if lang.assign_op(id).is_some() {
            return None; // x += x 有读语义差异，不动
        }
        let ch = lang.children(id);
        let (t, v) = (*ch.first()?, *ch.get(1)?);
        if lang.kind(t) != NodeKind::VarRef || lang.kind(v) != NodeKind::VarRef {
            return None;
        }
        if !lang.same_binding(t, v) || !lang.is_local_var(t) {
            return None;
        }
        // 删除目标必须是**语句**：`x = x;` 的 AST 形态是 ExprStmt[Assign]
        // ——删 Assign 留下空 ExprStmt 壳，后续规则按 children[0] 遍历
        // 即 panic（ddc weixin gp0.d1 复现：clinit 里 R8 寄存器噪声
        // `file = file;` 触发）。嵌套赋值位（`y = (x = x)` 的值孩子）删
        // 中树节点直接留洞——拒绝。Block 直挂 Assign 的语言形态照旧删
        // Assign 本身（Block 孩子删除安全）。
        let (parent, _) = walk.parents.get(&id)?;
        match lang.kind(*parent) {
            NodeKind::ExprStmt => Some(Edit::Delete { node: *parent }),
            NodeKind::Block => Some(Edit::Delete { node: id }),
            _ => None,
        }
    }
}

// ---------------------------------------------------------------------------
// 算术恒等式（仅精确整数，浮点有 -0.0/NaN 陷阱）：
// x+0 → x, 0+x → x, x-0 → x, x*1 → x, 1*x → x, x/1 → x
// ---------------------------------------------------------------------------


// ---------------------------------------------------------------------------
// 死声明删除：`T x;`（零使用、零求值）或 `T x = V;`（V 纯/MayRead 且
// 零使用）。声明之后整个区域对 x 零事件（读/写/遮蔽/不透明全无）。
// 反编译器把寄存器噪声折叠后常残留零用途声明（如 PMD 噪声文件消解
// 后的 `int a;`）。匿名类捕获由 opaque 守卫拒绝（对抗波 4 教训）。
// ---------------------------------------------------------------------------

pub struct DeadDecl;

impl<L: Lang> Rule<L> for DeadDecl {
    fn name(&self) -> &'static str {
        "dead_decl"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::VarDecl]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk } = ctx;
        if lang.kind(id) != NodeKind::VarDecl {
            return None;
        }
        if !lang.dead_decl_allowed(id) {
            return None;
        }
        let ch = lang.children(id).to_vec();
        // 有 init：V 的求值会被一并删除——仅纯/只读可删（MayThrow 不行：
        // `int x = 1/0;` 删掉会失去异常）
        if let Some(&v) = ch.first() {
            if lang.effect(v) > Effect::MayRead {
                return None;
            }
            if lang.value_may_throw_unresolved(v) {
                return None;
            }
            // 自引用 init（JS TDZ：`let x = x + 1` 抛 ReferenceError）——
            // 删除声明会吞掉异常（Test262 let/const use-before-init 抓获）。
            let name_s = lang.var_name(id).unwrap_or("");
            if subtree_contains(&*lang, v, |n| {
                lang.kind(n) == NodeKind::VarRef && lang.var_name(n) == Some(name_s)
            }) {
                return None;
            }
        }
        let parent = walk.parent(id)?;
        if lang.kind(parent) != NodeKind::Block {
            return None;
        }
        let idx = walk.index(id)?;
        let stmts = lang.children(parent).to_vec();
        // **句法扫描**（任何同名 VarRef 引用 → 保留）：事件模型对复合
        // 赋值目标内的读有盲区（`arr[j] += 5` 的 j——scan_region 的
        // Assign 分支只递归 RHS），零事件 ≠ 零引用（differential Adv5
        // 抓获：删掉 j 后 arr[j] 悬空）。写目标本身也是 VarRef，同样被
        // 句法扫描覆盖。
        let name_str = lang.var_name(id)?.to_string();
        let name_key = lang.var_key(id)?;
        for &s in &stmts[idx + 1..] {
            // 原文/类体可能读该名字（不可见）→ 不可判定零用途
            if lang.stmt_has_opaque_region(s) {
                return None;
            }
            match (lang.stmt_mentions(s, name_key), lang.stmt_opaque(s)) {
                (Some(mentions), Some(opaque)) => {
                    // Any VarRef (including event-blind write targets),
                    // any same-name declaration, or an opaque region.
                    if mentions || opaque {
                        return None;
                    }
                }
                _ => {
                    // 遮蔽 / 不透明（匿名类捕获的原文引用无 AST 节点）
                    // 仍走事件扫描
                    if subtree_contains(&*lang, s, |n| {
                        lang.kind(n) == NodeKind::VarRef
                            && lang.var_name(n) == Some(name_str.as_str())
                    }) {
                        return None;
                    }
                    let mut w = Watch::<L>::new(name_key, &[]);
                    scan_region(&*lang, s, &mut w);
                    if w.opaque || w.shadowed {
                        return None;
                    }
                }
            }
        }
        Some(Edit::Delete { node: id })
    }
}

// ---------------------------------------------------------------------------
// 重赋值折回声明（寄存器累加器形态）：
//   T x = e; …（对 x 零事件）…; x = x op K;  →  T x = e op K;
//   （复合形态 `x op= K` 与展开形态 `x = x op K` 均可）
//
// e 的求值位置**不动**（仍在声明处）；被删除的只是后面那条纯 delta
// 语句；声明与折回点之间对 x 零事件 ⇒ 中间值从未被观察 ⇒ 恒安全。
// 语义：x 的终值 (e op K) 由同一算符作用于同一操作数计算——精确
// （整数环绕/浮点舍入均逐位一致）。
// 拒绝：Div/Rem 且 K==0（折回后异常从后面的语句提前到声明处——
// 中间语句的副作用顺序改变）；K 非字面量；op 为短路/比较。
// 之后 local_propagation 可继续把 e op K 内联到唯一使用处。
// ---------------------------------------------------------------------------

pub struct AssignBackFold;

impl<L: Lang> Rule<L> for AssignBackFold {
    fn name(&self) -> &'static str {
        "assign_back_fold"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::VarDecl]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk } = ctx;
        if lang.kind(id) != NodeKind::VarDecl {
            return None;
        }
        let ch = lang.children(id).to_vec();
        let (&init, has_init) = (ch.first()?, !ch.is_empty());
        if !has_init {
            return None; // 无 init 用 decl_assign_merge / dead_decl 处理
        }
        let name_key = lang.var_key(id)?;
        // 窄类型（byte/short/char 等）复合赋值含隐式收窄——折回声明
        // 产出超域非法常量或丢静态类型（differential Adv2 抓获：
        // `char ch = 'a'; ch += 2` 折后 println 打 99 而非 'c'）
        if !lang.is_wide_decl(id) {
            return None;
        }
        let parent = walk.parent(id)?;
        if lang.kind(parent) != NodeKind::Block {
            return None;
        }
        let idx = walk.index(id)?;
        let stmts = lang.children(parent).to_vec();

        // 扫描声明之后的语句：对 x 的首个引用必须恰是「重赋值」。
        // 句法扫描兜底事件模型盲区（复合赋值目标内的读——`arr[j]+=5`
        // 的 j 对事件索引不可见）；首个引用非重赋值形态 → 中间值被
        // 观察，拒绝。
        let name_str = lang.var_name(id)?.to_string();
        for &s in &stmts[idx + 1..] {
            // Indexed: VarRef mention (including event-blind assignment
            // targets) or a Shadow declaration, plus opaque subtrees.
            let touched = match (lang.stmt_mentions(s, name_key), lang.stmt_opaque(s)) {
                (Some(mentions), Some(opaque)) => mentions || opaque,
                _ => {
                    let mut w = Watch::<L>::new(name_key, &[]);
                    scan_region(&*lang, s, &mut w);
                    let referenced = subtree_contains(&*lang, s, |n| {
                        lang.kind(n) == NodeKind::VarRef
                            && lang.var_name(n) == Some(name_str.as_str())
                    });
                    referenced || w.opaque || w.shadowed
                }
            };
            // 不透明区间（原文/类表达式/未知调用）可能读 x——不可折回
            //（Test262 TypedArray speciesctor：class 体内读 flag，
            // 折回把 `flag=false; …; flag=true` 变成声明即 true）
            if lang.effect(s) == Effect::Unknown {
                return None;
            }
            if !touched {
                continue; // 与 x 无关的语句，跳过
            }
            // 首个触碰 x 的语句必须是重赋值 `x = x op K`
            let assign = match lang.kind(s) {
                NodeKind::Assign => s,
                NodeKind::ExprStmt => {
                    let c = lang.children(s);
                    if c.len() == 1 && lang.kind(c[0]) == NodeKind::Assign {
                        c[0]
                    } else {
                        return None;
                    }
                }
                _ => return None, // 读/遮蔽/不透明先出现 → 中间值被观察
            };
            let ach = lang.children(assign).to_vec();
            if ach.len() != 2 {
                return None;
            }
            let (target, rhs) = (ach[0], ach[1]);
            if lang.kind(target) != NodeKind::VarRef || !lang.same_binding(id, target) {
                return None;
            }
            // (op, K)：复合形态（RHS 即字面量操作数）/ 展开形态
            // （RHS = Binary{x op K}）
            let (op, k) = match lang.assign_op(assign) {
                Some(op) => (op, lang.literal(rhs).and_then(|x| x.as_int())?),
                None => {
                    if lang.kind(rhs) != NodeKind::Binary {
                        return None;
                    }
                    let op = lang.bin_op(rhs)?;
                    let rc = lang.children(rhs).to_vec();
                    if rc.len() != 2 || lang.kind(rc[0]) != NodeKind::VarRef {
                        return None;
                    }
                    // 左操作数必须是 x 自身（同一绑定）；右操作数字面量
                    if !lang.same_binding(id, rc[0]) {
                        return None;
                    }
                    (op, lang.literal(rc[1]).and_then(|x| x.as_int())?)
                }
            };
            if op.is_short_circuit() || op.is_comparison() {
                return None;
            }
            // Div/Rem 且 K==0：折回后异常提前 → 中间语句副作用顺序改变
            if matches!(op, BinOp::Div | BinOp::Rem) && k == 0 {
                return None;
            }
            // long 域声明的 K 提升（H1 抓获：`long g=2147483647;
            // g=g+1` 折成 int 字面量对 → i32 环绕 -2147483648，静默
            // 错值）。init 若是 int 字面量也一并提升——保证 const_fold
            // 走 i64 域
            let lit = if lang.is_long_decl(id) {
                lang.build_int(k, true)
            } else {
                lang.build_int(k, false)
            };
            let init = if lang.is_long_decl(id) {
                match lang.literal(init) {
                    Some(crate::kind::LitRef::Int(v)) => lang.build_int(v, true),
                    _ => init,
                }
            } else {
                init
            };
            let merged = lang.build_bin(op, init, lit);
            // 声明原位换 init（Splice 孩子），语句删除
            return Some(Edit::Multi(vec![
                Edit::Delete { node: s },
                Edit::Splice { node: id, index: 0, remove: ch.len(), insert: vec![merged] },
            ]));
        }
        None
    }
}

pub struct ArithIdentity;

impl<L: Lang> Rule<L> for ArithIdentity {
    fn name(&self) -> &'static str {
        "arith_identity"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Binary]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk: _ } = ctx;
        if lang.kind(id) != NodeKind::Binary {
            return None;
        }
        let op = lang.bin_op(id)?;
        let ch = lang.children(id);
        let (l, r) = (*ch.first()?, *ch.get(1)?);
        let int_lit = |n: L::Id, v: i64| -> bool {
            matches!(lang.literal(n), Some(l) if l.as_int() == Some(v))
        };
        let x = match op {
            BinOp::Add if int_lit(r, 0) => l,
            BinOp::Add if int_lit(l, 0) => r,
            BinOp::Sub if int_lit(r, 0) => l,
            BinOp::Mul if int_lit(r, 1) => l,
            BinOp::Mul if int_lit(l, 1) => r,
            BinOp::Div if int_lit(r, 1) => l,
            _ => return None,
        };
        if !lang.is_exact_int(x) {
            return None;
        }
        Some(Edit::Replace {
            target: id,
            with: x,
        })
    }
}

// ---------------------------------------------------------------------------
// 局部变量传播：
//   T x = V; …; [唯一一次读 x，且区间内无对相关变量的写、无遮蔽]
//   → 用 V 替换该次读，删除声明。
//
// 安全条件：
//   - effect(V) ≤ MayRead（纯值/局部读）：可自由移动，只要求区间内的
//     显式局部写不与 V 读到的变量相交；
//   - effect(V) > MayRead（可能抛/写/未知）：仅当声明与唯一使用**相邻**、
//     使用处于直线语句（return/expr-stmt/assign/throw）、路径无条件求值、
//     且 use 之前求值的兄弟子树效果 ≤ MayRead 时才内联
//     （保持 V 先于语句内其他效果求值）。
// ---------------------------------------------------------------------------


/// 语句区间内名字的**句法** VarRef 出现次数（逐节点计数；含赋值目标、
/// 数组下标基——事件模型对这些位置的读不可见）。传播类规则的引用
/// 计数兜底。注意与「语句级存在性」区分：同一语句里出现两次要计 2。
///
/// `key` 用于有索引语言的一次二分查询；未索引节点回退递归计数。
fn count_name_refs<L: Lang>(
    lang: &L,
    stmts: &[L::Id],
    key: L::NameKey,
    name: &str,
) -> usize {
    let mut n: usize = 0;
    for &s in stmts {
        match lang.stmt_var_ref_count(s, key) {
            Some(c) => n += c as usize,
            None => count_var_refs(lang, s, name, &mut n),
        }
    }
    n
}

/// null 字面量不可安全内联的使用位：沿 Paren 链向上（Paren 不改变静
/// 态类型），首遇 Call/New（实参或 callee 位）或 Member/Index/InstanceOf
/// 头位即拒绝——裸 null 在这些位置丢失声明类型带来的重载消歧依据
///（`sb.append(null)` 对 append(CharSequence)/append(char[]) 二义），
/// 或干脆非法（`null.m()`）。Cast 及其余节点重建类型语境 → 安全。
/// 传播类规则（local/assign）共用的位置门。
fn null_unsafe_position<L: Lang>(lang: &L, walk: &Walk<L>, use_id: L::Id) -> bool {
    let mut cur = use_id;
    while let Some(&(p, _)) = walk.parents.get(&cur) {
        match lang.kind(p) {
            NodeKind::Paren => cur = p,
            NodeKind::Call
            | NodeKind::New
            | NodeKind::Member
            | NodeKind::Index
            | NodeKind::InstanceOf => return true,
            _ => return false,
        }
    }
    false
}

/// 有声明类型的值（VarRef）不可内联进**类型判定位**：cast 操作数与
/// instanceof 被测式的可转换性依赖被替换变量的静态类型——`Object o =
/// k3Var; String s = (String) o;` 合法，内联成 `(String) k3Var` 后引用
/// 类型不相关 → 不可编译（ddc weixin dt/k 复现：R8 宽化赋值的 Object
/// 中转变量被传播进 narrowing cast）。沿 Paren 链向上，首遇 Cast /
/// InstanceOf 即命中；其余位置类型语境与变量无关 → 放行。字面量自带
/// 类型不受此限（null 走 null_unsafe_position）。
///
/// 返回命中的 Cast/InstanceOf 节点（形状层）；类型相容性由 Lang 钩子
/// var_cast_position_ok 裁决——声明类型未解析（反编译自由名）或与目标
/// 类型同名 → 恒可编译，放行（golden redundant_typed_cast：
/// `String s = value; return (String) s;` → `(String) value`）。
fn typed_position<L: Lang>(lang: &L, walk: &Walk<L>, use_id: L::Id) -> Option<L::Id> {
    let mut cur = use_id;
    while let Some(&(p, _)) = walk.parents.get(&cur) {
        match lang.kind(p) {
            NodeKind::Paren => cur = p,
            NodeKind::Cast | NodeKind::InstanceOf => return Some(p),
            _ => return None,
        }
    }
    None
}

fn count_var_refs<L: Lang>(lang: &L, node: L::Id, name: &str, n: &mut usize) {
    if lang.kind(node) == NodeKind::VarRef && lang.var_name(node) == Some(name) {
        *n += 1;
    }
    for &c in lang.children(node) {
        count_var_refs(lang, c, name, n);
    }
}

pub struct LocalPropagation;

impl<L: Lang> Rule<L> for LocalPropagation {
    fn name(&self) -> &'static str {
        "local_propagation"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::VarDecl]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk } = ctx;
        if lang.kind(id) != NodeKind::VarDecl {
            return None;
        }
        let name = lang.var_name(id)?.to_string();
        let ch = lang.children(id);
        if ch.len() != 1 {
            return None; // 必须有且仅有 init
        }
        let value = ch[0];
        // 裸数组字面量只在声明/赋值 RHS 位置合法（Java/多数语言同此）；
        // 传播到任意表达式位置会产出非法源码，且多站点传播会破坏数组对象
        // 同一性 → 拒绝（显式 new T[]{…} 形态可安全传播）
        if lang.kind(value) == NodeKind::ArrayLit {
            return None;
        }
        // lambda / 方法引用值：内联进 receiver 位产出非法 Java
        //（(() -> {}).run() / String::length.apply()——对抗波 2 抓获）。
        // 参数位合法但罕见——统一保守拒绝。
        if matches!(lang.kind(value), NodeKind::Lambda | NodeKind::MethodRef) {
            return None;
        }
        // 自引用 init（int x = x + 1 之类）直接放弃
        if subtree_contains(&*lang, value, |n| {
            lang.kind(n) == NodeKind::VarRef && lang.var_name(n) == Some(name.as_str())
        }) {
            return None;
        }
        let parent = walk.parent(id)?;
        if lang.kind(parent) != NodeKind::Block {
            return None;
        }
        let index = walk.index(id)?;
        let stmts: Vec<L::Id> = lang.children(parent).to_vec();
        if index + 1 > stmts.len() {
            return None;
        }

        // 字符类型声明：内联值必须是 Char 字面量（其他 int 型值——字面量
        // 或表达式——会让 println/拼接观察到 int 而非 char：
        // `char c = 98; println(c)` 打 'b'，内联后打 98）。int 字面量
        // init 就地转换为 Char 字面量再传播（Java char 域 u16；转换
        // 随声明删除获得正成本——纯归一化规则过不了严格降本门）。
        let mut value = value;
        if lang.is_char_decl(id) {
            match lang.literal(value) {
                Some(LitRef::Char(_)) => {}
                Some(LitRef::Int(v)) => {
                    // Java char = UTF-16 单元：域 [0, 0xFFFF]（代理区也拒，
                    // 打印不合法）
                    let c = u32::try_from(v)
                        .ok()
                        .filter(|&u| (0..=0xFFFF).contains(&u) && !(0xD800..=0xDFFF).contains(&u))
                        .and_then(char::from_u32)?;
                    value = lang.build_char(c);
                }
                _ => return None,
            }
        }
        // long 域声明 + int 字面量 init：按 i32 域传播会让后续算术在
        // i32 环绕（`long n = 5; n + 2147483647` 曾折成 -2147483644，
        // 正确值 2147483652——边界攻击 t06d 抓获）。就地转 Long 字面量
        //（5L）传播，const_fold 走 i64 域
        if lang.is_long_decl(id) {
            if let Some(LitRef::Int(v)) = lang.literal(value) {
                value = lang.build_int(v, true);
            }
        }
        // byte/short 窄域声明：Int 字面量内联到**调用实参位**是非法收窄
        //（JLS 常量收窄仅限赋值上下文——t04h 抓获：eat(127, 1023) 进
        // byte/short 形参不可编译）。使用位置不可判 → 保守拒绝传播
        //（char 域已有 Int→Char 转换——Char 字面量到 char 形参合法）
        if !lang.is_wide_decl(id) && !lang.is_char_decl(id) {
            return None;
        }

        if !lang.local_propagation_value_ok(id, value) {
            return None;
        }
        // 扫描 decl 之后的区域（用途/遮蔽全区间；写冲突窗口见下）。
        // 句法引用计数兜底事件盲区：赋值目标内的读（t[k]=… 的 t）
        // 对事件索引不可见——bd.java 差分抓获：load 被替换而 store
        // 目标残留，输出引用已删除的变量。
        let name_str = lang.var_name(id)?.to_string();
        let name_key = lang.var_key(id)?;
        // **λ 捕获守卫 + 句法引用计数**（jOOQ Tools.hasAmbiguousNames
        // 抓获：`Set names = new HashSet<>(); return anyMatch(fields, f ->
        // !names.add(...))` 曾被内联成 `f -> !new HashSet<>().add(...)`——
        // 每次谓词新建 Set，add 恒 true，重名检测恒 false）→ 拒绝传播。
        // Indexed languages answer both facts per statement in one binary
        // search; the recursive pair is the fallback for unindexed nodes.
        // 两个条件任一成立即拒绝，故可提前退出。
        let mut captured = false;
        let mut syntactic_refs = 0usize;
        for &st in &stmts[index + 1..] {
            let (refs, cap) = match (
                lang.stmt_var_ref_count(st, name_key),
                lang.stmt_lambda_captures(st, name_key),
            ) {
                (Some(refs), Some(cap)) => (refs as usize, cap),
                _ => {
                    let mut n = 0usize;
                    count_var_refs(&*lang, st, &name_str, &mut n);
                    let cap = subtree_contains(&*lang, st, |n| {
                        if lang.is_function_like(n) {
                            return subtree_contains(&*lang, n, |m| {
                                lang.kind(m) == NodeKind::VarRef
                                    && lang.var_name(m) == Some(name_str.as_str())
                            });
                        }
                        false
                    });
                    (n, cap)
                }
            };
            syntactic_refs += refs;
            captured |= cap;
            if captured || syntactic_refs > 1 {
                break;
            }
        }
        if captured || syntactic_refs != 1 {
            return None;
        }
        let mut wa = Watch::<L>::new(name_key, &[]);
        for &s in &stmts[index + 1..] {
            // 类体/原文里的读不可见——声明删除会让其悬挂
            if lang.stmt_has_opaque_region(s) {
                return None;
            }
            scan_region(&*lang, s, &mut wa);
        }
        if wa.opaque || wa.shadowed || wa.uses.len() != 1 {
            return None;
        }
        let use_id = wa.uses[0];

        // 纯自赋值（x = x）形态拒绝传播：让 self_assign 先删。传播会把
        // RHS 换成声明值制造悬挂写 `i = 1`——store_kill 删声明后
        // dead_store 因锚点缺失无法清理（golden self_assign_removed
        // 抓获：期望 keep(); 实得 i = 1; keep();）
        if let Some(&(p, _)) = walk.parents.get(&use_id) {
            if lang.kind(p) == NodeKind::Assign && lang.assign_op(p).is_none() {
                let pch = lang.children(p);
                if pch.len() == 2
                    && lang.kind(pch[0]) == NodeKind::VarRef
                    && lang.same_binding(pch[0], use_id)
                    && pch[1] == use_id
                {
                    return None;
                }
            }
        }
        // raw 泛型接收位守卫（Lang 钩子）：声明的静态类型与 init 的静态
        // 类型在成员解析上不等价（Java raw `List l = m(); l.add("x")`
        // 内联后 CAP#1 硬错——mockito 抓获）。使用位的父/头信息由
        // walk 提供（Lang 侧无父指针）
        {
            let (up, head) = match walk.parents.get(&use_id) {
                Some(&(p, _)) => {
                    (Some(p), lang.children(p).first() == Some(&use_id))
                }
                None => (None, false),
            };
            if lang.receiver_propagation_unsound(id, value, up, head) {
                return None;
            }
        }
        // null 字面量**与基本类型字面量**不得内联进需要引用类型的表达式
        // 位：调用接收者（`null.setArgName(...)` / `65.getClass()`——后者
        // R13 P0-2 抓获：`Object o = +c; o.getClass()` 常量传播产出
        // `65.getClass()` 非法 Java）；for-each 可迭代位；数组访问基座
        //（`null[0]`）；字段/数组基座同理。字符串字面量接收者合法
        //（`"a".length()`）不在拒绝域
        if matches!(
            lang.literal(value),
            Some(
                LitRef::Null
                    | LitRef::Int(_)
                    | LitRef::Long(_)
                    | LitRef::Float(_)
                    | LitRef::Double(_)
                    | LitRef::Char(_)
                    | LitRef::Bool(_)
            )
        ) {
            let mut cur = use_id;
            while let Some(&(p, _)) = walk.parents.get(&cur) {
                let at_head = lang.children(p).first() == Some(&cur);
                match lang.kind(p) {
                    // null.m() / null[i]（Member 基座与 Index 基座）
                    NodeKind::Member | NodeKind::Index if at_head => return None,
                    // 静态调用 callee 位（防御）
                    NodeKind::Call if at_head => return None,
                    // instanceof 被测式（R14 族 C 抓获：`Object o = 3.5f;
                    // o instanceof Double` 曾内联成 `3.5f instanceof Double`
                    // ——基元模式预览语法不可编译）
                    NodeKind::InstanceOf if at_head => return None,
                    // for-each 可迭代位（ForEach 的第 1 个孩子 = iterable）
                    NodeKind::ForEach => {
                        let ch = lang.children(p);
                        if ch.first() == Some(&cur) {
                            return None;
                        }
                    }
                    // synchronized (null…) 锁位（唯一孩子）
                    NodeKind::Synchronized => {
                        let ch = lang.children(p);
                        if ch.len() == 1 && ch[0] == cur {
                            return None;
                        }
                    }
                    _ => {}
                }
                cur = p;
            }
        }
        // null 字面量实参位：`String s = null; sb.append(s)` 合法——s 的
        // 声明类型是重载消歧依据；内联成 `sb.append(null)` 后
        // append(CharSequence) 与 append(char[]) 二义，不可编译
        //（ddc 输出 PurchasesOrchestrator 复现：R8 null 复位赋值 + 单次
        // 使用恰好落在重载族调用上）。沿 Paren 链向上（Paren 不改变静
        // 态类型），首遇 Call/New 实参位即拒绝；Cast 及其余节点重建类
        // 型语境（`(String) s` 重新钉住重载）→ 放行
        if matches!(lang.literal(value), Some(LitRef::Null))
            && null_unsafe_position(&*lang, walk, use_id)
        {
            return None;
        }
        // VarRef 值不得内联进 cast/instanceof 类型判定位（声明类型即合法
        // 性依据——同 null 实参位的族）；声明类型未解析或与目标类型
        // 同名时恒可编译（var_cast_position_ok 钩子裁决）
        if lang.kind(value) == NodeKind::VarRef
            && typed_position(&*lang, walk, use_id)
                .is_some_and(|t| !lang.var_cast_position_ok(value, t))
        {
            return None;
        }
        // value 读到的名字键（写冲突兴趣集；reads 含 name 自身——一票否决）
        let mut watch_keys: Vec<L::NameKey> = Vec::new();
        collect_read_keys(&*lang, value, &mut watch_keys);
        watch_keys.push(name_key);
        // 写冲突窗口 = [decl 后, 使用语句]：纯值移动到使用点，
        // 使用点之后的写不影响（值已被消费）。首个含 use 的语句即使用语句
        //（uses[0] 是遍历序首个读，与 subtree_contains 等价）
        let mut wb = Watch::<L>::new(name_key, &watch_keys);
        let mut use_stmt_idx = None;
        for (off, &s) in stmts[index + 1..].iter().enumerate() {
            let uses_before = wb.uses.len();
            scan_region(&*lang, s, &mut wb);
            if wb.uses.len() > uses_before {
                use_stmt_idx = Some(index + 1 + off);
                break; // 使用语句之后的写不参与冲突判定
            }
        }
        // 字段读移动守卫：值中含**非局部** VarRef（可变字段——跨调用/跨
        // 线程可变）：把读从声明点移动到使用点会改变读时机（d.java 抓获：
        // `boolean var2 = b; for (a var1 : a()) { var1.i(); if (var2)… }` 折
        // 成 if (b)——入口快照变成每次调用后现读，b 有跨文件写点）。
        // 窗口内语句须全 Pure（无调用/无写/无读）。
        // Call 的 callee VarRef 是方法名（不同命名空间）——不算字段读
        if subtree_contains(&*lang, value, |n| {
            if lang.kind(n) != NodeKind::VarRef || lang.is_local_var(n) {
                return false;
            }
            // 仅**本单元声明的字段**（局部/参数/未解析名——如手工 AST
            // 无作用域——不算）
            if !lang.var_name(n).is_some_and(|nm| lang.is_field_name(nm)) {
                return false;
            }
            let is_callee = match walk.parent(n) {
                Some(p) => {
                    matches!(lang.kind(p), NodeKind::Call) && lang.children(p).first() == Some(&n)
                }
                None => false,
            };
            !is_callee
        }) {
            if let Some(ui) = use_stmt_idx {
                for &s in &stmts[index + 1..=ui] {
                    if lang.effect(s) != Effect::Pure {
                        return None;
                    }
                }
            }
        }
        // 声明删除前提：use 之后不得再出现对该名字的任何引用（读/写/遮蔽）。
        // 值的传播只关心窗口内写，但**删声明**要求名字彻底无残留引用——
        // 使用点之后的死写（如 v = "y"）同样引用声明，残留会让输出失去声明。
        if let Some(ui) = use_stmt_idx {
            // 句法版：任何残留引用（含事件不可见的目标位）都拒绝删声明
            if count_name_refs(&*lang, &stmts[ui + 1..], name_key, &name_str) != 0 {
                return None;
            }
            let mut wc = Watch::<L>::new(name_key, &[name_key]);
            for &s in &stmts[ui + 1..] {
                scan_region(&*lang, s, &mut wc);
                if wc.opaque || !wc.uses.is_empty() || wc.wrote(name_key) || wc.shadowed {
                    return None;
                }
            }
        }
        let ve = lang.effect(value);

        if ve <= Effect::MayRead {
            // 自由移动：区间内的显式局部写不得触碰 V 读到的变量
            // （对声明变量自身的写也包含在 writes 里，一票否决）
            if let Some(target_key) =
                use_stmt_target_write_key(&*lang, &stmts[index + 1..], use_id)
            {
                wb.clear_wrote(target_key);
            }
            if wb.opaque || wb.any_write() {
                return None;
            }
            return Some(Edit::Multi(vec![
                Edit::Replace {
                    target: use_id,
                    with: value,
                },
                Edit::Delete { node: id },
            ]));
        }

        // 有副作用的 V：只允许相邻 + 直线 + 前缀可读
        if index + 1 >= stmts.len() {
            return None;
        }
        let stmt = stmts[index + 1];
        if !subtree_contains(&*lang, stmt, |n| n == use_id) {
            return None;
        }
        match lang.kind(stmt) {
            // VarDecl 同样合法：init 就是语句的全部求值，V 内联进 init 的
            // 求值时机与原声明位置一致（相邻保证无插入效果）
            NodeKind::Return
            | NodeKind::ExprStmt
            | NodeKind::Assign
            | NodeKind::Throw
            | NodeKind::VarDecl => {}
            _ => return None,
        }
        if !straight_path(&*lang, stmt, use_id) {
            return None;
        }
        if !prefix_effects_readable(&*lang, stmt, use_id) {
            return None;
        }
        Some(Edit::Multi(vec![
            Edit::Replace {
                target: use_id,
                with: value,
            },
            Edit::Delete { node: id },
        ]))
    }
}

/// 区域扫描观察：主名字的读 + 小兴趣集的写命中 + 同名遮蔽。
///
/// 规则预先声明感兴趣的**写名字**（通常 ≤4：主名字 + value 读到的变量），
/// 收集期直接判中——零哈希、零集合重建。对比旧版：每次检查重建
/// `HashSet<&str>`（分配/重哈希/释放）曾占 release 运行时间 ~50%。
pub(crate) struct Watch<L: Lang> {
    /// 区域含不透明（Raw）内容：读/写集不可证明——所有守卫保守拒绝
    opaque: bool,
    /// 主名字键：读收集 + 遮蔽判定（整数等值比较）
    name: L::NameKey,
    /// 写冲突兴趣名字键（典型 2~4 个）
    watch: Vec<L::NameKey>,
    /// 与 watch 一一对应：窗口内是否出现过该名字的写
    writes_hit: Vec<bool>,
    shadowed: bool,
    /// 主名字的读（遍历序）
    uses: Vec<L::Id>,
}

impl<L: Lang> Watch<L> {
    pub(crate) fn new(name: L::NameKey, watch: &[L::NameKey]) -> Self {
        Watch {
            opaque: false,
            name,
            watch: watch.to_vec(),
            writes_hit: vec![false; watch.len()],
            shadowed: false,
            uses: Vec::new(),
        }
    }
    /// 名字键 `k` 在窗口内是否被写（聚合判断）
    pub(crate) fn wrote(&self, k: L::NameKey) -> bool {
        self.watch
            .iter()
            .zip(&self.writes_hit)
            .any(|(w, h)| *h && *w == k)
    }
    /// 任一兴趣名字被写
    pub(crate) fn any_write(&self) -> bool {
        self.writes_hit.iter().any(|h| *h)
    }
    /// 清除对 `k` 的写命中（use 语句自身对目标的写不参与冲突判定）
    pub(crate) fn clear_wrote(&mut self, k: L::NameKey) {
        for (w, h) in self.watch.iter().zip(self.writes_hit.iter_mut()) {
            if *w == k {
                *h = false;
            }
        }
    }
}

/// 扫描一个语句子树：主名字的**读**、兴趣名字的**写命中**、同名声明遮蔽。
fn scan_region<L: Lang>(lang: &L, node: L::Id, w: &mut Watch<L>) {
    lang.debug_verify_events(node);
    if let Some(events) = lang.region_events(node) {
        // 主名字区间：Use 收集 + Shadow 命中 + Write 命中（name 可能在 watch 里）
        let lo = events.partition_point(|e| e.key < w.name);
        let hi = events.partition_point(|e| e.key <= w.name);
        for e in &events[lo..hi] {
            match e.kind {
                crate::kind::EventKind::Use => w.uses.push(e.node),
                crate::kind::EventKind::Shadow => w.shadowed = true,
                crate::kind::EventKind::Write => {
                    for (wn, hit) in w.watch.iter().zip(w.writes_hit.iter_mut()) {
                        if *wn == w.name {
                            *hit = true;
                        }
                    }
                }
                // 防御：索引层保证含 Raw 的语句不建索引（键排序分区看不见
                // Opaque 键），此处理论上不可达
                crate::kind::EventKind::Opaque => w.opaque = true,
            }
        }
        // 其余 watch 键区间：只找 Write
        for (i, wk) in w.watch.iter().enumerate() {
            if *wk == w.name {
                continue; // 已随主名字区间处理
            }
            let lo = events.partition_point(|e| e.key < *wk);
            let hi = events.partition_point(|e| e.key <= *wk);
            for e in &events[lo..hi] {
                if e.kind == crate::kind::EventKind::Write {
                    w.writes_hit[i] = true;
                    break; // 区间内任一 Write 即命中
                }
            }
        }
        return;
    }
    // 递归路径也查不透明（索引被失效后落到此处的防御）
    if lang.is_opaque(node) {
        w.opaque = true;
        return;
    }
    // 无索引语言：原递归遍历（事件序一致；键比较同上）
    match lang.kind(node) {
        NodeKind::Assign => {
            let ch = lang.children(node);
            if let Some(&t) = ch.first() {
                if lang.kind(t) == NodeKind::VarRef {
                    if let Some(k) = lang.var_key(t) {
                        for (wn, hit) in w.watch.iter().zip(w.writes_hit.iter_mut()) {
                            if *wn == k {
                                *hit = true;
                            }
                        }
                    }
                    // 复合赋值读旧值（简单赋值是纯写）
                    if lang.assign_op(node).is_some() {
                        scan_region(lang, t, w);
                    }
                } else {
                    // 成员/索引目标：基座/键是读
                    scan_region(lang, t, w);
                }
            }
            if let Some(&v) = ch.get(1) {
                scan_region(lang, v, w);
            }
        }
        NodeKind::Unary if lang.un_op(node).is_some_and(|o| o.is_incdec()) => {
            let ch = lang.children(node);
            if let Some(&t) = ch.first() {
                if lang.kind(t) == NodeKind::VarRef {
                    if let Some(k) = lang.var_key(t) {
                        for (wn, hit) in w.watch.iter().zip(w.writes_hit.iter_mut()) {
                            if *wn == k {
                                *hit = true;
                            }
                        }
                    }
                    // ++/-- 也读旧值
                    scan_region(lang, t, w);
                } else {
                    scan_region(lang, t, w);
                }
            }
        }
        NodeKind::VarDecl => {
            if let Some(k) = lang.var_key(node) {
                if k == w.name {
                    w.shadowed = true;
                }
                for (wn, hit) in w.watch.iter().zip(w.writes_hit.iter_mut()) {
                    if *wn == k {
                        *hit = true;
                    }
                }
            }
            for &c in lang.children(node) {
                scan_region(lang, c, w);
            }
        }
        // ForEach 循环变量 / Catch 绑定：在子作用域声明了同名变量 → 视为遮蔽；
        // ForEach 每轮给循环变量赋值 → 也算显式写。
        NodeKind::ForEach | NodeKind::Catch => {
            if let Some(k) = lang.var_key(node) {
                if k == w.name {
                    w.shadowed = true;
                }
                if lang.kind(node) == NodeKind::ForEach {
                    for (wn, hit) in w.watch.iter().zip(w.writes_hit.iter_mut()) {
                        if *wn == k {
                            *hit = true;
                        }
                    }
                }
            }
            for &c in lang.children(node) {
                scan_region(lang, c, w);
            }
        }
        NodeKind::VarRef => {
            if lang.var_key(node) == Some(w.name) {
                w.uses.push(node);
            }
        }
        NodeKind::Raw => {
            // 不可解析原文：读/写集未知
            w.opaque = true;
        }
        _ => {
            for &c in lang.children(node) {
                scan_region(lang, c, w);
            }
        }
    }
}

/// root → target 的路径上不得有条件求值结构（Ternary / If / 短路右支）。
fn straight_path<L: Lang>(lang: &L, root: L::Id, target: L::Id) -> bool {
    fn go<L: Lang>(lang: &L, node: L::Id, target: L::Id) -> bool {
        if node == target {
            return true;
        }
        match lang.kind(node) {
            NodeKind::Ternary | NodeKind::If => return false,
            NodeKind::Binary if lang.bin_op(node).is_some_and(|o| o.is_short_circuit()) => {
                return false
            }
            _ => {}
        }
        for &c in lang.children(node) {
            if subtree_contains(lang, c, |n| n == target) {
                return go(lang, c, target);
            }
        }
        false
    }
    go(lang, root, target)
}


// ---------------------------------------------------------------------------
// 短路折叠：
//   false && x → false / true || x → true（x 原本就不求值，恒安全）
//   true && x → x / false || x → x / x && true → x / x || false → x（x 保留，恒安全）
//   x && false → false / x || true → true（x 的求值被丢弃，需 effect ≤ MayRead）
// ---------------------------------------------------------------------------

/// id 是否位于标签模板 tag 位（Custom(Template) 的 child 0）。
fn in_tag_position<L: Lang>(lang: &L, walk: &crate::walk::Walk<L>, id: L::Id) -> bool {
    let Some((p, i)) = walk.parents.get(&id) else { return false };
    *i == 0
        && lang.kind(*p) == NodeKind::Custom(1) // custom::TEMPLATE
        && lang.is_tagged_template(*p)
}

pub struct BoolShortCircuit;

impl<L: Lang> Rule<L> for BoolShortCircuit {
    fn name(&self) -> &'static str {
        "bool_short_circuit"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Binary]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk } = ctx;
        if lang.kind(id) != NodeKind::Binary {
            return None;
        }
        let op = lang.bin_op(id)?;
        if !op.is_short_circuit() {
            return None;
        }
        // `delete (a && b)` 的 delete 语义是「非引用 → true」，折叠成
        // `delete b` 会真的尝试删除（返回值改变）；
        // 标签位折臂会改变 tag 的 Reference/this 绑定。
        if in_tag_position(&*lang, &walk, id) {
            return None;
        }
        if let Some((p, i)) = walk.parents.get(&id) {
            if *i == 0 && lang.kind(*p) == NodeKind::Unary && lang.un_op(*p) == Some(UnOp::Delete) {
                return None;
            }
        }
        let ch = lang.children(id);
        let (l, r) = (*ch.first()?, *ch.get(1)?);
        let is_lit = |n: L::Id, v: bool| matches!(lang.literal(n), Some(LitRef::Bool(b)) if b == v);
        // 左侧字面量：短路后 x 可能根本不求值 → 直接取字面量结果
        let with = match (op, is_lit(l, true), is_lit(l, false)) {
            (BinOp::And, _, true) => l,  // false && x → false（x 不求值）
            (BinOp::Or, true, _) => l,   // true || x → true（x 不求值）
            (BinOp::And, true, _) => r,  // true && x → x
            (BinOp::Or, _, true) => r,   // false || x → x
            _ => {
                // 右侧字面量：x && true → x / x || false → x（x 保留求值，恒安全）；
                // x && false → false / x || true → true（x 的求值被丢弃，需可丢弃）
                // x && true → x / x || false → x **仅当 x 是布尔值**：
                // JS 里 `sym && true` 对 truthy 的非布尔 x 返回 true 而不是
                // x（Test262 logical-and/symbol 抓获；Java 类型系统恒真）。
                match (op, is_lit(r, true), is_lit(r, false)) {
                    (BinOp::And, true, _) if lang.is_bool(l) => l,
                    (BinOp::Or, _, true) if lang.is_bool(l) => l,
                    // x && false → false / x || true → true **仅当 x 布尔**：
                    // JS 里 truthy/falsy 非布尔 x 返回 x（`null && false`
                    // === null），且未解析引用求值会抛 ReferenceError
                    //（Test262 logical 家族抓获）
                    (BinOp::And, _, true) | (BinOp::Or, true, _)
                        if lang.is_bool(l) && lang.effect(l) <= Effect::MayRead => {
                            r
                        }
                    _ => return None,
                }
            }
        };
        Some(Edit::Replace {
            target: id,
            with,
        })
    }
}

// ---------------------------------------------------------------------------
// 比较取反：!(a == b) → a != b（恒安全，含 NaN/引用）；
// !(a < b) → a >= b 等次序比较仅精确整数（浮点 NaN 下不成立）。
// ---------------------------------------------------------------------------

pub struct NotCompare;

impl<L: Lang> Rule<L> for NotCompare {
    fn name(&self) -> &'static str {
        "not_compare"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Unary]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk: _ } = ctx;
        if lang.kind(id) != NodeKind::Unary || lang.un_op(id) != Some(UnOp::Not) {
            return None;
        }
        let inner = *lang.children(id).first()?;
        if lang.kind(inner) != NodeKind::Binary {
            return None;
        }
        let op = lang.bin_op(inner)?;
        let neg = match op {
            BinOp::Eq => BinOp::Ne,
            BinOp::Ne => BinOp::Eq,
            BinOp::Lt => BinOp::Ge,
            BinOp::Le => BinOp::Gt,
            BinOp::Gt => BinOp::Le,
            BinOp::Ge => BinOp::Lt,
            _ => return None,
        };
        if op.is_short_circuit() {
            return None;
        }
        // 次序比较的取反仅对精确整数成立（NaN: !(a<b)=true 但 a>=b=false）
        if matches!(op, BinOp::Lt | BinOp::Le | BinOp::Gt | BinOp::Ge) {
            let ch = lang.children(inner);
            if !lang.is_exact_int(*ch.first()?) || !lang.is_exact_int(*ch.get(1)?) {
                return None;
            }
        }
        let ch = lang.children(inner).to_vec();
        let rebuilt = lang.build_bin(neg, ch[0], ch[1]);
        Some(Edit::Replace {
            target: id,
            with: rebuilt,
        })
    }
}

// ---------------------------------------------------------------------------
// 三元折叠：true ? a : b → a；false ? a : b → b（另一支原本不求值，恒安全）；
//           c ? a : a → a（c 的求值被丢弃，需 effect ≤ MayRead）
// ---------------------------------------------------------------------------

/// 折臂敏感位：从 `id` 向上穿过 Paren/Ternary（类型透明链），最终落在
/// **推断声明**（Java 10 `var`）的 init 上 → true。var 的静态类型由整个
/// init 表达式推断（三元 = lub(两臂)）——折臂改类型。其余容器（实参/
/// 二元操作数/Cast/λ 体）的静态类型由外层结构决定，内层折臂不影响推断。
fn in_inferred_var_init<L: Lang>(lang: &L, walk: &Walk<L>, mut id: L::Id) -> bool {
    while let Some(&(p, _)) = walk.parents.get(&id) {
        match lang.kind(p) {
            NodeKind::Paren | NodeKind::Ternary => id = p,
            NodeKind::VarDecl => return lang.is_inferred_decl(p),
            _ => return false,
        }
    }
    false
}

pub struct TernaryFold;

impl<L: Lang> Rule<L> for TernaryFold {
    fn name(&self) -> &'static str {
        "ternary_fold"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Ternary]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk } = ctx;
        if lang.kind(id) != NodeKind::Ternary {
            return None;
        }
        if in_tag_position(&*lang, &walk, id) {
            return None;
        }
        let ch = lang.children(id);
        let (c, a, b) = (*ch.first()?, *ch.get(1)?, *ch.get(2)?);
        match lang.literal(c) {
            // R15 P0-1（ClassNotFoundExceptionDueToPrunedCodeTest o1/o2）
            // 抓获：`var o1 = false ? new Object(){} : null` 曾折成
            // `var o1 = null`——var 推断从匿名类类型变 null 类型，
            // `o1 == o1` 不可编译。推断声明 init 位（类型透明链上）
            // 拒绝折臂。
            Some(LitRef::Bool(true)) => {
                if !in_inferred_var_init(&*lang, walk, id) {
                    return Some(Edit::Replace { target: id, with: a });
                }
            }
            Some(LitRef::Bool(false))
                if !in_inferred_var_init(&*lang, walk, id) => {
                    return Some(Edit::Replace { target: id, with: b });
                }
            _ => {}
        }
        // c ? a : a：两臂结构全同 → 折叠不改变静态类型（lub(a,a)=a），
        // var init 位同样安全
        if crate::analysis::structurally_equal(&*lang, a, b) && lang.effect(c) <= Effect::MayRead {
            return Some(Edit::Replace { target: id, with: a });
        }
        None
    }
}

// ---------------------------------------------------------------------------
// 三元布尔：c ? true : false → c；c ? false : true → !c（c 为原始 boolean）
// ---------------------------------------------------------------------------

pub struct TernaryBool;

impl<L: Lang> Rule<L> for TernaryBool {
    fn name(&self) -> &'static str {
        "ternary_bool"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Ternary]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk: _ } = ctx;
        if lang.kind(id) != NodeKind::Ternary {
            return None;
        }
        let ch = lang.children(id);
        let (c, a, b) = (*ch.first()?, *ch.get(1)?, *ch.get(2)?);
        let lb = |n: L::Id| matches!(lang.literal(n), Some(LitRef::Bool(v)) if v);
        let is_false = |n: L::Id| matches!(lang.literal(n), Some(LitRef::Bool(v)) if !v);
        if !lang.is_bool(c) {
            return None;
        }
        let with = if lb(a) && is_false(b) {
            c
        } else if is_false(a) && lb(b) {
            lang.build_unary(UnOp::Not, c)
        } else {
            return None;
        };
        Some(Edit::Replace {
            target: id,
            with,
        })
    }
}

// ---------------------------------------------------------------------------
// 整数常量折叠 + 字符串拼接：形态匹配在引擎（双字面量 + 非短路/比较
// 算符），数值/拼接**语义**全权在 Lang::fold_lit_bin（Java：JLS i32/i64
// 环绕 + 移位掩码 + 除零不折 + "a"+1 拼接；其他语言各自定义）。
// ---------------------------------------------------------------------------

pub struct ConstFoldBin;

impl<L: Lang> Rule<L> for ConstFoldBin {
    fn name(&self) -> &'static str {
        "const_fold_bin"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Binary]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk: _ } = ctx;
        if lang.kind(id) != NodeKind::Binary {
            return None;
        }
        let op = lang.bin_op(id)?;
        if op.is_short_circuit() || op.is_comparison() {
            return None;
        }
        let ch = lang.children(id);
        let (l, r) = (*ch.first()?, *ch.get(1)?);
        // 双字面量才考虑折叠（具体哪些组合可折由语言语义决定）
        lang.literal(l)?;
        lang.literal(r)?;
        let with = lang.fold_lit_bin(op, l, r)?;
        Some(Edit::Replace {
            target: id,
            with,
        })
    }
}

// ---------------------------------------------------------------------------
// 零元素 / 平凡算术：
//   0 * x → 0、x * 0 → 0（x 求值被丢弃：需 effect ≤ MayRead 且精确整数——
//     浮点 0.0*NaN=NaN）
//   x - x → 0（同名局部整型变量）
//   x << 0 / x >> 0 / x >>> 0 → x（位移 0 恒等，x 保留，恒安全）
// ---------------------------------------------------------------------------

pub struct ArithZero;

impl<L: Lang> Rule<L> for ArithZero {
    fn name(&self) -> &'static str {
        "arith_zero"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Binary]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk: _ } = ctx;
        if lang.kind(id) != NodeKind::Binary {
            return None;
        }
        let op = lang.bin_op(id)?;
        let ch = lang.children(id);
        let (l, r) = (*ch.first()?, *ch.get(1)?);
        let int_lit = |n: L::Id, v: i64| lang.literal(n).and_then(|x| x.as_int()) == Some(v);

        // 移位 0：x << 0 → x（恒等）。**需精确整数域**：JS 的移位是
        // ToInt32 胁迫——`-4294967295 << 0 === 1` 而非原值（Test262
        // left-shift 族抓获）。Java 的 int/long 解析为真 → 中性。
        if matches!(op, BinOp::Shl | BinOp::Shr | BinOp::UShr) && int_lit(r, 0) {
            if !lang.is_exact_int(l) {
                return None;
            }
            return Some(Edit::Replace {
                target: id,
                with: l,
            });
        }
        // 0 * x → 0 / x * 0 → 0
        if op == BinOp::Mul {
            let (zero, other) = if int_lit(l, 0) {
                (l, r)
            } else if int_lit(r, 0) {
                (r, l)
            } else {
                return None;
            };
            if !lang.is_exact_int(other) || lang.effect(other) > Effect::MayRead {
                return None;
            }
            return Some(Edit::Replace {
                target: id,
                with: zero,
            });
        }
        // x - x → 0（同名局部变量，整型）
        if op == BinOp::Sub
            && lang.kind(l) == NodeKind::VarRef
            && lang.kind(r) == NodeKind::VarRef
            && lang.same_binding(l, r)
            && lang.is_local_var(l)
            && lang.is_exact_int(l)
        {
            let with = lang.build_int(0, false);
            return Some(Edit::Replace {
                target: id,
                with,
            });
        }
        None
    }
}

// ---------------------------------------------------------------------------
// 死赋值：x = a; x = b; → x = b;（a 的求值被丢弃：局部变量 x + effect(a) ≤ MayRead）
// 含声明形：int x = a; x = b; → int x = b;
// ---------------------------------------------------------------------------


// ---------------------------------------------------------------------------
// 逆运算对抵消（寄存器噪声）：`x op= K;` 紧跟 `x op⁻¹= K;`（同局部变量、
// 同字面量常量、相邻语句、精确整型）→ 两条全删。
//   `x += K; x -= K` → 删（模 2^w 加法群恒等）
//   `x ^= K; x ^= K` → 删（XOR 自逆）
//   `x -= K; x += K` → 删（对称）
// 展开形态 `x = x + K; x = x - K`（RHS = Binary(x op K)）与混合形态
// （`x += K; x = x - K`）同样处理——反编译器常输出展开形态。
// 相邻 ⇒ 两句之间无任何求值 ⇒ 删除恒安全（x 终值不变；x 的旧值
// 不被其他求值观察到）。
// 守卫：x 为局部变量（is_local_var——字段写有可见副作用）；K 为
// 无副作用字面量；x 精确整型（浮点 (x+K)-K 在溢出/Inf 上 ≠ x）。
// 真实样本：PMD 语料 PMDTaskTestExample.java（a+=1; a-=1 交替 50 次
// 完全不动——寄存器噪声经典形态，109 行全为噪声）。
// ---------------------------------------------------------------------------

pub struct InverseAssignPair;

impl<L: Lang> Rule<L> for InverseAssignPair {
    fn name(&self) -> &'static str {
        "inverse_assign_pair"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Block]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, .. } = ctx;
        let ch = lang.children(id).to_vec();
        if ch.len() < 2 {
            return None;
        }
        for w in 0..ch.len() - 1 {
            let (s1, s2) = (ch[w], ch[w + 1]);
            // 窗口左端不是赋值语句 → 滑到下一窗口（不能整check放弃）
            let Some(a1) = norm_assign(lang, s1) else { continue };
            let Some(a2) = norm_assign(lang, s2) else { continue };
            // 同一局部变量（同一绑定；名字键仅作快速预筛）
            if a1.key != a2.key
                || !lang.same_binding(a1.target, a2.target)
                || !a1.local
                || !a2.local
            {
                continue;
            }
            // 精确整型（模 2^w 群恒等性成立；浮点舍入破坏交换律）
            if !a1.exact_int || !a2.exact_int {
                continue;
            }
            // 同一常量 + 互逆算符
            if a1.k != a2.k {
                continue;
            }
            let inverse = match (a1.op, a2.op) {
                (BinOp::Add, BinOp::Sub) | (BinOp::Sub, BinOp::Add) => true,
                (BinOp::BitXor, BinOp::BitXor) => true,
                _ => false,
            };
            if !inverse {
                continue;
            }
            // 删除两条语句：先删靠后的（索引不失效），再删靠前的
            return Some(Edit::Multi(vec![
                Edit::Delete { node: s2 },
                Edit::Delete { node: s1 },
            ]));
        }
        None
    }
}

/// 归一化一条赋值语句 → `(变量键, 算符, 常量)`。
/// 接受复合形态（`x op= K`，RHS 即字面量）与展开形态（`x = x op K`，
/// RHS = Binary{op, VarRef(x), K}）。
struct NormAssign<L: Lang> {
    key: L::NameKey,
    target: L::Id,
    op: BinOp,
    k: i64,
    local: bool,
    /// 精确整型（浮点 (x+K)-K 在舍入/Inf 上 ≠ x，必须拒绝）
    exact_int: bool,
}

fn norm_assign<L: Lang>(lang: &L, stmt: L::Id) -> Option<NormAssign<L>> {
    // 语句形态：ExprStmt{Assign} 或裸 Assign
    let a = match lang.kind(stmt) {
        NodeKind::ExprStmt => {
            let c = lang.children(stmt);
            if c.len() == 1 && lang.kind(c[0]) == NodeKind::Assign {
                c[0]
            } else {
                return None;
            }
        }
        NodeKind::Assign => stmt,
        _ => return None,
    };
    let ch = lang.children(a).to_vec();
    if ch.len() != 2 {
        return None;
    }
    let (target, rhs) = (ch[0], ch[1]);
    if lang.kind(target) != NodeKind::VarRef {
        return None;
    }
    let key = lang.var_key(target)?;
    let local = lang.is_local_var(target);
    let exact_int = lang.is_exact_int(target);
    // RHS 的求值效果：字面量 Pure；Binary(VarRef, lit) 是 Pure/MayRead——
    // 展开形态对 x 旧值的读正是我们要抵消的一部分（两句间的观察点
    // 为空），但 Binary 内若非纯形态拒绝
    match lang.assign_op(a) {
        // 复合形态：`x op= K`——RHS 就是操作数
        Some(op) => {
            let k = lang.literal(rhs).and_then(|x| x.as_int())?;
            Some(NormAssign { key, target, op, k, local, exact_int })
        }
        // 展开形态：`x = x op K`——RHS = Binary{op, VarRef(x), K}
        None => {
            if lang.kind(rhs) != NodeKind::Binary {
                return None;
            }
            let op = lang.bin_op(rhs)?;
            let rc = lang.children(rhs).to_vec();
            if rc.len() != 2 {
                return None;
            }
            // 左操作数必须是 x 自身（同一绑定）
            if lang.kind(rc[0]) != NodeKind::VarRef || !lang.same_binding(target, rc[0]) {
                return None;
            }
            let k = lang.literal(rc[1]).and_then(|x| x.as_int())?;
            Some(NormAssign { key, target, op, k, local, exact_int })
        }
    }
}

pub struct DeadStore;

impl<L: Lang> Rule<L> for DeadStore {
    fn name(&self) -> &'static str {
        "dead_store"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::VarDecl, NodeKind::Assign, NodeKind::ExprStmt]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk } = ctx;
        // 归一化第一条语句：VarDecl / Assign / ExprStmt{Assign}
        let (decl_form, first_assign) = match lang.kind(id) {
            NodeKind::VarDecl => {
                if lang.children(id).is_empty() {
                    return None; // 无 init 的声明不适用
                }
                (true, None)
            }
            NodeKind::Assign => (false, Some(id)),
            NodeKind::ExprStmt => {
                let ch = lang.children(id);
                if ch.len() == 1 && lang.kind(ch[0]) == NodeKind::Assign {
                    (false, Some(ch[0]))
                } else {
                    return None;
                }
            }
            _ => return None,
        };
        if decl_form && !lang.dead_decl_allowed(id) {
            // 声明形态的死存储删除同受语言侧约束（TS 的 let/const 删除
            // 改变 TDZ/词法遮蔽——annexB block-decl 家族抓获）
            return None;
        }
        let first_value = if decl_form {
            *lang.children(id).first()?
        } else {
            let a = first_assign?;
            if lang.assign_op(a).is_some() {
                return None; // 复合赋值有读语义
            }
            *lang.children(a).get(1)?
        };
        if lang.value_may_throw_unresolved(first_value) {
            return None;
        }

        // 父 Block + 相邻下一条语句
        let parent = walk.parent(id)?;
        if lang.kind(parent) != NodeKind::Block {
            return None;
        }
        let idx = walk.index(id)?;
        let stmts = lang.children(parent).to_vec();
        let next_stmt = *stmts.get(idx + 1)?;

        // 归一化第二条语句：Assign / ExprStmt{Assign}（简单赋值）；
        // 不是赋值 → 落入「零用途死存储」形态（见下）
        let second = match lang.kind(next_stmt) {
            NodeKind::Assign => Some(next_stmt),
            NodeKind::ExprStmt => {
                let ch = lang.children(next_stmt);
                if ch.len() == 1 && lang.kind(ch[0]) == NodeKind::Assign {
                    Some(ch[0])
                } else {
                    None
                }
            }
            _ => None,
        };
        let Some(second) = second else {
            // 零用途死存储：后继对该变量**无读且无写**（含遮蔽重声明），
            // 存入的值永不流出 → 整条删除。值效果 ≤ MayRead 才可丢弃。
            // （批量化后可达的形态：传播先消费了击杀写，init 残留为死值；
            //   旧逐编辑时序下由相邻对形态逐步吸收，未暴露此缺口。）
            // **作用域前提**：名字必须声明于本块（块级作用域，出块不可见）。
            // 外层声明的名字可能在本块之后被外层代码读取（if/while 分支里
            // 对外层变量的赋值就是典型）→ 扫描只覆盖本块尾部，必须拒绝。
            // （property.rs seed=29 的随机程序抓获此漏洞。）
            let (name_node, target_local) = if decl_form {
                (id, true)
            } else {
                let t = *lang.children(first_assign?).first()?;
                if lang.kind(t) != NodeKind::VarRef {
                    return None;
                }
                (t, lang.is_local_var(t))
            };
            // 赋值形态：目标必须是本局部（声明形态天然是局部）
            if !target_local {
                return None;
            }
            let name = lang.var_name(name_node)?.to_string();
            let declared_in_this_block = if decl_form {
                true // 检查对象就是本块的声明
            } else if parent == walk.root {
                // 方法体根块：尾部扫描覆盖到方法末尾——参数/局部变量的
                // 零后续事件即可判定为死（方法外不可见）
                true
            } else {
                // 嵌套块：名字必须声明于本块（外层声明的名字可能被
                // 本块之后的外层代码读取）
                stmts[..idx].iter().any(|&s| {
                    lang.kind(s) == NodeKind::VarDecl && lang.var_name(s) == Some(name.as_str())
                })
            };
            if !declared_in_this_block {
                return None;
            }
            // 块内**之前**的原文/类体可能捕获该名字并在之后调用时读
            //（`class A{constructor(){flag…}}; flag=true; new A()`——
            // 之后语句看不到类体里的读；Test262 TypedArray 家族抓获）
            for &s in &stmts[..idx] {
                if lang.stmt_has_opaque_region(s) {
                    return None;
                }
            }
            let name_key0 = lang.var_key(name_node)?;
            // Indexed: a statement that neither mentions the name (any
            // VarRef, including event-blind assignment targets, or a
            // shadowing declaration) nor contains opaque text cannot read,
            // write or shadow it. The recursive event scan remains the
            // fallback for unindexed nodes.
            for &s in &stmts[idx + 1..] {
                if lang.stmt_has_opaque_region(s) {
                    return None;
                }
                match (lang.stmt_mentions(s, name_key0), lang.stmt_opaque(s)) {
                    (Some(mentions), Some(opaque)) => {
                        if mentions || opaque {
                            return None;
                        }
                    }
                    _ => {
                        let mut w0 = Watch::<L>::new(name_key0, &[name_key0]);
                        scan_region(&*lang, s, &mut w0);
                        if w0.opaque
                            || !w0.uses.is_empty()
                            || w0.wrote(name_key0)
                            || w0.shadowed
                        {
                            return None;
                        }
                        if subtree_contains(&*lang, s, |n| {
                            lang.kind(n) == NodeKind::VarRef
                                && lang.var_name(n) == Some(name.as_str())
                        }) {
                            return None;
                        }
                    }
                }
            }
            if lang.effect(first_value) > Effect::MayRead {
                return None;
            }
            return Some(Edit::Delete { node: id });
        };
        if lang.assign_op(second).is_some() {
            return None;
        }
        let nch = lang.children(second);
        let (target, value) = (*nch.first()?, *nch.get(1)?);
        if lang.kind(target) != NodeKind::VarRef {
            return None;
        }

        // 变量一致性 + 局部性
        if decl_form {
            if !lang.same_binding(id, target) {
                return None;
            }
        } else {
            let a = first_assign?;
            let t = *lang.children(a).first()?;
            if lang.kind(t) != NodeKind::VarRef
                || !lang.same_binding(t, target)
                || !lang.is_local_var(t)
            {
                return None;
            }
        }
        // b 中不得引用 x（x = b 依赖旧值时不可删第一条）。
        // 含 Raw/原文区域时 AST 看不见其中的读（`r2 = yield r2`）——
        // 保守拒绝，否则会删掉前一条仍被读取的存储。
        if lang.stmt_has_opaque_region(value)
            || subtree_contains(&*lang, value, |n| {
                lang.kind(n) == NodeKind::VarRef && lang.same_binding(n, target)
            })
        {
            return None;
        }
        // a 的求值被丢弃：effect ≤ MayRead
        if lang.effect(first_value) > Effect::MayRead {
            return None;
        }

        if decl_form {
            // int x = a; x = b; → init 换成 b，删除第二条语句
            Some(Edit::Multi(vec![
                Edit::Replace {
                    target: first_value,
                    with: value,
                },
                Edit::Delete {
                    node: next_stmt,
                },
            ]))
        } else {
            // x = a; x = b; → 删除第一条语句（id 本身：Assign 或 ExprStmt）
            Some(Edit::Delete { node: id })
        }
    }
}

// ---------------------------------------------------------------------------
// 不可达语句删除（块内 return/throw/break/continue 之后的语句）。
// 语义安全（永不执行），但属于 DCE 范畴——**默认不启用**，经 all_rules() 选配。
// ---------------------------------------------------------------------------

/// 块展平：Block 的直接孩子是 Block 且**内层无任何声明**（VarDecl/
/// ForEach/Catch 绑定引入作用域）→ 内层语句上提。典型形态：try 剥壳、
/// CFF 还原后的裸嵌套块。无声明 ⇒ 无遮蔽 ⇒ 语义等价。
pub struct BlockFlatten;

impl<L: Lang> Rule<L> for BlockFlatten {
    fn name(&self) -> &'static str {
        "block_flatten"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Block]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk } = ctx;
        let parent = walk.parent(id)?;
        if lang.kind(parent) != NodeKind::Block {
            return None;
        }
        // 内层不得有任何声明（作用域引入）
        let inner = lang.children(id).to_vec();
        let declares = |n: L::Id| -> bool {
            matches!(
                lang.kind(n),
                NodeKind::VarDecl | NodeKind::ForEach | NodeKind::Catch
            )
        };
        if inner.iter().any(|&s| declares(s)) {
            return None;
        }
        // JS：直接孩子的函数声明同理（Annex B 提升/初始化时机改变）
        if lang.block_hoists_decl(id) {
            return None;
        }
        // 模式绑定流作用域（R15 三角化根因 1——BindingsTest1 抓获）：
        // instanceof 绑定（藏在 while/for/if 条件子树里，不在直接孩子的
        // declares 三类中）的作用域按 JLS 6.3 可越出语句到 enclosing
        // block——上提曾使两个匿名块内的 String s 绑定落进同一方法域
        //（javac「变量 s 重复定义」）
        if lang.has_pattern_bindings(id) {
            return None;
        }
        let idx = walk.index(id)?;
        Some(Edit::Splice {
            node: parent,
            index: idx,
            remove: 1,
            insert: inner,
        })
    }
}

/// 数值双重取负：-(-x) → x（补码回绕下恒等：-(-MIN) ≡ MIN ≡ x）；
/// -(-lit) → |lit|（解析器不折负字面量的外层负）。
pub struct DoubleNegFold;

impl<L: Lang> Rule<L> for DoubleNegFold {
    fn name(&self) -> &'static str {
        "double_neg_fold"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Unary]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, .. } = ctx;
        if lang.un_op(id) != Some(UnOp::Neg) {
            return None;
        }
        let inner = *lang.children(id).first()?;
        if lang.un_op(inner) == Some(UnOp::Neg) {
            let x = *lang.children(inner).first()?;
            // `-(-x)` 仅对精确整数恒等：布尔/字符串/null 会被 ToNumber
            //（`-(-true)` = 1 而不是 true）；浮点 -0 也会改变符号。
            if !lang.is_exact_int(x) {
                return None;
            }
            return Some(Edit::Replace {
                target: id,
                with: x,
            });
        }
        // -(-lit)：负数值字面量取外层负 → 正字面量（宽度溢出时语言侧
        // 拒绝——Java i32 的 `-(-2147483648)` 无法表示）
        if let Some((v, wide)) = lang.literal(inner).and_then(|l| match l {
            LitRef::Int(v) => Some((v, false)),
            LitRef::Long(v) => Some((v, true)),
            _ => None,
        }) {
            if v < 0 {
                let n = lang.fold_lit_neg(v, wide)?;
                let with = lang.build_int(n, wide);
                return Some(Edit::Replace {
                    target: id,
                    with,
                });
            }
        }
        None
    }
}

pub struct UnreachableAfterTerminal;

impl<L: Lang> Rule<L> for UnreachableAfterTerminal {
    fn name(&self) -> &'static str {
        "unreachable_after_terminal"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Block]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk: _ } = ctx;
        if lang.kind(id) != NodeKind::Block {
            return None;
        }
        let children = lang.children(id).to_vec();
        let terminal = |n: L::Id| -> bool {
            matches!(
                lang.kind(n),
                NodeKind::Return | NodeKind::Throw | NodeKind::Break | NodeKind::Continue
            )
        };
        let mut cut: Option<(usize, usize)> = None;
        for (i, &s) in children.iter().enumerate() {
            if terminal(s) && i + 1 < children.len() {
                cut = Some((i + 1, children.len() - i - 1));
                break;
            }
        }
        let (index, remove) = cut?;
        Some(Edit::Splice {
            node: id,
            index,
            remove,
            insert: Vec::new(),
        })
    }
}


// ---------------------------------------------------------------------------
// 比较常量折叠（反混淆核心）：1 < 2 → true、2 == 3 → false。
// 仅整数（含 long）与布尔字面量；浮点（NaN 语义）与字符串（引用比较）不折。
// 连锁：boolean b = 2 > 1; if (b) {...} → 不透明谓词被完全击穿。
// ---------------------------------------------------------------------------

pub struct CmpConstFold;

impl<L: Lang> Rule<L> for CmpConstFold {
    fn name(&self) -> &'static str {
        "cmp_const_fold"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Binary]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk: _ } = ctx;
        if lang.kind(id) != NodeKind::Binary {
            return None;
        }
        let op = lang.bin_op(id)?;
        if !op.is_comparison() {
            return None;
        }
        let ch = lang.children(id);
        let (l, r) = (*ch.first()?, *ch.get(1)?);
        // 双字面量才考虑折叠；可比较种类与比较语义由语言决定
        // （Java：整数/字符/布尔 Eq/Ne；字符串是引用比较 → 不折）
        lang.literal(l)?;
        lang.literal(r)?;
        let v = lang.fold_lit_cmp(op, l, r)?;
        let with = lang.build_bool(v);
        Some(Edit::Replace {
            target: id,
            with,
        })
    }
}

// ---------------------------------------------------------------------------
// 位运算恒等式（混淆器高频产物）：
//   x ^ 0 → x、0 ^ x → x、x | 0 → x、0 | x → x、x & -1 → x、-1 & x → x（x 保留，恒安全）
//   x | -1 → -1、x & 0 → 0、0 & x → 0（x 求值被丢弃 → effect ≤ MayRead）
//   x ^ x → 0（同名局部整型变量）
// ---------------------------------------------------------------------------

pub struct BitIdentity;

impl<L: Lang> Rule<L> for BitIdentity {
    fn name(&self) -> &'static str {
        "bit_identity"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Binary]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk: _ } = ctx;
        if lang.kind(id) != NodeKind::Binary {
            return None;
        }
        let op = lang.bin_op(id)?;
        if !matches!(op, BinOp::BitAnd | BinOp::BitOr | BinOp::BitXor) {
            return None;
        }
        let ch = lang.children(id);
        let (l, r) = (*ch.first()?, *ch.get(1)?);
        let int_lit = |n: L::Id| lang.literal(n).and_then(|x| x.as_int());

        // x ^ x → 0（同名局部变量）
        if op == BinOp::BitXor
            && lang.kind(l) == NodeKind::VarRef
            && lang.kind(r) == NodeKind::VarRef
            && lang.same_binding(l, r)
            && lang.is_local_var(l)
            && lang.is_exact_int(l)
        {
            let with = lang.build_int(0, false);
            return Some(Edit::Replace {
                target: id,
                with,
            });
        }

        // 字面量在任一侧（互换尝试）
        for (x, k) in [(l, r), (r, l)] {
            let Some(kv) = int_lit(k) else {
                continue;
            };
            if !lang.is_exact_int(x) {
                continue;
            }
            let identity = match (op, kv) {
                (BinOp::BitXor, 0) => Some(x),          // x ^ 0 → x
                (BinOp::BitOr, 0) => Some(x),           // x | 0 → x
                (BinOp::BitAnd, -1) => Some(x),         // x & -1 → x
                (BinOp::BitAnd, 0) | (BinOp::BitOr, -1) => {
                    // x & 0 → 0 / x | -1 → -1：丢弃 x 求值，需可丢弃
                    if lang.effect(x) > Effect::MayRead {
                        None
                    } else {
                        Some(k)
                    }
                }
                _ => None,
            };
            if let Some(with) = identity {
                return Some(Edit::Replace {
                    target: id,
                    with,
                });
            }
        }
        None
    }
}

// ---------------------------------------------------------------------------
// 算术/异或重结合（混淆器经典）：(x + K1) - K2 → x ± K、(x ^ K1) ^ K2 → x ^ K。
// 整数回绕算术下恒安全；K == 0 时直接得 x（(x + 5) - 5 → x、(x ^ 84) ^ 84 → x）。
// ---------------------------------------------------------------------------

pub struct ArithReassoc;

impl<L: Lang> Rule<L> for ArithReassoc {
    fn name(&self) -> &'static str {
        "arith_reassoc"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Binary]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk: _ } = ctx;
        if lang.kind(id) != NodeKind::Binary {
            return None;
        }
        let op2 = lang.bin_op(id)?;
        if !matches!(op2, BinOp::Add | BinOp::Sub | BinOp::BitXor) {
            return None;
        }
        let och = lang.children(id).to_vec();
        let inner = och[0];
        if lang.kind(inner) != NodeKind::Binary {
            return None;
        }
        let op1 = lang.bin_op(inner)?;
        let ich = lang.children(inner).to_vec();
        let x = ich[0];

        // 字符串拼接重结合：(x + "K1") + "K2" → x + "K1K2"
        // （拼接满足结合律且各操作数恰按序求值一次；x 任意类型）。
        // **整链折叠**：左倾链 (((x+"a")+"a")+"a")… 一刀折成 x+"aaa…"——
        // 逐层折每 pass 一层是 O(n²)（openjdk DeepStringConcat 32001 项
        // 78s/32k passes 抓获；整链后 1 编辑收敛）
        if op1 == BinOp::Add && op2 == BinOp::Add {
            if let (Some(LitRef::Str(k1)), Some(LitRef::Str(k2))) =
                (lang.literal(ich[1]), lang.literal(och[1]))
            {
                // 收集左倾链（外→内），要求每层 (child + "str") 同构
                let mut chain: Vec<L::Id> = vec![id];
                let mut cur = id;
                loop {
                    let cch = lang.children(cur).to_vec();
                    let next = cch[0];
                    let deeper = lang.kind(next) == NodeKind::Binary
                        && lang.bin_op(next) == Some(BinOp::Add)
                        && matches!(lang.literal(lang.children(next).to_vec()[1]),
                            Some(LitRef::Str(_)));
                    if deeper {
                        chain.push(next);
                        cur = next;
                    } else {
                        break;
                    }
                }
                // base = 最内层的左操作数；parts 内→外 = 源顺序
                let innermost = *chain.last().unwrap();
                let ich2 = lang.children(innermost).to_vec();
                let base = ich2[0];
                let mut joined = String::new();
                for &n in chain.iter().rev() {
                    if let Some(LitRef::Str(k)) =
                        lang.literal(lang.children(n).to_vec()[1])
                    {
                        joined.push_str(k);
                    }
                }
                let _ = (k1, k2); // 已并入 joined
                let lit = lang.build_str(&joined);
                let with = lang.build_bin(BinOp::Add, base, lit);
                return Some(Edit::Replace {
                    target: id,
                    with,
                });
            }
        }

        // 整数重结合：(x ± K1) ± K2 → x ± K、(x ^ K1) ^ K2 → x ^ K
        // 常量合并与宽度回绕语义在 Lang::reassoc_delta（Java：i32/i64
        // JLS 环绕——i32 合并溢出不回绕会产出非法字面量）
        let k1 = lang.literal(ich[1]).and_then(|x| x.as_int())?;
        let k2 = lang.literal(och[1]).and_then(|x| x.as_int())?;
        if !lang.is_exact_int(x) {
            return None;
        }
        let is_long = matches!(lang.literal(ich[1]), Some(LitRef::Long(_)))
            || matches!(lang.literal(och[1]), Some(LitRef::Long(_)));
        let with = match lang.reassoc_delta(op1, k1, op2, k2, is_long)? {
            ReassocOutcome::Neutral => x,
            ReassocOutcome::Lit(lit, op) => lang.build_bin(op, x, lit),
        };
        Some(Edit::Replace {
            target: id,
            with,
        })
    }
}

// ---------------------------------------------------------------------------
// if→三元归并（反编译器/混淆器经典形态）：
//   if (c) return a; else return b;    →  return c ? a : b;
//   if (c) return a; return b;（收尾） →  return c ? a : b;
// 恒安全：c 单次求值、a/b 条件求值在两种形态下一致（三元是惰性的）。
// 布尔特例由 ternary_bool 继续收敛成 return c / return !c。
// ---------------------------------------------------------------------------

pub struct IfToTernary;

impl<L: Lang> Rule<L> for IfToTernary {
    fn name(&self) -> &'static str {
        "if_to_ternary"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::If]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk } = ctx;
        if lang.kind(id) != NodeKind::If {
            return None;
        }
        let ch = lang.children(id).to_vec();
        let c = ch[0];

        // 提取"单 return 语句"分支：Block{Return} 或裸 Return
        let single_ret = |n: L::Id| -> Option<Option<L::Id>> {
            match lang.kind(n) {
                NodeKind::Return => Some(lang.children(n).first().copied()),
                NodeKind::Block if lang.children(n).len() == 1 => {
                    let inner = lang.children(n)[0];
                    if lang.kind(inner) == NodeKind::Return {
                        Some(lang.children(inner).first().copied())
                    } else {
                        None
                    }
                }
                _ => None,
            }
        };

        if ch.len() == 3 {
            // 完整 if-else：两支都是单 return
            let a = single_ret(ch[1])?;
            let b = single_ret(ch[2])?;
            let with = match (a, b) {
                (Some(va), Some(vb)) => {
                    let t = lang.build_ternary(c, va, vb);
                    lang.build_return(Some(t))
                }
                (None, None) => lang.build_return(None),
                _ => return None,
            };
            return Some(Edit::Replace {
                target: id,
                with,
            });
        }

        // if-then + 紧随的收尾 return（必须是父块最后一条语句）
        if ch.len() != 2 {
            return None;
        }
        let a = single_ret(ch[1])?;
        let parent = walk.parent(id)?;
        if lang.kind(parent) != NodeKind::Block {
            return None;
        }
        let idx = walk.index(id)?;
        let stmts = lang.children(parent).to_vec();
        if idx + 2 != stmts.len() {
            return None; // return 必须是最后一条
        }
        let tail = stmts[idx + 1];
        if lang.kind(tail) != NodeKind::Return {
            return None;
        }
        let b = lang.children(tail).first().copied();
        let with = match (a, b) {
            (Some(va), Some(vb)) => {
                let t = lang.build_ternary(c, va, vb);
                lang.build_return(Some(t))
            }
            (None, None) => lang.build_return(None),
            _ => return None,
        };
        Some(Edit::Splice {
            node: parent,
            index: idx,
            remove: 2,
            insert: vec![with],
        })
    }
}

// ---------------------------------------------------------------------------
// if→三元赋值归并：
//   if (c) { x = a; } else { x = b; }  →  x = c ? a : b;
// 恒安全（x 为变量名，两种形态各求值一次；a/b 条件求值一致）。
// ---------------------------------------------------------------------------

pub struct IfAssignTernary;

impl<L: Lang> Rule<L> for IfAssignTernary {
    fn name(&self) -> &'static str {
        "if_assign_ternary"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::If]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk: _ } = ctx;
        if lang.kind(id) != NodeKind::If {
            return None;
        }
        let ch = lang.children(id).to_vec();
        if ch.len() != 3 {
            return None;
        }
        let c = ch[0];
        // 单赋值分支：Block{Assign} / Block{ExprStmt{Assign}} / 裸（ExprStmt 包裹的）Assign
        let single_assign = |n: L::Id| -> Option<(L::Id, L::Id)> {
            let unwrap = |i: L::Id| -> L::Id {
                if lang.kind(i) == NodeKind::ExprStmt {
                    let ch = lang.children(i);
                    if ch.len() == 1 && lang.kind(ch[0]) == NodeKind::Assign {
                        return ch[0];
                    }
                }
                i
            };
            let inner = match lang.kind(n) {
                NodeKind::Assign => n,
                NodeKind::ExprStmt => unwrap(n),
                NodeKind::Block if lang.children(n).len() == 1 => unwrap(lang.children(n)[0]),
                _ => return None,
            };
            if lang.kind(inner) != NodeKind::Assign {
                return None;
            }
            if lang.assign_op(inner).is_some() {
                return None;
            }
            let a = lang.children(inner).to_vec();
            if lang.kind(a[0]) != NodeKind::VarRef {
                return None;
            }
            Some((a[0], a[1]))
        };
        let (t1, va) = single_assign(ch[1])?;
        let (t2, vb) = single_assign(ch[2])?;
        if !lang.same_binding(t1, t2) || lang.var_name(t1).is_none() {
            return None;
        }
        let t = lang.build_ternary(c, va, vb);
        let with = lang.build_assign(t1, t);
        Some(Edit::Replace {
            target: id,
            with,
        })
    }
}


// ---------------------------------------------------------------------------
// 声明-赋值合并（反编译器把声明与赋值拆开的形态）：
//   int x; x = 5;  →  int x = 5;
// 守卫：相邻同块、简单赋值、value 不引用 x（依赖旧值则非法）。
// 之后 local_propagation 可继续把 init 内联到唯一使用处。
// ---------------------------------------------------------------------------

pub struct DeclAssignMerge;

impl<L: Lang> Rule<L> for DeclAssignMerge {
    fn name(&self) -> &'static str {
        "decl_assign_merge"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::VarDecl]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk } = ctx;
        if lang.kind(id) != NodeKind::VarDecl || !lang.children(id).is_empty() {
            return None; // 必须是无 init 声明
        }
        let name = lang.var_name(id)?.to_string();
        let parent = walk.parent(id)?;
        if lang.kind(parent) != NodeKind::Block {
            return None;
        }
        let idx = walk.index(id)?;
        let stmts = lang.children(parent).to_vec();
        let next = *stmts.get(idx + 1)?;
        // 下一条：Assign(x, v) / ExprStmt{Assign(x, v)}（简单赋值）
        let assign = match lang.kind(next) {
            NodeKind::Assign => next,
            NodeKind::ExprStmt => {
                let ch = lang.children(next);
                if ch.len() == 1 && lang.kind(ch[0]) == NodeKind::Assign {
                    ch[0]
                } else {
                    return None;
                }
            }
            _ => return None,
        };
        if lang.assign_op(assign).is_some() {
            return None;
        }
        let ach = lang.children(assign).to_vec();
        let (target, value) = (ach[0], ach[1]);
        if lang.kind(target) != NodeKind::VarRef || lang.var_name(target) != Some(name.as_str()) {
            return None;
        }
        // value 不得引用 x（读旧值）
        if subtree_contains(&*lang, value, |n| {
            lang.kind(n) == NodeKind::VarRef && lang.var_name(n) == Some(name.as_str())
        }) {
            return None;
        }
        Some(Edit::Multi(vec![
            Edit::Splice {
                node: id,
                index: 0,
                remove: 0,
                insert: vec![value],
            },
            Edit::Delete { node: next },
        ]))
    }
}

// ---------------------------------------------------------------------------
// 赋值传播（寄存器拷贝消除，jadx/jcdc 产物）：
//   x = v; …唯一一次读 x（区间内无对 x / v 读集的写、无遮蔽）…
//   → 用 v 替换该次读，删除赋值。
// 安全锚点：x 必须在本块内有前置声明（赋值无作用域，块外读取会破坏语义）。
// v 纯/只读 → 自由移动（写冲突检查）；v 有副作用 → 相邻 + 直线 + 可读前缀。
// ---------------------------------------------------------------------------

pub struct AssignPropagation;

impl<L: Lang> Rule<L> for AssignPropagation {
    fn name(&self) -> &'static str {
        "assign_propagation"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Assign, NodeKind::ExprStmt]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk } = ctx;
        // 语句形态：Assign / ExprStmt{Assign}
        let (assign, stmt_node) = match lang.kind(id) {
            NodeKind::Assign => (id, id),
            NodeKind::ExprStmt => {
                let ch = lang.children(id);
                if ch.len() == 1 && lang.kind(ch[0]) == NodeKind::Assign {
                    (ch[0], id)
                } else {
                    return None;
                }
            }
            _ => return None,
        };
        if lang.assign_op(assign).is_some() {
            return None;
        }
        let ach = lang.children(assign).to_vec();
        let (target, value) = (ach[0], ach[1]);
        if lang.kind(target) != NodeKind::VarRef || !lang.is_local_var(target) {
            return None;
        }
        // 裸数组字面量只在声明/赋值 RHS 位置合法，传播会产出非法源码 → 拒绝
        if lang.kind(value) == NodeKind::ArrayLit {
            return None;
        }
        // lambda / 方法引用值同 ArrayLit：receiver 位非法 → 保守拒绝
        if matches!(lang.kind(value), NodeKind::Lambda | NodeKind::MethodRef) {
            return None;
        }
        let name = lang.var_name(target)?.to_string();

        let parent = walk.parent(stmt_node)?;
        if lang.kind(parent) != NodeKind::Block {
            return None;
        }
        let idx = walk.index(stmt_node)?;
        let stmts = lang.children(parent).to_vec();

        // 锚点：x 在本块 idx 之前有 VarDecl（保证 x 不会逃逸到块外）
        let anchored = stmts[..idx].iter().any(|&s| {
            lang.kind(s) == NodeKind::VarDecl && lang.var_name(s) == Some(name.as_str())
        });
        if !anchored {
            return None;
        }

        // value 不引用 x（x = x + 1 之类）
        if subtree_contains(&*lang, value, |n| {
            lang.kind(n) == NodeKind::VarRef && lang.var_name(n) == Some(name.as_str())
        }) {
            return None;
        }

        // 扫描赋值之后的区域（用途/遮蔽全区间；写冲突窗口见下）。
        // 句法引用计数兜底（同 local_propagation——bd.java 教训）
        let name_key = lang.var_key(target)?;
        if count_name_refs(&*lang, &stmts[idx + 1..], name_key, &name) != 1 {
            return None;
        }
        let mut wa = Watch::<L>::new(name_key, &[]);
        for &s in &stmts[idx + 1..] {
            if lang.stmt_has_opaque_region(s) {
                return None;
            }
            scan_region(&*lang, s, &mut wa);
        }
        if wa.opaque || wa.shadowed || wa.uses.len() != 1 {
            return None;
        }
        let use_id = wa.uses[0];
        // null 字面量：接收者位（AST 形态 Call→Member→[recv]，VarRef 的
        // 父是 Member）或静态调用 callee（防御）之外，**实参位**同样拒绝
        // ——`str2 = null; append(str2)` 合法（str2 声明类型钉住重载），
        // 内联成 `append(null)` 后 append(CharSequence)/append(char[])
        // 二义（ddc 输出 PurchasesOrchestrator 复现）。null_unsafe_position
        // 沿 Paren 链判位；Cast 包裹（(String) s）重建类型 → 放行
        if matches!(lang.literal(value), Some(LitRef::Null))
            && null_unsafe_position(&*lang, walk, use_id)
        {
            return None;
        }
        // VarRef 值不得内联进 cast/instanceof 类型判定位（声明类型即合法
        // 性依据——同 null 实参位的族）；声明类型未解析或与目标类型
        // 同名时恒可编译（var_cast_position_ok 钩子裁决）
        if lang.kind(value) == NodeKind::VarRef
            && typed_position(&*lang, walk, use_id)
                .is_some_and(|t| !lang.var_cast_position_ok(value, t))
        {
            return None;
        }
        // 写冲突兴趣集：value 读到的名字键 + x 自身（自身被写 → 覆盖，拒绝）
        let mut watch_keys: Vec<L::NameKey> = Vec::new();
        collect_read_keys(&*lang, value, &mut watch_keys);
        watch_keys.push(name_key);
        let mut wb = Watch::<L>::new(name_key, &watch_keys);
        for &s in &stmts[idx + 1..] {
            let uses_before = wb.uses.len();
            scan_region(&*lang, s, &mut wb);
            if uses_before < wb.uses.len() {
                break; // 使用语句之后的写不参与冲突判定
            }
        }
        // x 自身在窗口内被写 → 赋值会被覆盖，拒绝（先于目标写排除）
        if wb.wrote(name_key) {
            return None;
        }
        let ve = lang.effect(value);

        if ve <= Effect::MayRead {
            if let Some(target_key) = use_stmt_target_write_key(&*lang, &stmts[idx + 1..], use_id) {
                wb.clear_wrote(target_key);
            }
            if wb.any_write() {
                return None;
            }
            return Some(Edit::Multi(vec![
                Edit::Replace {
                    target: use_id,
                    with: value,
                },
                Edit::Delete { node: stmt_node },
            ]));
        }

        // 有副作用：相邻 + 直线 + 可读前缀（与 local_propagation 同判据）
        if idx + 1 >= stmts.len() {
            return None;
        }
        let stmt = stmts[idx + 1];
        if !subtree_contains(&*lang, stmt, |n| n == use_id) {
            return None;
        }
        match lang.kind(stmt) {
            NodeKind::Return
            | NodeKind::ExprStmt
            | NodeKind::Assign
            | NodeKind::Throw
            | NodeKind::VarDecl => {}
            _ => return None,
        }
        if !straight_path(&*lang, stmt, use_id) {
            return None;
        }
        if !prefix_effects_readable(&*lang, stmt, use_id) {
            return None;
        }
        Some(Edit::Multi(vec![
            Edit::Replace {
                target: use_id,
                with: value,
            },
            Edit::Delete { node: stmt_node },
        ]))
    }
}


/// 单一使用点若恰为 `target = use` 的 RHS，则该语句对 target 的写发生在
/// use 求值**之后**——从写冲突集中排除（否则 `int v = x + s; x = v;` 这类
/// 循环尾寄存器回拷永远无法内联）。
/// 排除目标必须≠被传播变量自身：`int i = 1; i = i;` 里排除 `i` 会删掉声明
/// 留下无绑定的赋值（该场景由 self_assign 处理）。
fn use_stmt_target_write_key<L: Lang>(
    lang: &L,
    stmts_after: &[L::Id],
    use_id: L::Id,
) -> Option<L::NameKey> {
    for &s in stmts_after {
        if subtree_contains(lang, s, |n| n == use_id) {
            let assign = match lang.kind(s) {
                NodeKind::Assign => s,
                NodeKind::ExprStmt => {
                    let ch = lang.children(s);
                    if ch.len() == 1 && lang.kind(ch[0]) == NodeKind::Assign {
                        ch[0]
                    } else {
                        return None;
                    }
                }
                _ => return None,
            };
            if lang.assign_op(assign).is_some() {
                return None;
            }
            let ch = lang.children(assign);
            if ch.len() == 2 && ch[1] == use_id && lang.kind(ch[0]) == NodeKind::VarRef {
                return lang.var_key(ch[0]);
            }
            return None;
        }
    }
    None
}

/// 收集 `id` 子树中 VarRef 读到的全部名字键（去重，线性）。
/// 替代旧 reads_vars 的 HashSet 版：value 的读名字典型 ≤10，线性查重更快。
fn collect_read_keys<L: Lang>(lang: &L, id: L::Id, out: &mut Vec<L::NameKey>) {
    let mut stack = vec![id];
    while let Some(n) = stack.pop() {
        if lang.kind(n) == NodeKind::VarRef {
            if let Some(k) = lang.var_key(n) {
                if !out.contains(&k) {
                    out.push(k);
                }
            }
        }
        for &c in lang.children(n) {
            stack.push(c);
        }
    }
}


// ---------------------------------------------------------------------------
// 尾部 continue 删除（反编译器产物）：
//   while (c) { …; continue; }  →  while (c) { … }
// 循环体最后一条无标签 continue 与顺序落入下一轮等价。恒安全。
// ---------------------------------------------------------------------------

pub struct TrailingContinue;

impl<L: Lang> Rule<L> for TrailingContinue {
    fn name(&self) -> &'static str {
        "trailing_continue"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::While, NodeKind::For, NodeKind::DoWhile, NodeKind::ForEach]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk: _ } = ctx;
        // While/For/DoWhile/ForEach 的 body 为 Block 且末语句为无标签 continue
        let body = match lang.kind(id) {
            NodeKind::While | NodeKind::For | NodeKind::DoWhile | NodeKind::ForEach => {
                *lang.children(id).last()?
            }
            _ => return None,
        };
        if lang.kind(body) != NodeKind::Block {
            return None;
        }
        let ch = lang.children(body).to_vec();
        let last = *ch.last()?;
        if lang.kind(last) != NodeKind::Continue || !lang.children(last).is_empty() {
            // JavaAST 的 Continue 标签在负载里，children 为空即无标签（引擎层约定）
            // 对带标签的语言实现：children[last] 非空即有标签 → 不动
            if lang.kind(last) != NodeKind::Continue {
                return None;
            }
        }
        Some(Edit::Delete { node: last })
    }
}

// ---------------------------------------------------------------------------
// 多用途拷贝传播（寄存器副本消除）：
//   x = y;（y 为简单 VarRef，x 有本块声明锚点，之后无对 x/y 的写、无遮蔽）
//   → 把 x 的**所有**后续读替换为 y，删除赋值。
// 守卫 y 不被写：替换后各使用点读的是 y 的当前值，须与原 x 的值一致。
// ---------------------------------------------------------------------------

pub struct MultiUseCopyPropagation;

impl<L: Lang> Rule<L> for MultiUseCopyPropagation {
    fn name(&self) -> &'static str {
        "multi_use_copy"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Assign, NodeKind::ExprStmt]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk } = ctx;
        // 语句形态：Assign(x, VarRef y) / ExprStmt{Assign(...)}
        let (assign, stmt_node) = match lang.kind(id) {
            NodeKind::Assign => (id, id),
            NodeKind::ExprStmt => {
                let ch = lang.children(id);
                if ch.len() == 1 && lang.kind(ch[0]) == NodeKind::Assign {
                    (ch[0], id)
                } else {
                    return None;
                }
            }
            _ => return None,
        };
        if lang.assign_op(assign).is_some() {
            return None;
        }
        let ach = lang.children(assign).to_vec();
        let (target, value) = (ach[0], ach[1]);
        if lang.kind(target) != NodeKind::VarRef || lang.kind(value) != NodeKind::VarRef {
            return None;
        }
        // lambda / 方法引用值：内联进 receiver 位非法（对抗波 2）→ 保守拒绝
        if matches!(lang.kind(value), NodeKind::Lambda | NodeKind::MethodRef) {
            return None;
        }
        if !lang.is_local_var(target) {
            return None;
        }
        let name = lang.var_name(target)?.to_string();
        let src = lang.var_name(value)?.to_string();
        if name == src {
            return None;
        }

        let parent = walk.parent(stmt_node)?;
        if lang.kind(parent) != NodeKind::Block {
            return None;
        }
        let idx = walk.index(stmt_node)?;
        let stmts = lang.children(parent).to_vec();

        // 锚点：x 在本块 idx 之前有 VarDecl
        let anchored = stmts[..idx].iter().any(|&s| {
            lang.kind(s) == NodeKind::VarDecl && lang.var_name(s) == Some(name.as_str())
        });
        if !anchored {
            return None;
        }

        // 扫描后续：收集 x 的全部读；对 x 的任何写 / 对 y 的任何写 / 遮蔽 → 拒绝
        let name_key = lang.var_key(target)?;
        let src_key = lang.var_key(value)?;
        let mut w1 = Watch::<L>::new(name_key, &[name_key, src_key]);
        for &s in &stmts[idx + 1..] {
            scan_region(&*lang, s, &mut w1);
        }
        if w1.opaque || w1.shadowed || w1.uses.is_empty() {
            return None;
        }
        if w1.wrote(name_key) || w1.wrote(src_key) {
            return None;
        }
        // 句法守卫：x 在赋值后的句法出现总数（含数组写 base——x[i]=v
        // 的 x 不产生事件，事件模型盲区）必须等于事件 uses 数。
        // 出现 > uses ⇒ 存在事件看不见的引用（bd.java 差分二轮抓获：
        // var10004 = var154 的读被替换、var10004[i]= 写目标残留 →
        // 读写裂脑 → 类初始化 AIOOBE）
        let synth = count_name_refs(&*lang, &stmts[idx + 1..], name_key, &name);
        if synth != w1.uses.len() {
            return None;
        }
        // y 也不得在【赋值前】与 x 指向不同值（x=y 之前 y 已是其所值，无需检查）；
        // 但 x=y 之间不能有对 y 的写（相邻语句，天然无中间）——赋值本身就是当前值 ✓

        // VarRef 值（y）不得内联进 cast/instanceof 类型判定位：被替换
        // 变量（x）的声明类型是这些位置可转换性的依据——`Object x =
        // k3Var; s = (String) x;` 合法，替换成 `(String) k3Var` 引用类型
        // 不相关即不可编译（ddc weixin dt/k 复现）。任一使用点命中 →
        // 整体拒绝（多用途替换不可拆单点）。
        if w1.uses.iter().any(|&u| {
            typed_position(&*lang, walk, u).is_some_and(|t| !lang.var_cast_position_ok(value, t))
        }) {
            return None;
        }

        let mut edits: Vec<Edit<L>> = w1.uses
            .iter()
            .map(|&u| Edit::Replace {
                target: u,
                with: value,
            })
            .collect();
        edits.push(Edit::Delete { node: stmt_node });
        Some(Edit::Multi(edits))
    }
}


// ---------------------------------------------------------------------------
// 尾部裸 return 删除（DAD/androguard 伪影）：
//   void 方法/构造器体最后一条无值 return 与自然结束等价，删除。
// 仅匹配根块（方法体）末语句，避免动到块中间的提前 return。
// ---------------------------------------------------------------------------

pub struct TrailingReturn;

impl<L: Lang> Rule<L> for TrailingReturn {
    fn name(&self) -> &'static str {
        "trailing_return"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Return]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let root = ctx.root();
        let parent = ctx.parent(id)?;
        let lang = ctx.lang;
        if lang.kind(id) != NodeKind::Return || !lang.children(id).is_empty() {
            return None;
        }
        if parent != root || lang.kind(parent) != NodeKind::Block {
            return None;
        }
        if lang.children(parent).last() != Some(&id) {
            return None;
        }
        Some(Edit::Delete { node: id })
    }
}


// ---------------------------------------------------------------------------
// 三元-布尔运算归并（反混淆产物）：
//   c ? a : false → c && a、c ? true : a → c || a（a 为布尔）
//   c 单次求值、a 条件求值，两种形态完全一致。
// ---------------------------------------------------------------------------

pub struct TernaryBoolOp;

impl<L: Lang> Rule<L> for TernaryBoolOp {
    fn name(&self) -> &'static str {
        "ternary_bool_op"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::Ternary]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk: _ } = ctx;
        if lang.kind(id) != NodeKind::Ternary {
            return None;
        }
        let ch = lang.children(id);
        let (c, a, b) = (*ch.first()?, *ch.get(1)?, *ch.get(2)?);
        // c 必须是语法级布尔：`1 ? true : false` 得 true，而 `1 || false`
        // 得 1（truthy 非布尔条件改变结果类型）。随机差分 case 2074 抓获。
        if !lang.is_bool(c) {
            return None;
        }
        let lb = |n: L::Id| matches!(lang.literal(n), Some(LitRef::Bool(v)) if v);
        let is_false = |n: L::Id| matches!(lang.literal(n), Some(LitRef::Bool(v)) if !v);
        let with = if is_false(b) && lang.is_bool(a) {
            lang.build_bin(BinOp::And, c, a)
        } else if lb(b) && lang.is_bool(a) {
            // c ? a : true ≡ !c || a —— 少见，跳过保持保守
            return None;
        } else if lb(a) && lang.is_bool(b) {
            // c ? true : b ≡ c || b
            lang.build_bin(BinOp::Or, c, b)
        } else if is_false(a) && lang.is_bool(b) {
            // c ? false : b ≡ !c && b —— 少见，跳过保持保守
            return None;
        } else {
            return None;
        };
        Some(Edit::Replace {
            target: id,
            with,
        })
    }
}


// ---------------------------------------------------------------------------
// 远距存储消除（寄存器预声明清理，ddc/jcdc 产物）：
//   int x = 0; …（无 x 事件）…; x = 1;   →  int x = 1;（字面量提升）
//   int x = 0; …（无 x 事件）…; x = v;   →  int x;（init 死亡，剥除）
//   x = 0; …（无 x 事件）…; x = v;       →  删除第一条（值 ≤ MayRead）
// 守卫：首个 x 事件必须是写；语句内 x 出现若非简单赋值目标即视为读（阻断）。
// ---------------------------------------------------------------------------

pub struct StoreKill;

/// 扫描一个节点在 `name` 上的事件。
/// `dominating`：当前子树是否在必经路径上（If/While 的 cond、Ternary 的
/// cond、For 的 init/cond、非短路二元两侧 = true；分支体/循环体 = false）。
///
/// 语言侧若提供语句级预计算索引（[`Lang::stmt_kill_event`]），调用方
/// 应优先查询；本函数是未索引节点的回退路径。
pub fn scan_kill_event<L: Lang>(
    lang: &L,
    node: L::Id,
    name: &str,
    dominating: bool,
) -> StmtKill<L::Id> {
    match lang.kind(node) {
        NodeKind::VarRef => {
            if lang.var_name(node) == Some(name) {
                StmtKill::Read
            } else {
                StmtKill::None
            }
        }
        NodeKind::VarDecl => {
            if lang.var_name(node) == Some(name) {
                StmtKill::Shadow
            } else {
                // 其他变量声明：init 必经求值
                for &c in lang.children(node) {
                    let e = scan_kill_event(lang, c, name, dominating);
                    if !matches!(e, StmtKill::None) {
                        return e;
                    }
                }
                StmtKill::None
            }
        }
        NodeKind::Assign => {
            let ch = lang.children(node);
            // RHS 先求值
            if let Some(&val) = ch.get(1) {
                let e = scan_kill_event(lang, val, name, dominating);
                if matches!(e, StmtKill::Read | StmtKill::Shadow) {
                    return e;
                }
            }
            let Some(&target) = ch.first() else {
                return StmtKill::None;
            };
            if lang.kind(target) == NodeKind::VarRef {
                if lang.assign_op(node).is_some() {
                    // 复合赋值 `x += 1` 读旧值
                    if lang.var_name(target) == Some(name) {
                        return StmtKill::Read;
                    }
                } else if lang.var_name(target) == Some(name) {
                    if dominating {
                        return StmtKill::DomWrite(node);
                    }
                    return StmtKill::None; // 条件写：不击杀，也不算读
                }
            } else {
                // 成员/索引目标：基座与键是读（`a[i] = v` 读 a、i）
                let e = scan_kill_event(lang, target, name, false);
                if matches!(e, StmtKill::Read | StmtKill::Shadow) {
                    return e;
                }
            }
            StmtKill::None
        }
        NodeKind::If | NodeKind::While | NodeKind::Ternary => {
            // cond 必经；分支体/循环体不必然——但其中的【读】仍保活 init，
            // 必须扫描（以非支配标记：读传播，写不击杀）
            let ch = lang.children(node);
            if let Some(&c) = ch.first() {
                let e = scan_kill_event(lang, c, name, true);
                if !matches!(e, StmtKill::None) {
                    return e;
                }
            }
            for &c in &ch[1..] {
                let e = scan_kill_event(lang, c, name, false);
                if matches!(e, StmtKill::Read | StmtKill::Shadow) {
                    return e;
                }
            }
            StmtKill::None
        }
        NodeKind::Binary => {
            let Some(op) = lang.bin_op(node) else {
                return StmtKill::None;
            };
            let ch = lang.children(node);
            if op.is_short_circuit() {
                // 左支必经；右支条件求值（读→保活；写→不击杀）
                if let Some(&l) = ch.first() {
                    let e = scan_kill_event(lang, l, name, true);
                    if !matches!(e, StmtKill::None) {
                        return e;
                    }
                }
                if let Some(&r) = ch.get(1) {
                    let e = scan_kill_event(lang, r, name, false);
                    if matches!(e, StmtKill::Read | StmtKill::Shadow) {
                        return e;
                    }
                }
                StmtKill::None
            } else {
                for &c in ch {
                    let e = scan_kill_event(lang, c, name, dominating);
                    if !matches!(e, StmtKill::None) {
                        return e;
                    }
                }
                StmtKill::None
            }
        }
        NodeKind::For | NodeKind::ForEach | NodeKind::Switch | NodeKind::Try => {
            // 必经部分（For 的 inits+cond / ForEach 的 iterable / Switch 的
            // subject）以支配标记扫；其余（循环体/case 体/try 块）以非支配
            // 标记扫——其中的读仍保活
            let ch = lang.children(node);
            let dom_len = match lang.kind(node) {
                NodeKind::For => ch.len().saturating_sub(2),
                NodeKind::ForEach | NodeKind::Switch => 1,
                _ => 0,
            };
            for &c in &ch[..dom_len] {
                let e = scan_kill_event(lang, c, name, true);
                if !matches!(e, StmtKill::None) {
                    return e;
                }
            }
            for &c in &ch[dom_len..] {
                let e = scan_kill_event(lang, c, name, false);
                if matches!(e, StmtKill::Read | StmtKill::Shadow) {
                    return e;
                }
            }
            StmtKill::None
        }
        _ => {
            for &c in lang.children(node) {
                let e = scan_kill_event(lang, c, name, dominating);
                if !matches!(e, StmtKill::None) {
                    return e;
                }
            }
            StmtKill::None
        }
    }
}

/// One DFS of a statement computing the dominating=true kill event for
/// every name that has one. Recursion order matches [`scan_kill_event`];
/// the first event per name wins. Used by language crates to build a
/// per-statement index so StoreKill does not rescan subtrees per candidate.
pub fn collect_stmt_kill_events<L: Lang>(lang: &L, root: L::Id) -> Vec<(L::NameKey, StmtKill<L::Id>)> {
    use std::collections::BTreeMap;
    enum Act<Id> {
        Enter(Id, bool),
        AfterAssign(Id, bool),
    }
    let mut events: BTreeMap<L::NameKey, StmtKill<L::Id>> = BTreeMap::new();
    let mut stack = vec![Act::Enter(root, true)];
    while let Some(act) = stack.pop() {
        match act {
            Act::AfterAssign(id, dominating) => {
                if lang.assign_op(id).is_some() {
                    continue;
                }
                let Some(&target) = lang.children(id).first() else {
                    continue;
                };
                if lang.kind(target) != NodeKind::VarRef {
                    continue;
                }
                let Some(key) = lang.var_key(target) else {
                    continue;
                };
                if dominating {
                    events.entry(key).or_insert(StmtKill::DomWrite(id));
                }
            }
            Act::Enter(node, dominating) => match lang.kind(node) {
                NodeKind::VarRef => {
                    if let Some(key) = lang.var_key(node) {
                        events.entry(key).or_insert(StmtKill::Read);
                    }
                }
                NodeKind::VarDecl => {
                    if let Some(key) = lang.var_key(node) {
                        events.entry(key).or_insert(StmtKill::Shadow);
                    }
                    for &c in lang.children(node).iter().rev() {
                        stack.push(Act::Enter(c, dominating));
                    }
                }
                NodeKind::Assign => {
                    let ch = lang.children(node);
                    stack.push(Act::AfterAssign(node, dominating));
                    if let Some(&val) = ch.get(1) {
                        stack.push(Act::Enter(val, dominating));
                    }
                    // 目标是 VarRef：简单赋值是纯写（不产生 Use），
                    // 复合赋值要读旧值；成员/索引目标整体按读扫描。
                    if let Some(&target) = ch.first() {
                        if lang.kind(target) != NodeKind::VarRef
                            || lang.assign_op(node).is_some()
                        {
                            stack.push(Act::Enter(target, false));
                        }
                    }
                }
                NodeKind::If | NodeKind::While | NodeKind::Ternary => {
                    let ch = lang.children(node);
                    for (i, &c) in ch.iter().enumerate().rev() {
                        stack.push(Act::Enter(c, i == 0));
                    }
                }
                NodeKind::Binary => {
                    let ch = lang.children(node);
                    let short = lang.bin_op(node).is_some_and(|op| op.is_short_circuit());
                    for (i, &c) in ch.iter().enumerate().rev() {
                        let child_dom = if short { i == 0 } else { dominating };
                        stack.push(Act::Enter(c, child_dom));
                    }
                }
                NodeKind::For | NodeKind::ForEach | NodeKind::Switch | NodeKind::Try => {
                    let ch = lang.children(node);
                    let dom_len = match lang.kind(node) {
                        NodeKind::For => ch.len().saturating_sub(2),
                        NodeKind::ForEach | NodeKind::Switch => 1,
                        _ => 0,
                    };
                    for (i, &c) in ch.iter().enumerate().rev() {
                        stack.push(Act::Enter(c, i < dom_len));
                    }
                }
                _ => {
                    for &c in lang.children(node).iter().rev() {
                        stack.push(Act::Enter(c, dominating));
                    }
                }
            },
        }
    }
    events.into_iter().collect()
}

impl<L: Lang> Rule<L> for StoreKill {
    fn name(&self) -> &'static str {
        "store_kill"
    }
    fn kinds(&self) -> &'static [NodeKind] {
        &[NodeKind::VarDecl, NodeKind::Assign, NodeKind::ExprStmt]
    }
    fn check(&self, ctx: RewriteCtx<'_, L>, id: L::Id) -> Option<Edit<L>> {
        let RewriteCtx { lang, walk } = ctx;
        // 形态：VarDecl(带 init) 或 Assign（裸/ExprStmt 包裹）
        let (is_decl, stmt_node, first_value) = match lang.kind(id) {
            NodeKind::VarDecl => {
                if lang.children(id).is_empty() {
                    return None;
                }
                (true, id, *lang.children(id).first()?)
            }
            NodeKind::Assign => {
                if lang.assign_op(id).is_some() {
                    return None;
                }
                (false, id, *lang.children(id).get(1)?)
            }
            NodeKind::ExprStmt => {
                let ch = lang.children(id);
                if ch.len() == 1 && lang.kind(ch[0]) == NodeKind::Assign {
                    let a = ch[0];
                    if lang.assign_op(a).is_some() {
                        return None;
                    }
                    (false, id, *lang.children(a).get(1)?)
                } else {
                    return None;
                }
            }
            _ => return None,
        };
        let (name, name_key) = if is_decl {
            (lang.var_name(id)?.to_string(), lang.var_key(id)?)
        } else {
            let a = match lang.kind(id) {
                NodeKind::Assign => id,
                _ => lang.children(id)[0],
            };
            let t = *lang.children(a).first()?;
            if lang.kind(t) != NodeKind::VarRef || !lang.is_local_var(t) {
                return None;
            }
            (lang.var_name(t)?.to_string(), lang.var_key(t)?)
        };

        let parent = walk.parent(stmt_node)?;
        if lang.kind(parent) != NodeKind::Block {
            return None;
        }
        let idx = walk.index(stmt_node)?;
        let stmts = lang.children(parent).to_vec();

        // 扫描：读（任意路径）→ 保活；支配写 → 击杀；条件写 → 继续
        let mut killed_by: Option<(usize, L::Id)> = None;
        for (si, &s) in stmts.iter().enumerate().skip(idx + 1) {
            // 原文/类体等不可见区域可能读本变量——击杀不得跨越
            //（Test262 TypedArray：class 构造器读 flag，首个 store 被杀，
            // flag 变成声明即 true）
            if lang.stmt_has_opaque_region(s) {
                return None;
            }
            let event = lang
                .stmt_kill_event(s, name_key)
                .unwrap_or_else(|| scan_kill_event(&*lang, s, &name, true));
            match event {
                StmtKill::None => continue,
                StmtKill::Read | StmtKill::Shadow => return None,
                StmtKill::DomWrite(assign) => {
                    killed_by = Some((si, assign));
                    break;
                }
            }
        }
        let (si, kill_assign) = killed_by?;
        // 击杀语句级守卫：击杀赋值必须就是**顶层语句本身**（或其
        // ExprStmt 包裹）。嵌套击杀（do-while 体/裸块/for-init 里的写）
        // 曾按 stmts[si] 删整条外层语句——连兄弟副作用一起删
        //（RealTimeSequencer.pump() 的 133 行 do-while 曾被整删成
        // 3 行；ConstantPool.writeTags 双 for+switch 全删——JDK 语料
        // ≥9 文件）。同语句内击杀点之后的读也无法用语句级窗口证明
        // 不可达 → 保守拒绝嵌套击杀（顶层击杀不受影响）
        {
            let kill_container = match walk.parents.get(&kill_assign) {
                Some(&(p, _)) if lang.kind(p) == NodeKind::ExprStmt => p,
                _ => kill_assign,
            };
            if kill_container != stmts[si] {
                return None;
            }
        }
        // 异常窗口守卫：被击杀点到击杀点之间的语句若可能抛（效果 >
        // MayRead），异常路径会绕过击杀点直达 finally/catch——后者可能
        // 在「中间值」状态读到本变量（finally-flag 恢复模式：
        // v=true; setSoTimeout(1); v=false; finally{if(v)…}——差分
        // 抓获：v=true 曾被杀，异常路径的恢复逻辑变死代码）。
        //
        // 精化：异常只可能被**包围本块的 Try** 的 handler 截获；无包围
        // Try 时异常直接出方法，局部变量对外不可见——中间值无人可观
        // 察，窗口内的不纯语句不构成障碍（ddc 0.1.33 形态：声明序列 →
        // new 对象 → 击杀赋值，构造调用恰在窗口内）。窗口内嵌套 try
        // 的 handler 读已由 per-statement 事件扫描以 Read 击退，不在
        // 此列。
        let mut has_try_ancestor = false;
        {
            let mut cur = stmt_node;
            while let Some(&(p, _)) = walk.parents.get(&cur) {
                if lang.kind(p) == NodeKind::Try {
                    has_try_ancestor = true;
                    break;
                }
                cur = p;
            }
        }
        if has_try_ancestor {
            for &s in &stmts[idx + 1..si] {
                if lang.effect(s) > Effect::MayRead {
                    return None;
                }
            }
        }

        if is_decl {
            // 击杀赋值的值是字面量 → 提升进声明 init
            // （原 init 被替换丢弃——同样必须可丢弃，副作用调用不得删！）
            if lang.effect(first_value) > Effect::MayRead {
                return None;
            }
            let value = *lang.children(kill_assign).get(1)?;
            if is_literal(lang, value) {
                let assign_stmt = stmts[si];
                return Some(Edit::Multi(vec![
                    Edit::Splice {
                        node: id,
                        index: 0,
                        remove: 1,
                        insert: vec![value],
                    },
                    Edit::Delete { node: assign_stmt },
                ]));
            }
            // 否则：剥除 init（裸声明）——init 必须可丢弃（副作用调用不得删！）。
            // 推断类型声明（Java `var`）剥 init 后非法——拒绝
            //（obf_fuzz 差分 seed 4 抓获：`var flag;` 编译失败）
            if lang.is_inferred_decl(id) {
                return None;
            }
            if lang.effect(first_value) > Effect::MayRead {
                return None;
            }
            return Some(Edit::Splice {
                node: id,
                index: 0,
                remove: 1,
                insert: Vec::new(),
            });
        }
        // 赋值形态：首值必须可丢弃
        if lang.effect(first_value) > Effect::MayRead {
            return None;
        }
        Some(Edit::Delete { node: stmt_node })
    }
}

/// value 是否为字面量（提升安全：无求值位置问题）。
fn is_literal<L: Lang>(lang: &L, n: L::Id) -> bool {
    matches!(lang.kind(n), NodeKind::Literal)
}

// ---------------------------------------------------------------------------
// 注册表
// ---------------------------------------------------------------------------

/// 默认规则（保守集：不做 DCE 类清理）。
pub fn default_rules<L: Lang>() -> Vec<Box<dyn Rule<L>>> {
    vec![
        Box::new(DoubleNegFold),
        Box::new(BlockFlatten),
        Box::new(ParenRemoval),
        Box::new(ConstCondition),
        Box::new(BooleanReturn),
        Box::new(IfToTernary),
        Box::new(IfAssignTernary),
        Box::new(IfElseEmpty),
        Box::new(BoolCompare),
        Box::new(DoubleNot),
        Box::new(BoolNotFold),
        Box::new(BoolShortCircuit),
        Box::new(NotCompare),
        Box::new(TernaryFold),
        Box::new(TernaryBool),
        Box::new(TernaryBoolOp),
        Box::new(ConstFoldBin),
        Box::new(CmpConstFold),
        Box::new(BitIdentity),
        Box::new(ArithReassoc),
        Box::new(SelfAssign),
        Box::new(ArithIdentity),
        Box::new(ArithZero),
        Box::new(DeclAssignMerge),
        Box::new(LocalPropagation),
        Box::new(AssignPropagation),
        Box::new(MultiUseCopyPropagation),
        Box::new(TrailingReturn),
        Box::new(InverseAssignPair),
        Box::new(DeadDecl),
        Box::new(AssignBackFold),
        Box::new(DeadStore),
        Box::new(StoreKill),
    ]
}

/// 全量规则（含 DCE 类选配项；按设计文档"可选配，默认不启用"）。
pub fn all_rules<L: Lang>() -> Vec<Box<dyn Rule<L>>> {
    let mut rules = default_rules();
    rules.push(Box::new(UnreachableAfterTerminal));
    rules
}