go_dreamfactory/lego/sys/gin/render/html.go
2022-06-07 20:18:22 +08:00

85 lines
1.8 KiB
Go

package render
import (
"html/template"
"net/http"
)
var htmlContentType = []string{"text/html; charset=utf-8"}
/*
Delims 表示一组用于 HTML 模板渲染的左右分隔符。
*/
type Delims struct {
// Left delimiter, defaults to {{.
Left string
// Right delimiter, defaults to }}.
Right string
}
/*
包含模板引用及其分隔符
*/
type HTMLRender interface {
Instance(string, interface{}) Render
}
type HTMLProduction struct {
Template *template.Template
Delims Delims
}
func (r HTMLProduction) Instance(name string, data interface{}) Render {
return HTML{
Template: r.Template,
Name: name,
Data: data,
}
}
type HTMLDebug struct {
Files []string
Glob string
Delims Delims
FuncMap template.FuncMap
}
// 返回一个实现Render接口的HTML实例。
func (r HTMLDebug) Instance(name string, data interface{}) Render {
return HTML{
Template: r.loadTemplate(),
Name: name,
Data: data,
}
}
func (r HTMLDebug) loadTemplate() *template.Template {
if r.FuncMap == nil {
r.FuncMap = template.FuncMap{}
}
if len(r.Files) > 0 {
return template.Must(template.New("").Delims(r.Delims.Left, r.Delims.Right).Funcs(r.FuncMap).ParseFiles(r.Files...))
}
if r.Glob != "" {
return template.Must(template.New("").Delims(r.Delims.Left, r.Delims.Right).Funcs(r.FuncMap).ParseGlob(r.Glob))
}
panic("the HTML debug render was created without files or glob pattern")
}
type HTML struct {
Template *template.Template
Name string
Data interface{}
}
func (r HTML) Render(w http.ResponseWriter) error {
r.WriteContentType(w)
if r.Name == "" {
return r.Template.Execute(w, r.Data)
}
return r.Template.ExecuteTemplate(w, r.Name, r.Data)
}
func (r HTML) WriteContentType(w http.ResponseWriter) {
writeContentType(w, htmlContentType)
}