napi-rs/test_module/src/lib.rs

#[macro_use]
extern crate napi;
#[macro_use]
extern crate napi_derive;

use napi::{CallContext, Error, JsString, JsUnknown, Module, Result, Status};

#[cfg(napi4)]
mod libuv;
#[cfg(napi4)]
mod napi4;
#[cfg(napi5)]
mod napi5;
#[cfg(napi4)]
mod tokio_rt;

mod buffer;
mod either;
mod external;
mod function;
mod napi_version;
mod symbol;
mod task;

use buffer::{buffer_to_string, get_buffer_length};
use either::{dynamic_argument_length, either_number_string};
use external::{create_external, get_external_count};
use function::{call_function, call_function_with_this};
#[cfg(napi4)]
use libuv::read_file::uv_read_file;
#[cfg(napi4)]
use napi4::{test_threadsafe_function, test_tokio_readfile, test_tsfn_error};
#[cfg(napi5)]
use napi5::is_date::test_object_is_date;
use napi_version::get_napi_version;
use symbol::{create_named_symbol, create_symbol_from_js_string, create_unnamed_symbol};
use task::test_spawn_thread;
#[cfg(napi4)]
use tokio_rt::{error_from_tokio_future, test_execute_tokio_readfile};

register_module!(test_module, init);

fn init(module: &mut Module) -> Result<()> {
  module.create_named_method("testThrow", test_throw)?;
  module.create_named_method("testThrowWithReason", test_throw_with_reason)?;
  module.create_named_method("testSpawnThread", test_spawn_thread)?;
  module.create_named_method("createExternal", create_external)?;
  module.create_named_method("getExternalCount", get_external_count)?;
  module.create_named_method("getBufferLength", get_buffer_length)?;
  module.create_named_method("bufferToString", buffer_to_string)?;
  module.create_named_method("createNamedSymbol", create_named_symbol)?;
  module.create_named_method("createUnnamedSymbol", create_unnamed_symbol)?;
  module.create_named_method("createSymbolFromJsString", create_symbol_from_js_string)?;
  module.create_named_method("getNapiVersion", get_napi_version)?;
  module.create_named_method("testCallFunction", call_function)?;
  module.create_named_method("testCallFunctionWithThis", call_function_with_this)?;
  module.create_named_method("eitherNumberString", either_number_string)?;
  module.create_named_method("dynamicArgumentLength", dynamic_argument_length)?;
  #[cfg(napi4)]
  module.create_named_method("testExecuteTokioReadfile", test_execute_tokio_readfile)?;
  #[cfg(napi4)]
  module.create_named_method("testTsfnError", test_tsfn_error)?;
  #[cfg(napi4)]
  module.create_named_method("testThreadsafeFunction", test_threadsafe_function)?;
  #[cfg(napi4)]
  module.create_named_method("testTokioReadfile", test_tokio_readfile)?;
  #[cfg(napi4)]
  module.create_named_method("testTokioError", error_from_tokio_future)?;
  #[cfg(napi4)]
  module.create_named_method("uvReadFile", uv_read_file)?;
  #[cfg(napi5)]
  module.create_named_method("testObjectIsDate", test_object_is_date)?;
  Ok(())
}

#[js_function]
fn test_throw(_ctx: CallContext) -> Result<JsUnknown> {
  Err(Error::from_status(Status::GenericFailure))
}

#[js_function(1)]
fn test_throw_with_reason(ctx: CallContext) -> Result<JsUnknown> {
  let reason = ctx.get::<JsString>(0)?;
  Err(Error::new(
    Status::GenericFailure,
    reason.as_str()?.to_owned(),
  ))
}

#[js_function]
pub fn test_throw_with_panic(_ctx: CallContext) -> Result<JsUnknown> {
  panic!("don't panic.");
}