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>
60 lines
1.5 KiB
Go
60 lines
1.5 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 (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/open-policy-agent/opa/ast"
|
|
"github.com/open-policy-agent/opa/topdown/builtins"
|
|
)
|
|
|
|
func builtinRegoParseModule(a, b ast.Value) (ast.Value, error) {
|
|
|
|
filename, err := builtins.StringOperand(a, 1)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
input, err := builtins.StringOperand(b, 1)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
module, err := ast.ParseModule(string(filename), string(input))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var buf bytes.Buffer
|
|
if err := json.NewEncoder(&buf).Encode(module); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
term, err := ast.ParseTerm(buf.String())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return term.Value, nil
|
|
}
|
|
|
|
func registerRegoMetadataBuiltinFunction(builtin *ast.Builtin) {
|
|
f := func(BuiltinContext, []*ast.Term, func(*ast.Term) error) error {
|
|
// The compiler should replace all usage of this function, so the only way to get here is within a query;
|
|
// which cannot define rules.
|
|
return fmt.Errorf("the %s function must only be called within the scope of a rule", builtin.Name)
|
|
}
|
|
RegisterBuiltinFunc(builtin.Name, f)
|
|
}
|
|
|
|
func init() {
|
|
RegisterFunctionalBuiltin2(ast.RegoParseModule.Name, builtinRegoParseModule)
|
|
registerRegoMetadataBuiltinFunction(ast.RegoMetadataChain)
|
|
registerRegoMetadataBuiltinFunction(ast.RegoMetadataRule)
|
|
}
|