77 lines
1.4 KiB
Go
77 lines
1.4 KiB
Go
package functions
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"reflect"
|
|
)
|
|
|
|
type Function interface {
|
|
Name() string
|
|
Types() []string
|
|
Definition() string
|
|
WrapFn(context.Context) func(...any) (any, error)
|
|
}
|
|
|
|
type RawFunc[A Args, R any] func(context.Context, A) (R, error)
|
|
|
|
type Args interface {
|
|
Validate() error
|
|
}
|
|
|
|
type functionImpl[A Args, R any] struct {
|
|
name string
|
|
fn RawFunc[A, R]
|
|
definition string
|
|
types []string
|
|
}
|
|
|
|
func (b *functionImpl[A, R]) Name() string {
|
|
return b.name
|
|
}
|
|
|
|
func (b *functionImpl[A, R]) Types() []string {
|
|
return b.types
|
|
}
|
|
|
|
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
|
|
}
|
|
}
|