Bump cel-go to v0.12.0
This commit is contained in:
Cici Huang
285
vendor/github.com/google/cel-go/ext/guards.go
generated
vendored
285
vendor/github.com/google/cel-go/ext/guards.go
generated
vendored
@@ -17,287 +17,34 @@ package ext
|
||||
import (
|
||||
"github.com/google/cel-go/common/types"
|
||||
"github.com/google/cel-go/common/types/ref"
|
||||
"github.com/google/cel-go/common/types/traits"
|
||||
"github.com/google/cel-go/interpreter/functions"
|
||||
)
|
||||
|
||||
// function invocation guards for common call signatures within extension functions.
|
||||
|
||||
func callInBytesOutString(fn func([]byte) (string, error)) functions.UnaryOp {
|
||||
return func(val ref.Val) ref.Val {
|
||||
vVal, ok := val.(types.Bytes)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(val)
|
||||
}
|
||||
str, err := fn([]byte(vVal))
|
||||
if err != nil {
|
||||
return types.NewErr(err.Error())
|
||||
}
|
||||
return types.String(str)
|
||||
func intOrError(i int64, err error) ref.Val {
|
||||
if err != nil {
|
||||
return types.NewErr(err.Error())
|
||||
}
|
||||
return types.Int(i)
|
||||
}
|
||||
|
||||
func callInStrOutBytes(fn func(string) ([]byte, error)) functions.UnaryOp {
|
||||
return func(val ref.Val) ref.Val {
|
||||
vVal, ok := val.(types.String)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(val)
|
||||
}
|
||||
byt, err := fn(string(vVal))
|
||||
if err != nil {
|
||||
return types.NewErr(err.Error())
|
||||
}
|
||||
return types.Bytes(byt)
|
||||
func bytesOrError(bytes []byte, err error) ref.Val {
|
||||
if err != nil {
|
||||
return types.NewErr(err.Error())
|
||||
}
|
||||
return types.Bytes(bytes)
|
||||
}
|
||||
|
||||
func callInStrOutStr(fn func(string) (string, error)) functions.UnaryOp {
|
||||
return func(val ref.Val) ref.Val {
|
||||
vVal, ok := val.(types.String)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(val)
|
||||
}
|
||||
str, err := fn(string(vVal))
|
||||
if err != nil {
|
||||
return types.NewErr(err.Error())
|
||||
}
|
||||
return types.String(str)
|
||||
func stringOrError(str string, err error) ref.Val {
|
||||
if err != nil {
|
||||
return types.NewErr(err.Error())
|
||||
}
|
||||
return types.String(str)
|
||||
}
|
||||
|
||||
func callInStrIntOutStr(fn func(string, int64) (string, error)) functions.BinaryOp {
|
||||
return func(val, arg ref.Val) ref.Val {
|
||||
vVal, ok := val.(types.String)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(val)
|
||||
}
|
||||
argVal, ok := arg.(types.Int)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(arg)
|
||||
}
|
||||
out, err := fn(string(vVal), int64(argVal))
|
||||
if err != nil {
|
||||
return types.NewErr(err.Error())
|
||||
}
|
||||
return types.String(out)
|
||||
}
|
||||
}
|
||||
|
||||
func callInStrStrOutInt(fn func(string, string) (int64, error)) functions.BinaryOp {
|
||||
return func(val, arg ref.Val) ref.Val {
|
||||
vVal, ok := val.(types.String)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(val)
|
||||
}
|
||||
argVal, ok := arg.(types.String)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(arg)
|
||||
}
|
||||
out, err := fn(string(vVal), string(argVal))
|
||||
if err != nil {
|
||||
return types.NewErr(err.Error())
|
||||
}
|
||||
return types.Int(out)
|
||||
}
|
||||
}
|
||||
|
||||
func callInStrStrOutListStr(fn func(string, string) ([]string, error)) functions.BinaryOp {
|
||||
return func(val, arg ref.Val) ref.Val {
|
||||
vVal, ok := val.(types.String)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(val)
|
||||
}
|
||||
argVal, ok := arg.(types.String)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(arg)
|
||||
}
|
||||
out, err := fn(string(vVal), string(argVal))
|
||||
if err != nil {
|
||||
return types.NewErr(err.Error())
|
||||
}
|
||||
return types.DefaultTypeAdapter.NativeToValue(out)
|
||||
}
|
||||
}
|
||||
|
||||
func callInStrIntIntOutStr(fn func(string, int64, int64) (string, error)) functions.FunctionOp {
|
||||
return func(args ...ref.Val) ref.Val {
|
||||
if len(args) != 3 {
|
||||
return types.NoSuchOverloadErr()
|
||||
}
|
||||
vVal, ok := args[0].(types.String)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(args[0])
|
||||
}
|
||||
arg1Val, ok := args[1].(types.Int)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(args[1])
|
||||
}
|
||||
arg2Val, ok := args[2].(types.Int)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(args[2])
|
||||
}
|
||||
out, err := fn(string(vVal), int64(arg1Val), int64(arg2Val))
|
||||
if err != nil {
|
||||
return types.NewErr(err.Error())
|
||||
}
|
||||
return types.String(out)
|
||||
}
|
||||
}
|
||||
|
||||
func callInStrStrStrOutStr(fn func(string, string, string) (string, error)) functions.FunctionOp {
|
||||
return func(args ...ref.Val) ref.Val {
|
||||
if len(args) != 3 {
|
||||
return types.NoSuchOverloadErr()
|
||||
}
|
||||
vVal, ok := args[0].(types.String)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(args[0])
|
||||
}
|
||||
arg1Val, ok := args[1].(types.String)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(args[1])
|
||||
}
|
||||
arg2Val, ok := args[2].(types.String)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(args[2])
|
||||
}
|
||||
out, err := fn(string(vVal), string(arg1Val), string(arg2Val))
|
||||
if err != nil {
|
||||
return types.NewErr(err.Error())
|
||||
}
|
||||
return types.String(out)
|
||||
}
|
||||
}
|
||||
|
||||
func callInStrStrIntOutInt(fn func(string, string, int64) (int64, error)) functions.FunctionOp {
|
||||
return func(args ...ref.Val) ref.Val {
|
||||
if len(args) != 3 {
|
||||
return types.NoSuchOverloadErr()
|
||||
}
|
||||
vVal, ok := args[0].(types.String)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(args[0])
|
||||
}
|
||||
arg1Val, ok := args[1].(types.String)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(args[1])
|
||||
}
|
||||
arg2Val, ok := args[2].(types.Int)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(args[2])
|
||||
}
|
||||
out, err := fn(string(vVal), string(arg1Val), int64(arg2Val))
|
||||
if err != nil {
|
||||
return types.NewErr(err.Error())
|
||||
}
|
||||
return types.Int(out)
|
||||
}
|
||||
}
|
||||
|
||||
func callInStrStrIntOutListStr(fn func(string, string, int64) ([]string, error)) functions.FunctionOp {
|
||||
return func(args ...ref.Val) ref.Val {
|
||||
if len(args) != 3 {
|
||||
return types.NoSuchOverloadErr()
|
||||
}
|
||||
vVal, ok := args[0].(types.String)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(args[0])
|
||||
}
|
||||
arg1Val, ok := args[1].(types.String)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(args[1])
|
||||
}
|
||||
arg2Val, ok := args[2].(types.Int)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(args[2])
|
||||
}
|
||||
out, err := fn(string(vVal), string(arg1Val), int64(arg2Val))
|
||||
if err != nil {
|
||||
return types.NewErr(err.Error())
|
||||
}
|
||||
return types.DefaultTypeAdapter.NativeToValue(out)
|
||||
}
|
||||
}
|
||||
|
||||
func callInStrStrStrIntOutStr(fn func(string, string, string, int64) (string, error)) functions.FunctionOp {
|
||||
return func(args ...ref.Val) ref.Val {
|
||||
if len(args) != 4 {
|
||||
return types.NoSuchOverloadErr()
|
||||
}
|
||||
vVal, ok := args[0].(types.String)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(args[0])
|
||||
}
|
||||
arg1Val, ok := args[1].(types.String)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(args[1])
|
||||
}
|
||||
arg2Val, ok := args[2].(types.String)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(args[2])
|
||||
}
|
||||
arg3Val, ok := args[3].(types.Int)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(args[3])
|
||||
}
|
||||
out, err := fn(string(vVal), string(arg1Val), string(arg2Val), int64(arg3Val))
|
||||
if err != nil {
|
||||
return types.NewErr(err.Error())
|
||||
}
|
||||
return types.String(out)
|
||||
}
|
||||
}
|
||||
|
||||
func callInListStrOutStr(fn func([]string) (string, error)) functions.UnaryOp {
|
||||
return func(args1 ref.Val) ref.Val {
|
||||
vVal, ok := args1.(traits.Lister)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(args1)
|
||||
}
|
||||
strings := make([]string, vVal.Size().Value().(int64))
|
||||
i := 0
|
||||
for it := vVal.Iterator(); it.HasNext() == types.True; {
|
||||
next := it.Next()
|
||||
v, ok := next.(types.String)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(next)
|
||||
}
|
||||
strings[i] = string(v)
|
||||
i++
|
||||
}
|
||||
out, err := fn(strings)
|
||||
if err != nil {
|
||||
return types.NewErr(err.Error())
|
||||
}
|
||||
return types.DefaultTypeAdapter.NativeToValue(out)
|
||||
}
|
||||
}
|
||||
|
||||
func callInListStrStrOutStr(fn func([]string, string) (string, error)) functions.BinaryOp {
|
||||
return func(args1, args2 ref.Val) ref.Val {
|
||||
vVal, ok := args1.(traits.Lister)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(args1)
|
||||
}
|
||||
arg1Val, ok := args2.(types.String)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(args2)
|
||||
}
|
||||
strings := make([]string, vVal.Size().Value().(int64))
|
||||
i := 0
|
||||
for it := vVal.Iterator(); it.HasNext() == types.True; {
|
||||
next := it.Next()
|
||||
v, ok := next.(types.String)
|
||||
if !ok {
|
||||
return types.MaybeNoSuchOverloadErr(next)
|
||||
}
|
||||
strings[i] = string(v)
|
||||
i++
|
||||
}
|
||||
out, err := fn(strings, string(arg1Val))
|
||||
if err != nil {
|
||||
return types.NewErr(err.Error())
|
||||
}
|
||||
return types.DefaultTypeAdapter.NativeToValue(out)
|
||||
func listStringOrError(strs []string, err error) ref.Val {
|
||||
if err != nil {
|
||||
return types.NewErr(err.Error())
|
||||
}
|
||||
return types.DefaultTypeAdapter.NativeToValue(strs)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user