Add support for registering import module callback functions in quick…

…js module

internal/jsbuiltin/src/00_module.js

@@ -193,4 +193,25 @@ return module;
   require.cache = {};
   jssh.require = require;
   jssh.requiremodule = requiremodule;
+
+  const importModuleCallbacks = [];
+
+  const registerImportModuleCallback = (callback) => {
+    if (typeof callback !== "function") {
+      throw new TypeError("callback must be a function");
+    }
+    importModuleCallbacks.push(callback);
+  };
+
+  const callImportModuleCallbacks = (name, dir) => {
+    for (const callback of importModuleCallbacks) {
+      const result = callback(name, dir);
+      if (result) {
+        return result;
+      }
+    }
+  };
+
+  jssh.registerImportModuleCallback = registerImportModuleCallback;
+  jssh.callImportModuleCallbacks = callImportModuleCallbacks;
 }

internal/jsexecutor/jsexecutor.go

@@ -15,7 +15,9 @@ type JSFunction = func(ctx *JSContext, this JSValue, args []JSValue) JSValue //
 
 // NewJSRuntime 创建新的JSRuntime实例
 func NewJSRuntime() JSRuntime {
-	return quickjs.NewRuntime()
+	runtime := quickjs.NewRuntime()
+	runtime.SetModuleLoader(moduleLoader)
+	return runtime
 }
 
 // IsGoFunction 判断是否为Go的函数类型
@@ -238,3 +240,10 @@ func anySliceToJSValue(ctx *quickjs.Context, v reflect.Value, vt reflect.Type) q
 	}
 	return arr
 }
+
+// RegisterModuleLoader 注册模块加载器
+func RegisterModuleLoader(loader func(ctx *JSContext, moduleName string) (string, error)) {
+	moduleLoader = loader
+}
+
+var moduleLoader func(ctx *JSContext, moduleName string) (string, error)

quickjs/quickjs.go

@@ -21,7 +21,8 @@ import (
 import "C"
 
 type Runtime struct {
-	ref *C.JSRuntime
+	ref           *C.JSRuntime
+	moduleLoader  func(ctx *Context, moduleName string) (string, error)
 }
 
 func NewRuntime() Runtime {
@@ -42,7 +43,17 @@ func (r Runtime) NewContext() *Context {
 	C.JS_AddIntrinsicOperators(ref)
 	C.JS_EnableBignumExt(ref, C.int(1))
 
-	return &Context{ref: ref}
+	ctx := &Context{ref: ref}
+
+	if r.moduleLoader != nil {
+		C.JS_SetModuleLoaderFunc(r.ref, nil, (*C.JSModuleLoaderFunc)(C.InvokeModuleLoader), unsafe.Pointer(ctx))
+	}
+
+	return ctx
+}
+
+func (r *Runtime) SetModuleLoader(loader func(ctx *Context, moduleName string) (string, error)) {
+	r.moduleLoader = loader
 }
 
 func (r Runtime) ExecutePendingJob() (Context, error) {
@@ -400,7 +411,7 @@ func (v Value) SetByInt64(idx int64, val Value) {
 }
 
 func (v Value) SetByUint32(idx uint32, val Value) {
-	C.JS_SetPropertyUint32(v.ctx.ref, v.ref, C.uint32_t(idx), val.ref)
+	C.JS_SetPropertyUint32(v.ctx.ref, v.ref, C.uint32_t(idx))
 }
 
 func (v Value) Len() int64 { return v.Get("length").Int64() }