fix #1 Allow the use of custom template functions

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>
gin-contrib · Dec 20, 2016 · 0216d6b · 0216d6b
1 parent 599adf1
commit 0216d6b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/multitemplate.go b/multitemplate.go
@@ -42,6 +42,13 @@ func (r *Render) AddFromString(name, templateString string) *template.Template {
 	return tmpl
 }
 
+func (r Render) AddFromFilesFuncs(name string, funcMap template.FuncMap, files ...string) *template.Template {
+	tname := filepath.Base(files[0])
+	tmpl := template.Must(template.New(tname).Funcs(funcMap).ParseFiles(files...))
+	r.Add(name, tmpl)
+	return tmpl
+}
+
 func (r Render) Instance(name string, data interface{}) render.Render {
 	return render.HTML{
 		Template: r[name],