denoland-deno/test_napi/src/async.rs

// Copyright 2018-2023 the Deno authors. All rights reserved. MIT license.

use napi_sys::Status::napi_ok;
use napi_sys::ValueType::napi_function;
use napi_sys::*;
use std::os::raw::{c_char, c_void};
use std::ptr;

pub struct Baton {
  called: bool,
  func: napi_ref,
  task: napi_async_work,
}

unsafe extern "C" fn execute(_env: napi_env, data: *mut c_void) {
  let baton: &mut Baton = &mut *(data as *mut Baton);
  assert!(!baton.called);
  assert!(!baton.func.is_null());

  baton.called = true;
}

unsafe extern "C" fn complete(
  env: napi_env,
  status: napi_status,
  data: *mut c_void,
) {
  assert!(status == napi_ok);
  let baton: Box<Baton> = Box::from_raw(data as *mut Baton);
  assert!(baton.called);
  assert!(!baton.func.is_null());

  let mut global: napi_value = ptr::null_mut();
  assert!(napi_get_global(env, &mut global) == napi_ok);

  let mut callback: napi_value = ptr::null_mut();
  assert!(napi_get_reference_value(env, baton.func, &mut callback) == napi_ok);

  let mut _result: napi_value = ptr::null_mut();
  assert!(
    napi_call_function(env, global, callback, 0, ptr::null(), &mut _result)
      == napi_ok
  );

  assert!(napi_delete_reference(env, baton.func) == napi_ok);
  assert!(napi_delete_async_work(env, baton.task) == napi_ok);
}

extern "C" fn test_async_work(
  env: napi_env,
  info: napi_callback_info,
) -> napi_value {
  let (args, argc, _) = crate::get_callback_info!(env, info, 1);
  assert_eq!(argc, 1);

  let mut ty = -1;
  assert!(unsafe { napi_typeof(env, args[0], &mut ty) } == napi_ok);
  assert_eq!(ty, napi_function);

  let mut resource_name: napi_value = ptr::null_mut();
  assert!(
    unsafe {
      napi_create_string_utf8(
        env,
        "test_async_resource\0".as_ptr() as *const c_char,
        usize::MAX,
        &mut resource_name,
      )
    } == napi_ok
  );

  let mut async_work: napi_async_work = ptr::null_mut();

  let mut func: napi_ref = ptr::null_mut();
  assert!(
    unsafe { napi_create_reference(env, args[0], 1, &mut func) } == napi_ok
  );
  let baton = Box::new(Baton {
    called: false,
    func,
    task: async_work,
  });

  assert!(
    unsafe {
      napi_create_async_work(
        env,
        ptr::null_mut(),
        resource_name,
        Some(execute),
        Some(complete),
        Box::into_raw(baton) as *mut c_void,
        &mut async_work,
      )
    } == napi_ok
  );
  assert!(unsafe { napi_queue_async_work(env, async_work) } == napi_ok);

  ptr::null_mut()
}

pub fn init(env: napi_env, exports: napi_value) {
  let properties = &[crate::new_property!(
    env,
    "test_async_work\0",
    test_async_work
  )];

  unsafe {
    napi_define_properties(env, exports, properties.len(), properties.as_ptr())
  };
}