OxideJS-New/crates/oxide_compiler/src/counter/expr/literal.rs

use super::*;

impl Compiler {
    /// Domain entry point: count any literal expression. The central
    /// `count_expression` match delegates all literal variants here in one arm.
    /// Numeric/String/Boolean/Null share the default single-`LOAD_CONST` shape;
    /// RegExp emits three instructions.
    pub(in crate::counter) fn count_literal(&self, expr: &Expression, ctx: &mut CompileCtx) {
        match expr {
            Expression::RegExpLiteral(_) => self.count_regexp_literal(ctx),
            _ => self.count_default_expression(ctx),
        }
    }

    pub(in crate::counter) fn count_this_expression(&self, ctx: &mut CompileCtx) {
        ctx.alloc_reg();
        ctx.projected_pc += 1; // LOAD_VAR from reg 254
    }

    pub(in crate::counter) fn count_identifier_expression(&self, expr: &Expression, ctx: &mut CompileCtx) {
        let Expression::Identifier(ident) = expr else {
            return;
        };
        let name = ident.name.as_str();

        // Check if this is an upvalue reference (nested closure)
        for up in &ctx.current_upvalue_captures {
            if up.name == name {
                ctx.alloc_reg();
                ctx.count_instr(); // LOAD_UPVALUE
                return;
            }
        }

        // Check if the owner's binding is captured (cell access)
        if ctx.scopes.symbols.lookup_is_captured(name) {
            ctx.alloc_reg();
            ctx.count_instr(); // CELL_GET
            return;
        }

        if CompileCtx::is_known_builtin(name) {
            let _ = ctx.lookup_or_builtin(name);
        }
        ctx.alloc_reg();
        ctx.projected_pc += 1; // LOAD_VAR
    }

    fn count_regexp_literal(&self, ctx: &mut CompileCtx) {
        ctx.alloc_reg(); // pattern reg (LOAD_CONST)
        ctx.projected_pc += 1;
        ctx.alloc_reg(); // flags reg (LOAD_CONST)
        ctx.projected_pc += 1;
        ctx.alloc_reg(); // result reg (CREATE_REGEXP)
        ctx.projected_pc += 1;
    }

    pub(in crate::counter) fn count_default_expression(&self, ctx: &mut CompileCtx) {
        ctx.alloc_reg();
        ctx.projected_pc += 1; // LOAD_CONST or LOAD_VAR
    }
}