diff --git a/schema/message.go b/schema/message.go index d3601208..892abe70 100644 --- a/schema/message.go +++ b/schema/message.go @@ -28,6 +28,7 @@ import ( "github.com/nikolalohinski/gonja" "github.com/nikolalohinski/gonja/config" + "github.com/nikolalohinski/gonja/exec" "github.com/nikolalohinski/gonja/nodes" "github.com/nikolalohinski/gonja/parser" "github.com/slongfield/pyfmt" @@ -1868,6 +1869,8 @@ const ( jinjaExtends = "extends" jinjaImport = "import" jinjaFrom = "from" + jinjaFile = "file" + jinjaFileSet = "fileset" ) func getJinjaEnv() (*gonja.Environment, error) { @@ -1911,6 +1914,24 @@ func getJinjaEnv() (*gonja.Environment, error) { return } } + if jinjaEnv.Filters.Exists(jinjaFile) { + err = jinjaEnv.Filters.Replace(jinjaFile, func(e *exec.Evaluator, in *exec.Value, params *exec.VarArgs) *exec.Value { + return exec.AsValue(fmt.Errorf("keyword[file] has been disabled")) + }) + if err != nil { + envInitErr = fmt.Errorf(formatInitError, err) + return + } + } + if jinjaEnv.Filters.Exists(jinjaFileSet) { + err = jinjaEnv.Filters.Replace(jinjaFileSet, func(e *exec.Evaluator, in *exec.Value, params *exec.VarArgs) *exec.Value { + return exec.AsValue(fmt.Errorf("keyword[fileset] has been disabled")) + }) + if err != nil { + envInitErr = fmt.Errorf(formatInitError, err) + return + } + } }) return jinjaEnv, envInitErr }