forked from tal-tech/rigger
/
render.go
122 lines (107 loc) · 2.35 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
119
120
121
122
package xorm
import (
"bytes"
"io/ioutil"
"os"
"path"
"strings"
"text/template"
"xorm.io/xorm/schemas"
"github.com/lunny/log"
)
type Render struct {
isMultiFile bool
genDir string
prefix string
model string
tables []*schemas.Table
langTmpl LangTmpl
}
func (r *Render) Do(tmpl *template.Template, newFileName, ext string) error {
var (
w *os.File
err error
)
if !r.isMultiFile {
w, err = os.Create(path.Join(r.genDir, newFileName))
if err != nil {
log.Errorf("%v", err)
return err
}
imports := r.langTmpl.GenImports(r.tables)
tbls := make([]*schemas.Table, 0)
for _, table := range r.tables {
//[SWH|+]
if r.prefix != "" {
table.Name = strings.TrimPrefix(table.Name, r.prefix)
}
tbls = append(tbls, table)
}
newbytes := bytes.NewBufferString("")
t := &Tmpl{Tables: tbls, Imports: imports, Models: r.model}
err = tmpl.Execute(newbytes, t)
if err != nil {
log.Errorf("%v", err)
return err
}
tplcontent, err := ioutil.ReadAll(newbytes)
if err != nil {
log.Errorf("%v", err)
return err
}
var source string
if r.langTmpl.Formater != nil {
source, err = r.langTmpl.Formater(string(tplcontent))
if err != nil {
log.Errorf("%v", err)
return err
}
} else {
source = string(tplcontent)
}
w.WriteString(source)
w.Close()
} else {
for _, table := range r.tables {
//[SWH|+]
if r.prefix != "" {
table.Name = strings.TrimPrefix(table.Name, r.prefix)
}
filename := table2Obj(table.Name)
// imports
tbs := []*schemas.Table{table}
imports := r.langTmpl.GenImports(tbs)
w, err := os.Create(path.Join(r.genDir, filename+ext))
if err != nil {
log.Errorf("%v", err)
return err
}
defer w.Close()
newbytes := bytes.NewBufferString("")
t := &Tmpl{Tables: tbs, Imports: imports, Models: r.model}
err = tmpl.Execute(newbytes, t)
if err != nil {
log.Errorf("%v", err)
return err
}
tplcontent, err := ioutil.ReadAll(newbytes)
if err != nil {
log.Errorf("%v", err)
return err
}
var source string
if r.langTmpl.Formater != nil {
source, err = r.langTmpl.Formater(string(tplcontent))
if err != nil {
log.Errorf("%v-%v", err, string(tplcontent))
//return err
}
} else {
source = string(tplcontent)
}
w.WriteString(source)
w.Close()
}
}
return nil
}