Initial QSfera import
This commit is contained in:
+60
@@ -0,0 +1,60 @@
|
||||
// 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/v1/ast"
|
||||
"github.com/open-policy-agent/opa/v1/topdown/builtins"
|
||||
)
|
||||
|
||||
func builtinRegoParseModule(_ BuiltinContext, operands []*ast.Term, iter func(*ast.Term) error) error {
|
||||
|
||||
filename, err := builtins.StringOperand(operands[0].Value, 1)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
input, err := builtins.StringOperand(operands[1].Value, 1)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// FIXME: Use configured rego-version?
|
||||
module, err := ast.ParseModule(string(filename), string(input))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
var buf bytes.Buffer
|
||||
if err := json.NewEncoder(&buf).Encode(module); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
term, err := ast.ParseTerm(buf.String())
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return iter(term)
|
||||
}
|
||||
|
||||
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() {
|
||||
RegisterBuiltinFunc(ast.RegoParseModule.Name, builtinRegoParseModule)
|
||||
registerRegoMetadataBuiltinFunction(ast.RegoMetadataChain)
|
||||
registerRegoMetadataBuiltinFunction(ast.RegoMetadataRule)
|
||||
}
|
||||
Reference in New Issue
Block a user