feat(napi): refactor Env::get_global

This commit is contained in:
LongYinan 2020-10-02 23:23:21 +08:00
parent 2338daf9e1
commit a532451fc0
No known key found for this signature in database
GPG key ID: C3666B7FC82ADAD7
7 changed files with 256 additions and 231 deletions

View file

@ -494,10 +494,10 @@ impl Env {
result result
} }
pub fn get_global(&self) -> Result<JsObject> { pub fn get_global(&self) -> Result<JsGlobal> {
let mut raw_global = ptr::null_mut(); let mut raw_global = ptr::null_mut();
check_status(unsafe { sys::napi_get_global(self.0, &mut raw_global) })?; check_status(unsafe { sys::napi_get_global(self.0, &mut raw_global) })?;
Ok(JsObject::from_raw_unchecked(self.0, raw_global)) Ok(JsGlobal::from_raw_unchecked(self.0, raw_global))
} }
pub fn get_napi_version(&self) -> Result<u32> { pub fn get_napi_version(&self) -> Result<u32> {

View file

@ -4,7 +4,7 @@ use std::ptr;
use super::Value; use super::Value;
use crate::error::check_status; use crate::error::check_status;
use crate::{sys, JsUnknown, Ref, Result}; use crate::{sys, JsUnknown, NapiValue, Ref, Result};
#[repr(transparent)] #[repr(transparent)]
#[derive(Debug, Clone, Copy)] #[derive(Debug, Clone, Copy)]
@ -49,8 +49,8 @@ impl JsArrayBufferValue {
self.value self.value
} }
pub fn into_unknown(self) -> Result<JsUnknown> { pub fn into_unknown(self) -> JsUnknown {
self.value.into_unknown() JsUnknown::from_raw_unchecked(self.value.0.env, self.value.0.value)
} }
} }

View file

@ -6,7 +6,7 @@ use super::Value;
#[cfg(feature = "serde-json")] #[cfg(feature = "serde-json")]
use super::ValueType; use super::ValueType;
use crate::error::check_status; use crate::error::check_status;
use crate::{sys, JsUnknown, Ref, Result}; use crate::{sys, JsUnknown, NapiValue, Ref, Result};
#[repr(transparent)] #[repr(transparent)]
#[derive(Debug, Clone, Copy)] #[derive(Debug, Clone, Copy)]
@ -68,8 +68,8 @@ impl JsBufferValue {
self.value self.value
} }
pub fn into_unknown(self) -> Result<JsUnknown> { pub fn into_unknown(self) -> JsUnknown {
self.value.into_unknown() JsUnknown::from_raw_unchecked(self.value.0.env, self.value.0.value)
} }
} }

View file

@ -1,4 +1,3 @@
use std::mem;
use std::ptr; use std::ptr;
use super::Value; use super::Value;
@ -38,10 +37,10 @@ impl JsFunction {
Status::Unknown, Status::Unknown,
"Get raw this failed".to_owned(), "Get raw this failed".to_owned(),
))?; ))?;
let mut raw_args = unsafe { mem::MaybeUninit::<[sys::napi_value; 8]>::uninit().assume_init() }; let raw_args = args
for (i, arg) in args.into_iter().enumerate() { .iter()
raw_args[i] = arg.0.value; .map(|arg| arg.0.value)
} .collect::<Vec<sys::napi_value>>();
let mut return_value = ptr::null_mut(); let mut return_value = ptr::null_mut();
check_status(unsafe { check_status(unsafe {
sys::napi_call_function( sys::napi_call_function(
@ -49,7 +48,7 @@ impl JsFunction {
raw_this, raw_this,
self.0.value, self.0.value,
args.len() as u64, args.len() as u64,
&raw_args[0], raw_args.as_ptr(),
&mut return_value, &mut return_value,
) )
})?; })?;

View file

@ -0,0 +1,29 @@
use std::convert::TryInto;
use super::*;
use crate::Env;
#[repr(transparent)]
#[derive(Debug)]
pub struct JsGlobal(pub(crate) Value);
#[repr(transparent)]
#[derive(Debug)]
pub struct JsTimeout(pub(crate) Value);
impl JsGlobal {
pub fn set_interval(&self, handler: JsFunction, interval: f64) -> Result<JsTimeout> {
let func: JsFunction = self.get_named_property("setInterval")?;
func
.call(
None,
&[
handler.into_unknown(),
Env::from_raw(self.0.env)
.create_double(interval)?
.into_unknown(),
],
)
.and_then(|ret| ret.try_into())
}
}

View file

@ -1,4 +1,5 @@
use std::convert::{From, TryFrom}; use std::convert::{From, TryFrom};
use std::ffi::CString;
use std::ptr; use std::ptr;
use crate::error::check_status; use crate::error::check_status;
@ -17,6 +18,7 @@ mod buffer;
mod either; mod either;
mod escapable_handle_scope; mod escapable_handle_scope;
mod function; mod function;
mod global;
mod number; mod number;
mod object; mod object;
mod object_property; mod object_property;
@ -37,6 +39,7 @@ pub(crate) use de::De;
pub use either::Either; pub use either::Either;
pub use escapable_handle_scope::EscapableHandleScope; pub use escapable_handle_scope::EscapableHandleScope;
pub use function::JsFunction; pub use function::JsFunction;
pub use global::*;
pub use number::JsNumber; pub use number::JsNumber;
pub use object::JsObject; pub use object::JsObject;
pub use object_property::Property; pub use object_property::Property;
@ -120,8 +123,8 @@ macro_rules! impl_js_value_methods {
($js_value:ident) => { ($js_value:ident) => {
impl $js_value { impl $js_value {
#[inline] #[inline]
pub fn into_unknown(self) -> Result<JsUnknown> { pub fn into_unknown(self) -> JsUnknown {
JsUnknown::from_raw(self.0.env, self.0.value) JsUnknown::from_raw_unchecked(self.0.env, self.0.value)
} }
#[inline] #[inline]
@ -219,6 +222,206 @@ macro_rules! impl_js_value_methods {
}; };
} }
macro_rules! impl_object_methods {
($js_value:ident) => {
impl $js_value {
pub fn set_property<V>(&mut self, key: JsString, value: V) -> Result<()>
where
V: NapiValue,
{
check_status(unsafe {
sys::napi_set_property(self.0.env, self.0.value, key.0.value, value.raw())
})
}
pub fn get_property<K, T>(&self, key: &K) -> Result<T>
where
K: NapiValue,
T: NapiValue,
{
let mut raw_value = ptr::null_mut();
check_status(unsafe {
sys::napi_get_property(self.0.env, self.0.value, key.raw(), &mut raw_value)
})?;
T::from_raw(self.0.env, raw_value)
}
pub fn set_named_property<T>(&mut self, name: &str, value: T) -> Result<()>
where
T: NapiValue,
{
let key = CString::new(name)?;
check_status(unsafe {
sys::napi_set_named_property(self.0.env, self.0.value, key.as_ptr(), value.raw())
})
}
pub fn get_named_property<T>(&self, name: &str) -> Result<T>
where
T: NapiValue,
{
let key = CString::new(name)?;
let mut raw_value = ptr::null_mut();
check_status(unsafe {
sys::napi_get_named_property(self.0.env, self.0.value, key.as_ptr(), &mut raw_value)
})?;
T::from_raw(self.0.env, raw_value)
}
pub fn has_named_property<S>(&self, name: S) -> Result<bool>
where
S: AsRef<str>,
{
let mut result = false;
let key = CString::new(name.as_ref())?;
check_status(unsafe {
sys::napi_has_named_property(self.0.env, self.0.value, key.as_ptr(), &mut result)
})?;
Ok(result)
}
pub fn delete_property<S>(&mut self, name: S) -> Result<bool>
where
S: NapiValue,
{
let mut result = false;
check_status(unsafe {
sys::napi_delete_property(self.0.env, self.0.value, name.raw(), &mut result)
})?;
Ok(result)
}
pub fn delete_named_property(&mut self, name: &str) -> Result<bool> {
let mut result = false;
let key_str = CString::new(name)?;
let mut js_key = ptr::null_mut();
check_status(unsafe {
sys::napi_create_string_utf8(self.0.env, key_str.as_ptr(), name.len() as _, &mut js_key)
})?;
check_status(unsafe {
sys::napi_delete_property(self.0.env, self.0.value, js_key, &mut result)
})?;
Ok(result)
}
pub fn has_own_property(&self, key: &str) -> Result<bool> {
let mut result = false;
let string = CString::new(key)?;
let mut js_key = ptr::null_mut();
check_status(unsafe {
sys::napi_create_string_utf8(self.0.env, string.as_ptr(), key.len() as _, &mut js_key)
})?;
check_status(unsafe {
sys::napi_has_own_property(self.0.env, self.0.value, js_key, &mut result)
})?;
Ok(result)
}
pub fn has_own_property_js<K>(&self, key: K) -> Result<bool>
where
K: NapiValue,
{
let mut result = false;
check_status(unsafe {
sys::napi_has_own_property(self.0.env, self.0.value, key.raw(), &mut result)
})?;
Ok(result)
}
pub fn has_property(&self, name: &str) -> Result<bool> {
let string = CString::new(name)?;
let mut js_key = ptr::null_mut();
let mut result = false;
check_status(unsafe {
sys::napi_create_string_utf8(self.0.env, string.as_ptr(), name.len() as _, &mut js_key)
})?;
check_status(unsafe {
sys::napi_has_property(self.0.env, self.0.value, js_key, &mut result)
})?;
Ok(result)
}
pub fn has_property_js<K>(&self, name: K) -> Result<bool>
where
K: NapiValue,
{
let mut result = false;
check_status(unsafe {
sys::napi_has_property(self.0.env, self.0.value, name.raw(), &mut result)
})?;
Ok(result)
}
pub fn get_property_names<T>(&self) -> Result<T>
where
T: NapiValue,
{
let mut raw_value = ptr::null_mut();
let status =
unsafe { sys::napi_get_property_names(self.0.env, self.0.value, &mut raw_value) };
check_status(status)?;
T::from_raw(self.0.env, raw_value)
}
pub fn get_prototype<T>(&self) -> Result<T>
where
T: NapiValue,
{
let mut result = ptr::null_mut();
check_status(unsafe { sys::napi_get_prototype(self.0.env, self.0.value, &mut result) })?;
T::from_raw(self.0.env, result)
}
pub fn set_element<T>(&mut self, index: u32, value: T) -> Result<()>
where
T: NapiValue,
{
check_status(unsafe { sys::napi_set_element(self.0.env, self.0.value, index, value.raw()) })
}
pub fn has_element(&self, index: u32) -> Result<bool> {
let mut result = false;
check_status(unsafe {
sys::napi_has_element(self.0.env, self.0.value, index, &mut result)
})?;
Ok(result)
}
pub fn delete_element(&mut self, index: u32) -> Result<bool> {
let mut result = false;
check_status(unsafe {
sys::napi_delete_element(self.0.env, self.0.value, index, &mut result)
})?;
Ok(result)
}
pub fn get_element<T>(&self, index: u32) -> Result<T>
where
T: NapiValue,
{
let mut raw_value = ptr::null_mut();
check_status(unsafe {
sys::napi_get_element(self.0.env, self.0.value, index, &mut raw_value)
})?;
T::from_raw(self.0.env, raw_value)
}
pub fn define_properties(&mut self, properties: &[Property]) -> Result<()> {
check_status(unsafe {
sys::napi_define_properties(
self.0.env,
self.0.value,
properties.len() as _,
properties
.iter()
.map(|property| property.raw())
.collect::<Vec<sys::napi_property_descriptor>>()
.as_ptr(),
)
})
}
pub fn get_array_length(&self) -> Result<u32> {
if self.is_array()? != true {
return Err(Error::new(
Status::ArrayExpected,
"Object is not array".to_owned(),
));
}
self.get_array_length_unchecked()
}
#[inline]
pub fn get_array_length_unchecked(&self) -> Result<u32> {
let mut length: u32 = 0;
check_status(unsafe { sys::napi_get_array_length(self.0.env, self.raw(), &mut length) })?;
Ok(length)
}
}
};
}
pub trait NapiValue: Sized { pub trait NapiValue: Sized {
fn from_raw(env: sys::napi_env, value: sys::napi_value) -> Result<Self>; fn from_raw(env: sys::napi_env, value: sys::napi_value) -> Result<Self>;
@ -236,9 +439,16 @@ impl_js_value_methods!(JsBuffer);
impl_js_value_methods!(JsNumber); impl_js_value_methods!(JsNumber);
impl_js_value_methods!(JsString); impl_js_value_methods!(JsString);
impl_js_value_methods!(JsObject); impl_js_value_methods!(JsObject);
impl_js_value_methods!(JsGlobal);
impl_js_value_methods!(JsFunction); impl_js_value_methods!(JsFunction);
impl_js_value_methods!(JsExternal); impl_js_value_methods!(JsExternal);
impl_js_value_methods!(JsSymbol); impl_js_value_methods!(JsSymbol);
impl_js_value_methods!(JsTimeout);
impl_object_methods!(JsObject);
impl_object_methods!(JsBuffer);
impl_object_methods!(JsArrayBuffer);
impl_object_methods!(JsGlobal);
use ValueType::*; use ValueType::*;
@ -250,6 +460,8 @@ impl_napi_value_trait!(JsArrayBuffer, Object);
impl_napi_value_trait!(JsNumber, Number); impl_napi_value_trait!(JsNumber, Number);
impl_napi_value_trait!(JsString, String); impl_napi_value_trait!(JsString, String);
impl_napi_value_trait!(JsObject, Object); impl_napi_value_trait!(JsObject, Object);
impl_napi_value_trait!(JsGlobal, Object);
impl_napi_value_trait!(JsTimeout, Object);
impl_napi_value_trait!(JsFunction, Function); impl_napi_value_trait!(JsFunction, Function);
impl_napi_value_trait!(JsExternal, External); impl_napi_value_trait!(JsExternal, External);
impl_napi_value_trait!(JsSymbol, Symbol); impl_napi_value_trait!(JsSymbol, Symbol);

View file

@ -1,220 +1,5 @@
use std::ffi::CString;
use std::ptr;
use super::Value; use super::Value;
use crate::error::check_status;
use crate::{sys, Error, JsString, NapiValue, Property, Result, Status};
#[repr(transparent)] #[repr(transparent)]
#[derive(Debug)] #[derive(Debug)]
pub struct JsObject(pub(crate) Value); pub struct JsObject(pub(crate) Value);
impl JsObject {
pub fn set_property<V>(&mut self, key: JsString, value: V) -> Result<()>
where
V: NapiValue,
{
check_status(unsafe {
sys::napi_set_property(self.0.env, self.0.value, key.0.value, value.raw())
})
}
pub fn get_property<K, T>(&self, key: &K) -> Result<T>
where
K: NapiValue,
T: NapiValue,
{
let mut raw_value = ptr::null_mut();
check_status(unsafe {
sys::napi_get_property(self.0.env, self.0.value, key.raw(), &mut raw_value)
})?;
T::from_raw(self.0.env, raw_value)
}
pub fn set_named_property<T>(&mut self, name: &str, value: T) -> Result<()>
where
T: NapiValue,
{
let key = CString::new(name)?;
check_status(unsafe {
sys::napi_set_named_property(self.0.env, self.0.value, key.as_ptr(), value.raw())
})
}
pub fn get_named_property<T>(&self, name: &str) -> Result<T>
where
T: NapiValue,
{
let key = CString::new(name)?;
let mut raw_value = ptr::null_mut();
check_status(unsafe {
sys::napi_get_named_property(self.0.env, self.0.value, key.as_ptr(), &mut raw_value)
})?;
T::from_raw(self.0.env, raw_value)
}
pub fn has_named_property<S>(&self, name: S) -> Result<bool>
where
S: AsRef<str>,
{
let mut result = false;
let key = CString::new(name.as_ref())?;
check_status(unsafe {
sys::napi_has_named_property(self.0.env, self.0.value, key.as_ptr(), &mut result)
})?;
Ok(result)
}
pub fn delete_property<S>(&mut self, name: S) -> Result<bool>
where
S: NapiValue,
{
let mut result = false;
check_status(unsafe {
sys::napi_delete_property(self.0.env, self.0.value, name.raw(), &mut result)
})?;
Ok(result)
}
pub fn delete_named_property(&mut self, name: &str) -> Result<bool> {
let mut result = false;
let key_str = CString::new(name)?;
let mut js_key = ptr::null_mut();
check_status(unsafe {
sys::napi_create_string_utf8(self.0.env, key_str.as_ptr(), name.len() as _, &mut js_key)
})?;
check_status(unsafe {
sys::napi_delete_property(self.0.env, self.0.value, js_key, &mut result)
})?;
Ok(result)
}
pub fn has_own_property(&self, key: &str) -> Result<bool> {
let mut result = false;
let string = CString::new(key)?;
let mut js_key = ptr::null_mut();
check_status(unsafe {
sys::napi_create_string_utf8(self.0.env, string.as_ptr(), key.len() as _, &mut js_key)
})?;
check_status(unsafe {
sys::napi_has_own_property(self.0.env, self.0.value, js_key, &mut result)
})?;
Ok(result)
}
pub fn has_own_property_js<K>(&self, key: K) -> Result<bool>
where
K: NapiValue,
{
let mut result = false;
check_status(unsafe {
sys::napi_has_own_property(self.0.env, self.0.value, key.raw(), &mut result)
})?;
Ok(result)
}
pub fn has_property(&self, name: &str) -> Result<bool> {
let string = CString::new(name)?;
let mut js_key = ptr::null_mut();
let mut result = false;
check_status(unsafe {
sys::napi_create_string_utf8(self.0.env, string.as_ptr(), name.len() as _, &mut js_key)
})?;
check_status(unsafe { sys::napi_has_property(self.0.env, self.0.value, js_key, &mut result) })?;
Ok(result)
}
pub fn has_property_js<K>(&self, name: K) -> Result<bool>
where
K: NapiValue,
{
let mut result = false;
check_status(unsafe {
sys::napi_has_property(self.0.env, self.0.value, name.raw(), &mut result)
})?;
Ok(result)
}
pub fn get_property_names<T>(&self) -> Result<T>
where
T: NapiValue,
{
let mut raw_value = ptr::null_mut();
let status = unsafe { sys::napi_get_property_names(self.0.env, self.0.value, &mut raw_value) };
check_status(status)?;
T::from_raw(self.0.env, raw_value)
}
pub fn get_prototype<T>(&self) -> Result<T>
where
T: NapiValue,
{
let mut result = ptr::null_mut();
check_status(unsafe { sys::napi_get_prototype(self.0.env, self.0.value, &mut result) })?;
T::from_raw(self.0.env, result)
}
pub fn set_element<T>(&mut self, index: u32, value: T) -> Result<()>
where
T: NapiValue,
{
check_status(unsafe { sys::napi_set_element(self.0.env, self.0.value, index, value.raw()) })
}
pub fn has_element(&self, index: u32) -> Result<bool> {
let mut result = false;
check_status(unsafe { sys::napi_has_element(self.0.env, self.0.value, index, &mut result) })?;
Ok(result)
}
pub fn delete_element(&mut self, index: u32) -> Result<bool> {
let mut result = false;
check_status(unsafe {
sys::napi_delete_element(self.0.env, self.0.value, index, &mut result)
})?;
Ok(result)
}
pub fn get_element<T>(&self, index: u32) -> Result<T>
where
T: NapiValue,
{
let mut raw_value = ptr::null_mut();
check_status(unsafe {
sys::napi_get_element(self.0.env, self.0.value, index, &mut raw_value)
})?;
T::from_raw(self.0.env, raw_value)
}
pub fn define_properties(&mut self, properties: &[Property]) -> Result<()> {
check_status(unsafe {
sys::napi_define_properties(
self.0.env,
self.0.value,
properties.len() as _,
properties
.iter()
.map(|property| property.raw())
.collect::<Vec<sys::napi_property_descriptor>>()
.as_ptr(),
)
})
}
pub fn get_array_length(&self) -> Result<u32> {
if self.is_array()? != true {
return Err(Error::new(
Status::ArrayExpected,
"Object is not array".to_owned(),
));
}
self.get_array_length_unchecked()
}
#[inline]
pub fn get_array_length_unchecked(&self) -> Result<u32> {
let mut length: u32 = 0;
check_status(unsafe { sys::napi_get_array_length(self.0.env, self.raw(), &mut length) })?;
Ok(length)
}
}