async

internal/runtime/pkg/builtin/builtin.go

@@ -13,16 +13,26 @@ type Builtin struct {
 	Name       string
 	Function   func(*goja.Runtime) func(goja.FunctionCall) goja.Value
 	Definition string
+	isPromise  bool
 }
 
 type EmptyArgs struct{}
 
+type RegisterOption func(*Builtin) error
+
+func WithPromise() RegisterOption {
+	return func(b *Builtin) error {
+		b.isPromise = true
+		return nil
+	}
+}
+
 var (
 	builtinRegistry = make(map[string]Builtin)
 	registryMutex   sync.RWMutex
 )
 
-func RegisterBuiltin[T any, R any](name string, fn any) {
+func RegisterBuiltin[T any, R any](name string, fn any, opts ...RegisterOption) {
 	var zeroT T
 	tType := reflect.TypeOf(zeroT)
 
@@ -31,18 +41,35 @@ func RegisterBuiltin[T any, R any](name string, fn any) {
 	}
 
 	fnType := reflect.TypeOf(fn)
-	wrapper := createWrapper[T](fn, fnType)
+
+	isPromise := false
+	for _, opt := range opts {
+		if opt != nil {
+			isPromise = true
+			break
+		}
+	}
+
+	wrapper := createWrapper[T](fn, fnType, isPromise)
 
 	registryMutex.Lock()
-	builtinRegistry[name] = Builtin{
+	b := Builtin{
 		Name:       name,
 		Function:   wrapper,
-		Definition: generateTypeScriptDefinition(name, tType, fnType),
+		Definition: generateTypeScriptDefinition(name, tType, fnType, isPromise),
 	}
+	for _, opt := range opts {
+		if opt != nil {
+			if err := opt(&b); err != nil {
+				panic(fmt.Sprintf("builtin %s: option error: %v", name, err))
+			}
+		}
+	}
+	builtinRegistry[name] = b
 	registryMutex.Unlock()
 }
 
-func createWrapper[T any](fn any, fnType reflect.Type) func(*goja.Runtime) func(goja.FunctionCall) goja.Value {
+func createWrapper[T any](fn any, fnType reflect.Type, isPromise bool) func(*goja.Runtime) func(goja.FunctionCall) goja.Value {
 	return func(vm *goja.Runtime) func(goja.FunctionCall) goja.Value {
 		return func(call goja.FunctionCall) goja.Value {
 			var args T
@@ -83,19 +110,53 @@ func createWrapper[T any](fn any, fnType reflect.Type) func(*goja.Runtime) func(
 
 			if err, isError := results[len(results)-1].Interface().(error); isError {
 				if err != nil {
+					if isPromise {
+						return createRejectedPromise(vm, err)
+					}
 					panic(err)
 				}
 				if len(results) == 1 {
+					if isPromise {
+						return createResolvedPromise(vm)
+					}
 					return goja.Undefined()
 				}
+				if isPromise {
+					return createResolvedPromise(vm, results[0])
+				}
 				return convertGoValueToJS(vm, results[0])
 			}
 
+			if isPromise {
+				return createResolvedPromise(vm, results[0])
+			}
+
 			return convertGoValueToJS(vm, results[0])
 		}
 	}
 }
 
+func createResolvedPromise(vm *goja.Runtime, value ...reflect.Value) goja.Value {
+	promise, resolve, _ := vm.NewPromise()
+	go func() {
+		if len(value) > 0 {
+			jsValue := convertGoValueToJS(vm, value[0])
+			_ = resolve(jsValue)
+		} else {
+			_ = resolve(goja.Undefined())
+		}
+	}()
+	return vm.ToValue(promise)
+}
+
+func createRejectedPromise(vm *goja.Runtime, err error) goja.Value {
+	promise, _, reject := vm.NewPromise()
+	go func() {
+		_ = reject(vm.ToValue(err.Error()))
+	}()
+	return vm.ToValue(promise)
+}
+
 func convertJSValueToGo(vm *goja.Runtime, jsValue goja.Value, targetType reflect.Type) (any, error) {
 	if goja.IsNull(jsValue) {
 		if targetType.Kind() == reflect.Pointer || targetType.Kind() == reflect.Map {
@@ -292,7 +353,7 @@ func getFieldName(field reflect.StructField) string {
 	return field.Name
 }
 
-func generateTypeScriptDefinition(name string, argsType reflect.Type, fnType reflect.Type) string {
+func generateTypeScriptDefinition(name string, argsType reflect.Type, fnType reflect.Type, isPromise bool) string {
 	if argsType.Kind() != reflect.Struct {
 		return ""
 	}
@@ -321,6 +382,10 @@ func generateTypeScriptDefinition(name string, argsType reflect.Type, fnType ref
 		}
 	}
 
+	if isPromise {
+		returnSignature = fmt.Sprintf("Promise<%s>", returnSignature)
+	}
+
 	return fmt.Sprintf("declare function %s(%s): %s;", name, strings.Join(params, ", "), returnSignature)
 }
 
@@ -367,6 +432,11 @@ func goTypeToTSType(t reflect.Type, isPointer bool) string {
 			fields = append(fields, fmt.Sprintf("%s: %s", name, tsType))
 		}
 		return fmt.Sprintf("{ %s }", strings.Join(fields, "; "))
+	case reflect.Pointer:
+		if t.Elem().Kind() == reflect.Struct {
+			return goTypeToTSType(t.Elem(), false)
+		}
+		return "any"
 	default:
 		return "any"
 	}