use bumpalo::collections::vec::Vec;
use bumpalo::collections::CollectIn;
use bumpalo::Bump;
use roc_module::low_level::LowLevel;
use roc_module::symbol::{IdentIds, ModuleId, Symbol};
use roc_target::TargetInfo;

use crate::ir::{
    Call, CallSpecId, CallType, Expr, HostExposedLayouts, JoinPointId, ModifyRc, PassedFunction,
    Proc, ProcLayout, SelfRecursive, Stmt, UpdateModeId,
};
use crate::layout::{
    Builtin, InLayout, LambdaName, Layout, LayoutInterner, LayoutRepr, Niche, STLayoutInterner,
    UnionLayout,
};

mod equality;
mod refcount;

const LAYOUT_BOOL: InLayout = Layout::BOOL;
const LAYOUT_UNIT: InLayout = Layout::UNIT;

const ARG_1: Symbol = Symbol::ARG_1;
const ARG_2: Symbol = Symbol::ARG_2;
const ARG_3: Symbol = Symbol::ARG_3;

/// "Infinite" reference count, for static values
/// Ref counts are encoded as negative numbers where isize::MIN represents 1
pub const REFCOUNT_MAX: usize = 0;

#[derive(Clone, Copy, Debug, PartialEq, Eq)]
pub enum HelperOp {
    Inc,
    Dec,
    IndirectInc,
    IndirectDec,
    DecRef(JoinPointId),
    Reset,
    ResetRef,
    Eq,
}

impl HelperOp {
    fn is_decref(&self) -> bool {
        matches!(self, Self::DecRef(_))
    }

    fn is_dec(&self) -> bool {
        matches!(self, Self::Dec)
    }

    fn is_inc(&self) -> bool {
        matches!(self, Self::Inc)
    }
}

#[derive(Debug)]
struct Specialization<'a> {
    op: HelperOp,
    layout: InLayout<'a>,
    symbol: Symbol,
    proc: Option<Proc<'a>>,
}

#[derive(Debug)]
pub struct Context<'a> {
    new_linker_data: Vec<'a, (Symbol, ProcLayout<'a>)>,
    recursive_union: Option<UnionLayout<'a>>,
    op: HelperOp,
}

/// Generate specialized helper procs for code gen
/// ----------------------------------------------
///
/// Some low level operations need specialized helper procs to traverse data structures at runtime.
/// This includes refcounting, hashing, and equality checks.
///
/// For example, when checking List equality, we need to visit each element and compare them.
/// Depending on the type of the list elements, we may need to recurse deeper into each element.
/// For tag unions, we may need branches for different tag IDs, etc.
///
/// This module creates specialized helper procs for all such operations and types used in the program.
///
/// The backend drives the process, in two steps:
/// 1) When it sees the relevant node, it calls CodeGenHelp to get the replacement IR.
///    CodeGenHelp returns IR for a call to the helper proc, and remembers the specialization.
/// 2) After the backend has generated code for all user procs, it takes the IR for all of the
///    specialized helpers procs, and generates target code for them too.
///
pub struct CodeGenHelp<'a> {
    arena: &'a Bump,
    home: ModuleId,
    target_info: TargetInfo,
    layout_isize: InLayout<'a>,
    specializations: Vec<'a, Specialization<'a>>,
    debug_recursion_depth: usize,
}

impl<'a> CodeGenHelp<'a> {
    pub fn new(arena: &'a Bump, target_info: TargetInfo, home: ModuleId) -> Self {
        let layout_isize = Layout::isize(target_info);

        CodeGenHelp {
            arena,
            home,
            target_info,
            layout_isize,
            specializations: Vec::with_capacity_in(16, arena),
            debug_recursion_depth: 0,
        }
    }

    pub fn take_procs(&mut self) -> Vec<'a, Proc<'a>> {
        let procs_iter = self
            .specializations
            .drain(0..)
            .map(|spec| spec.proc.unwrap());
        Vec::from_iter_in(procs_iter, self.arena)
    }

    // ============================================================================
    //
    //              CALL GENERATED PROCS
    //
    // ============================================================================

    /// Expand a `Refcounting` node to a `Let` node that calls a specialized helper proc.
    /// The helper procs themselves are to be generated later with `generate_procs`
    pub fn expand_refcount_stmt(
        &mut self,
        ident_ids: &mut IdentIds,
        layout_interner: &mut STLayoutInterner<'a>,
        layout: InLayout<'a>,
        modify: &ModifyRc,
        following: &'a Stmt<'a>,
    ) -> (&'a Stmt<'a>, Vec<'a, (Symbol, ProcLayout<'a>)>) {
        let op = match modify {
            ModifyRc::Inc(..) => HelperOp::Inc,
            ModifyRc::Dec(_) => HelperOp::Dec,
            ModifyRc::DecRef(_) => {
                let jp_decref = JoinPointId(self.create_symbol(ident_ids, "jp_decref"));
                HelperOp::DecRef(jp_decref)
            }
        };

        let mut ctx = Context {
            new_linker_data: Vec::new_in(self.arena),
            recursive_union: None,
            op,
        };

        let rc_stmt = refcount::refcount_stmt(
            self,
            ident_ids,
            &mut ctx,
            layout_interner,
            layout,
            modify,
            following,
        );
        (rc_stmt, ctx.new_linker_data)
    }

    pub fn call_reset_refcount(
        &mut self,
        ident_ids: &mut IdentIds,
        layout_interner: &mut STLayoutInterner<'a>,
        layout: InLayout<'a>,
        argument: Symbol,
    ) -> (Expr<'a>, Vec<'a, (Symbol, ProcLayout<'a>)>) {
        self.call_refcount(ident_ids, layout_interner, layout, argument, false)
    }

    /**
    Call a resetref operation. It is similar to reset except it does not recursively decrement it's children when unique.
    */
    pub fn call_resetref_refcount(
        &mut self,
        ident_ids: &mut IdentIds,
        layout_interner: &mut STLayoutInterner<'a>,
        layout: InLayout<'a>,
        argument: Symbol,
    ) -> (Expr<'a>, Vec<'a, (Symbol, ProcLayout<'a>)>) {
        self.call_refcount(ident_ids, layout_interner, layout, argument, true)
    }

    /**
    Call either a reset or a resetref refcount operation.
    */
    fn call_refcount(
        &mut self,
        ident_ids: &mut IdentIds,
        layout_interner: &mut STLayoutInterner<'a>,
        layout: InLayout<'a>,
        argument: Symbol,
        resetref: bool,
    ) -> (Expr<'a>, Vec<'a, (Symbol, ProcLayout<'a>)>) {
        let mut ctx = Context {
            new_linker_data: Vec::new_in(self.arena),
            recursive_union: None,
            op: if resetref {
                HelperOp::ResetRef
            } else {
                HelperOp::Reset
            },
        };

        let proc_name = self.find_or_create_proc(ident_ids, &mut ctx, layout_interner, layout);

        let arguments = self.arena.alloc([argument]);
        let ret_layout = layout;
        let arg_layouts = self.arena.alloc([layout]);
        let expr = Expr::Call(Call {
            call_type: CallType::ByName {
                name: LambdaName::no_niche(proc_name),
                ret_layout,
                arg_layouts,
                specialization_id: CallSpecId::BACKEND_DUMMY,
            },
            arguments,
        });

        (expr, ctx.new_linker_data)
    }

    /// Generate a refcount increment procedure, *without* a Call expression.
    /// *This method should be rarely used* - only when the proc is to be called from Zig.
    /// Otherwise you want to generate the Proc and the Call together, using another method.
    pub fn gen_refcount_proc(
        &mut self,
        ident_ids: &mut IdentIds,
        layout_interner: &mut STLayoutInterner<'a>,
        layout: InLayout<'a>,
        op: HelperOp,
    ) -> (Symbol, Vec<'a, (Symbol, ProcLayout<'a>)>) {
        let mut ctx = Context {
            new_linker_data: Vec::new_in(self.arena),
            recursive_union: None,
            op,
        };

        let proc_name = self.find_or_create_proc(ident_ids, &mut ctx, layout_interner, layout);

        (proc_name, ctx.new_linker_data)
    }

    /// Replace a generic `Lowlevel::Eq` call with a specialized helper proc.
    /// The helper procs themselves are to be generated later with `generate_procs`
    pub fn call_specialized_equals(
        &mut self,
        ident_ids: &mut IdentIds,
        layout_interner: &mut STLayoutInterner<'a>,
        layout: InLayout<'a>,
        arguments: &'a [Symbol],
    ) -> (Expr<'a>, Vec<'a, (Symbol, ProcLayout<'a>)>) {
        let mut ctx = Context {
            new_linker_data: Vec::new_in(self.arena),
            recursive_union: None,
            op: HelperOp::Eq,
        };

        let expr = self
            .call_specialized_op(ident_ids, &mut ctx, layout_interner, layout, arguments)
            .unwrap();

        (expr, ctx.new_linker_data)
    }

    // ============================================================================
    //
    //              CALL SPECIALIZED OP
    //
    // ============================================================================

    fn call_specialized_op(
        &mut self,
        ident_ids: &mut IdentIds,
        ctx: &mut Context<'a>,
        layout_interner: &mut STLayoutInterner<'a>,
        called_layout: InLayout<'a>,
        arguments: &'a [Symbol],
    ) -> Option<Expr<'a>> {
        use HelperOp::*;

        // debug_assert!(self.debug_recursion_depth < 100);
        self.debug_recursion_depth += 1;

        let layout = if matches!(
            layout_interner.get(called_layout).repr,
            LayoutRepr::RecursivePointer(_)
        ) {
            let union_layout = ctx.recursive_union.unwrap();
            layout_interner.insert_no_semantic(LayoutRepr::Union(union_layout))
        } else {
            called_layout
        };

        if layout_needs_helper_proc(layout_interner, layout, ctx.op) {
            let arena = self.arena;
            let proc_name = self.find_or_create_proc(ident_ids, ctx, layout_interner, layout);

            let (ret_layout, arg_layouts): (InLayout<'a>, &'a [InLayout<'a>]) = {
                let arg = self.replace_rec_ptr(ctx, layout_interner, layout);
                let box_arg = layout_interner.insert_no_semantic(LayoutRepr::Boxed(arg));

                match ctx.op {
                    Dec | DecRef(_) => (LAYOUT_UNIT, self.arena.alloc([arg])),
                    Reset | ResetRef => (layout, self.arena.alloc([layout])),
                    Inc => (LAYOUT_UNIT, self.arena.alloc([arg, self.layout_isize])),
                    IndirectDec => (LAYOUT_UNIT, arena.alloc([box_arg])),
                    IndirectInc => (LAYOUT_UNIT, arena.alloc([box_arg, self.layout_isize])),
                    Eq => (LAYOUT_BOOL, self.arena.alloc([arg, arg])),
                }
            };

            Some(Expr::Call(Call {
                call_type: CallType::ByName {
                    name: LambdaName::no_niche(proc_name),
                    ret_layout,
                    arg_layouts,
                    specialization_id: CallSpecId::BACKEND_DUMMY,
                },
                arguments,
            }))
        } else if ctx.op == HelperOp::Eq {
            Some(Expr::Call(Call {
                call_type: CallType::LowLevel {
                    op: LowLevel::Eq,
                    update_mode: UpdateModeId::BACKEND_DUMMY,
                },
                arguments,
            }))
        } else {
            None
        }
    }

    fn find_or_create_proc(
        &mut self,
        ident_ids: &mut IdentIds,
        ctx: &mut Context<'a>,
        layout_interner: &mut STLayoutInterner<'a>,
        orig_layout: InLayout<'a>,
    ) -> Symbol {
        use HelperOp::*;

        let layout = self.replace_rec_ptr(ctx, layout_interner, orig_layout);

        let found = self
            .specializations
            .iter()
            .find(|spec| spec.op == ctx.op && spec.layout == layout);

        if let Some(spec) = found {
            return spec.symbol;
        }

        // Procs can be recursive, so we need to create the symbol before the body is complete
        // But with nested recursion, that means Symbols and Procs can end up in different orders.
        // We want the same order, especially for function indices in Wasm. So create an empty slot and fill it in later.
        let (proc_symbol, proc_layout) =
            self.create_proc_symbol(ident_ids, layout_interner, ctx, layout);
        ctx.new_linker_data.push((proc_symbol, proc_layout));
        let spec_index = self.specializations.len();
        self.specializations.push(Specialization {
            op: ctx.op,
            layout,
            symbol: proc_symbol,
            proc: None,
        });

        // Recursively generate the body of the Proc and sub-procs
        let (ret_layout, body) = match ctx.op {
            Inc | Dec | DecRef(_) => (
                LAYOUT_UNIT,
                refcount::refcount_generic(
                    self,
                    ident_ids,
                    ctx,
                    layout_interner,
                    layout,
                    Symbol::ARG_1,
                ),
            ),
            IndirectInc | IndirectDec => (
                LAYOUT_UNIT,
                refcount::refcount_indirect(
                    self,
                    ident_ids,
                    ctx,
                    layout_interner,
                    layout,
                    Symbol::ARG_1,
                ),
            ),
            Reset => (
                layout,
                refcount::refcount_reset_proc_body(
                    self,
                    ident_ids,
                    ctx,
                    layout_interner,
                    layout,
                    Symbol::ARG_1,
                ),
            ),
            ResetRef => (
                layout,
                refcount::refcount_resetref_proc_body(
                    self,
                    ident_ids,
                    ctx,
                    layout_interner,
                    layout,
                    Symbol::ARG_1,
                ),
            ),
            Eq => (
                LAYOUT_BOOL,
                equality::eq_generic(self, ident_ids, ctx, layout_interner, layout),
            ),
        };

        let args: &'a [(InLayout<'a>, Symbol)] = {
            let roc_value = (layout, ARG_1);
            match ctx.op {
                Inc => {
                    let inc_amount = (self.layout_isize, ARG_2);
                    self.arena.alloc([roc_value, inc_amount])
                }
                Dec | DecRef(_) | Reset | ResetRef => self.arena.alloc([roc_value]),
                IndirectInc => {
                    let box_layout = layout_interner.insert_no_semantic(LayoutRepr::Boxed(layout));
                    let inc_amount = (self.layout_isize, ARG_2);
                    self.arena.alloc([(box_layout, ARG_1), inc_amount])
                }
                IndirectDec => {
                    let box_layout = layout_interner.insert_no_semantic(LayoutRepr::Boxed(layout));
                    self.arena.alloc([(box_layout, ARG_1)])
                }
                Eq => self.arena.alloc([roc_value, (layout, ARG_2)]),
            }
        };

        self.specializations[spec_index].proc = Some(Proc {
            name: LambdaName::no_niche(proc_symbol),
            args,
            body,
            closure_data_layout: None,
            ret_layout,
            is_self_recursive: SelfRecursive::NotSelfRecursive,
            host_exposed_layouts: HostExposedLayouts::NotHostExposed,
        });

        proc_symbol
    }

    fn create_proc_symbol(
        &self,
        ident_ids: &mut IdentIds,
        layout_interner: &mut STLayoutInterner<'a>,
        ctx: &mut Context<'a>,
        layout: InLayout<'a>,
    ) -> (Symbol, ProcLayout<'a>) {
        let debug_name = format!(
            "#help{}_{:?}_{:?}",
            self.specializations.len(),
            ctx.op,
            layout
        )
        .replace("Builtin", "");
        let proc_symbol: Symbol = self.create_symbol(ident_ids, &debug_name);

        let proc_layout = match ctx.op {
            HelperOp::Inc => ProcLayout {
                arguments: self.arena.alloc([layout, self.layout_isize]),
                result: LAYOUT_UNIT,
                niche: Niche::NONE,
            },
            HelperOp::Dec => ProcLayout {
                arguments: self.arena.alloc([layout]),
                result: LAYOUT_UNIT,
                niche: Niche::NONE,
            },
            HelperOp::IndirectInc => {
                let box_layout = layout_interner.insert_no_semantic(LayoutRepr::Boxed(layout));

                ProcLayout {
                    arguments: self.arena.alloc([box_layout, self.layout_isize]),
                    result: LAYOUT_UNIT,
                    niche: Niche::NONE,
                }
            }
            HelperOp::IndirectDec => {
                let box_layout = layout_interner.insert_no_semantic(LayoutRepr::Boxed(layout));

                ProcLayout {
                    arguments: self.arena.alloc([box_layout]),
                    result: LAYOUT_UNIT,
                    niche: Niche::NONE,
                }
            }
            HelperOp::Reset => ProcLayout {
                arguments: self.arena.alloc([layout]),
                result: layout,
                niche: Niche::NONE,
            },
            HelperOp::DecRef(_) => unreachable!("No generated Proc for DecRef"),
            HelperOp::ResetRef => unreachable!("No generated Proc for ResetRef"),
            HelperOp::Eq => ProcLayout {
                arguments: self.arena.alloc([layout, layout]),
                result: LAYOUT_BOOL,
                niche: Niche::NONE,
            },
        };

        (proc_symbol, proc_layout)
    }

    fn create_symbol(&self, ident_ids: &mut IdentIds, debug_name: &str) -> Symbol {
        let ident_id = ident_ids.add_str(debug_name);
        Symbol::new(self.home, ident_id)
    }

    // When creating or looking up Specializations, we need to replace RecursivePointer
    // with the particular Union layout it represents at this point in the tree.
    // For example if a program uses `RoseTree a : [Tree a (List (RoseTree a))]`
    // then it could have both `RoseTree I64` and `RoseTree Str`. In this case it
    // needs *two* specializations for `List(RecursivePointer)`, not just one.
    fn replace_rec_ptr(
        &mut self,
        ctx: &Context<'a>,
        layout_interner: &mut STLayoutInterner<'a>,
        layout: InLayout<'a>,
    ) -> InLayout<'a> {
        let lay = layout_interner.get(layout);
        let repr = match lay.repr {
            LayoutRepr::Builtin(Builtin::List(v)) => {
                let v = self.replace_rec_ptr(ctx, layout_interner, v);
                LayoutRepr::Builtin(Builtin::List(v))
            }

            LayoutRepr::Builtin(_) => return layout,

            LayoutRepr::Struct { field_layouts } => {
                let mut new_field_layouts = Vec::with_capacity_in(field_layouts.len(), self.arena);
                for f in field_layouts.iter() {
                    new_field_layouts.push(self.replace_rec_ptr(ctx, layout_interner, *f));
                }
                LayoutRepr::Struct {
                    field_layouts: new_field_layouts.into_bump_slice(),
                }
            }

            LayoutRepr::Union(UnionLayout::NonRecursive(tags)) => {
                let mut new_tags = Vec::with_capacity_in(tags.len(), self.arena);
                for fields in tags {
                    let mut new_fields = Vec::with_capacity_in(fields.len(), self.arena);
                    for field in fields.iter() {
                        new_fields.push(self.replace_rec_ptr(ctx, layout_interner, *field))
                    }
                    new_tags.push(new_fields.into_bump_slice());
                }
                LayoutRepr::Union(UnionLayout::NonRecursive(new_tags.into_bump_slice()))
            }

            LayoutRepr::Union(_) => {
                // we always fully unroll recursive types. That means tha when we find a
                // recursive tag union we can replace it with the layout
                return layout;
            }

            LayoutRepr::Boxed(inner) => {
                let inner = self.replace_rec_ptr(ctx, layout_interner, inner);
                LayoutRepr::Boxed(inner)
            }

            LayoutRepr::LambdaSet(lambda_set) => {
                return self.replace_rec_ptr(ctx, layout_interner, lambda_set.representation)
            }

            // This line is the whole point of the function
            LayoutRepr::RecursivePointer(_) => LayoutRepr::Union(ctx.recursive_union.unwrap()),
        };

        layout_interner.insert(Layout::new(repr, lay.semantic()))
    }

    fn union_tail_recursion_fields(
        &self,
        union_in_layout: InLayout<'a>,
        union: UnionLayout<'a>,
    ) -> Option<Vec<'a, Option<usize>>> {
        use UnionLayout::*;
        match union {
            NonRecursive(_) => None,

            Recursive(tags) => self.union_tail_recursion_fields_help(union_in_layout, tags),

            NonNullableUnwrapped(field_layouts) => {
                self.union_tail_recursion_fields_help(union_in_layout, &[field_layouts])
            }

            NullableWrapped {
                other_tags: tags, ..
            } => self.union_tail_recursion_fields_help(union_in_layout, tags),

            NullableUnwrapped { other_fields, .. } => {
                self.union_tail_recursion_fields_help(union_in_layout, &[other_fields])
            }
        }
    }

    fn union_tail_recursion_fields_help(
        &self,
        in_layout: InLayout<'a>,
        tags: &[&'a [InLayout<'a>]],
    ) -> Option<Vec<'a, Option<usize>>> {
        let tailrec_indices = tags
            .iter()
            .map(|fields| fields.iter().position(|f| *f == in_layout))
            .collect_in::<Vec<_>>(self.arena);

        if tailrec_indices.iter().any(|i| i.is_some()) {
            None
        } else {
            Some(tailrec_indices)
        }
    }
}

pub struct CallerProc<'a> {
    pub proc_symbol: Symbol,
    pub proc_layout: ProcLayout<'a>,
    pub proc: Proc<'a>,
}

impl<'a> CallerProc<'a> {
    fn create_symbol(home: ModuleId, ident_ids: &mut IdentIds, debug_name: &str) -> Symbol {
        let ident_id = ident_ids.add_str(debug_name);
        Symbol::new(home, ident_id)
    }

    fn create_caller_proc_symbol(
        home: ModuleId,
        ident_ids: &mut IdentIds,
        operation: &str,
        wrapped_function: Symbol,
    ) -> Symbol {
        let debug_name = format!("#help_{}_{}_{:?}", "caller", operation, wrapped_function,);

        Self::create_symbol(home, ident_ids, &debug_name)
    }

    pub fn new(
        arena: &'a Bump,
        home: ModuleId,
        ident_ids: &mut IdentIds,
        layout_interner: &mut STLayoutInterner<'a>,
        passed_function: &PassedFunction<'a>,
        capture_layout: Option<InLayout<'a>>,
    ) -> Self {
        let mut ctx = Context {
            new_linker_data: Vec::new_in(arena),
            recursive_union: None,
            op: HelperOp::Eq,
        };

        let box_capture_layout = if let Some(capture_layout) = capture_layout {
            layout_interner.insert_no_semantic(LayoutRepr::Boxed(capture_layout))
        } else {
            layout_interner.insert_no_semantic(LayoutRepr::Boxed(Layout::UNIT))
        };

        let box_argument_layout = layout_interner
            .insert_no_semantic(LayoutRepr::Boxed(passed_function.argument_layouts[0]));

        let box_return_layout =
            layout_interner.insert_no_semantic(LayoutRepr::Boxed(passed_function.return_layout));

        let proc_layout = ProcLayout {
            arguments: arena.alloc([box_capture_layout, box_argument_layout, box_return_layout]),
            result: Layout::UNIT,
            niche: Niche::NONE,
        };

        let proc_symbol =
            Self::create_caller_proc_symbol(home, ident_ids, "map", passed_function.name.name());

        ctx.new_linker_data.push((proc_symbol, proc_layout));

        let unbox_capture = Expr::ExprUnbox {
            symbol: Symbol::ARG_1,
        };

        let unbox_argument = Expr::ExprUnbox {
            symbol: Symbol::ARG_2,
        };

        let unboxed_capture = Self::create_symbol(home, ident_ids, "unboxed_capture");
        let unboxed_argument = Self::create_symbol(home, ident_ids, "unboxed_argument");
        let call_result = Self::create_symbol(home, ident_ids, "call_result");
        let unit_symbol = Self::create_symbol(home, ident_ids, "unit_symbol");
        let ignored = Self::create_symbol(home, ident_ids, "ignored");

        let call = Expr::Call(Call {
            call_type: CallType::ByName {
                name: passed_function.name,
                ret_layout: passed_function.return_layout,
                arg_layouts: passed_function.argument_layouts,
                specialization_id: passed_function.specialization_id,
            },
            arguments: if capture_layout.is_some() {
                arena.alloc([unboxed_argument, unboxed_capture])
            } else {
                arena.alloc([unboxed_argument])
            },
        });

        let ptr_write = Expr::Call(Call {
            call_type: CallType::LowLevel {
                op: LowLevel::PtrWrite,
                update_mode: UpdateModeId::BACKEND_DUMMY,
            },
            arguments: arena.alloc([Symbol::ARG_3, call_result]),
        });

        let mut body = Stmt::Let(
            unboxed_argument,
            unbox_argument,
            passed_function.argument_layouts[0],
            arena.alloc(Stmt::Let(
                call_result,
                call,
                passed_function.return_layout,
                arena.alloc(Stmt::Let(
                    ignored,
                    ptr_write,
                    box_return_layout,
                    arena.alloc(Stmt::Let(
                        unit_symbol,
                        Expr::Struct(&[]),
                        Layout::UNIT,
                        arena.alloc(Stmt::Ret(unit_symbol)),
                    )),
                )),
            )),
        );

        if let Some(capture_layout) = capture_layout {
            body = Stmt::Let(
                unboxed_capture,
                unbox_capture,
                capture_layout,
                arena.alloc(body),
            );
        }

        let args: &'a [(InLayout<'a>, Symbol)] = {
            arena.alloc([
                (box_capture_layout, ARG_1),
                (box_argument_layout, ARG_2),
                (box_return_layout, ARG_3),
            ])
        };

        let proc = Proc {
            name: LambdaName::no_niche(proc_symbol),
            args,
            body,
            closure_data_layout: None,
            ret_layout: Layout::UNIT,
            is_self_recursive: SelfRecursive::NotSelfRecursive,
            host_exposed_layouts: HostExposedLayouts::NotHostExposed,
        };

        if false {
            let allocator = ven_pretty::BoxAllocator;
            let doc = proc
                .to_doc::<_, (), _>(
                    &allocator,
                    layout_interner,
                    true,
                    crate::ir::Parens::NotNeeded,
                )
                .1
                .pretty(80)
                .to_string();

            println!("{}", doc);
        }

        Self {
            proc_symbol,
            proc_layout,
            proc,
        }
    }
}

fn let_lowlevel<'a>(
    arena: &'a Bump,
    result_layout: InLayout<'a>,
    result: Symbol,
    op: LowLevel,
    arguments: &[Symbol],
    next: &'a Stmt<'a>,
) -> Stmt<'a> {
    Stmt::Let(
        result,
        Expr::Call(Call {
            call_type: CallType::LowLevel {
                op,
                update_mode: UpdateModeId::BACKEND_DUMMY,
            },
            arguments: arena.alloc_slice_copy(arguments),
        }),
        result_layout,
        next,
    )
}

fn layout_needs_helper_proc<'a>(
    layout_interner: &STLayoutInterner<'a>,
    layout: InLayout<'a>,
    op: HelperOp,
) -> bool {
    match layout_interner.get(layout).repr {
        LayoutRepr::Builtin(
            Builtin::Int(_) | Builtin::Float(_) | Builtin::Bool | Builtin::Decimal,
        ) => false,
        LayoutRepr::Builtin(Builtin::Str) => {
            // Str type can use either Zig functions or generated IR, since it's not generic.
            // Eq uses a Zig function, refcount uses generated IR.
            // Both are fine, they were just developed at different times.
            matches!(op, HelperOp::Inc | HelperOp::Dec | HelperOp::DecRef(_))
        }
        LayoutRepr::Builtin(Builtin::List(_)) => true,
        LayoutRepr::Struct { .. } => true, // note: we do generate a helper for Unit, with just a Stmt::Ret
        LayoutRepr::Union(UnionLayout::NonRecursive(tags)) => !tags.is_empty(),
        LayoutRepr::Union(_) => true,
        LayoutRepr::LambdaSet(_) => true,
        LayoutRepr::RecursivePointer(_) => false,
        LayoutRepr::Boxed(_) => true,
    }
}