mg-transport-core/core/template.go

45 lines
1.1 KiB
Go
Raw Normal View History

2019-09-04 15:22:27 +03:00
package core
import (
"html/template"
"github.com/gin-contrib/multitemplate"
2019-09-18 10:28:55 +03:00
"github.com/gobuffalo/packr/v2"
2019-09-04 15:22:27 +03:00
)
// Renderer wraps multitemplate.Renderer in order to make it easier to use
type Renderer struct {
multitemplate.Renderer
2019-09-18 10:28:55 +03:00
TemplatesBox *packr.Box
2019-09-04 15:22:27 +03:00
FuncMap template.FuncMap
}
// NewRenderer is a Renderer constructor
func NewRenderer(funcMap template.FuncMap) Renderer {
return Renderer{
Renderer: multitemplate.NewRenderer(),
FuncMap: funcMap,
}
}
// Push is an AddFromFilesFuncs wrapper
func (r *Renderer) Push(name string, files ...string) *template.Template {
2019-09-18 10:28:55 +03:00
if r.TemplatesBox == nil {
return r.AddFromFilesFuncs(name, r.FuncMap, files...)
} else {
return r.addFromBox(name, r.FuncMap, files...)
}
2019-09-04 15:22:27 +03:00
}
2019-09-18 10:28:55 +03:00
// addFromBox adds embedded template
func (r *Renderer) addFromBox(name string, funcMap template.FuncMap, files ...string) *template.Template {
var filesData []string
for _, file := range files {
if data, err := r.TemplatesBox.FindString(file); err == nil {
filesData = append(filesData, data)
}
}
return r.AddFromStringsFuncs(name, funcMap, filesData...)
}