use std::mem; use std::ops::{Deref, DerefMut}; use std::ptr; use super::{Value, ValueType}; use crate::{bindgen_runtime::TypeName, check_status, sys, JsUnknown, NapiValue, Ref, Result}; pub struct JsBuffer(pub(crate) Value); impl TypeName for JsBuffer { fn type_name() -> &'static str { "Buffer" } fn value_type() -> ValueType { ValueType::Object } } pub struct JsBufferValue { pub(crate) value: JsBuffer, data: mem::ManuallyDrop>, } impl JsBuffer { pub fn into_value(self) -> Result { let mut data = ptr::null_mut(); let mut len: usize = 0; check_status!(unsafe { sys::napi_get_buffer_info(self.0.env, self.0.value, &mut data, &mut len) })?; Ok(JsBufferValue { data: mem::ManuallyDrop::new(unsafe { Vec::from_raw_parts(data as *mut _, len, len) }), value: self, }) } pub fn into_ref(self) -> Result> { Ref::new(self.0, 1, self.into_value()?) } } impl JsBufferValue { #[cfg(feature = "serde-json")] pub(crate) fn from_raw(env: sys::napi_env, value: sys::napi_value) -> Result { let mut data = ptr::null_mut(); let mut len = 0usize; check_status!(unsafe { sys::napi_get_buffer_info(env, value, &mut data, &mut len as *mut usize as *mut _) })?; Ok(Self { value: JsBuffer(Value { env, value, value_type: ValueType::Object, }), data: mem::ManuallyDrop::new(unsafe { Vec::from_raw_parts(data as *mut _, len, len) }), }) } pub fn new(value: JsBuffer, data: mem::ManuallyDrop>) -> Self { JsBufferValue { value, data } } pub fn into_raw(self) -> JsBuffer { self.value } pub fn into_unknown(self) -> JsUnknown { unsafe { JsUnknown::from_raw_unchecked(self.value.0.env, self.value.0.value) } } } impl AsRef<[u8]> for JsBufferValue { fn as_ref(&self) -> &[u8] { self.data.as_slice() } } impl AsMut<[u8]> for JsBufferValue { fn as_mut(&mut self) -> &mut [u8] { self.data.as_mut_slice() } } impl Deref for JsBufferValue { type Target = [u8]; fn deref(&self) -> &[u8] { self.data.as_slice() } } impl DerefMut for JsBufferValue { fn deref_mut(&mut self) -> &mut Self::Target { self.data.as_mut_slice() } }