-
Notifications
You must be signed in to change notification settings - Fork 8
/
template_renderer.go
106 lines (88 loc) · 2.76 KB
/
template_renderer.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
package render
import (
"bytes"
"sort"
"text/template"
"github.com/gobuffalo/packr"
"github.com/rotisserie/eris"
)
// exported interface for using to render templates
type TemplateRenderer interface {
RenderCustomTemplates(customTemplates map[string]string, customFuncs template.FuncMap, data interface{}) ([]OutFile, error)
}
// map of template files to the file they render to
type inputTemplates map[string]OutFile
func (r *inputTemplates) add(t inputTemplates) {
if *r == nil {
*r = make(map[string]OutFile)
}
for file, out := range t {
(*r)[file] = out
}
}
type templateRenderer struct {
// core templates, baked into the binary
templates packr.Box
// custom delimiters to use for evaluations
left, right string
}
var DefaultTemplateRenderer = templateRenderer{
// a packr.Box for reading the conents of ../templates
// note that this code uses relative path
// and will need to be updated if the relative
// path from this file to the templates dir changes
templates: packr.NewBox("../templates"),
}
func (r templateRenderer) renderCoreTemplates(templatesToRender inputTemplates, data interface{}) ([]OutFile, error) {
var renderedFiles []OutFile
for tmplPath, out := range templatesToRender {
content, err := r.renderCoreTemplate(tmplPath, data)
if err != nil {
return nil, err
}
out.Content = content
renderedFiles = append(renderedFiles, out)
}
return renderedFiles, nil
}
func (r templateRenderer) renderCoreTemplate(path string, data interface{}) (string, error) {
templateText, err := r.templates.FindString(path)
if err != nil {
return "", err
}
return r.executeTemplate(path, templateText, nil, data)
}
func (r templateRenderer) RenderCustomTemplates(customTemplates map[string]string, customFuncs template.FuncMap, data interface{}) ([]OutFile, error) {
var renderedFiles []OutFile
for outPath, templateText := range customTemplates {
content, err := r.executeTemplate(outPath, templateText, customFuncs, data)
if err != nil {
return nil, eris.Wrapf(err, "error executing template for %s", outPath)
}
out := OutFile{
Path: outPath,
Content: content,
}
renderedFiles = append(renderedFiles, out)
}
sort.SliceStable(renderedFiles, func(i, j int) bool {
return renderedFiles[i].Path < renderedFiles[j].Path
})
return renderedFiles, nil
}
func (r templateRenderer) executeTemplate(name, templateText string, extraFuncs template.FuncMap, data interface{}) (string, error) {
funcs := makeTemplateFuncs(extraFuncs)
tmpl := template.New(name).Funcs(funcs)
if r.left != "" {
tmpl = tmpl.Delims(r.left, r.right)
}
tmpl, err := tmpl.Parse(templateText)
if err != nil {
return "", err
}
buf := &bytes.Buffer{}
if err := tmpl.Funcs(funcs).Execute(buf, data); err != nil {
return "", err
}
return buf.String(), nil
}