quickjs_rusty/
module_loader.rs

1use std::ffi::{c_char, c_void, CStr, CString};
2use std::ptr::null_mut;
3
4use anyhow::Result;
5use libquickjs_ng_sys as q;
6
7use super::compile::compile_module;
8
9/// Custom module loader function, passes (module_name, opaque) and returns module code
10/// If the module code is not found, return None
11pub type JSModuleLoaderFunc = Box<dyn Fn(&str, *mut c_void) -> Result<String>>;
12/// Custom module normalize function, passes (module_base_name, module_name, opaque)
13/// and returns normalized module name (or None if not found)
14pub type JSModuleNormalizeFunc = Box<dyn Fn(&str, &str, *mut c_void) -> Result<String>>;
15
16pub struct ModuleLoader {
17    pub loader: JSModuleLoaderFunc,
18    pub normalize: Option<JSModuleNormalizeFunc>,
19    pub opaque: *mut c_void,
20}
21
22pub unsafe extern "C" fn js_module_loader(
23    ctx: *mut q::JSContext,
24    module_name: *const c_char,
25    opaque: *mut c_void,
26) -> *mut q::JSModuleDef {
27    let wrapper = &*(opaque as *mut ModuleLoader);
28    let opaque = wrapper.opaque;
29    let loader = &wrapper.loader;
30
31    let module_name = CStr::from_ptr(module_name).to_string_lossy().to_string();
32    let module_code = match loader(&module_name, opaque) {
33        Ok(v) => v,
34        Err(err) => {
35            throw_internal_error(ctx, &err.to_string());
36            return null_mut() as *mut q::JSModuleDef;
37        }
38    };
39
40    match compile_module(ctx, &module_code, &module_name) {
41        Ok(v) => {
42            let module_def = q::JS_Ext_GetPtr(v.value);
43            module_def as *mut q::JSModuleDef
44        }
45        Err(e) => {
46            throw_internal_error(ctx, &e.to_string());
47            null_mut() as *mut q::JSModuleDef
48        }
49    }
50}
51
52pub unsafe extern "C" fn js_module_normalize(
53    ctx: *mut q::JSContext,
54    module_base_name: *const c_char,
55    module_name: *const c_char,
56    opaque: *mut c_void,
57) -> *mut c_char {
58    let wrapper = &*(opaque as *mut ModuleLoader);
59    let opaque = wrapper.opaque;
60    let normalize = &wrapper.normalize;
61
62    let module_base_name = CStr::from_ptr(module_base_name).to_str().unwrap();
63    let module_name = CStr::from_ptr(module_name).to_str().unwrap();
64
65    if let Some(module_normalize_func) = normalize {
66        let mut normalized_module_name =
67            match module_normalize_func(module_base_name, module_name, opaque) {
68                Ok(v) => v,
69                Err(err) => {
70                    throw_internal_error(ctx, &err.to_string());
71                    return null_mut() as *mut c_char;
72                }
73            };
74        normalized_module_name.push('\0');
75        let m = q::js_malloc(ctx, normalized_module_name.len());
76        std::ptr::copy(
77            normalized_module_name.as_ptr(),
78            m as *mut u8,
79            normalized_module_name.len(),
80        );
81        m as *mut c_char
82    } else {
83        log::warn!("module normalize func not set");
84        null_mut() as *mut c_char
85    }
86}
87
88#[inline]
89fn throw_internal_error(ctx: *mut q::JSContext, err: &str) {
90    let err = CString::new(err).unwrap();
91    unsafe {
92        q::JS_ThrowInternalError(ctx, err.as_ptr() as *const c_char);
93    }
94}
quickjs_rusty/module_loader.rs

quickjs_rusty/
module_loader.rs
