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
|
package tmpl
import (
"html/template"
"log"
"net/http"
"os"
"path"
)
// Bundle map of templates
type Bundle map[string]*template.Template
// Tmpl template renderer
type Tmpl interface {
GetTemplate(name string) *template.Template
Render(w http.ResponseWriter, templateName string, data interface{}) error
}
// HTML rendering bundle
type HTML struct {
bundle Bundle
}
// NewHTMLTmpl load a bundle
func NewHTMLTmpl(basePath string) (*HTML, error) {
print(basePath)
b, err := loadTemplates(basePath)
if err != nil {
return nil, err
}
return &HTML{
b,
}, err
}
func loadTemplates(basePath string) (Bundle, error) {
templates := make(Bundle)
baseTmplPath := path.Join(basePath, "parts", "_layout.tmpl")
baseTemplate := template.New("base")
baseTemplate, err := baseTemplate.ParseFiles(baseTmplPath)
if err != nil {
log.Fatal(err)
return nil, err
}
partsPath := path.Join(basePath, "pages")
partsDir, err := os.Open(partsPath)
if err != nil {
log.Fatal(err)
return nil, err
}
parts, err := partsDir.Readdir(-1)
if err != nil {
log.Fatal(err)
return nil, err
}
for _, part := range parts {
t, err := baseTemplate.Clone()
if err != nil {
log.Fatal(err)
return nil, err
}
templates[part.Name()] = template.Must(
t.ParseFiles(path.Join(partsPath, part.Name())))
}
return templates, nil
}
// GetTemplate load template by name
func (b *HTML) GetTemplate(name string) *template.Template {
return b.bundle[name]
}
// Render render templates
func (b *HTML) Render(w http.ResponseWriter,
templateName string,
data interface{}) error {
tpl := b.GetTemplate(templateName)
err := tpl.ExecuteTemplate(w, "base", data)
if err != nil {
return err
}
return nil
}
|
