Курнат Андрей
132 lines
3.8 KiB
Go
132 lines
3.8 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 (
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/open-policy-agent/opa/v1/ast"
|
|
"github.com/open-policy-agent/opa/v1/topdown/builtins"
|
|
)
|
|
|
|
func builtinToNumber(_ BuiltinContext, operands []*ast.Term, iter func(*ast.Term) error) error {
|
|
switch a := operands[0].Value.(type) {
|
|
case ast.Null:
|
|
return iter(ast.InternedTerm(0))
|
|
case ast.Boolean:
|
|
if a {
|
|
return iter(ast.InternedTerm(1))
|
|
}
|
|
return iter(ast.InternedTerm(0))
|
|
case ast.Number:
|
|
return iter(operands[0])
|
|
case ast.String:
|
|
strValue := string(a)
|
|
|
|
if it := ast.InternedIntNumberTermFromString(strValue); it != nil {
|
|
return iter(it)
|
|
}
|
|
|
|
trimmedVal := strings.TrimLeft(strValue, "+-")
|
|
lowerCaseVal := strings.ToLower(trimmedVal)
|
|
|
|
if lowerCaseVal == "inf" || lowerCaseVal == "infinity" || lowerCaseVal == "nan" {
|
|
return builtins.NewOperandTypeErr(1, operands[0].Value, "valid number string")
|
|
}
|
|
|
|
_, err := strconv.ParseFloat(strValue, 64)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return iter(ast.NewTerm(ast.Number(a)))
|
|
}
|
|
return builtins.NewOperandTypeErr(1, operands[0].Value, "null", "boolean", "number", "string")
|
|
}
|
|
|
|
// Deprecated: deprecated in v0.13.0.
|
|
func builtinToArray(_ BuiltinContext, operands []*ast.Term, iter func(*ast.Term) error) error {
|
|
switch val := operands[0].Value.(type) {
|
|
case *ast.Array:
|
|
return iter(ast.NewTerm(val))
|
|
case ast.Set:
|
|
arr := make([]*ast.Term, val.Len())
|
|
i := 0
|
|
val.Foreach(func(term *ast.Term) {
|
|
arr[i] = term
|
|
i++
|
|
})
|
|
return iter(ast.NewTerm(ast.NewArray(arr...)))
|
|
default:
|
|
return builtins.NewOperandTypeErr(1, operands[0].Value, "array", "set")
|
|
}
|
|
}
|
|
|
|
// Deprecated: deprecated in v0.13.0.
|
|
func builtinToSet(_ BuiltinContext, operands []*ast.Term, iter func(*ast.Term) error) error {
|
|
switch val := operands[0].Value.(type) {
|
|
case *ast.Array:
|
|
s := ast.NewSet()
|
|
val.Foreach(func(v *ast.Term) {
|
|
s.Add(v)
|
|
})
|
|
return iter(ast.NewTerm(s))
|
|
case ast.Set:
|
|
return iter(ast.NewTerm(val))
|
|
default:
|
|
return builtins.NewOperandTypeErr(1, operands[0].Value, "array", "set")
|
|
}
|
|
}
|
|
|
|
// Deprecated: deprecated in v0.13.0.
|
|
func builtinToString(_ BuiltinContext, operands []*ast.Term, iter func(*ast.Term) error) error {
|
|
switch val := operands[0].Value.(type) {
|
|
case ast.String:
|
|
return iter(ast.NewTerm(val))
|
|
default:
|
|
return builtins.NewOperandTypeErr(1, operands[0].Value, "string")
|
|
}
|
|
}
|
|
|
|
// Deprecated: deprecated in v0.13.0.
|
|
func builtinToBoolean(_ BuiltinContext, operands []*ast.Term, iter func(*ast.Term) error) error {
|
|
switch val := operands[0].Value.(type) {
|
|
case ast.Boolean:
|
|
return iter(ast.NewTerm(val))
|
|
default:
|
|
return builtins.NewOperandTypeErr(1, operands[0].Value, "boolean")
|
|
}
|
|
}
|
|
|
|
// Deprecated: deprecated in v0.13.0.
|
|
func builtinToNull(_ BuiltinContext, operands []*ast.Term, iter func(*ast.Term) error) error {
|
|
switch val := operands[0].Value.(type) {
|
|
case ast.Null:
|
|
return iter(ast.NewTerm(val))
|
|
default:
|
|
return builtins.NewOperandTypeErr(1, operands[0].Value, "null")
|
|
}
|
|
}
|
|
|
|
// Deprecated: deprecated in v0.13.0.
|
|
func builtinToObject(_ BuiltinContext, operands []*ast.Term, iter func(*ast.Term) error) error {
|
|
switch val := operands[0].Value.(type) {
|
|
case ast.Object:
|
|
return iter(ast.NewTerm(val))
|
|
default:
|
|
return builtins.NewOperandTypeErr(1, operands[0].Value, "object")
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
RegisterBuiltinFunc(ast.ToNumber.Name, builtinToNumber)
|
|
RegisterBuiltinFunc(ast.CastArray.Name, builtinToArray)
|
|
RegisterBuiltinFunc(ast.CastSet.Name, builtinToSet)
|
|
RegisterBuiltinFunc(ast.CastString.Name, builtinToString)
|
|
RegisterBuiltinFunc(ast.CastBoolean.Name, builtinToBoolean)
|
|
RegisterBuiltinFunc(ast.CastNull.Name, builtinToNull)
|
|
RegisterBuiltinFunc(ast.CastObject.Name, builtinToObject)
|
|
}
|