ef03b1e3df
Upgrade dependent version: github.com/open-policy-agent/opa v0.18.0 -> v0.45.0 Signed-off-by: hongzhouzi <hongzhouzi@kubesphere.io> Signed-off-by: hongzhouzi <hongzhouzi@kubesphere.io>
95 lines
2.1 KiB
Go
95 lines
2.1 KiB
Go
// Copyright 2018 The OPA Authors. All rights reserved.
|
|
// Use of this source code is governed by an Apache2
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package topdown
|
|
|
|
import (
|
|
"github.com/open-policy-agent/opa/ast"
|
|
"github.com/open-policy-agent/opa/topdown/builtins"
|
|
)
|
|
|
|
func builtinArrayConcat(a, b ast.Value) (ast.Value, error) {
|
|
arrA, err := builtins.ArrayOperand(a, 1)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
arrB, err := builtins.ArrayOperand(b, 2)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
arrC := make([]*ast.Term, arrA.Len()+arrB.Len())
|
|
|
|
i := 0
|
|
arrA.Foreach(func(elemA *ast.Term) {
|
|
arrC[i] = elemA
|
|
i++
|
|
})
|
|
|
|
arrB.Foreach(func(elemB *ast.Term) {
|
|
arrC[i] = elemB
|
|
i++
|
|
})
|
|
|
|
return ast.NewArray(arrC...), nil
|
|
}
|
|
|
|
func builtinArraySlice(a, i, j ast.Value) (ast.Value, error) {
|
|
arr, err := builtins.ArrayOperand(a, 1)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
startIndex, err := builtins.IntOperand(i, 2)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
stopIndex, err := builtins.IntOperand(j, 3)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Clamp stopIndex to avoid out-of-range errors. If negative, clamp to zero.
|
|
// Otherwise, clamp to length of array.
|
|
if stopIndex < 0 {
|
|
stopIndex = 0
|
|
} else if stopIndex > arr.Len() {
|
|
stopIndex = arr.Len()
|
|
}
|
|
|
|
// Clamp startIndex to avoid out-of-range errors. If negative, clamp to zero.
|
|
// Otherwise, clamp to stopIndex to avoid to avoid cases like arr[1:0].
|
|
if startIndex < 0 {
|
|
startIndex = 0
|
|
} else if startIndex > stopIndex {
|
|
startIndex = stopIndex
|
|
}
|
|
|
|
return arr.Slice(startIndex, stopIndex), nil
|
|
}
|
|
|
|
func builtinArrayReverse(bctx BuiltinContext, operands []*ast.Term, iter func(*ast.Term) error) error {
|
|
arr, err := builtins.ArrayOperand(operands[0].Value, 1)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
length := arr.Len()
|
|
reversedArr := make([]*ast.Term, length)
|
|
|
|
for index := 0; index < length; index++ {
|
|
reversedArr[index] = arr.Elem(length - index - 1)
|
|
}
|
|
|
|
return iter(ast.ArrayTerm(reversedArr...))
|
|
}
|
|
|
|
func init() {
|
|
RegisterFunctionalBuiltin2(ast.ArrayConcat.Name, builtinArrayConcat)
|
|
RegisterFunctionalBuiltin3(ast.ArraySlice.Name, builtinArraySlice)
|
|
RegisterBuiltinFunc(ast.ArrayReverse.Name, builtinArrayReverse)
|
|
}
|