feat: implment create and deref JsBuffer

This commit is contained in:
LongYinan 2019-05-16 00:41:22 +08:00
parent 40e8ba32a9
commit 211ce1e523
No known key found for this signature in database
GPG key ID: C3666B7FC82ADAD7
2 changed files with 50 additions and 32 deletions

View file

@ -3,8 +3,10 @@ use std::any::TypeId;
use std::ffi::CString; use std::ffi::CString;
use std::marker::PhantomData; use std::marker::PhantomData;
use std::mem; use std::mem;
use std::ops::{Deref, DerefMut};
use std::os::raw::{c_char, c_void}; use std::os::raw::{c_char, c_void};
use std::ptr; use std::ptr;
use std::slice;
use std::string::String as RustString; use std::string::String as RustString;
mod executor; mod executor;
@ -45,6 +47,9 @@ pub struct Object;
#[derive(Clone, Copy, Debug)] #[derive(Clone, Copy, Debug)]
pub struct Function; pub struct Function;
#[derive(Clone, Copy, Debug)]
pub struct Buffer;
#[derive(Clone, Copy, Debug)] #[derive(Clone, Copy, Debug)]
pub struct Value<'env, T> { pub struct Value<'env, T> {
env: &'env Env, env: &'env Env,
@ -263,6 +268,14 @@ impl Env {
Value::from_raw(self, raw_value) Value::from_raw(self, raw_value)
} }
pub fn create_buffer(&self, length: usize) -> Value<Buffer> {
let mut raw_value = ptr::null_mut();
let mut _data = ptr::null_mut();
let status = unsafe { sys::napi_create_buffer(self.0, length, _data, &mut raw_value) };
debug_assert!(Status::from(status) == Status::Ok);
Value::from_raw(self, raw_value)
}
pub fn create_function<'a, 'b>( pub fn create_function<'a, 'b>(
&'a self, &'a self,
name: &'b str, name: &'b str,
@ -479,6 +492,12 @@ impl ValueType for Object {
} }
} }
impl ValueType for Buffer {
fn matches_raw_type(raw_type: sys::napi_valuetype) -> bool {
raw_type == sys::napi_valuetype::napi_external
}
}
impl ValueType for Function { impl ValueType for Function {
fn matches_raw_type(raw_type: sys::napi_valuetype) -> bool { fn matches_raw_type(raw_type: sys::napi_valuetype) -> bool {
raw_type == sys::napi_valuetype::napi_function raw_type == sys::napi_valuetype::napi_function
@ -669,13 +688,8 @@ impl<'env> Value<'env, Object> {
pub fn get_property_names<T: ValueType>(&self) -> Result<Value<T>> { pub fn get_property_names<T: ValueType>(&self) -> Result<Value<T>> {
let mut raw_value = ptr::null_mut(); let mut raw_value = ptr::null_mut();
let status = unsafe { let status =
sys::napi_get_property_names( unsafe { sys::napi_get_property_names(self.raw_env(), self.raw_value(), &mut raw_value) };
self.raw_env(),
self.raw_value(),
&mut raw_value
)
};
check_status(status)?; check_status(status)?;
Value::<Any>::from_raw(self.env, raw_value).try_into() Value::<Any>::from_raw(self.env, raw_value).try_into()
} }
@ -686,27 +700,15 @@ impl<'env> Value<'env, Object> {
pub fn get_index<T: ValueType>(&self, index: u32) -> Result<Value<T>> { pub fn get_index<T: ValueType>(&self, index: u32) -> Result<Value<T>> {
let mut raw_value = ptr::null_mut(); let mut raw_value = ptr::null_mut();
let status = unsafe { let status =
sys::napi_get_element( unsafe { sys::napi_get_element(self.raw_env(), self.raw_value(), index, &mut raw_value) };
self.raw_env(),
self.raw_value(),
index,
&mut raw_value
)
};
check_status(status)?; check_status(status)?;
Value::<Any>::from_raw(self.env, raw_value).try_into() Value::<Any>::from_raw(self.env, raw_value).try_into()
} }
pub fn is_array(&self) -> Result<bool> { pub fn is_array(&self) -> Result<bool> {
let mut is_array = false; let mut is_array = false;
let status = unsafe { let status = unsafe { sys::napi_is_array(self.raw_env(), self.raw_value(), &mut is_array) };
sys::napi_is_array(
self.raw_env(),
self.raw_value(),
&mut is_array,
)
};
check_status(status)?; check_status(status)?;
Ok(is_array) Ok(is_array)
} }
@ -715,16 +717,11 @@ impl<'env> Value<'env, Object> {
if self.is_array()? != true { if self.is_array()? != true {
return Err(Error { return Err(Error {
status: Status::ArrayExpected, status: Status::ArrayExpected,
}) });
} }
let mut length: u32 = 0; let mut length: u32 = 0;
let status = unsafe { let status =
sys::napi_get_array_length( unsafe { sys::napi_get_array_length(self.raw_env(), self.raw_value(), &mut length) };
self.raw_env(),
self.raw_value(),
&mut length
)
};
check_status(status)?; check_status(status)?;
Ok(length) Ok(length)
} }
@ -738,6 +735,28 @@ impl<'env> Value<'env, Object> {
} }
} }
impl<'env> Deref for Value<'env, Buffer> {
type Target = [u8];
fn deref(&self) -> &[u8] {
let mut data = ptr::null_mut();
let size = ptr::null_mut();
let status = unsafe { sys::napi_get_buffer_info(self.env.0, self.raw_value, &mut data, size) };
debug_assert!(Status::from(status) == Status::Ok);
unsafe { slice::from_raw_parts(data as *const _, *size) }
}
}
impl<'env> DerefMut for Value<'env, Buffer> {
fn deref_mut(&mut self) -> &mut [u8] {
let mut data = ptr::null_mut();
let size = ptr::null_mut();
let status = unsafe { sys::napi_get_buffer_info(self.env.0, self.raw_value, &mut data, size) };
debug_assert!(Status::from(status) == Status::Ok);
unsafe { slice::from_raw_parts_mut(data as *mut _, *size) }
}
}
impl<'env> Value<'env, Function> { impl<'env> Value<'env, Function> {
pub fn call( pub fn call(
&self, &self,

View file

@ -2,7 +2,6 @@
#![allow(non_camel_case_types)] #![allow(non_camel_case_types)]
#![allow(non_snake_case)] #![allow(non_snake_case)]
#![allow(dead_code)] #![allow(dead_code)]
#![cfg_attr(feature = "cargo-clippy", allow(expl_impl_clone_on_copy))]
include!("bindings.rs"); include!("bindings.rs");