forked from sunfmin/mangotemplate
/
template.go
125 lines (106 loc) · 2.74 KB
/
template.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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
package mangotemplate
import (
"errors"
"fmt"
"html/template"
"io"
"io/ioutil"
"regexp"
"strings"
)
var (
AutoReload = false
TemplatePath = "templates/"
TemplateSuffix = ".html"
templateKeyword = regexp.MustCompile(`\{\{\ *template\ +\"([^\}]*)\"[^\}]*\}\}`)
templateFuncMap = map[*template.Template]template.FuncMap{}
)
func Funcs(tpl *template.Template, funcmap template.FuncMap) {
tpl.Funcs(funcmap)
templateFuncMap[tpl] = funcmap
}
func Render(preloadedTpl *template.Template, wr io.Writer, name string, data interface{}) (err error) {
if !AutoReload {
err = preloadedTpl.ExecuteTemplate(wr, name, data)
return
}
tpl := template.New(name)
addFunc(tpl, preloadedTpl)
err = parseTemplates(tpl)
if err != nil {
fmt.Println("== mangotemplate: Could not reload template, will use preloaded template instead.")
fmt.Println(err.Error())
err = preloadedTpl.ExecuteTemplate(wr, name, data)
return
}
err = tpl.Execute(wr, data)
check(err)
return
}
func addFunc(tpl *template.Template, preloadedTpl *template.Template) {
funcmap, ok := templateFuncMap[preloadedTpl]
if ok {
tpl.Funcs(funcmap)
}
}
func parseTemplates(tpl *template.Template) (err error) {
templatesToBeParsed := []string{tpl.Name()}
for len(templatesToBeParsed) != 0 {
name := templatesToBeParsed[0]
templatesToBeParsed = templatesToBeParsed[1:]
parsed := false
for _, t := range tpl.Templates() {
if tpl.Name() != name && t.Name() == name {
parsed = true
break
}
}
if !parsed {
content, err := readTemplate(name)
if err != nil {
return err
}
_, err = tpl.Parse(content)
check(err)
for _, matched := range templateKeyword.FindAllStringSubmatch(content, -1) {
templatesToBeParsed = append(templatesToBeParsed, matched[1])
}
}
}
return
}
func check(err error) {
if err != nil {
panic(err)
}
}
func readTemplate(name string) (result string, err error) {
paths := templatePaths(name)
for _, path := range paths {
var content []byte
content, err = ioutil.ReadFile(path)
if err == nil {
result = string(content)
return
}
}
err = errors.New("Could not find template [" + name + "] from: " + strings.Join(paths, ", "))
return
}
// Example
// input: index/menu
// output:
// [templates/index/menu.html,
// templates/index/_menu.html,
// templates/layout/index/menu.html,
// templates/layout/index/_menu.html]
func templatePaths(name string) []string {
names := strings.Split(name, "/")
partialName := strings.Join(names[:len(names)-1], "/") + "/_" + names[len(names)-1]
return []string{
TemplatePath + name + TemplateSuffix,
TemplatePath + partialName + TemplateSuffix,
TemplatePath + "layout/" + name + TemplateSuffix,
TemplatePath + "layout/" + partialName + TemplateSuffix,
}
}