-
Notifications
You must be signed in to change notification settings - Fork 1
/
serverHandlerDynamicTemplate.go
387 lines (355 loc) · 11.6 KB
/
serverHandlerDynamicTemplate.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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
package viot
import(
"bufio"
"text/template"
"path/filepath"
"fmt"
"bytes"
"io/ioutil"
"strings"
"errors"
"os"
"io"
"reflect"
"context"
)
var errTemplateNotParse = errors.New("viot: The template has not been parsed yet!")
//标头-模本-处理动态页面文件
type shdtHeader struct{
P *ServerHandlerDynamic
filePath []string // 文件路径, map[文件名或别名]文件路径
delimLeft,delimRight string // 语法识别符
}
//openFile 打开文件内容
// dirPath string 目录
// map[string]string 内容,map[文件名]文件内容
// error 错误,如果文件不能打开读取
func (h *shdtHeader) openIncludeFile(rootPath, pagePath string) (map[string]string, error){
var(
dirPath = filepath.Dir(pagePath)
filePath string
fileBase string
fileContent = make(map[string]string)
c []byte
err error
)
for _, v := range h.filePath {
if v[0] == '/' || v[0] == '\\' {
filePath = filepath.Clean(v)
}else{
filePath = filepath.Join(dirPath, v)
filePath = filepath.Clean(filePath)
}
filePath = filepath.Join(rootPath, filePath)
c, err = ioutil.ReadFile(filePath)
if err != nil {
return nil, fmt.Errorf("viot: Dynamically embedded template file read failed(%s)", err.Error())
}
fileBase = filepath.Base(filePath)
if h.P != nil && h.P.ReplaceParse != nil {
c = h.P.ReplaceParse(filePath, c)
}
fileContent[fileBase] = string(c)
}
return fileContent, nil
}
//serverHandlerDynamicTemplate 模本-处理动态页面文件
type serverHandlerDynamicTemplate struct {
P *ServerHandlerDynamic
rootPath string // 文件目录
pagePath string // 文件名称
fileName string
t *template.Template
}
func (T *serverHandlerDynamicTemplate) ParseText(name, content string) error {
T.fileName = name
r := strings.NewReader(content)
return T.Parse(r)
}
func (T *serverHandlerDynamicTemplate) ParseFile(path string) error {
//文件名
file, err := os.Open(path)
if err != nil {
return err
}
defer file.Close()
b, err := ioutil.ReadAll(file)
if err != nil {
return err
}
T.fileName = filepath.Base(path)
r := bytes.NewReader(b)
return T.Parse(r)
}
func (T *serverHandlerDynamicTemplate) SetPath(root, page string) {
T.rootPath = root
T.pagePath = page
if T.fileName == "" {
T.fileName = filepath.Base(T.pagePath)
}
}
func (T *serverHandlerDynamicTemplate) Parse(rd io.Reader) (err error) {
var(
h shdtHeader //文件头
c string //内容
libs map[string]string //文件头嵌入的内容
)
//解析文件头和主体数据
r := bufio.NewReader(rd)
h, c, err = T.separation(r)
if err != nil {
return
}
libs, err = h.openIncludeFile(T.rootPath, T.pagePath)
if err != nil {
return
}
//模板文件内容载入Tmplate
t := template.New(T.fileName)
t.Delims(h.delimLeft, h.delimRight)
t.Funcs(TemplateFunc)
//解析主体内容
c = T.format(h.delimLeft, h.delimRight, c)
_, err = t.Parse(c)
if err != nil {
return
}
//解析子内容
T.t, err = T.loadTmpl(h.delimLeft, h.delimRight, t, libs)
return
}
func (T *serverHandlerDynamicTemplate) Execute(out io.Writer, in interface{}) error {
if T.t == nil {
return errTemplateNotParse
}
//执行模板
if tdot, ok := in.(DotContexter); ok {
tdot.WithContext(context.WithValue(tdot.Context(), "Template", &ServerHandlerDynamicTemplateExtend{Template:T.t}))
}
return T.t.ExecuteTemplate(out, T.fileName, in)
}
//separation 解析模本,头,内容
// shdtHeader 模本标头
// string 内容,动态语法
// error 错误,如果语法无法解析
func (T *serverHandlerDynamicTemplate) separation(buf *bufio.Reader) (shdtHeader, string, error) {
var (
line []byte
h = shdtHeader{
P : T.P,
delimLeft : "{{",
delimRight : "}}",
}
)
for {
l, isPrefix, err := buf.ReadLine()
if err != nil {
return shdtHeader{}, "", err
}
//空行后面是内容
if len(l) == 0 {
break
}
line = append(line, l...)
if isPrefix {
continue
}
//清除字符前面 //
i := bytes.IndexByte(line, '=')
if i < 0 {
return shdtHeader{}, "", fmt.Errorf("viot: Error parsing file header(%s)", string(line))
}
key := string(bytes.Trim(line[:i], "\t "))
i++ // skip colon
value := string(bytes.Trim(line[i:], "\t "))
if value == "" || value == "/" || value == "\\" {
return shdtHeader{}, "", fmt.Errorf("viot: Error parsing file header(%s)", string(line))
}
switch key {
case "//file":
h.filePath = append(h.filePath, value)
case "//delimLeft":
h.delimLeft = value
case "//delimRight":
h.delimRight = value
}
line = line[:0]
}
b, err := ioutil.ReadAll(buf)
if err != nil {
return shdtHeader{}, "", fmt.Errorf("viot: Error reading file body data(%s)", err.Error())
}
return h, string(b), nil
}
//loadTmpl 模本载入
// delimLeft, delimRight string 语法识别符
// t *template.Template 模本对象
// f map[string]string 模本内容,map[文件名]文件内容,这是文件头嵌入的模本文件内容。
// *template.Template 模本对象
// error 错误
func (T *serverHandlerDynamicTemplate) loadTmpl(delimLeft, delimRight string, t *template.Template, f map[string]string) (*template.Template, error) {
var tmpl *template.Template
if t == nil {
return t, errors.New("viot: The parent template is nil")
}
for k, v := range f {
tmpl = t.New(k)
v = T.format(delimLeft, delimRight, v)
_, err := tmpl.Parse(v)
if err != nil {
return nil, err
}
}
return t, nil
}
//format 语法整合
// delimLeft string 语法识别符(左)
// delimRight string 语法识别符(右)
// c string 语法内容
// string 整理后的语法
func (T *serverHandlerDynamicTemplate) format(delimLeft, delimRight, c string) string {
var bytesBuffer strings.Builder
for _, line := range []string{"\r\n", "\n", "\r"} {
if strings.Contains(c, line) {
var syntax bool
clines := strings.Split(c, line)
clinesL := len(clines)-1
for k, cline := range clines {
clineTrim := strings.Trim(cline, " \t")
leftHas := strings.HasSuffix(clineTrim, delimLeft)
rightHas := strings.HasPrefix(clineTrim, delimRight)
switch true {
case leftHas && rightHas:
//格式:\r\n }}abcx{{
clineTrim = strings.TrimPrefix(clineTrim, delimRight)
clineTrim = strings.TrimSuffix(clineTrim, delimLeft)
//写入内容,非语法
bytesBuffer.WriteString(clineTrim)
syntax = true
continue
case leftHas:
//格式:abcx{{
cline = strings.TrimRight(cline, " \t")
cline = strings.TrimSuffix(cline, delimLeft)
//写入内容,非语法
bytesBuffer.WriteString(cline)
syntax = true
continue
case rightHas:
//格式:}}12345
cline = strings.TrimLeft(cline, " \t")
cline = strings.TrimPrefix(cline, delimRight)
syntax = false
}
if syntax {
if clineTrim == "" || strings.HasPrefix(clineTrim, "//") {continue}
cline = fmt.Sprint(delimLeft, cline, delimRight)
}else{
if clinesL != k {
cline = fmt.Sprint(cline, line)
}
}
bytesBuffer.WriteString(cline)
}
//退出,已经确定换行符,不再继续
break
}
}
if bytesBuffer.Len() != 0 {
return bytesBuffer.String()
}else{
return c
}
}
type part struct{
input []reflect.Value
output []reflect.Value
}
func (T *part) Args(i int) interface{} {
if i == -1 {
var ret []interface{}
for _, in := range T.input {
ret = append(ret, typeSelect(in))
}
return ret
}
if len(T.input) > i {
v := T.input[i]
return typeSelect(v)
}
return nil
}
func (T *part) Result(out ...interface{}){
for _, arg := range out {
T.output = append(T.output, reflect.ValueOf(arg))
}
}
//这是个额外扩展,由于模板不能实现函数创建,只能在这里构造一个支持创建函数。
//在创建的函数内部,需要使用 Args 方法读取参数,使用 Result 方法返回结果。
//仅限用于template模板,自定义模板不支持
type ServerHandlerDynamicTemplateExtend struct{
*template.Template
}
//NewFunc 构建一个新的函数,仅限在template中使用
// func([]reflect.Value) []reflect.Value) 新的函数
func (T *ServerHandlerDynamicTemplateExtend) NewFunc(name string) (f func([]reflect.Value) []reflect.Value, err error) {
if T.Template == nil {
return nil, errTemplateNotParse
}
if T.Template.Lookup(name) == nil {
return nil, fmt.Errorf("viot: Template content not found {{define \"%s\"}} ... {{end}} , Calling this method does not support", name)
}
return func(in []reflect.Value) []reflect.Value {
p := &part{input: in,}
err := T.Template.ExecuteTemplate(ioutil.Discard, name, p)
if err != nil {
panic(err)
}
return p.output
}, nil
}
//Call 执行函数
// f func([]reflect.Value) []reflect.Value 由NewFunc创建的函数
// args ...interface{} 可变参数
// []interface{} 返回结果
func (T *ServerHandlerDynamicTemplateExtend) Call(f func([]reflect.Value) []reflect.Value, args ...interface{}) []interface{} {
var(
inv []reflect.Value
ret []interface{}
)
for _, arg := range args {
inv = append(inv, reflect.ValueOf(arg))
}
ef := execFunc{}
if err := ef.add(f, inv); err != nil {
panic(err)
}
//NewFunc 执行后返回是[]reflect.Value
for _, result := range ef.exec() {
//已100%确认变量的类型为reflect.Value
//1,查看func (T *serverHandlerDynamicTemplateExtend) NewFunc(name string) (f func([]reflect.Value) []reflect.Value, err error)
//2,查看func (T *execFunc) exec() (ret []interface{})
for _, rv := range result.([]reflect.Value) {
ret = append(ret, typeSelect(rv))
}
}
return ret
}
//ExecuteTemplate 调用模板
// out io.Writer 模板内容返回内容写入到out
// name string 模板名称
// in interface{} 模板点,带外模板的内容
// error 执行语法错误
func (T *ServerHandlerDynamicTemplateExtend) ExecuteTemplate(out io.Writer, name string, in interface{}) error {
if T.Template == nil {
return errTemplateNotParse
}
if T.Template.Lookup(name) == nil {
return fmt.Errorf("viot: Template content not found {{define \"%s\"}} ... {{end}} , Calling this method does not support", name)
}
//执行模板
if tdot, ok := in.(DotContexter); ok {
tdot.WithContext(context.WithValue(tdot.Context(), "Template", &ServerHandlerDynamicTemplateExtend{Template:T.Template}))
}
return T.Template.ExecuteTemplate(out, name, in)
}