-
Notifications
You must be signed in to change notification settings - Fork 0
/
render.go
118 lines (114 loc) · 2.45 KB
/
render.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
106
107
108
109
110
111
112
113
114
115
116
117
118
package generator
import (
"bufio"
"bytes"
"github.com/gobuffalo/packr/v2"
"go.uber.org/zap"
"os"
"path/filepath"
"text/template"
)
func RenderWithStruct(outPath, tmplDir, tmplName string, data interface{}, cover bool) {
var err error
var buff bytes.Buffer
var content string
var tmpl *template.Template
if tmplDir != "" {
tmpl, err = template.ParseFiles(filepath.Join(tmplDir, tmplName))
if err != nil {
zap.S().Info(err)
return
}
} else {
content, err := packr.New("lazy-templates", "../templates").FindString(tmplName)
if err != nil {
zap.S().Info(err)
return
}
tmpl, err = template.New(tmplName).Parse(content)
if err != nil {
zap.S().Info(err)
return
}
}
err = tmpl.Execute(&buff, data)
if err != nil {
zap.S().Info(err)
return
}
var fs *os.File
defer fs.Close()
_, err = os.Stat(outPath)
content = buff.String()
if err != nil { // file not exist
fs, err = os.Create(outPath) //create file
} else { // file exist
if !cover {
fs, err = os.OpenFile(outPath, os.O_WRONLY|os.O_APPEND, 0666)
} else {
fs, err = os.OpenFile(outPath, os.O_WRONLY|os.O_TRUNC, 0666)
}
}
if err != nil {
zap.S().Info(err)
return
}
writer := bufio.NewWriter(fs)
writer.WriteString(content)
writer.Flush()
}
func RenderWithMap(outPath, tmplDir, tmplName string, data map[string]interface{}, cover bool) {
zap.S().Debug(outPath)
var err error
var buff bytes.Buffer
var content string
var tmpl *template.Template
if tmplDir != "" {
tmpl, err = template.ParseFiles(filepath.Join(tmplDir, tmplName))
if err != nil {
zap.S().Info(err)
return
}
} else {
content, err := packr.New("lazy-templates", "../templates").FindString(tmplName)
if err != nil {
zap.S().Info(err)
return
}
tmpl, err = template.New(tmplName).Parse(content)
if err != nil {
zap.S().Info(err)
return
}
}
var fs *os.File
var isNew = false
defer fs.Close()
_, err = os.Stat(outPath)
if err != nil {
isNew = true
}
data["NewFile"] = isNew
err = tmpl.Execute(&buff, data)
if err != nil {
zap.S().Info(err)
return
}
content = buff.String()
if isNew { // file not exist
fs, err = os.Create(outPath) //create file
} else { // file exist
if !cover {
fs, err = os.OpenFile(outPath, os.O_WRONLY|os.O_APPEND, 0666)
} else {
fs, err = os.OpenFile(outPath, os.O_WRONLY|os.O_TRUNC, 0666)
}
}
if err != nil {
zap.S().Info(err)
return
}
writer := bufio.NewWriter(fs)
writer.WriteString(content)
writer.Flush()
}