-
Notifications
You must be signed in to change notification settings - Fork 4
/
view.go
54 lines (44 loc) · 1.12 KB
/
view.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
package marmoset
import (
"html/template"
"os"
"path/filepath"
"regexp"
"strings"
)
var templates *template.Template
// P is just a short hand of `map[string]interface{}`
// I don't want to write `map[string]interface{}` so repeatedly... :(
type P map[string]interface{}
// LoadViews ...
func LoadViews(p string) *template.Template {
// var viewpath string
// if filepath.IsAbs(p) {
// viewpath = p
// } else {
// _, f, _, _ := runtime.Caller(1)
// viewpath = path.Join(path.Dir(f), p) + "/"
// }
viewpath := p
exp := regexp.MustCompile("[^/]+\\.html$")
pool := template.New("")
filepath.Walk(viewpath, func(fullpath string, info os.FileInfo, err error) error {
if exp.MatchString(fullpath) {
name := strings.Replace(strings.Replace(fullpath, viewpath, "", -1), filepath.Ext(fullpath), "", -1)
name = strings.Trim(name, "/")
tmp, err := template.ParseFiles(fullpath)
if err != nil {
panic(err)
}
if _, err = pool.AddParseTree(name, tmp.Tree); err != nil {
panic(err)
}
}
return nil
})
templates = pool
return templates
}
func UseTemplate(tpl *template.Template) {
templates = tpl
}