Initial QSfera import
This commit is contained in:
+45
@@ -0,0 +1,45 @@
|
||||
package topdown
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"strings"
|
||||
"text/template"
|
||||
|
||||
"github.com/open-policy-agent/opa/v1/ast"
|
||||
"github.com/open-policy-agent/opa/v1/topdown/builtins"
|
||||
)
|
||||
|
||||
func renderTemplate(_ BuiltinContext, operands []*ast.Term, iter func(*ast.Term) error) error {
|
||||
preContentTerm, err := builtins.StringOperand(operands[0].Value, 1)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
templateVariablesTerm, err := builtins.ObjectOperand(operands[1].Value, 2)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
var templateVariables map[string]any
|
||||
|
||||
if err := ast.As(templateVariablesTerm, &templateVariables); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
tmpl, err := template.New("template").Parse(string(preContentTerm))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
var buf bytes.Buffer
|
||||
if err := tmpl.Execute(&buf, templateVariables); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
res := strings.ReplaceAll(buf.String(), "<no value>", "<undefined>")
|
||||
return iter(ast.StringTerm(res))
|
||||
}
|
||||
|
||||
func init() {
|
||||
RegisterBuiltinFunc(ast.RenderTemplate.Name, renderTemplate)
|
||||
}
|
||||
Reference in New Issue
Block a user