rspack_napi 0.102.9

#![allow(clippy::not_unsafe_ptr_arg_deref)]

use std::{cell::Cell, ptr, rc::Rc};

use napi::{
  Env, JsValue, Result,
  bindgen_prelude::{JsObjectValue, Object, ToNapiValue, check_status},
  sys::{self, napi_env},
};

use crate::OneShotRef;

pub struct WeakRef {
  // Own the zero-count N-API handle without rooting the JS value.
  reference: OneShotRef,
  deleted: Rc<Cell<bool>>,
}

impl WeakRef {
  pub fn new(env: napi_env, object: &mut Object) -> Result<Self> {
    let mut raw_ref = ptr::null_mut();
    check_status!(unsafe { sys::napi_create_reference(env, object.raw(), 0, &mut raw_ref) })?;
    let reference = OneShotRef::from_napi_ref(env, raw_ref)?;

    let deleted = Rc::new(Cell::new(false));
    let deleted_clone = deleted.clone();
    object.add_finalizer((), (), move |_ctx| {
      deleted_clone.set(true);
    })?;

    Ok(Self { reference, deleted })
  }

  pub fn as_object(&self, env: &Env) -> Result<Object<'static>> {
    let napi_val = unsafe { ToNapiValue::to_napi_value(env.raw(), self)? };
    Ok(Object::from_raw(env.raw(), napi_val))
  }
}

impl ToNapiValue for &WeakRef {
  unsafe fn to_napi_value(env: sys::napi_env, val: Self) -> Result<sys::napi_value> {
    if val.deleted.get() {
      return Err(napi::Error::new(
        napi::Status::InvalidArg,
        "WeakRef has been deleted",
      ));
    }
    let result = unsafe { ToNapiValue::to_napi_value(env, &val.reference)? };
    // GC can clear a weak handle before the finalizer marks it deleted.
    if result.is_null() {
      return Err(napi::Error::new(
        napi::Status::InvalidArg,
        "WeakRef has been deleted",
      ));
    }
    Ok(result)
  }
}

impl ToNapiValue for &mut WeakRef {
  unsafe fn to_napi_value(env: sys::napi_env, val: Self) -> Result<sys::napi_value> {
    unsafe { ToNapiValue::to_napi_value(env, &*val) }
  }
}