forked from mirror/gin
81 lines
1.7 KiB
Go
81 lines
1.7 KiB
Go
// Copyright 2014 Manu Martinez-Almeida. All rights reserved.
|
|
// Use of this source code is governed by a MIT style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package render
|
|
|
|
import (
|
|
"html/template"
|
|
"net/http"
|
|
)
|
|
|
|
type Delims struct {
|
|
Left string
|
|
Right string
|
|
}
|
|
|
|
type HTMLRender interface {
|
|
Instance(string, interface{}) Render
|
|
}
|
|
|
|
type HTMLProduction struct {
|
|
Template *template.Template
|
|
Delims Delims
|
|
}
|
|
|
|
type HTMLDebug struct {
|
|
Files []string
|
|
Glob string
|
|
Delims Delims
|
|
FuncMap template.FuncMap
|
|
}
|
|
|
|
type HTML struct {
|
|
Template *template.Template
|
|
Name string
|
|
Data interface{}
|
|
}
|
|
|
|
var htmlContentType = []string{"text/html; charset=utf-8"}
|
|
|
|
func (r HTMLProduction) Instance(name string, data interface{}) Render {
|
|
return HTML{
|
|
Template: r.Template,
|
|
Name: name,
|
|
Data: data,
|
|
}
|
|
}
|
|
|
|
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 len(r.Glob) > 0 {
|
|
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")
|
|
}
|
|
|
|
func (r HTML) Render(w http.ResponseWriter) error {
|
|
r.WriteContentType(w)
|
|
|
|
if len(r.Name) == 0 {
|
|
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)
|
|
}
|