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();
        &REGISTRY
    }
}

/// submit property for including in global registry
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);
    }
}