napi_calm_down/js_values/
buffer.rs

1use std::mem;
2use std::ops::{Deref, DerefMut};
3use std::ptr;
4
5use super::{Value, ValueType};
6use crate::bindgen_runtime::ValidateNapiValue;
7use crate::{
8  bindgen_runtime::TypeName, check_status, sys, Error, JsUnknown, NapiValue, Ref, Result, Status,
9};
10
11pub struct JsBuffer(pub(crate) Value);
12
13impl TypeName for JsBuffer {
14  fn type_name() -> &'static str {
15    "Buffer"
16  }
17
18  fn value_type() -> ValueType {
19    ValueType::Object
20  }
21}
22
23impl ValidateNapiValue for JsBuffer {
24  unsafe fn validate(env: sys::napi_env, napi_val: sys::napi_value) -> Result<sys::napi_value> {
25    let mut is_buffer = false;
26    check_status!(unsafe { sys::napi_is_buffer(env, napi_val, &mut is_buffer) })?;
27    if !is_buffer {
28      return Err(Error::new(
29        Status::InvalidArg,
30        "Value is not a buffer".to_owned(),
31      ));
32    }
33    Ok(ptr::null_mut())
34  }
35}
36
37pub struct JsBufferValue {
38  pub(crate) value: JsBuffer,
39  data: mem::ManuallyDrop<Vec<u8>>,
40}
41
42impl JsBuffer {
43  pub fn into_value(self) -> Result<JsBufferValue> {
44    let mut data = ptr::null_mut();
45    let mut len: usize = 0;
46    check_status!(unsafe {
47      sys::napi_get_buffer_info(self.0.env, self.0.value, &mut data, &mut len)
48    })?;
49    Ok(JsBufferValue {
50      data: mem::ManuallyDrop::new(unsafe { Vec::from_raw_parts(data as *mut _, len, len) }),
51      value: self,
52    })
53  }
54
55  pub fn into_ref(self) -> Result<Ref<JsBufferValue>> {
56    Ref::new(self.0, 1, self.into_value()?)
57  }
58}
59
60impl JsBufferValue {
61  pub fn new(value: JsBuffer, data: mem::ManuallyDrop<Vec<u8>>) -> Self {
62    JsBufferValue { value, data }
63  }
64
65  pub fn into_raw(self) -> JsBuffer {
66    self.value
67  }
68
69  pub fn into_unknown(self) -> JsUnknown {
70    unsafe { JsUnknown::from_raw_unchecked(self.value.0.env, self.value.0.value) }
71  }
72}
73
74impl AsRef<[u8]> for JsBufferValue {
75  fn as_ref(&self) -> &[u8] {
76    self.data.as_slice()
77  }
78}
79
80impl AsMut<[u8]> for JsBufferValue {
81  fn as_mut(&mut self) -> &mut [u8] {
82    self.data.as_mut_slice()
83  }
84}
85
86impl Deref for JsBufferValue {
87  type Target = [u8];
88
89  fn deref(&self) -> &Self::Target {
90    self.data.as_slice()
91  }
92}
93
94impl DerefMut for JsBufferValue {
95  fn deref_mut(&mut self) -> &mut Self::Target {
96    self.data.as_mut_slice()
97  }
98}