template: support custom functions

Signed-off-by: Tw <tw19881113@gmail.com>
This commit is contained in:
Tw 2017-04-18 16:17:30 +08:00
parent 236341f78b
commit f77a7a805a
2 changed files with 8 additions and 1 deletions

View File

@ -244,6 +244,9 @@ func (c Context) Markdown(filename string) (string, error) {
return string(markdown), nil
}
// TemplateFuncs contains user defined functions
var TemplateFuncs = template.FuncMap{}
// ContextInclude opens filename using fs and executes a template with the context ctx.
// This does the same thing that Context.Include() does, but with the ability to provide
// your own context so that the included files can have access to additional fields your
@ -261,7 +264,8 @@ func ContextInclude(filename string, ctx interface{}, fs http.FileSystem) (strin
return "", err
}
tpl, err := template.New(filename).Parse(string(body))
tpl := template.New(filename).Funcs(TemplateFuncs)
tpl, err = tpl.Parse(string(body))
if err != nil {
return "", err
}

View File

@ -44,6 +44,9 @@ func (t Templates) ServeHTTP(w http.ResponseWriter, r *http.Request) (int, error
tpl.Delims(rule.Delims[0], rule.Delims[1])
}
// Add custom functions
tpl.Funcs(httpserver.TemplateFuncs)
// Build the template
templatePath := filepath.Join(t.Root, fpath)
tpl, err := tpl.ParseFiles(templatePath)