75
vendor/github.com/open-policy-agent/opa/topdown/array.go
generated
vendored
Normal file
75
vendor/github.com/open-policy-agent/opa/topdown/array.go
generated
vendored
Normal file
@@ -0,0 +1,75 @@
|
||||
// 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.Array, 0, len(arrA)+len(arrB))
|
||||
|
||||
for _, elemA := range arrA {
|
||||
arrC = append(arrC, elemA)
|
||||
}
|
||||
|
||||
for _, elemB := range arrB {
|
||||
arrC = append(arrC, elemB)
|
||||
}
|
||||
|
||||
return 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
|
||||
}
|
||||
|
||||
// Return empty array if bounds cannot be clamped sensibly.
|
||||
if (startIndex >= stopIndex) || (startIndex <= 0 && stopIndex <= 0) {
|
||||
return arr[0:0], nil
|
||||
}
|
||||
|
||||
// Clamp bounds to avoid out-of-range errors.
|
||||
if startIndex < 0 {
|
||||
startIndex = 0
|
||||
}
|
||||
|
||||
if stopIndex > len(arr) {
|
||||
stopIndex = len(arr)
|
||||
}
|
||||
|
||||
arrb := arr[startIndex:stopIndex]
|
||||
|
||||
return arrb, nil
|
||||
|
||||
}
|
||||
|
||||
func init() {
|
||||
RegisterFunctionalBuiltin2(ast.ArrayConcat.Name, builtinArrayConcat)
|
||||
RegisterFunctionalBuiltin3(ast.ArraySlice.Name, builtinArraySlice)
|
||||
}
|
||||
Reference in New Issue
Block a user