1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
use std::ptr;
use log::debug;
use inventory::Collect;
use inventory::submit;
use inventory::iter;
use inventory::Registry;
use crate::Property;
use crate::val::JsExports;
use crate::sys::napi_value;
use crate::sys::napi_env;
use crate::NjError;
type ClassCallback = fn(&mut JsExports) -> Result<(), NjError>;
enum NapiRegister {
Property(Property),
Callback(ClassCallback),
}
impl Collect for NapiRegister {
fn registry() -> &'static Registry<Self> {
static REGISTRY: Registry<NapiRegister> = Registry::new();
®ISTRY
}
}
pub fn submit_property(value: Property) {
submit::<NapiRegister>(NapiRegister::Property(value))
}
pub fn submit_register_callback(callback: ClassCallback) {
submit::<NapiRegister>(NapiRegister::Callback(callback));
}
#[no_mangle]
pub extern "C" fn init_modules(env: napi_env, exports: napi_value) -> napi_value {
debug!("initializing modules");
let mut js_exports = JsExports::new(env, exports);
let mut prop_builder = js_exports.prop_builder();
for register in iter::<NapiRegister> {
match register {
NapiRegister::Property(property) => {
debug!("registering property: {:#?}", property);
prop_builder.mut_append(property.to_owned());
}
NapiRegister::Callback(callback) => {
debug!("invoking register callback");
if let Err(err) = callback(&mut js_exports) {
panic!("error invoking JS callback: {}", err);
}
}
}
}
js_exports
.define_property(prop_builder)
.expect("property should not fail");
exports
}
#[crate::ctor]
fn init_module() {
use crate::c_str;
use crate::sys::NAPI_VERSION;
use crate::sys::napi_module;
use crate::sys::napi_module_register;
static mut _module: napi_module = napi_module {
nm_version: NAPI_VERSION as i32,
nm_flags: 0,
nm_filename: c_str!("lib.rs").as_ptr() as *const i8,
nm_register_func: Some(init_modules),
nm_modname: c_str!("rust_module").as_ptr() as *const i8,
nm_priv: ptr::null_mut(),
reserved: [
ptr::null_mut(),
ptr::null_mut(),
ptr::null_mut(),
ptr::null_mut(),
],
};
crate::init_logger();
unsafe {
napi_module_register(&mut _module);
}
}