napi-rs/test_module/src/either.rs
LongYinan 1a3621b727
feat(napi): major upgrades for napi@1
1. inline everything
2. change `check_status` and `type_of` to macro
3. provide #[module_exports] macro
4. remove debug and repr[transparent] for ffi struct
2020-11-26 11:31:49 +08:00

35 lines
1.1 KiB
Rust

use std::convert::TryInto;
use napi::{CallContext, Either, JsNumber, JsObject, JsString, Result};
#[js_function(1)]
pub fn either_number_string(ctx: CallContext) -> Result<Either<JsNumber, JsString>> {
let arg = ctx.get::<Either<JsNumber, JsString>>(0)?;
match arg {
Either::A(n) => {
let n: u32 = n.try_into()?;
ctx.env.create_uint32(n + 100).map(Either::A)
}
Either::B(s) => {
let content = format!("Either::B({})", s.into_utf8()?.as_str()?);
ctx.env.create_string_from_std(content).map(Either::B)
}
}
}
#[js_function(1)]
pub fn dynamic_argument_length(ctx: CallContext) -> Result<JsNumber> {
let value: Option<JsNumber> = ctx.try_get::<JsNumber>(0)?.into();
if let Some(n) = value {
let n: u32 = n.try_into()?;
ctx.env.create_uint32(n + 100)
} else {
ctx.env.create_uint32(42)
}
}
pub fn register_js(exports: &mut JsObject) -> Result<()> {
exports.create_named_method("eitherNumberString", either_number_string)?;
exports.create_named_method("dynamicArgumentLength", dynamic_argument_length)?;
Ok(())
}