napi-rs/napi/src/js_values/arraybuffer.rs

use std::convert::TryFrom;
use std::convert::TryInto;
use std::mem;
use std::ops::Deref;
use std::os::raw::c_void;
use std::ptr;

use super::{Value, ValueType};
use crate::error::check_status;
use crate::{sys, Error, JsUnknown, NapiValue, Ref, Result, Status};

#[repr(transparent)]
#[derive(Debug)]
pub struct JsArrayBuffer(pub(crate) Value);

#[derive(Debug)]
pub struct JsArrayBufferValue {
  pub(crate) value: JsArrayBuffer,
  data: mem::ManuallyDrop<Vec<u8>>,
}

#[repr(transparent)]
#[derive(Debug)]
pub struct JsTypedArray(pub(crate) Value);

#[derive(Debug)]
pub struct JsTypedArrayValue {
  pub arraybuffer: JsArrayBuffer,
  data: *mut c_void,
  pub byte_offset: u64,
  pub length: u64,
  pub typedarray_type: TypedArrayType,
}

#[repr(transparent)]
#[derive(Debug)]
pub struct JsDataView(pub(crate) Value);

#[derive(Debug)]
pub struct JsDataViewValue {
  pub arraybuffer: JsArrayBuffer,
  data: *mut c_void,
  pub byte_offset: u64,
  pub length: u64,
}

#[derive(Debug)]
pub enum TypedArrayType {
  Int8,
  Uint8,
  Uint8Clamped,
  Int16,
  Uint16,
  Int32,
  Uint32,
  Float32,
  Float64,
  #[cfg(napi6)]
  BigInt64,
  #[cfg(napi6)]
  BigUint64,
}

impl TryFrom<sys::napi_typedarray_type> for TypedArrayType {
  type Error = Error;

  fn try_from(value: sys::napi_typedarray_type) -> Result<Self> {
    match value {
      sys::napi_typedarray_type::napi_int8_array => Ok(Self::Int8),
      sys::napi_typedarray_type::napi_uint8_array => Ok(Self::Uint8),
      sys::napi_typedarray_type::napi_uint8_clamped_array => Ok(Self::Uint8Clamped),
      sys::napi_typedarray_type::napi_int16_array => Ok(Self::Int16),
      sys::napi_typedarray_type::napi_uint16_array => Ok(Self::Uint16),
      sys::napi_typedarray_type::napi_int32_array => Ok(Self::Int32),
      sys::napi_typedarray_type::napi_uint32_array => Ok(Self::Uint32),
      sys::napi_typedarray_type::napi_float32_array => Ok(Self::Float32),
      sys::napi_typedarray_type::napi_float64_array => Ok(Self::Float64),
      #[cfg(napi6)]
      sys::napi_typedarray_type::napi_bigint64_array => Ok(Self::BigInt64),
      #[cfg(napi6)]
      sys::napi_typedarray_type::napi_biguint64_array => Ok(Self::BigUint64),
      _ => Err(Error::from_status(Status::Unknown)),
    }
  }
}

impl From<TypedArrayType> for sys::napi_typedarray_type {
  fn from(value: TypedArrayType) -> Self {
    match value {
      TypedArrayType::Int8 => sys::napi_typedarray_type::napi_int8_array,
      TypedArrayType::Uint8 => sys::napi_typedarray_type::napi_uint8_array,
      TypedArrayType::Uint8Clamped => sys::napi_typedarray_type::napi_uint8_clamped_array,
      TypedArrayType::Int16 => sys::napi_typedarray_type::napi_int16_array,
      TypedArrayType::Uint16 => sys::napi_typedarray_type::napi_uint16_array,
      TypedArrayType::Int32 => sys::napi_typedarray_type::napi_int32_array,
      TypedArrayType::Uint32 => sys::napi_typedarray_type::napi_uint32_array,
      TypedArrayType::Float32 => sys::napi_typedarray_type::napi_float32_array,
      TypedArrayType::Float64 => sys::napi_typedarray_type::napi_float64_array,
      #[cfg(napi6)]
      TypedArrayType::BigInt64 => sys::napi_typedarray_type::napi_bigint64_array,
      #[cfg(napi6)]
      TypedArrayType::BigUint64 => sys::napi_typedarray_type::napi_biguint64_array,
    }
  }
}

impl JsArrayBuffer {
  #[cfg(napi7)]
  pub fn detach(self) -> Result<()> {
    check_status(unsafe { sys::napi_detach_arraybuffer(self.0.env, self.0.value) })
  }

  #[cfg(napi7)]
  pub fn is_detached(&self) -> Result<bool> {
    let mut is_detached = false;
    check_status(unsafe {
      sys::napi_is_detached_arraybuffer(self.0.env, self.0.value, &mut is_detached)
    })?;
    Ok(is_detached)
  }

  pub fn into_value(self) -> Result<JsArrayBufferValue> {
    let mut data = ptr::null_mut();
    let mut len: u64 = 0;
    check_status(unsafe {
      sys::napi_get_arraybuffer_info(self.0.env, self.0.value, &mut data, &mut len)
    })?;
    Ok(JsArrayBufferValue {
      data: mem::ManuallyDrop::new(unsafe {
        Vec::from_raw_parts(data as *mut _, len as usize, len as usize)
      }),
      value: self,
    })
  }

  pub fn into_typedarray(
    self,
    typedarray_type: TypedArrayType,
    length: u64,
    byte_offset: u64,
  ) -> Result<JsTypedArray> {
    let mut typedarray_value = ptr::null_mut();
    check_status(unsafe {
      sys::napi_create_typedarray(
        self.0.env,
        typedarray_type.into(),
        length,
        self.0.value,
        byte_offset,
        &mut typedarray_value,
      )
    })?;
    Ok(JsTypedArray(Value {
      env: self.0.env,
      value: typedarray_value,
      value_type: ValueType::Object,
    }))
  }

  pub fn into_dataview(self, length: u64, byte_offset: u64) -> Result<JsDataView> {
    let mut dataview_value = ptr::null_mut();
    check_status(unsafe {
      sys::napi_create_dataview(
        self.0.env,
        length,
        self.0.value,
        byte_offset,
        &mut dataview_value,
      )
    })?;
    Ok(JsDataView(Value {
      env: self.0.env,
      value: dataview_value,
      value_type: ValueType::Object,
    }))
  }

  #[inline]
  pub fn into_ref(self) -> Result<Ref<JsArrayBufferValue>> {
    Ref::new(self.0, 1, self.into_value()?)
  }
}

impl JsArrayBufferValue {
  pub fn new(value: JsArrayBuffer, data: Vec<u8>) -> Self {
    JsArrayBufferValue {
      value,
      data: mem::ManuallyDrop::new(data),
    }
  }

  pub fn into_raw(self) -> JsArrayBuffer {
    self.value
  }

  pub fn into_unknown(self) -> JsUnknown {
    JsUnknown::from_raw_unchecked(self.value.0.env, self.value.0.value)
  }
}

impl AsRef<[u8]> for JsArrayBufferValue {
  fn as_ref(&self) -> &[u8] {
    self.data.as_slice()
  }
}

impl Deref for JsArrayBufferValue {
  type Target = [u8];

  fn deref(&self) -> &[u8] {
    self.data.as_slice()
  }
}

impl JsTypedArray {
  /// get TypeArray info
  /// https://nodejs.org/api/n-api.html#n_api_napi_get_typedarray_info
  ///
  /// ***Warning***: Use caution while using this API since the underlying data buffer is managed by the VM.
  pub fn into_value(self) -> Result<JsTypedArrayValue> {
    let mut typedarray_type = sys::napi_typedarray_type::napi_int8_array;
    let mut len = 0u64;
    let mut data = ptr::null_mut();
    let mut arraybuffer_value = ptr::null_mut();
    let mut byte_offset = 0u64;
    check_status(unsafe {
      sys::napi_get_typedarray_info(
        self.0.env,
        self.0.value,
        &mut typedarray_type,
        &mut len,
        &mut data,
        &mut arraybuffer_value,
        &mut byte_offset,
      )
    })?;

    Ok(JsTypedArrayValue {
      data,
      length: len,
      byte_offset,
      typedarray_type: typedarray_type.try_into()?,
      arraybuffer: JsArrayBuffer::from_raw_unchecked(self.0.env, arraybuffer_value),
    })
  }
}

impl JsDataView {
  pub fn into_value(self) -> Result<JsDataViewValue> {
    let mut length = 0u64;
    let mut byte_offset = 0u64;
    let mut arraybuffer_value = ptr::null_mut();
    let mut data = ptr::null_mut();

    check_status(unsafe {
      sys::napi_get_dataview_info(
        self.0.env,
        self.0.value,
        &mut length,
        &mut data,
        &mut arraybuffer_value,
        &mut byte_offset,
      )
    })?;
    Ok(JsDataViewValue {
      arraybuffer: JsArrayBuffer::from_raw_unchecked(self.0.env, arraybuffer_value),
      byte_offset,
      length,
      data,
    })
  }
}