forked from hairyhenderson/gomplate
/
gomplate.go
65 lines (54 loc) · 1.36 KB
/
gomplate.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
package main
import (
"io"
"log"
"text/template"
)
func (g *Gomplate) createTemplate() *template.Template {
return template.New("template").Funcs(g.funcMap).Option("missingkey=error")
}
// Gomplate -
type Gomplate struct {
funcMap template.FuncMap
leftDelim string
rightDelim string
}
// RunTemplate -
func (g *Gomplate) RunTemplate(text string, out io.Writer) {
context := &Context{}
tmpl, err := g.createTemplate().Delims(g.leftDelim, g.rightDelim).Parse(text)
if err != nil {
log.Fatalf("Line %q: %v\n", text, err)
}
if err := tmpl.Execute(out, context); err != nil {
panic(err)
}
}
// NewGomplate -
func NewGomplate(data *Data, leftDelim, rightDelim string) *Gomplate {
return &Gomplate{
leftDelim: leftDelim,
rightDelim: rightDelim,
funcMap: initFuncs(data),
}
}
func runTemplate(o *GomplateOpts) error {
defer runCleanupHooks()
data := NewData(o.dataSources, o.dataSourceHeaders)
g := NewGomplate(data, o.lDelim, o.rDelim)
if o.inputDir != "" {
return processInputDir(o.inputDir, o.outputDir, g)
}
return processInputFiles(o.input, o.inputFiles, o.outputFiles, g)
}
// Called from process.go ...
func renderTemplate(g *Gomplate, inString string, outPath string) error {
outFile, err := openOutFile(outPath)
if err != nil {
return err
}
// nolint: errcheck
defer outFile.Close()
g.RunTemplate(inString, outFile)
return nil
}