submilli_engine/

lower_patterns.rs

//! Pre-infer pass eliminating destructuring patterns from [`Ast`].
//! Runs between [`crate::parse`] and [`crate::infer`]; downstream sees only plain
//! `Let`/`Const`/`ObjectRest`/`ForOf` and `ParamDecl` with `pattern: None`.
//! Synthesised array-pattern `IndexAccess` nodes are tagged in [`Ast::pattern_origins`]
//! so the typechecker can rephrase index errors as destructure-specific diagnostics.

use std::collections::{BTreeMap, BTreeSet};

use crate::compiler_error::{CompilerFailure, CompilerStage};
use crate::tree_height;

use crate::{
    Ast, Binding, BindingKind, ClassMember, Expr, ExprId, ExprKind, Ident, ParamDecl,
    PatternOrigin, Span, Stmt, StmtId, StmtKind,
};

pub fn lower(mut ast: Ast) -> Result<Ast, CompilerFailure> {
    // Callers may supply an AST that did not come from `parse_checked`. Pattern
    // lowering reports its failures in the inference stage it feeds.
    tree_height::check_syntax(&ast).map_err(|failure| failure.with_stage(CompilerStage::Infer))?;
    let mut ctx = LowerCtx { next_tmp: 0 };
    ctx.lower_arrows(&mut ast)?;
    ctx.lower_function_params(&mut ast)?;
    ctx.lower_class_member_params(&mut ast)?;
    ctx.lower_for_of_patterns(&mut ast)?;
    ctx.lower_for_pattern_inits(&mut ast)?;
    ctx.lower_pattern_stmts(&mut ast)?;
    Ok(ast)
}

/// Starts every `fresh` name, so none can collide with a source identifier.
const FRESH_MARKER: &str = "#pattern_";

/// The `fresh` prefix of a lowered destructured parameter.
const PATTERN_PARAM: &str = "p";

/// Whether `name` stands in for a destructured parameter, which has no source
/// name: either the fresh `#pattern_p_N` from lowering, or the parser's empty
/// placeholder on an interface method signature, which has no body to lower.
pub(crate) fn is_pattern_param(name: &str) -> bool {
    name.is_empty()
        || name
            .strip_prefix(FRESH_MARKER)
            .and_then(|rest| rest.strip_prefix(PATTERN_PARAM))
            .is_some_and(|rest| rest.starts_with('_'))
}

/// The `fresh` prefix of the value a destructuring declaration unpacks.
const PATTERN_VALUE: &str = "dst";

/// Whether `name` is the fresh `#pattern_dst_N` that holds the value a
/// destructuring declaration unpacks.
pub(crate) fn is_pattern_value(name: &str) -> bool {
    name.strip_prefix(FRESH_MARKER)
        .and_then(|rest| rest.strip_prefix(PATTERN_VALUE))
        .is_some_and(|rest| rest.starts_with('_'))
}

struct LowerCtx {
    next_tmp: u32,
}

/// What a pattern's bindings inherit from the declaration that holds it.
#[derive(Clone, Copy)]
struct PatternRoot {
    is_const: bool,
    /// Whether the declaration annotates the pattern's type, which its
    /// defaults must then fit.
    annotated: bool,
}

impl LowerCtx {
    fn fresh(&mut self, prefix: &str, span: Span) -> Result<Ident, CompilerFailure> {
        let n = self.next_tmp;
        self.next_tmp = self
            .next_tmp
            .checked_add(1)
            .ok_or_else(|| CompilerFailure::Limit {
                stage: CompilerStage::Infer,
                span: None,
                message: "pattern temporary capacity exceeded".into(),
                help: vec![],
            })?;
        Ok(Ident {
            // Distinct from source identifiers and the later desugar pass.
            name: format!("{FRESH_MARKER}{prefix}_{n}"),
            span,
        })
    }

    fn lower_arrows(&mut self, ast: &mut Ast) -> Result<(), CompilerFailure> {
        for id in ast
            .expr_ids()
            .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
        {
            let has_pattern = match &ast
                .try_expr(id)
                .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
                .kind
            {
                ExprKind::Arrow { params, .. } => params.iter().any(|p| p.pattern.is_some()),
                _ => false,
            };
            if has_pattern {
                self.lower_arrow(ast, id)?;
            }
        }

        Ok(())
    }

    fn lower_arrow(&mut self, ast: &mut Ast, id: ExprId) -> Result<(), CompilerFailure> {
        // Clone the kind to release the read borrow before we start
        // mutating other arena slots.
        let kind = ast
            .try_expr(id)
            .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
            .kind
            .clone();
        let ExprKind::Arrow {
            mut params,
            return_type,
            type_predicate,
            body,
        } = kind
        else {
            return Err(lowering_failure(
                "unexpected node kind during pattern lowering",
            ));
        };

        self.materialise_pattern_params(ast, &mut params)?;

        ast.try_expr_mut(id)
            .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
            .kind = ExprKind::Arrow {
            params,
            return_type,
            type_predicate,
            body,
        };

        Ok(())
    }

    fn lower_function_params(&mut self, ast: &mut Ast) -> Result<(), CompilerFailure> {
        for id in ast
            .stmt_ids()
            .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
        {
            let has_pattern = match &ast
                .try_stmt(id)
                .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
                .kind
            {
                StmtKind::Function { params, .. } => params.iter().any(|p| p.pattern.is_some()),
                _ => false,
            };
            if has_pattern {
                self.lower_function(ast, id)?;
            }
        }

        Ok(())
    }

    fn lower_function(&mut self, ast: &mut Ast, id: StmtId) -> Result<(), CompilerFailure> {
        let stmt_span = ast
            .try_stmt(id)
            .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
            .span;
        let kind = std::mem::replace(
            &mut ast
                .try_stmt_mut(id)
                .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
                .kind,
            StmtKind::Block(Vec::new()), // sentinel, restored before return
        );
        let StmtKind::Function {
            name,
            generics,
            mut params,
            return_type,
            type_predicate,
            body,
            doc,
        } = kind
        else {
            return Err(lowering_failure(
                "unexpected node kind during pattern lowering",
            ));
        };

        self.materialise_pattern_params(ast, &mut params)?;

        ast.try_stmt_mut(id)
            .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
            .kind = StmtKind::Function {
            name,
            generics,
            params,
            return_type,
            type_predicate,
            body,
            doc,
        };
        // Span is unchanged, but reassert for the linter.
        let _ = stmt_span;

        Ok(())
    }

    fn lower_class_member_params(&mut self, ast: &mut Ast) -> Result<(), CompilerFailure> {
        for id in ast
            .stmt_ids()
            .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
        {
            let StmtKind::ClassDecl { members, .. } = &mut ast
                .try_stmt_mut(id)
                .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
                .kind
            else {
                continue;
            };
            let mut taken = std::mem::take(members);
            let lowered = self.lower_member_params(ast, &mut taken);
            // Restore the members before reporting a failure, so the class
            // keeps its shape either way.
            let StmtKind::ClassDecl { members, .. } = &mut ast
                .try_stmt_mut(id)
                .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
                .kind
            else {
                return Err(lowering_failure(
                    "unexpected node kind during pattern lowering",
                ));
            };
            *members = taken;
            lowered?;
        }

        Ok(())
    }

    fn lower_member_params(
        &mut self,
        ast: &mut Ast,
        members: &mut [ClassMember],
    ) -> Result<(), CompilerFailure> {
        for member in members {
            let params = match member {
                ClassMember::Method { params, .. } | ClassMember::Constructor { params, .. } => {
                    params.as_mut_slice()
                }
                ClassMember::Accessor {
                    param: Some(param), ..
                } => std::slice::from_mut(param.as_mut()),
                ClassMember::Accessor { param: None, .. } | ClassMember::Field { .. } => continue,
            };
            self.materialise_pattern_params(ast, params)?;
        }

        Ok(())
    }

    fn lower_for_of_patterns(&mut self, ast: &mut Ast) -> Result<(), CompilerFailure> {
        for id in ast
            .stmt_ids()
            .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
        {
            if !matches!(
                ast.try_stmt(id)
                    .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
                    .kind,
                StmtKind::ForOfPattern { .. }
            ) {
                continue;
            }
            self.expand_for_of_pattern(ast, id)?;
        }

        Ok(())
    }

    fn expand_for_of_pattern(&mut self, ast: &mut Ast, id: StmtId) -> Result<(), CompilerFailure> {
        let stmt_span = ast
            .try_stmt(id)
            .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
            .span;
        let kind = std::mem::replace(
            &mut ast
                .try_stmt_mut(id)
                .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
                .kind,
            StmtKind::Block(Vec::new()),
        );
        let StmtKind::ForOfPattern {
            binding_kind,
            binding,
            ty,
            iter,
            body,
        } = kind
        else {
            return Err(lowering_failure(
                "unexpected node kind during pattern lowering",
            ));
        };

        let pattern_span = binding.span();
        let dst = self.fresh(PATTERN_VALUE, pattern_span)?;
        let is_const = matches!(binding_kind, BindingKind::Const);
        let root = PatternRoot {
            is_const,
            annotated: ty.is_some(),
        };
        let mut decompose = self.emit_decompose(
            ast,
            binding,
            dst.clone(),
            root,
            /*doc=*/ None,
            /*literal=*/ None,
        )?;
        let mut source_bindings = Vec::new();
        for &id in &decompose {
            match &ast
                .try_stmt(id)
                .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
                .kind
            {
                StmtKind::Let { name, .. }
                | StmtKind::Const { name, .. }
                | StmtKind::ObjectRest { name, .. } => source_bindings.push(name.clone()),
                _ => {}
            }
        }
        ast.for_of_pattern_bindings.insert(id, source_bindings);
        // The loop head and the user's body are distinct lexical scopes.
        decompose.push(body);
        let body = ast
            .try_push_stmt(Stmt {
                kind: StmtKind::Block(decompose),
                span: ast
                    .try_stmt(body)
                    .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
                    .span,
            })
            .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?;

        ast.try_stmt_mut(id)
            .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
            .kind = StmtKind::ForOf {
            binding_kind,
            name: dst,
            ty,
            iter,
            body,
        };
        let _ = stmt_span;

        Ok(())
    }

    fn lower_for_pattern_inits(&mut self, ast: &mut Ast) -> Result<(), CompilerFailure> {
        for id in ast
            .stmt_ids()
            .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
        {
            let StmtKind::For {
                init: Some(init), ..
            } = ast
                .try_stmt(id)
                .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
                .kind
            else {
                continue;
            };
            let is_let = match ast
                .try_stmt(init)
                .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
                .kind
            {
                StmtKind::LetPattern { .. } => true,
                StmtKind::ConstPattern { .. } => false,
                _ => continue,
            };
            self.hoist_for_pattern_init(ast, id, init, is_let)?;
        }

        Ok(())
    }

    /// `for (let [i, j] = a; c; u) body` becomes
    /// `{ const dst = a; let i = dst[0]; let j = dst[1]; for (; c; u) body }`.
    /// The bindings keep the loop's scope but lose the per-iteration copies a
    /// `for` head's `let` gets, which only a closure capturing them observes;
    /// inference rejects that case using [`Ast::for_pattern_init_bindings`].
    fn hoist_for_pattern_init(
        &mut self,
        ast: &mut Ast,
        id: StmtId,
        init: StmtId,
        is_let: bool,
    ) -> Result<(), CompilerFailure> {
        let mut hoisted = self.expand_pattern_stmt(ast, init)?;
        let loop_stmt = ast
            .try_stmt_mut(id)
            .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?;
        let span = loop_stmt.span;
        let kind = std::mem::replace(&mut loop_stmt.kind, StmtKind::Block(Vec::new()));
        let StmtKind::For {
            condition,
            update,
            body,
            ..
        } = kind
        else {
            return Err(lowering_failure(
                "unexpected node kind during pattern lowering",
            ));
        };
        let lowered_loop = ast
            .try_push_stmt(Stmt {
                kind: StmtKind::For {
                    init: None,
                    condition,
                    update,
                    body,
                },
                span,
            })
            .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?;
        if is_let {
            let bindings = let_names(ast, &hoisted)?;
            if !bindings.is_empty() {
                ast.for_pattern_init_bindings.insert(lowered_loop, bindings);
            }
        }
        hoisted.push(lowered_loop);
        ast.try_stmt_mut(id)
            .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
            .kind = StmtKind::Block(hoisted);

        Ok(())
    }

    fn lower_pattern_stmts(&mut self, ast: &mut Ast) -> Result<(), CompilerFailure> {
        let top = std::mem::take(&mut ast.top_level);
        ast.top_level = self.expand_list(ast, top)?;

        for id in ast
            .stmt_ids()
            .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
        {
            let needs = matches!(
                ast.try_stmt(id)
                    .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
                    .kind,
                StmtKind::Block(_)
            );
            if !needs {
                continue;
            }
            let StmtKind::Block(taken) = std::mem::replace(
                &mut ast
                    .try_stmt_mut(id)
                    .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
                    .kind,
                StmtKind::Block(Vec::new()),
            ) else {
                return Err(lowering_failure(
                    "unexpected node kind during pattern lowering",
                ));
            };
            let expanded = self.expand_list(ast, taken)?;
            ast.try_stmt_mut(id)
                .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
                .kind = StmtKind::Block(expanded);
        }

        Ok(())
    }

    fn expand_list(
        &mut self,
        ast: &mut Ast,
        ids: Vec<StmtId>,
    ) -> Result<Vec<StmtId>, CompilerFailure> {
        let mut out = Vec::with_capacity(ids.len());
        for id in ids {
            let is_pattern = matches!(
                ast.try_stmt(id)
                    .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
                    .kind,
                StmtKind::LetPattern { .. } | StmtKind::ConstPattern { .. }
            );
            if !is_pattern {
                out.push(id);
                continue;
            }
            let expanded = self.expand_pattern_stmt(ast, id)?;
            out.extend(expanded);
        }
        Ok(out)
    }

    fn expand_pattern_stmt(
        &mut self,
        ast: &mut Ast,
        id: StmtId,
    ) -> Result<Vec<StmtId>, CompilerFailure> {
        let span = ast
            .try_stmt(id)
            .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
            .span;
        let kind = std::mem::replace(
            &mut ast
                .try_stmt_mut(id)
                .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
                .kind,
            StmtKind::Block(Vec::new()), // sentinel; the slot is unreachable post-lowering
        );
        let (is_const, binding, ty, value, doc) = match kind {
            StmtKind::LetPattern {
                binding,
                ty,
                value,
                doc,
            } => (false, binding, ty, value, doc),
            StmtKind::ConstPattern {
                binding,
                ty,
                value,
                doc,
            } => (true, binding, ty, value, doc),
            _ => {
                return Err(lowering_failure(
                    "unexpected node kind during pattern lowering",
                ));
            }
        };

        let literal = match ty {
            None => mark_literal_source(ast, &binding, value, is_const)?,
            Some(_) => None,
        };
        let pattern_span = binding.span();
        let dst = self.fresh(PATTERN_VALUE, pattern_span)?;
        let annotated = ty.is_some();

        // temp is always `const` even for `let` patterns — it's never re-assigned
        let dst_stmt = ast
            .try_push_stmt(Stmt {
                kind: StmtKind::Const {
                    name: dst.clone(),
                    ty,
                    value,
                    doc: None,
                },
                span,
            })
            .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?;

        let mut out = vec![dst_stmt];
        let root = PatternRoot {
            is_const,
            annotated,
        };
        out.extend(self.emit_decompose(ast, binding, dst, root, doc, literal.as_ref())?);
        Ok(out)
    }

    fn materialise_pattern_params(
        &mut self,
        ast: &mut Ast,
        params: &mut [ParamDecl],
    ) -> Result<(), CompilerFailure> {
        // A padding default is typed before the patterns bind and before any
        // later parameter does, so one that reads such a name would resolve
        // to an outer binding: its slot and the tuple slots after it stay out
        // of the padding. An earlier plain parameter is already bound.
        let pattern_names: BTreeSet<String> = params
            .iter()
            .filter_map(|param| param.pattern.as_ref())
            .flat_map(|pattern| bound_names(pattern).into_iter().map(str::to_string))
            .collect();
        let mut unbound_by_param = Vec::with_capacity(params.len());
        let mut unbound = pattern_names;
        for param in params.iter().rev() {
            unbound_by_param.push(unbound.clone());
            if param.pattern.is_none() {
                unbound.insert(param.name.name.clone());
            }
        }
        unbound_by_param.reverse();
        for (param, unbound) in params.iter_mut().zip(unbound_by_param) {
            let Some(pattern) = param.pattern.take() else {
                continue;
            };
            let fresh = self.fresh(PATTERN_PARAM, pattern.span())?;
            param.name = fresh.clone();
            let root = PatternRoot {
                is_const: false,
                annotated: param.ty.is_some(),
            };
            // An unannotated parameter's default is typed as a declaration's
            // initializer is. The argument, not the default, is what the
            // pattern takes apart: a caller or a contextual type may supply
            // more elements or properties than the default has.
            let literal = match (&param.ty, param.default) {
                (None, Some(default)) => mark_literal_source(ast, &pattern, default, false)?,
                _ => None,
            };
            if let Some(default) = param.default {
                let literal_id = peel_parens(ast, default)?;
                match &literal {
                    Some(LiteralSource::Tuple(len)) => {
                        if let Some(mut pad_defaults) = defaults_past_end(&pattern, *len) {
                            let mut kept = 0;
                            for slot in &pad_defaults {
                                if let Some(slot_default) = slot
                                    && reads_any_of(ast, *slot_default, &unbound)?
                                {
                                    break;
                                }
                                kept += 1;
                            }
                            pad_defaults.truncate(kept);
                            if !pad_defaults.is_empty() {
                                ast.tuple_pattern_pad_defaults
                                    .insert(literal_id, pad_defaults);
                            }
                        }
                    }
                    Some(LiteralSource::Object(names)) => {
                        let mut pad_defaults = Vec::new();
                        for (name, pad_default) in defaults_absent_from(&pattern, names) {
                            if !reads_any_of(ast, pad_default, &unbound)? {
                                pad_defaults.push((name, pad_default));
                            }
                        }
                        if !pad_defaults.is_empty() {
                            ast.object_pattern_pad_defaults
                                .insert(literal_id, pad_defaults);
                        }
                    }
                    None => {}
                }
            }
            let statements =
                self.emit_decompose(ast, pattern, fresh, root, /*doc=*/ None, None)?;
            if ast
                .parameter_bindings
                .insert(param.name.span, statements)
                .is_some()
            {
                return Err(lowering_failure(
                    "duplicate parameter binding span during pattern lowering",
                ));
            }
        }

        Ok(())
    }

    /// `doc` attaches to the first emitted stmt only. `literal` describes the
    /// literal `source` holds, if the pattern types one.
    fn emit_decompose(
        &mut self,
        ast: &mut Ast,
        binding: Binding,
        source: Ident,
        root: PatternRoot,
        mut doc: Option<crate::DocComment>,
        literal: Option<&LiteralSource>,
    ) -> Result<Vec<StmtId>, CompilerFailure> {
        let is_const = root.is_const;
        let mut out = Vec::new();
        match binding {
            Binding::Object {
                fields,
                rest,
                span: _,
            } => {
                let bound_names: Vec<Ident> = fields.iter().map(|f| f.source.clone()).collect();
                for field in fields {
                    // A defaulted property the literal lacks reads `undefined`,
                    // as tsc reads it, unless the literal inherits it.
                    let absent = matches!(
                        literal,
                        Some(LiteralSource::Object(names)) if !names.contains(&field.source.name)
                    ) && !OBJECT_PROTOTYPE_MEMBERS
                        .contains(&field.source.name.as_str());
                    let access = if absent && field.default.is_some() {
                        push_void_zero(ast, field.span)?
                    } else {
                        let recv =
                            push_expr(ast, ExprKind::Identifier(source.clone()), field.span)?;
                        push_expr(
                            ast,
                            ExprKind::FieldAccess {
                                receiver: recv,
                                name: field.source.clone(),
                            },
                            field.span,
                        )?
                    };
                    let value = self.apply_binding_default(
                        ast,
                        access,
                        field.default,
                        root.annotated,
                        field.span,
                        &mut out,
                    )?;
                    let decl = ast
                        .try_push_stmt(Stmt {
                            kind: make_decl(is_const, field.local, value, doc.take()),
                            span: field.span,
                        })
                        .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?;
                    out.push(decl);
                }
                if let Some(rest_ident) = rest {
                    let src_expr = ast
                        .try_push_expr(Expr {
                            kind: ExprKind::Identifier(source.clone()),
                            span: rest_ident.span,
                        })
                        .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?;
                    let decl = ast
                        .try_push_stmt(Stmt {
                            kind: StmtKind::ObjectRest {
                                is_const,
                                name: rest_ident.clone(),
                                source: src_expr,
                                exclude: bound_names,
                                ty: None,
                                doc: doc.take(),
                            },
                            span: rest_ident.span,
                        })
                        .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?;
                    out.push(decl);
                }
            }
            Binding::Array {
                elems,
                defaults,
                rest,
                span: pattern_span,
            } => {
                if elems.len() != defaults.len() {
                    return Err(lowering_failure(
                        "array pattern defaults do not match element positions",
                    ));
                }
                let elems_len = elems.len();
                for (i, (slot, default)) in elems.into_iter().zip(defaults).enumerate() {
                    let Some(local) = slot else {
                        if default.is_some() {
                            return Err(lowering_failure("array pattern hole has a default value"));
                        }
                        continue;
                    };
                    // A tuple literal holds nothing past its end, so a defaulted
                    // slot there reads `undefined`, as tsc reads it, and the
                    // literal keeps its own length.
                    let past_tuple_end =
                        matches!(literal, Some(LiteralSource::Tuple(len)) if i >= *len);
                    let access = if default.is_some() && past_tuple_end {
                        push_void_zero(ast, local.span)?
                    } else {
                        let origin = PatternOrigin {
                            pattern_span,
                            slot_arity: elems_len,
                            defaulted: default.is_some(),
                        };
                        self.array_slot_read(ast, &source, i, origin, local.span)?
                    };
                    let value = self.apply_binding_default(
                        ast,
                        access,
                        default,
                        root.annotated,
                        local.span,
                        &mut out,
                    )?;
                    let decl = ast
                        .try_push_stmt(Stmt {
                            kind: make_decl(is_const, local.clone(), value, doc.take()),
                            span: local.span,
                        })
                        .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?;
                    out.push(decl);
                }
                if let Some(rest_ident) = rest {
                    let rest_value = match literal {
                        Some(&LiteralSource::Tuple(len)) if len > elems_len => {
                            remaining_elements(ast, &source, elems_len..len, rest_ident.span)?
                        }
                        _ => array_rest(ast, &source, elems_len, rest_ident.span)?,
                    };
                    let decl = ast
                        .try_push_stmt(Stmt {
                            kind: make_decl(is_const, rest_ident.clone(), rest_value, doc.take()),
                            span: rest_ident.span,
                        })
                        .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?;
                    out.push(decl);
                }
            }
        }
        Ok(out)
    }

    /// `source[index]`, tagged with `origin` so the typechecker words index
    /// errors for the pattern. A defaulted slot's read is guarded by length.
    fn array_slot_read(
        &mut self,
        ast: &mut Ast,
        source: &Ident,
        index: usize,
        origin: PatternOrigin,
        span: Span,
    ) -> Result<ExprId, CompilerFailure> {
        let receiver = push_expr(ast, ExprKind::Identifier(source.clone()), span)?;
        let index = push_expr(ast, ExprKind::Number(index as f64), span)?;
        let access = push_expr(ast, ExprKind::IndexAccess { receiver, index }, span)?;
        let defaulted = origin.defaulted;
        ast.pattern_origins.insert(access, origin);
        if !defaulted {
            return Ok(access);
        }
        self.guard_defaulted_array_read(ast, receiver, index, access, span)
    }

    /// A pattern default treats an absent element as undefined, while ordinary
    /// index expressions retain their bounds checks.
    fn guard_defaulted_array_read(
        &mut self,
        ast: &mut Ast,
        receiver: ExprId,
        index: ExprId,
        access: ExprId,
        span: Span,
    ) -> Result<ExprId, CompilerFailure> {
        let length = push_expr(
            ast,
            ExprKind::FieldAccess {
                receiver,
                name: Ident {
                    name: "length".to_string(),
                    span,
                },
            },
            span,
        )?;
        let in_bounds = push_expr(
            ast,
            ExprKind::Binary {
                op: crate::BinOp::Lt,
                lhs: index,
                rhs: length,
            },
            span,
        )?;
        let undefined = push_void_zero(ast, span)?;
        push_expr(
            ast,
            ExprKind::Ternary {
                cond: in_bounds,
                then_: access,
                else_: undefined,
            },
            span,
        )
    }

    /// Save the property/index read once, then evaluate its default only for undefined.
    /// `void 0` is immune to a source binding named `undefined`.
    fn apply_binding_default(
        &mut self,
        ast: &mut Ast,
        access: ExprId,
        default: Option<ExprId>,
        annotated: bool,
        span: Span,
        out: &mut Vec<StmtId>,
    ) -> Result<ExprId, CompilerFailure> {
        let Some(default) = default else {
            return Ok(access);
        };
        let saved = self.fresh("value", span)?;
        let declaration = ast
            .try_push_stmt(Stmt {
                kind: make_decl(true, saved.clone(), access, None),
                span,
            })
            .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?;
        out.push(declaration);
        let saved_value = push_expr(ast, ExprKind::Identifier(saved.clone()), span)?;
        let undefined = push_void_zero(ast, span)?;
        let condition = push_expr(
            ast,
            ExprKind::Binary {
                op: crate::BinOp::Eq,
                lhs: saved_value,
                rhs: undefined,
            },
            span,
        )?;
        let selected = push_expr(
            ast,
            ExprKind::Ternary {
                cond: condition,
                then_: default,
                else_: saved_value,
            },
            span,
        )?;
        ast.binding_defaults.insert(
            selected,
            crate::ast::BindingDefault {
                saved,
                default,
                annotated,
            },
        );
        Ok(selected)
    }
}

/// A literal a pattern takes apart, which TypeScript types from the pattern.
enum LiteralSource {
    /// An array literal of this many elements, typed as a tuple.
    Tuple(usize),
    /// An object literal with these property names.
    Object(BTreeSet<String>),
}

/// Describe the literal `value` that `binding` takes apart, and record how
/// inference types it: an array literal as a tuple, and an object literal
/// with each defaulted property the pattern reads optional, as tsc types it.
fn mark_literal_source(
    ast: &mut Ast,
    binding: &Binding,
    value: ExprId,
    is_const: bool,
) -> Result<Option<LiteralSource>, CompilerFailure> {
    let value = peel_parens(ast, value)?;
    let bound = bound_names(binding);
    let context_free =
        |default: ExprId| -> Result<bool, CompilerFailure> { names_none_of(ast, default, &bound) };
    if let Some(len) = tuple_source_len(ast, binding, value)? {
        if let Binding::Array { defaults, .. } = binding {
            let mut filled: BTreeMap<usize, ExprId> = BTreeMap::new();
            for (index, default) in defaults.iter().take(len).enumerate() {
                if let Some(default) = *default
                    && context_free(default)?
                {
                    filled.insert(index, default);
                }
            }
            if !filled.is_empty() {
                ast.tuple_pattern_defaults.insert(value, filled);
            }
        }
        ast.tuple_pattern_sources.insert(value);
        if is_const {
            ast.const_pattern_sources.insert(value);
        }
        return Ok(Some(LiteralSource::Tuple(len)));
    }
    if let Binding::Object { fields, .. } = binding
        && let Some(names) = object_literal_names(ast, value)?
    {
        let mut defaulted: BTreeMap<String, Option<ExprId>> = BTreeMap::new();
        for field in fields
            .iter()
            .filter(|field| names.contains(&field.source.name))
        {
            if let Some(default) = field.default {
                let context = context_free(default)?.then_some(default);
                defaulted.insert(field.source.name.clone(), context);
            }
        }
        if !defaulted.is_empty() {
            ast.object_pattern_sources.insert(value, defaulted);
        }
        if is_const {
            ast.const_pattern_sources.insert(value);
        }
        return Ok(Some(LiteralSource::Object(names)));
    }
    mark_operand_tuple_sources(ast, binding, value)?;
    Ok(None)
}

/// The names `binding` declares.
fn bound_names(binding: &Binding) -> BTreeSet<&str> {
    match binding {
        Binding::Object { fields, rest, .. } => fields
            .iter()
            .map(|field| field.local.name.as_str())
            .chain(rest.iter().map(|rest| rest.name.as_str()))
            .collect(),
        Binding::Array { elems, rest, .. } => elems
            .iter()
            .flatten()
            .chain(rest.iter())
            .map(|name| name.name.as_str())
            .collect(),
    }
}

/// Whether `default` can be typed before its pattern binds `names`, as the
/// context of the pattern source's literal: a literal, or a reference whose
/// root is none of them. Any other default may read a name the pattern binds,
/// which a trial ahead of the pattern would resolve to an outer binding.
fn names_none_of(
    ast: &Ast,
    default: ExprId,
    names: &BTreeSet<&str>,
) -> Result<bool, CompilerFailure> {
    let mut expr = peel_parens(ast, default)?;
    loop {
        let kind = &ast
            .try_expr(expr)
            .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
            .kind;
        match kind {
            ExprKind::Identifier(name) => return Ok(!names.contains(name.name.as_str())),
            ExprKind::FieldAccess { receiver, .. } => expr = peel_parens(ast, *receiver)?,
            ExprKind::Unary { operand, .. } | ExprKind::As { expr: operand, .. } => {
                expr = peel_parens(ast, *operand)?;
            }
            ExprKind::Number(_) | ExprKind::String(_) | ExprKind::Boolean(_) | ExprKind::Null => {
                return Ok(true);
            }
            _ => return Ok(false),
        }
    }
}

/// Whether `expr` mentions any of `names`, inside a nested function too,
/// other than where a nested function's parameter or own name shadows it.
/// A nested `let` or `const` of the name still counts as a mention.
fn reads_any_of(
    ast: &Ast,
    expr: ExprId,
    names: &BTreeSet<String>,
) -> Result<bool, CompilerFailure> {
    let span = ast
        .try_expr(expr)
        .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
        .span;
    let mut shadowing: Vec<(Span, &str)> = Vec::new();
    let mut mentions: Vec<(Span, &str)> = Vec::new();
    for inner in ast.exprs_within(span) {
        match &inner.kind {
            ExprKind::Identifier(name) if names.contains(&name.name) => {
                mentions.push((inner.span, name.name.as_str()));
            }
            ExprKind::Arrow { params, .. } => {
                for param in params {
                    let declared = match &param.pattern {
                        Some(pattern) => bound_names(pattern),
                        None => BTreeSet::from([param.name.name.as_str()]),
                    };
                    shadowing.extend(declared.into_iter().map(|name| (inner.span, name)));
                }
            }
            ExprKind::FunctionExpression {
                name: Some(name), ..
            } => shadowing.push((inner.span, name.name.as_str())),
            _ => {}
        }
    }
    Ok(mentions.iter().any(|(at, name)| {
        !shadowing
            .iter()
            .any(|(region, shadowed)| shadowed == name && region.encloses(*at))
    }))
}

/// The properties every object literal inherits from `Object.prototype`.
pub(crate) const OBJECT_PROTOTYPE_MEMBERS: &[&str] = &[
    "constructor",
    "hasOwnProperty",
    "isPrototypeOf",
    "propertyIsEnumerable",
    "toLocaleString",
    "toString",
    "valueOf",
    "__proto__",
    "__defineGetter__",
    "__defineSetter__",
    "__lookupGetter__",
    "__lookupSetter__",
];

/// The defaults of an array pattern's slots from `len` on, when each is a
/// named slot with a default.
fn defaults_past_end(binding: &Binding, len: usize) -> Option<Vec<Option<ExprId>>> {
    let Binding::Array {
        elems, defaults, ..
    } = binding
    else {
        return None;
    };
    // A hole reads nothing, so it pads with no default. A named slot without a
    // default leaves the tuple as it is: reading past its end is an error.
    let past: Vec<Option<ExprId>> = elems
        .iter()
        .zip(defaults)
        .skip(len)
        .map(|(slot, default)| match slot {
            None => Some(None),
            Some(_) => default.map(Some),
        })
        .collect::<Option<_>>()?;
    past.iter().any(Option::is_some).then_some(past)
}

/// The properties an object pattern gives defaults that a literal with
/// `names` lacks and doesn't inherit, with their defaults.
fn defaults_absent_from(binding: &Binding, names: &BTreeSet<String>) -> Vec<(String, ExprId)> {
    let Binding::Object { fields, .. } = binding else {
        return Vec::new();
    };
    fields
        .iter()
        .filter(|field| {
            !names.contains(&field.source.name)
                && !OBJECT_PROTOTYPE_MEMBERS.contains(&field.source.name.as_str())
        })
        .filter_map(|field| Some((field.source.name.clone(), field.default?)))
        .collect()
}

fn peel_parens(ast: &Ast, value: ExprId) -> Result<ExprId, CompilerFailure> {
    ast.try_peel_parens(value)
        .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))
}

/// An array literal the pattern may take apart as one operand of `value`, a
/// fallback (`x ?? []`, `x || []`) or a conditional's branch, is a tuple as in
/// tsc, where the pattern is the context of every such operand.
fn mark_operand_tuple_sources(
    ast: &mut Ast,
    binding: &Binding,
    value: ExprId,
) -> Result<(), CompilerFailure> {
    let operands = match &ast
        .try_expr(value)
        .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
        .kind
    {
        ExprKind::Paren(inner) => vec![*inner],
        ExprKind::Binary {
            op: crate::BinOp::NullishCoalesce | crate::BinOp::Or,
            rhs,
            ..
        } => vec![*rhs],
        ExprKind::Ternary { then_, else_, .. } => vec![*then_, *else_],
        _ => return Ok(()),
    };
    for operand in operands {
        if tuple_source_len(ast, binding, operand)?.is_some() {
            ast.tuple_pattern_sources.insert(operand);
        } else {
            mark_operand_tuple_sources(ast, binding, operand)?;
        }
    }
    Ok(())
}

/// The property names of `value` when it is an object literal whose every
/// property is named.
fn object_literal_names(
    ast: &Ast,
    value: ExprId,
) -> Result<Option<BTreeSet<String>>, CompilerFailure> {
    let ExprKind::ObjectLiteral { members } = &ast
        .try_expr(value)
        .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
        .kind
    else {
        return Ok(None);
    };
    let mut names = BTreeSet::new();
    for member in members {
        let crate::ObjectLiteralMember::Field(field) = member else {
            return Ok(None);
        };
        names.insert(field.name.name.clone());
    }
    Ok(Some(names))
}

/// The length of `value` when it is an array literal without spreads that
/// `binding`, an array pattern, takes apart slot by slot, so TypeScript types
/// it as a tuple. A pattern of only a rest element takes the literal whole, as
/// an array.
fn tuple_source_len(
    ast: &Ast,
    binding: &Binding,
    value: ExprId,
) -> Result<Option<usize>, CompilerFailure> {
    let Binding::Array { elems, .. } = binding else {
        return Ok(None);
    };
    if elems.is_empty() {
        return Ok(None);
    }
    let ExprKind::ArrayLiteral { elements } = &ast
        .try_expr(value)
        .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
        .kind
    else {
        return Ok(None);
    };
    let has_spread = elements
        .iter()
        .any(|element| matches!(element, crate::ArrayLiteralElement::Spread { .. }));
    Ok((!has_spread).then_some(elements.len()))
}

/// `source.slice(from, source.length)`: the elements an array pattern's rest
/// element collects.
fn array_rest(
    ast: &mut Ast,
    source: &Ident,
    from: usize,
    span: Span,
) -> Result<ExprId, CompilerFailure> {
    let from_lit = push_expr(ast, ExprKind::Number(from as f64), span)?;
    let slice_receiver = push_expr(ast, ExprKind::Identifier(source.clone()), span)?;
    let slice_callee = push_expr(
        ast,
        ExprKind::FieldAccess {
            receiver: slice_receiver,
            name: Ident {
                name: "slice".to_string(),
                span,
            },
        },
        span,
    )?;
    let length_receiver = push_expr(ast, ExprKind::Identifier(source.clone()), span)?;
    let length_access = push_expr(
        ast,
        ExprKind::FieldAccess {
            receiver: length_receiver,
            name: Ident {
                name: "length".to_string(),
                span,
            },
        },
        span,
    )?;
    let slice_call = push_expr(
        ast,
        ExprKind::Call {
            callee: slice_callee,
            type_args: None,
            args: vec![from_lit, length_access],
        },
        span,
    )?;
    Ok(slice_call)
}

/// `[source[from], …, source[to - 1]].slice(0, to - from)`: the rest of a
/// tuple-typed literal. The remaining elements form a tuple of their own, and
/// slicing it yields an array of their types: one that can grow, where
/// TypeScript types the rest as a tuple.
fn remaining_elements(
    ast: &mut Ast,
    source: &Ident,
    indices: std::ops::Range<usize>,
    span: Span,
) -> Result<ExprId, CompilerFailure> {
    let count = indices.len();
    let mut elements = Vec::with_capacity(count);
    for i in indices {
        let receiver = push_expr(ast, ExprKind::Identifier(source.clone()), span)?;
        let index = push_expr(ast, ExprKind::Number(i as f64), span)?;
        let access = push_expr(ast, ExprKind::IndexAccess { receiver, index }, span)?;
        elements.push(crate::ArrayLiteralElement::Value(access));
    }
    let remaining = push_expr(ast, ExprKind::ArrayLiteral { elements }, span)?;
    ast.tuple_pattern_sources.insert(remaining);
    let slice = push_expr(
        ast,
        ExprKind::FieldAccess {
            receiver: remaining,
            name: Ident {
                name: "slice".to_string(),
                span,
            },
        },
        span,
    )?;
    let from = push_expr(ast, ExprKind::Number(0.0), span)?;
    let to = push_expr(ast, ExprKind::Number(count as f64), span)?;
    push_expr(
        ast,
        ExprKind::Call {
            callee: slice,
            type_args: None,
            args: vec![from, to],
        },
        span,
    )
}

fn push_expr(ast: &mut Ast, kind: ExprKind, span: Span) -> Result<ExprId, CompilerFailure> {
    ast.try_push_expr(Expr { kind, span })
        .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))
}

/// `void 0`, which is `undefined` even where a source binding shadows the name.
fn push_void_zero(ast: &mut Ast, span: Span) -> Result<ExprId, CompilerFailure> {
    let zero = push_expr(ast, ExprKind::Number(0.0), span)?;
    push_expr(ast, ExprKind::Void { operand: zero }, span)
}

fn make_decl(
    is_const: bool,
    name: Ident,
    value: ExprId,
    doc: Option<crate::DocComment>,
) -> StmtKind {
    if is_const {
        StmtKind::Const {
            name,
            ty: None,
            value,
            doc,
        }
    } else {
        StmtKind::Let {
            name,
            ty: None,
            value,
            doc,
        }
    }
}

/// The names the `let` statements among `stmts` declare.
fn let_names(ast: &Ast, stmts: &[StmtId]) -> Result<Vec<Ident>, CompilerFailure> {
    let mut names = Vec::new();
    for &stmt in stmts {
        if let StmtKind::Let { name, .. } = &ast
            .try_stmt(stmt)
            .map_err(|error| error.into_compiler_failure(CompilerStage::Infer))?
            .kind
        {
            names.push(name.clone());
        }
    }
    Ok(names)
}

fn lowering_failure(message: &str) -> CompilerFailure {
    CompilerFailure::Internal {
        stage: CompilerStage::Infer,
        span: None,
        message: message.into(),
    }
}

#[cfg(test)]
mod tests {
    use super::lower;
    use crate::{Asi, Binding, Stmt, StmtKind, Token, TokenKind, parse};

    fn pipeline(source: &str) -> crate::Ast {
        let mut asi = Asi::new(source, crate::FileId(0));
        let mut tokens: Vec<Token> = Vec::new();
        loop {
            let t = asi.next_token();
            let is_eof = matches!(t.kind, TokenKind::Eof);
            tokens.push(t);
            if is_eof {
                break;
            }
        }
        let _ = asi.into_diagnostics();
        let (mut ast, diags) = parse(source, tokens, crate::FileId(0));
        assert!(diags.is_empty(), "unexpected parse diags: {diags:?}");
        ast = lower(ast).unwrap();
        ast
    }

    fn no_patterns_left(ast: &crate::Ast) {
        for i in 0..ast.stmts_len() {
            let s = ast.try_stmt(crate::StmtId(i as u32)).unwrap();
            assert!(
                !matches!(
                    s.kind,
                    StmtKind::LetPattern { .. }
                        | StmtKind::ConstPattern { .. }
                        | StmtKind::ForOfPattern { .. }
                ),
                "found unlowered pattern at stmt #{i}: {:?}",
                s.kind,
            );
            if let StmtKind::Function { params, .. } = &s.kind {
                for (pi, p) in params.iter().enumerate() {
                    assert!(
                        p.pattern.is_none(),
                        "function param #{pi} still has a pattern after lowering",
                    );
                }
            }
        }
        for i in 0..ast.exprs_len() {
            let e = ast.try_expr(crate::ExprId(i as u32)).unwrap();
            if let crate::ExprKind::Arrow { params, .. } = &e.kind {
                for (pi, p) in params.iter().enumerate() {
                    assert!(
                        p.pattern.is_none(),
                        "arrow param #{pi} still has a pattern after lowering",
                    );
                }
            }
        }
    }

    #[test]
    fn lowers_simple_object_destructure() {
        let ast = pipeline("const { a, b } = obj;");
        no_patterns_left(&ast);
        // top_level: __dst, a-decl, b-decl
        assert_eq!(ast.top_level.len(), 3);
        let names: Vec<String> = ast
            .top_level
            .iter()
            .map(|id| match &ast.try_stmt(*id).unwrap().kind {
                StmtKind::Const { name, .. } => name.name.clone(),
                other => panic!("expected Const, got {other:?}"),
            })
            .collect();
        assert!(names[0].starts_with("#pattern_dst_"));
        assert_eq!(names[1], "a");
        assert_eq!(names[2], "b");
    }

    #[test]
    fn lowers_renamed_object_destructure() {
        let ast = pipeline("const { a: x, b: y } = obj;");
        no_patterns_left(&ast);
        let names: Vec<String> = ast
            .top_level
            .iter()
            .map(|id| match &ast.try_stmt(*id).unwrap().kind {
                StmtKind::Const { name, .. } => name.name.clone(),
                other => panic!("expected Const, got {other:?}"),
            })
            .collect();
        assert_eq!(names[1], "x");
        assert_eq!(names[2], "y");
    }

    #[test]
    fn lowers_array_destructure() {
        let ast = pipeline("const [x, y] = arr;");
        no_patterns_left(&ast);
        assert_eq!(ast.top_level.len(), 3);
        if let StmtKind::Const { value, .. } = &ast.try_stmt(ast.top_level[1]).unwrap().kind {
            assert!(matches!(
                ast.try_expr(*value).unwrap().kind,
                crate::ExprKind::IndexAccess { .. }
            ));
        }
    }

    #[test]
    fn lowers_array_with_hole_and_rest() {
        let ast = pipeline("const [, x, ...rest] = arr;");
        no_patterns_left(&ast);
        // __dst, x (= arr[1]), rest (= arr.slice(2))
        assert_eq!(ast.top_level.len(), 3);
        if let StmtKind::Const { name, value, .. } = &ast.try_stmt(ast.top_level[1]).unwrap().kind {
            assert_eq!(name.name, "x");
            if let crate::ExprKind::IndexAccess { index, .. } = &ast.try_expr(*value).unwrap().kind
            {
                assert_eq!(
                    ast.try_expr(*index).unwrap().kind,
                    crate::ExprKind::Number(1.0)
                );
            } else {
                panic!("expected IndexAccess");
            }
        }
        if let StmtKind::Const { name, value, .. } = &ast.try_stmt(ast.top_level[2]).unwrap().kind {
            assert_eq!(name.name, "rest");
            match &ast.try_expr(*value).unwrap().kind {
                crate::ExprKind::Call { callee, args, .. } => {
                    assert_eq!(args.len(), 2, "slice takes (start, end)");
                    assert_eq!(
                        ast.try_expr(args[0]).unwrap().kind,
                        crate::ExprKind::Number(2.0)
                    );
                    match &ast.try_expr(*callee).unwrap().kind {
                        crate::ExprKind::FieldAccess { name, .. } => {
                            assert_eq!(name.name, "slice");
                        }
                        other => panic!("expected slice FieldAccess, got {other:?}"),
                    }
                }
                other => panic!("expected Call, got {other:?}"),
            }
        }
    }

    #[test]
    fn lowers_object_rest_to_object_rest() {
        let ast = pipeline("const { a, ...rest } = obj;");
        no_patterns_left(&ast);
        assert_eq!(ast.top_level.len(), 3);
        match &ast.try_stmt(ast.top_level[2]).unwrap().kind {
            StmtKind::ObjectRest {
                is_const,
                name,
                exclude,
                ..
            } => {
                assert!(*is_const);
                assert_eq!(name.name, "rest");
                assert_eq!(exclude.len(), 1);
                assert_eq!(exclude[0].name, "a");
            }
            other => panic!("expected ObjectRest, got {other:?}"),
        }
    }

    #[test]
    fn lowers_function_param_pattern() {
        let src = "function f({ a, b }: T): number { return a + b; }";
        let ast = pipeline(src);
        no_patterns_left(&ast);
        let Stmt {
            kind: StmtKind::Function { body, params, .. },
            ..
        } = ast.try_stmt(ast.top_level[0]).unwrap().clone()
        else {
            panic!("expected Function");
        };
        assert_eq!(params.len(), 1);
        assert!(params[0].pattern.is_none());
        assert!(params[0].name.name.starts_with("#pattern_p_"));
        let body_stmts = match &ast.try_stmt(body).unwrap().kind {
            StmtKind::Block(s) => s.clone(),
            _ => panic!("body is not a Block"),
        };
        assert_eq!(body_stmts.len(), 1);
        let bindings = ast.parameter_bindings.get(&params[0].name.span).unwrap();
        assert_eq!(bindings.len(), 2);
        if let StmtKind::Let { name, .. } = &ast.try_stmt(bindings[0]).unwrap().kind {
            assert_eq!(name.name, "a");
        } else {
            panic!("expected Let a");
        }
        if let StmtKind::Let { name, .. } = &ast.try_stmt(bindings[1]).unwrap().kind {
            assert_eq!(name.name, "b");
        } else {
            panic!("expected Let b");
        }
        assert!(matches!(
            ast.try_stmt(body_stmts[0]).unwrap().kind,
            StmtKind::Return(_)
        ));
    }

    #[test]
    fn parameter_object_rest_has_a_mutable_authored_binding() {
        let ast = pipeline("function f({ head, ...tail }: T): number { return tail.value; }");
        let StmtKind::Function { params, .. } = &ast.try_stmt(ast.top_level[0]).unwrap().kind
        else {
            panic!("expected function");
        };
        let bindings = ast.parameter_bindings.get(&params[0].name.span).unwrap();
        assert_eq!(bindings.len(), 2);
        let StmtKind::ObjectRest { is_const, name, .. } = &ast.try_stmt(bindings[1]).unwrap().kind
        else {
            panic!("expected object rest binding");
        };
        assert!(!*is_const);
        assert_eq!(name.name, "tail");
    }

    #[test]
    fn lowers_arrow_expr_body_with_pattern() {
        let src = "const f = ({a, b}: T): number => a + b;";
        let ast = pipeline(src);
        no_patterns_left(&ast);
        let mut found = false;
        for i in 0..ast.exprs_len() {
            let e = ast.try_expr(crate::ExprId(i as u32)).unwrap();
            if let crate::ExprKind::Arrow { params, body, .. } = &e.kind {
                assert!(matches!(body, crate::ArrowBody::Expr(_)));
                assert_eq!(
                    ast.parameter_bindings
                        .get(&params[0].name.span)
                        .unwrap()
                        .len(),
                    2
                );
                found = true;
            }
        }
        assert!(found, "no Arrow found in expr arena");
    }

    #[test]
    fn lowers_for_of_array_pattern() {
        // `for (const [a, b] of pairs) { … }` becomes
        // `for (const #pattern_dst_N of pairs) { const a = #pattern_dst_N[0];
        // const b = #pattern_dst_N[1]; { ... } }` after lowering.
        let src = "function main(): void { for (const [a, b] of pairs) { use(a, b); } }";
        let ast = pipeline(src);
        no_patterns_left(&ast);

        let Stmt {
            kind: StmtKind::Function { body, .. },
            ..
        } = ast.try_stmt(ast.top_level[0]).unwrap().clone()
        else {
            panic!("expected Function");
        };
        let body_stmts = match &ast.try_stmt(body).unwrap().kind {
            StmtKind::Block(s) => s.clone(),
            _ => panic!("function body is not a Block"),
        };
        assert_eq!(body_stmts.len(), 1, "main body has just the for-of");

        let (loop_name, loop_body) = match &ast.try_stmt(body_stmts[0]).unwrap().kind {
            StmtKind::ForOf { name, body, .. } => (name.name.clone(), *body),
            other => panic!("expected ForOf, got {other:?}"),
        };
        assert!(
            loop_name.starts_with("#pattern_dst_"),
            "loop var was rewritten to a synth temp; got {loop_name}",
        );

        let loop_body_stmts = match &ast.try_stmt(loop_body).unwrap().kind {
            StmtKind::Block(s) => s.clone(),
            _ => panic!("for-of body is not a Block"),
        };
        assert_eq!(loop_body_stmts.len(), 3);
        if let StmtKind::Const { name, value, .. } = &ast.try_stmt(loop_body_stmts[0]).unwrap().kind
        {
            assert_eq!(name.name, "a");
            assert!(matches!(
                ast.try_expr(*value).unwrap().kind,
                crate::ExprKind::IndexAccess { .. }
            ));
            // Origin-tagged so the typechecker can rephrase index errors.
            assert!(ast.pattern_origins.contains_key(value));
        } else {
            panic!("expected Const a");
        }
        if let StmtKind::Const { name, .. } = &ast.try_stmt(loop_body_stmts[1]).unwrap().kind {
            assert_eq!(name.name, "b");
        } else {
            panic!("expected Const b");
        }
    }

    #[test]
    fn lowers_let_pattern_preserves_mutability() {
        // __dst is const; per-field bindings follow source kind (`let` stays `let`)
        let ast = pipeline("let { a } = obj;");
        no_patterns_left(&ast);
        // __dst is Const; `a` is Let.
        assert!(matches!(
            ast.try_stmt(ast.top_level[0]).unwrap().kind,
            StmtKind::Const { .. }
        ));
        assert!(matches!(
            ast.try_stmt(ast.top_level[1]).unwrap().kind,
            StmtKind::Let { .. }
        ));
    }

    #[test]
    fn no_op_on_plain_program() {
        let src = "const x = 1; function main(): void { }";
        let before = {
            let mut asi = Asi::new(src, crate::FileId(0));
            let mut tokens: Vec<Token> = Vec::new();
            loop {
                let t = asi.next_token();
                let is_eof = matches!(t.kind, TokenKind::Eof);
                tokens.push(t);
                if is_eof {
                    break;
                }
            }
            let _ = asi.into_diagnostics();
            let (ast, _) = parse(src, tokens, crate::FileId(0));
            ast
        };
        let after = pipeline(src);
        assert_eq!(format!("{before:?}"), format!("{after:?}"));
    }

    #[test]
    fn lowers_pattern_in_nested_block() {
        let src = "function main(): void { const { a } = obj; }";
        let ast = pipeline(src);
        no_patterns_left(&ast);
    }

    #[test]
    fn idempotent() {
        let src = "const { a, b } = obj;";
        let mut ast1 = pipeline(src);
        let snapshot = format!("{ast1:?}");
        ast1 = lower(ast1).unwrap();
        assert_eq!(snapshot, format!("{ast1:?}"));
    }

    #[test]
    fn binding_span_preserved() {
        let (ast, _) = parse(
            "const { a } = obj;",
            tokens_of("const { a } = obj;"),
            crate::FileId(0),
        );
        match &ast.try_stmt(ast.top_level[0]).unwrap().kind {
            crate::StmtKind::ConstPattern {
                binding: Binding::Object { span, .. },
                ..
            } => {
                // `{ a }` starts at offset 6 and ends at 11 (inclusive of `}`).
                assert_eq!(*span, crate::Span::new(crate::FileId(0), 6, 11).unwrap());
            }
            other => panic!("expected ConstPattern Object, got {other:?}"),
        }
    }

    #[test]
    fn defaults_save_each_access_once_and_test_undefined() {
        for source in [
            "const { a = fallback() } = source();",
            "const [a = fallback()] = source();",
        ] {
            let ast = pipeline(source);
            assert_eq!(ast.top_level.len(), 3);
            let StmtKind::Const {
                value: saved_access,
                ..
            } = ast.try_stmt(ast.top_level[1]).unwrap().kind
            else {
                panic!("expected saved read")
            };
            assert!(matches!(
                ast.try_expr(saved_access).unwrap().kind,
                crate::ExprKind::FieldAccess { .. } | crate::ExprKind::Ternary { .. }
            ));
            let StmtKind::Const { value, .. } = ast.try_stmt(ast.top_level[2]).unwrap().kind else {
                panic!("expected binding")
            };
            let crate::ExprKind::Ternary { cond, then_, else_ } = ast.try_expr(value).unwrap().kind
            else {
                panic!("expected lazy default")
            };
            let crate::ExprKind::Binary {
                op: crate::BinOp::Eq,
                lhs,
                rhs,
            } = ast.try_expr(cond).unwrap().kind
            else {
                panic!("expected undefined comparison")
            };
            assert_eq!(lhs, else_);
            assert!(matches!(
                ast.try_expr(rhs).unwrap().kind,
                crate::ExprKind::Void { .. }
            ));
            assert!(matches!(
                ast.try_expr(then_).unwrap().kind,
                crate::ExprKind::Call { .. }
            ));
        }
    }

    #[test]
    fn array_defaults_keep_original_indices_and_origin_metadata() {
        let ast = pipeline("const [, a = 1] = values;");
        let StmtKind::Const { value, .. } = ast.try_stmt(ast.top_level[1]).unwrap().kind else {
            panic!("expected saved index read")
        };
        let crate::ExprKind::Ternary {
            cond,
            then_: access,
            else_,
        } = ast.try_expr(value).unwrap().kind
        else {
            panic!("expected bounds guard")
        };
        assert!(matches!(
            ast.try_expr(cond).unwrap().kind,
            crate::ExprKind::Binary {
                op: crate::BinOp::Lt,
                ..
            }
        ));
        assert!(matches!(
            ast.try_expr(else_).unwrap().kind,
            crate::ExprKind::Void { .. }
        ));
        let crate::ExprKind::IndexAccess { index, .. } = ast.try_expr(access).unwrap().kind else {
            panic!("expected guarded index")
        };
        assert!(matches!(
            ast.try_expr(index).unwrap().kind,
            crate::ExprKind::Number(1.0)
        ));
        assert_eq!(ast.pattern_origins.get(&access).unwrap().slot_arity, 2);
    }

    #[test]
    fn lowers_class_method_parameter_defaults() {
        let ast = pipeline("class C { method({ x = 1 }: T): number { return x; } }");
        let StmtKind::ClassDecl { ref members, .. } = ast.try_stmt(ast.top_level[0]).unwrap().kind
        else {
            panic!("expected class")
        };
        let crate::ClassMember::Method { params, body, .. } = &members[0] else {
            panic!("expected method")
        };
        assert!(params[0].pattern.is_none());
        let StmtKind::Block(ref statements) = ast.try_stmt(*body).unwrap().kind else {
            panic!("expected block")
        };
        assert_eq!(statements.len(), 1);
        assert_eq!(
            ast.parameter_bindings
                .get(&params[0].name.span)
                .unwrap()
                .len(),
            2
        );
    }

    fn tokens_of(source: &str) -> Vec<Token> {
        let mut asi = Asi::new(source, crate::FileId(0));
        let mut tokens = Vec::new();
        loop {
            let t = asi.next_token();
            let is_eof = matches!(t.kind, TokenKind::Eof);
            tokens.push(t);
            if is_eof {
                break;
            }
        }
        let _ = asi.into_diagnostics();
        tokens
    }
}