pre migrate

internal/functions/types.go

@@ -10,10 +10,11 @@ type Function interface {
 	Name() string
 	Types() []string
 	Definition() string
-	WrapFn(context.Context) func(...any) (any, error)
+	IsAsync() bool
+	Call(context.Context, []any) (any, error)
 }
 
-type RawFunc[A Args, R any] func(context.Context, A) (R, error)
+type GoFunc[A Args, R any] func(context.Context, A) (R, error)
 
 type Args interface {
 	Validate() error
@@ -21,9 +22,10 @@ type Args interface {
 
 type functionImpl[A Args, R any] struct {
 	name       string
-	fn         RawFunc[A, R]
+	fn         GoFunc[A, R]
 	definition string
 	types      []string
+	isAsync    bool
 }
 
 func (b *functionImpl[A, R]) Name() string {
@@ -38,39 +40,38 @@ func (b *functionImpl[A, R]) Definition() string {
 	return b.definition
 }
 
-func (b *functionImpl[A, R]) WrapFn(ctx context.Context) func(...any) (any, error) {
-	return func(allArgs ...any) (any, error) {
-		// Populate Arguments
-		var fnArgs A
-		aVal := reflect.ValueOf(&fnArgs).Elem()
-
-		// Populate Fields
-		for i := range min(aVal.NumField(), len(allArgs)) {
-			field := aVal.Field(i)
-
-			if !field.CanSet() {
-				return nil, errors.New("cannot set field")
-			}
-
-			argVal := reflect.ValueOf(allArgs[i])
-			if !argVal.Type().AssignableTo(field.Type()) {
-				return nil, errors.New("cannot assign field")
-			}
-
-			field.Set(argVal)
-		}
-
-		// Validate
-		if err := fnArgs.Validate(); err != nil {
-			return nil, errors.New("cannot validate args")
-		}
-
-		// Call Function
-		resp, err := b.fn(ctx, fnArgs)
-		if err != nil {
-			return nil, err
-		}
-
-		return resp, nil
-	}
+func (b *functionImpl[A, R]) IsAsync() bool {
+	return b.isAsync
+}
+
+func (b *functionImpl[A, R]) Function() any {
+	return b.fn
+}
+
+func (b *functionImpl[A, R]) Call(ctx context.Context, allArgs []any) (any, error) {
+	return b.CallGeneric(ctx, allArgs)
+}
+
+func (b *functionImpl[A, R]) CallGeneric(ctx context.Context, allArgs []any) (zeroR R, err error) {
+	// Populate Arguments
+	var fnArgs A
+	aVal := reflect.ValueOf(&fnArgs).Elem()
+
+	// Populate Fields
+	for i := range min(aVal.NumField(), len(allArgs)) {
+		field := aVal.Field(i)
+
+		if !field.CanSet() {
+			return zeroR, errors.New("cannot set field")
+		}
+
+		argVal := reflect.ValueOf(allArgs[i])
+		if !argVal.Type().AssignableTo(field.Type()) {
+			return zeroR, errors.New("cannot assign field")
+		}
+
+		field.Set(argVal)
+	}
+
+	return b.fn(ctx, fnArgs)
 }