naskya/napi-rs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

feat(napi-derive) Support #[napi(strict)] on &T and &mut T. (#1238)

Browse source
This commit is contained in:
Ivan Enderlin 2022-08-06 15:55:35 +02:00 committed by GitHub
parent 0f14799776
commit 6a9fdba8b6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 30 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
crates/backend/src/codegen/fn.rs
View file

@ -188,6 +188,28 @@ impl NapiFn {
path: &syn::PatType, path: &syn::PatType,
) -> TokenStream { ) -> TokenStream {
let ty = &*path.ty; let ty = &*path.ty;
let type_check = if self.return_if_invalid {
quote! {
if let Ok(maybe_promise) = <#ty as napi::bindgen_prelude::ValidateNapiValue>::validate(env, cb.get_arg(#index)) {
if !maybe_promise.is_null() {
return Ok(maybe_promise);
}
} else {
return Ok(std::ptr::null_mut());
}
}
} else if self.strict {
quote! {
let maybe_promise = <#ty as napi::bindgen_prelude::ValidateNapiValue>::validate(env, cb.get_arg(#index))?;
if !maybe_promise.is_null() {
return Ok(maybe_promise);
}
}
} else {
quote! {}
};
match ty { match ty {
syn::Type::Reference(syn::TypeReference { syn::Type::Reference(syn::TypeReference {
mutability: Some(_), mutability: Some(_),
@ -195,36 +217,21 @@ impl NapiFn {
.. ..
}) => { }) => {
quote! { quote! {
let #arg_name = <#elem as napi::bindgen_prelude::FromNapiMutRef>::from_napi_mut_ref(env, cb.get_arg(#index))?; let #arg_name = {
#type_check
<#elem as napi::bindgen_prelude::FromNapiMutRef>::from_napi_mut_ref(env, cb.get_arg(#index))?
};
} }
} }
syn::Type::Reference(syn::TypeReference { elem, .. }) => { syn::Type::Reference(syn::TypeReference { elem, .. }) => {
quote! { quote! {
let #arg_name = <#elem as napi::bindgen_prelude::FromNapiRef>::from_napi_ref(env, cb.get_arg(#index))?; let #arg_name = {
#type_check
<#elem as napi::bindgen_prelude::FromNapiRef>::from_napi_ref(env, cb.get_arg(#index))?
};
} }
} }
_ => { _ => {
let type_check = if self.return_if_invalid {
quote! {
if let Ok(maybe_promise) = <#ty as napi::bindgen_prelude::ValidateNapiValue>::validate(env, cb.get_arg(#index)) {
if !maybe_promise.is_null() {
return Ok(maybe_promise);
}
} else {
return Ok(std::ptr::null_mut());
}
}
} else if self.strict {
quote! {
let maybe_promise = <#ty as napi::bindgen_prelude::ValidateNapiValue>::validate(env, cb.get_arg(#index))?;
if !maybe_promise.is_null() {
return Ok(maybe_promise);
}
}
} else {
quote! {}
};
quote! { quote! {
let #arg_name = { let #arg_name = {
#type_check #type_check