rspack_napi 0.100.0-rc.2

rspack napi
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81

// Copied from https://github.com/napi-rs/napi-rs/blob/main/crates/napi/src/js_values/value_ref.rs
// 1. A new implementation has been added for creating a reference from raw napi_env and napi_value.
// 2. Implementation for &Ref and &mut Ref has been added to trait ToNapiValue.

#![allow(clippy::not_unsafe_ptr_arg_deref)]

use std::ptr;

use napi::{
  Result,
  bindgen_prelude::ToNapiValue,
  check_status, sys,
  sys::{napi_env, napi_value},
};

pub struct Ref {
  pub(crate) raw_ref: sys::napi_ref,
  pub(crate) count: u32,
}

#[allow(clippy::non_send_fields_in_send_ty)]
unsafe impl Send for Ref {}
unsafe impl Sync for Ref {}

impl Ref {
  pub fn new(env: napi_env, value: napi_value, ref_count: u32) -> Result<Ref> {
    let mut raw_ref = ptr::null_mut();
    assert_ne!(ref_count, 0, "Initial `ref_count` must be > 0");
    check_status!(unsafe { sys::napi_create_reference(env, value, ref_count, &mut raw_ref) })?;
    Ok(Ref {
      raw_ref,
      count: ref_count,
    })
  }

  pub fn reference(&mut self, env: napi_env) -> Result<u32> {
    check_status!(unsafe { sys::napi_reference_ref(env, self.raw_ref, &mut self.count) })?;
    Ok(self.count)
  }

  pub fn unref(&mut self, env: napi_env) -> Result<u32> {
    check_status!(unsafe { sys::napi_reference_unref(env, self.raw_ref, &mut self.count) })?;

    if self.count == 0 {
      check_status!(unsafe { sys::napi_delete_reference(env, self.raw_ref) })?;
    }
    Ok(self.count)
  }
}

#[cfg(debug_assertions)]
impl Drop for Ref {
  fn drop(&mut self) {
    debug_assert_eq!(
      self.count, 0,
      "Ref count is not equal to 0 while dropping Ref, potential memory leak"
    );
  }
}

impl ToNapiValue for &Ref {
  unsafe fn to_napi_value(env: sys::napi_env, val: Self) -> Result<sys::napi_value> {
    let mut result = ptr::null_mut();
    check_status!(
      unsafe { sys::napi_get_reference_value(env, val.raw_ref, &mut result) },
      "Failed to get reference value"
    )?;
    Ok(result)
  }
}

impl ToNapiValue for &mut Ref {
  unsafe fn to_napi_value(env: sys::napi_env, val: Self) -> Result<sys::napi_value> {
    let mut result = ptr::null_mut();
    check_status!(
      unsafe { sys::napi_get_reference_value(env, val.raw_ref, &mut result) },
      "Failed to get reference value"
    )?;
    Ok(result)
  }
}