use napi::{
  CallContext, CleanupEnvHook, ContextlessResult, Env, JsExternal, JsObject, JsUndefined, Result,
};

#[contextless_function]
fn add_cleanup_hook(mut env: Env) -> ContextlessResult<JsExternal> {
  let hook = env.add_env_cleanup_hook((), |_| {
    println!("cleanup hook executed");
  })?;
  env.create_external(hook, None).map(Some)
}

#[js_function(1)]
fn remove_cleanup_hook(ctx: CallContext) -> Result<JsUndefined> {
  let hook_external = ctx.get::<JsExternal>(0)?;
  let hook = *ctx
    .env
    .get_value_external::<CleanupEnvHook<()>>(&hook_external)?;
  ctx.env.remove_env_cleanup_hook(hook)?;
  ctx.env.get_undefined()
}

pub fn register_js(exports: &mut JsObject) -> Result<()> {
  exports.create_named_method("addCleanupHook", add_cleanup_hook)?;
  exports.create_named_method("removeCleanupHook", remove_cleanup_hook)?;
  Ok(())
}