-
Notifications
You must be signed in to change notification settings - Fork 6
/
template.go
96 lines (81 loc) · 2.1 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
package damsel
import (
"io/ioutil"
"path/filepath"
"dasa.cc/damsel/parse"
)
var Debug = false
var TemplateDir = ""
// SetPprint will force all document output to be pretty printed.
func SetPprint(b bool) {
parse.Pprint = b
}
type Template struct {
result []byte
}
// New returns a new template with no data.
func New() *Template {
t := &Template{}
return t
}
// Parse returns a new template and initializes with the []byte content.
func Parse(src []byte) (*Template, error) {
t := New()
err := t.Parse(src)
return t, err
}
// Parse initializes the template with the []byte content.
func (t *Template) Parse(src []byte) error {
s, err := parse.ActionParse(src)
if err != nil {
return err
}
t.result = s
return nil
}
// ParseString creates a new template and initializes with the string content.
func ParseString(src string) (*Template, error) {
t := New()
err := t.ParseString(src)
return t, err
}
// ParseString initializes the template with the string content.
func (t *Template) ParseString(src string) error {
return t.Parse([]byte(src))
}
// ParseFile creates a new template and initializes with the content of the given filename.
func ParseFile(filename string) (*Template, error) {
t := New()
err := t.ParseFile(filename)
return t, err
}
// ParseFile initializes the template with the content of the given filename.
func (t *Template) ParseFile(filename string) error {
b, err := ioutil.ReadFile(filepath.Join(TemplateDir, filename))
if err != nil {
return err
}
err = t.Parse(b)
return err
}
// ParseResult returns intermediary result. Integration with other template engines such as html/template should use
// this as source, passing that result on to parse.DocParse.
func (t *Template) ParseResult() []byte {
return t.result
}
// Result initiates the final parse phase and returns the document as a string.
func (t *Template) Result() (string, error) {
r, err := parse.DocParse(t.result)
if err != nil {
return "", err
}
return r, err
}
func init() {
parse.DefaultFuncMap = map[string]parse.ActionFn{
"js": js,
"css": css,
"extends": extends,
"include": include,
}
}