nj_core/
module.rs

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

nj_core/
module.rs
