Курнат Андрей
46 lines
996 B
Go
46 lines
996 B
Go
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)
|
|
}
|