OxideJS-New/crates/oxide_vm/src/builtins/number.rs

use oxide_types::value::JsValue;

use crate::coercion;
use crate::vm::Vm;
use oxide_runtime_api::NativeResult;

pub fn number_constructor(vm: &mut Vm, args: &[u8]) -> NativeResult {
    let n = if args.len() > 1 { coercion::to_number(vm.reg(args[1])) } else { 0.0 };
    let number_proto = vm.session().builtin_world().number_proto.as_ptr() as *mut oxide_types::object::JsObject;
    let is_ctor = if let Some(this_reg) = args.first().copied() {
        let this_val = vm.reg(this_reg);
        if this_val.is_object() {
            let ptr = this_val.as_js_object_ptr();
            if ptr.is_null() {
                false
            } else {
                let proto_ptr = unsafe { (*ptr).proto().as_js_object_ptr() };
                !proto_ptr.is_null() && std::ptr::eq(proto_ptr, number_proto)
            }
        } else {
            false
        }
    } else {
        false
    };

    if is_ctor {
        let this_val = vm.reg(args[0]);
        let obj = unsafe { &mut *this_val.as_js_object_ptr() };
        obj.type_tag = oxide_types::object::JsObject::OBJ_TYPE_NUMBER_OBJ;
        let boxed = if n.fract() == 0.0 && n.is_finite() && n >= i32::MIN as f64 && n <= i32::MAX as f64 {
            JsValue::int(n as i32)
        } else {
            JsValue::float(n)
        };
        obj.set_prop_at(0, boxed);
        return NativeResult::Ok(this_val);
    }

    if n.fract() == 0.0 && n.is_finite() && n >= i32::MIN as f64 && n <= i32::MAX as f64 {
        NativeResult::Ok(JsValue::int(n as i32))
    } else {
        NativeResult::Ok(JsValue::float(n))
    }
}

pub fn number_is_nan(vm: &mut Vm, args: &[u8]) -> NativeResult {
    if args.len() < 2 {
        return NativeResult::Ok(JsValue::bool(false));
    }
    let val = vm.reg(args[1]);
    if val.is_double() {
        NativeResult::Ok(JsValue::bool(val.as_double().is_nan()))
    } else {
        NativeResult::Ok(JsValue::bool(false))
    }
}

pub fn number_is_finite(vm: &mut Vm, args: &[u8]) -> NativeResult {
    if args.len() < 2 {
        return NativeResult::Ok(JsValue::bool(false));
    }
    let val = vm.reg(args[1]);
    if val.is_int() {
        return NativeResult::Ok(JsValue::bool(true));
    }
    if val.is_double() {
        return NativeResult::Ok(JsValue::bool(val.as_double().is_finite()));
    }
    let n = coercion::to_number(val);
    NativeResult::Ok(JsValue::bool(n.is_finite()))
}

pub fn number_parse_int(vm: &mut Vm, args: &[u8]) -> NativeResult {
    if args.len() < 2 {
        return NativeResult::Ok(JsValue::float(f64::NAN));
    }
    let s = coercion::to_string(vm.reg(args[1]));
    let s = s.trim();

    if s.is_empty() {
        return NativeResult::Ok(JsValue::float(f64::NAN));
    }

    let radix = if args.len() > 2 {
        let r = coercion::to_number(vm.reg(args[2])) as i32;
        if r == 0 {
            10
        } else {
            r.clamp(2, 36)
        }
    } else {
        10
    };

    let (rest, hex) = if s.starts_with("0x") || s.starts_with("0X") {
        if radix == 16 || radix == 0 || (args.len() <= 2) {
            (s[2..].to_string(), true)
        } else {
            (s.to_string(), false)
        }
    } else {
        (s.to_string(), false)
    };

    let actual_radix = if hex { 16u32 } else { radix as u32 };

    if let Ok(n) = i32::from_str_radix(&rest, actual_radix) {
        return NativeResult::Ok(JsValue::int(n));
    }

    NativeResult::Ok(JsValue::float(f64::NAN))
}

pub fn number_parse_float(vm: &mut Vm, args: &[u8]) -> NativeResult {
    if args.len() < 2 {
        return NativeResult::Ok(JsValue::float(f64::NAN));
    }
    let s = coercion::to_string(vm.reg(args[1]));
    let s = s.trim();

    if s.is_empty() {
        return NativeResult::Ok(JsValue::float(f64::NAN));
    }

    match fast_float::parse::<f64, _>(&s) {
        Ok(v) => NativeResult::Ok(JsValue::float(v)),
        Err(_) => NativeResult::Ok(JsValue::float(f64::NAN)),
    }
}

pub fn number_to_string(vm: &mut Vm, args: &[u8]) -> NativeResult {
    let n = coercion::to_number(vm.reg(args[0]));
    let radix = if args.len() > 1 {
        let r = coercion::to_number(vm.reg(args[1])) as u32;
        r.clamp(2, 36)
    } else {
        10u32
    };

    if radix == 10 {
        if n.is_nan() {
            return NativeResult::Ok(vm.new_string("NaN"));
        }
        if n.is_infinite() {
            if n.is_sign_positive() {
                return NativeResult::Ok(vm.new_string("Infinity"));
            }
            return NativeResult::Ok(vm.new_string("-Infinity"));
        }
        if n.fract() == 0.0 && n >= i32::MIN as f64 && n <= i32::MAX as f64 {
            return NativeResult::Ok(vm.new_string(&(n as i64).to_string()));
        }
        let mut buf = ryu::Buffer::new();
        NativeResult::Ok(vm.new_string(buf.format(n)))
    } else {
        if n.is_nan() {
            return NativeResult::Ok(vm.new_string("NaN"));
        }
        let nn = n as i64;
        let mut result = String::new();
        let mut value = nn.abs();
        if value == 0 {
            result.push('0');
        } else {
            let chars = "0123456789abcdefghijklmnopqrstuvwxyz";
            let mut digits = Vec::new();
            while value > 0 {
                digits.push(chars.as_bytes()[(value % radix as i64) as usize] as char);
                value /= radix as i64;
            }
            for ch in digits.iter().rev() {
                result.push(*ch);
            }
        }
        if nn < 0 {
            result.insert(0, '-');
        }
        NativeResult::Ok(vm.new_string(&result))
    }
}

pub fn number_to_fixed(vm: &mut Vm, args: &[u8]) -> NativeResult {
    let n = coercion::to_number(vm.reg(args[0]));
    let digits = if args.len() > 1 {
        coercion::to_number(vm.reg(args[1])) as usize
    } else {
        0usize
    }
    .min(20);

    let formatted = format!("{:.digits$}", n);
    NativeResult::Ok(vm.new_string(&formatted))
}

pub fn number_is_integer(vm: &mut Vm, args: &[u8]) -> NativeResult {
    let val = vm.reg(if args.len() > 1 { args[1] } else { args[0] });
    let n = coercion::to_number(val);
    NativeResult::Ok(JsValue::bool(n.trunc() == n && n.is_finite()))
}

pub fn number_is_safe_integer(vm: &mut Vm, args: &[u8]) -> NativeResult {
    let val = vm.reg(if args.len() > 1 { args[1] } else { args[0] });
    let n = coercion::to_number(val);
    let safe = n.trunc() == n && n.is_finite() && n >= -9007199254740991i64 as f64 && n <= 9007199254740991i64 as f64;
    NativeResult::Ok(JsValue::bool(safe))
}

pub fn number_to_exponential(vm: &mut Vm, args: &[u8]) -> NativeResult {
    let n = coercion::to_number(vm.reg(args[0]));
    let digits = if args.len() > 1 {
        coercion::to_number(vm.reg(args[1])) as usize
    } else {
        0usize
    }
    .min(100);
    if n.is_nan() {
        return NativeResult::Ok(vm.new_string("NaN"));
    }
    if n.is_infinite() {
        return NativeResult::Ok(vm.new_string(if n.is_sign_positive() { "Infinity" } else { "-Infinity" }));
    }
    let formatted = format!("{:.digits$e}", n);
    NativeResult::Ok(vm.new_string(&formatted))
}

pub fn number_to_precision(vm: &mut Vm, args: &[u8]) -> NativeResult {
    let n = coercion::to_number(vm.reg(args[0]));
    let precision = if args.len() > 1 {
        coercion::to_number(vm.reg(args[1])) as usize
    } else {
        0usize
    }
    .min(21);
    if n.is_nan() {
        return NativeResult::Ok(vm.new_string("NaN"));
    }
    if n.is_infinite() {
        return NativeResult::Ok(vm.new_string(if n.is_sign_positive() { "Infinity" } else { "-Infinity" }));
    }
    let formatted = format!("{:.precision$}", n);
    NativeResult::Ok(vm.new_string(&formatted))
}

pub fn number_value_of(vm: &mut Vm, args: &[u8]) -> NativeResult {
    let this_val = vm.reg(args[0]);
    if this_val.is_int() || this_val.is_double() {
        return NativeResult::Ok(this_val);
    }
    if this_val.is_object() {
        let ptr = this_val.as_js_object_ptr();
        if !ptr.is_null() {
            let obj = unsafe { &*ptr };
            if obj.is_number_obj() {
                return NativeResult::Ok(obj.get_prop_at(0));
            }
        }
    }
    NativeResult::Err(crate::builtins::error::create_type_error(
        vm,
        "Number.prototype.valueOf called on incompatible receiver",
    ))
}