allocator.rs

use std::sync::Arc;

use napi::{Env, bindgen_prelude::ToNapiValue, sys::napi_env};
use rspack_core::BindingCell;

use crate::{
  asset::AssetInfo,
  build_info::Assets,
  compilation::{CodeGenerationResult, CodeGenerationResults, Sources},
};

pub(crate) struct NapiAllocatorImpl;

impl NapiAllocatorImpl {
  pub fn new(_env: Env) -> Self {
    Self
  }
}

impl rspack_core::NapiAllocator for NapiAllocatorImpl {
  fn allocate_asset_info(
    &self,
    env: napi_env,
    val: &BindingCell<rspack_core::AssetInfo>,
  ) -> napi::Result<napi::sys::napi_value> {
    let asset_info: AssetInfo = (**val).clone().into();
    unsafe { ToNapiValue::to_napi_value(env, asset_info) }
  }

  fn allocate_code_generation_result(
    &self,
    env: napi_env,
    val: &BindingCell<rspack_core::CodeGenerationResult>,
  ) -> napi::Result<napi::sys::napi_value> {
    let code_generation_result = CodeGenerationResult::new(val.downgrade());
    unsafe { ToNapiValue::to_napi_value(env, code_generation_result) }
  }

  fn allocate_sources(
    &self,
    env: napi_env,
    val: &BindingCell<
      rustc_hash::FxHashMap<rspack_core::SourceType, rspack_core::rspack_sources::BoxSource>,
    >,
  ) -> napi::Result<napi::sys::napi_value> {
    let sources = Sources::new(val.downgrade());
    unsafe { ToNapiValue::to_napi_value(env, sources) }
  }

  fn allocate_code_generation_results(
    &self,
    env: napi_env,
    val: &BindingCell<rspack_core::CodeGenerationResults>,
  ) -> napi::Result<napi::sys::napi_value> {
    let code_generation_results = CodeGenerationResults::new(val.downgrade());
    unsafe { ToNapiValue::to_napi_value(env, code_generation_results) }
  }

  fn allocate_assets(
    &self,
    env: napi_env,
    val: &BindingCell<rustc_hash::FxHashMap<Arc<str>, rspack_core::CompilationAsset>>,
  ) -> napi::Result<napi::sys::napi_value> {
    let assets = Assets::new(val.downgrade());
    unsafe { ToNapiValue::to_napi_value(env, assets) }
  }
}