-
Notifications
You must be signed in to change notification settings - Fork 335
/
templating.go
105 lines (83 loc) · 1.96 KB
/
templating.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
package main
import (
"bytes"
"fmt"
"go/format"
"io"
"io/ioutil"
"os"
"path/filepath"
"strings"
"text/template"
"github.com/Masterminds/sprig"
)
func baseTemplate() *template.Template {
return template.New("").
Funcs(sprig.TxtFuncMap())
}
func loadTemplates(rTpl *template.Template, rootDir string) (*template.Template, error) {
cleanRoot := filepath.Clean(rootDir)
pfx := len(cleanRoot) + 1
return rTpl, filepath.Walk(cleanRoot, func(path string, info os.FileInfo, err error) error {
if err != nil {
return err
}
if info.IsDir() || !strings.HasSuffix(path, ".tpl") || err != nil {
return err
}
b, err := ioutil.ReadFile(path)
if err != nil {
return err
}
name := path[pfx:]
rTpl, err = rTpl.New(name).Parse(string(b))
return err
})
}
func writeFormattedGo(dst string, tpl *template.Template, payload interface{}) error {
return write(dst, tpl, payload, func(in io.ReadWriter) (out io.ReadWriter, err error) {
var (
org, bb []byte
)
if org, err = ioutil.ReadAll(in); err != nil {
return
}
cp := bytes.NewBuffer(org)
if bb, err = format.Source(cp.Bytes()); err != nil {
// output error and return un-formatted source
_, _ = fmt.Fprintf(os.Stderr, "%s fmt warn: %v\n", dst, err)
return cp, nil
}
return bytes.NewBuffer(bb), nil
})
}
func write(dst string, tpl *template.Template, payload interface{}, pp ...func(io.ReadWriter) (io.ReadWriter, error)) (err error) {
var (
output io.WriteCloser
buf io.ReadWriter
)
if tpl == nil {
return fmt.Errorf("could not find template for %s", dst)
}
buf = &bytes.Buffer{}
if err := tpl.Execute(buf, payload); err != nil {
return err
}
for _, proc := range pp {
if buf, err = proc(buf); err != nil {
return
}
}
if dst == "" || dst == "-" {
output = os.Stdout
} else {
if output, err = os.Create(dst); err != nil {
return err
}
defer output.Close()
}
if _, err = io.Copy(output, buf); err != nil {
return err
}
return nil
}