OxideJS-New/crates/oxide_kernel/src/kernel.rs

#![allow(clippy::arc_with_non_send_sync)]

use std::sync::Arc;

use oxide_types::mem::P;
use oxide_types::object::JsObject;
use oxide_types::value::JsValue;

use crate::builtin::BuiltinWorld;
use crate::code_forge::CodeForge;
use crate::prop_forge::PropForge;
use crate::shape_forge::{ShapeForge, EMPTY_SHAPE_ID};
use crate::string_forge::StringForge;

pub struct KernelConfig {
    pub min_pool_size: usize,
    pub max_pool_size: Option<usize>,
    pub max_dead_strings: Option<usize>,
    pub warmup_builtin_shapes: bool,
    pub warmup_builtin_code: bool,
    pub warmup_builtin_ic: bool,
}

impl KernelConfig {
    pub fn minimal() -> Self {
        Self {
            min_pool_size: 4,
            max_pool_size: Some(8),
            max_dead_strings: Some(10_000),
            warmup_builtin_shapes: true,
            warmup_builtin_code: false,
            warmup_builtin_ic: false,
        }
    }

    pub fn standard() -> Self {
        Self {
            min_pool_size: 8,
            max_pool_size: Some(32),
            max_dead_strings: Some(10_000),
            warmup_builtin_shapes: true,
            warmup_builtin_code: true,
            warmup_builtin_ic: false,
        }
    }

    pub fn full() -> Self {
        Self {
            min_pool_size: 16,
            max_pool_size: None,
            max_dead_strings: Some(5_000),
            warmup_builtin_shapes: true,
            warmup_builtin_code: true,
            warmup_builtin_ic: true,
        }
    }
}

impl Default for KernelConfig {
    fn default() -> Self {
        Self::minimal()
    }
}

pub struct OxideKernel {
    pub config: KernelConfig,
    pub string_forge: Arc<StringForge>,
    pub shape_forge: Arc<ShapeForge>,
    pub code_forge: Arc<CodeForge>,
    pub prop_forge: Arc<PropForge>,
    pub builtin_world: Arc<BuiltinWorld>,
    pub global_object: P<JsObject>,
}

impl OxideKernel {
    pub fn new(config: KernelConfig) -> Self {
        let string_forge = Arc::new(StringForge::new());
        let shape_forge = Arc::new(ShapeForge::new());
        let builtin_world = Arc::new(BuiltinWorld::new(&string_forge, &shape_forge));
        let code_forge = Arc::new(CodeForge::new());
        let prop_forge = Arc::new(PropForge::new());

        let mut global_obj = JsObject::new_empty(EMPTY_SHAPE_ID, JsValue::null());

        let si_nan = string_forge.intern("NaN").0;
        let si_undef = string_forge.intern("undefined").0;

        let nan_shape = shape_forge.make_shape(EMPTY_SHAPE_ID, si_nan);
        global_obj.set_shape_id(nan_shape);
        global_obj.set_prop_count(1);
        global_obj.set_inline_prop(0, JsValue::float(f64::NAN));

        let undef_shape = shape_forge.make_shape(nan_shape, si_undef);
        global_obj.set_shape_id(undef_shape);
        global_obj.set_prop_count(2);
        global_obj.set_inline_prop(1, JsValue::undefined());

        let global_object = P::new(global_obj);

        Self {
            config,
            string_forge,
            shape_forge,
            code_forge,
            prop_forge,
            builtin_world,
            global_object,
        }
    }

    pub fn string_forge(&self) -> &Arc<StringForge> {
        &self.string_forge
    }

    pub fn shape_forge(&self) -> &Arc<ShapeForge> {
        &self.shape_forge
    }

    pub fn code_forge(&self) -> &Arc<CodeForge> {
        &self.code_forge
    }

    pub fn prop_forge(&self) -> &Arc<PropForge> {
        &self.prop_forge
    }

    pub fn builtin_world(&self) -> &Arc<BuiltinWorld> {
        &self.builtin_world
    }

    pub fn config(&self) -> &KernelConfig {
        &self.config
    }

    pub fn global_object(&self) -> &P<JsObject> {
        &self.global_object
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::shape_forge::EMPTY_SHAPE_ID;

    #[test]
    fn test_kernel_new() {
        let kernel = OxideKernel::new(KernelConfig::minimal());
        let (idx, _) = kernel.string_forge().intern("test");
        assert!(idx > 0);
    }

    #[test]
    fn test_kernel_builtins_accessible() {
        let kernel = OxideKernel::new(KernelConfig::minimal());
        assert!(!kernel.builtin_world().object_proto.is_function());
        assert!(kernel.builtin_world().object_constructor.is_function());
    }

    #[test]
    fn test_kernel_shape_forge() {
        let kernel = OxideKernel::new(KernelConfig::minimal());
        assert!(kernel.shape_forge().get_shape(EMPTY_SHAPE_ID).is_some());
    }

    #[test]
    fn test_kernel_string_forge() {
        let kernel = OxideKernel::new(KernelConfig::minimal());
        let (i1, _) = kernel.string_forge().intern("hello");
        let (i2, _) = kernel.string_forge().intern("hello");
        assert_eq!(i1, i2);
    }

    #[test]
    fn test_kernel_config_presets() {
        assert_eq!(KernelConfig::minimal().max_pool_size, Some(8));
        assert_eq!(KernelConfig::standard().max_pool_size, Some(32));
        assert!(!KernelConfig::minimal().warmup_builtin_ic);
        assert!(KernelConfig::full().warmup_builtin_ic);
        assert_eq!(KernelConfig::full().max_pool_size, None);
    }
}