mirror of
https://github.com/jhillyerd/inbucket.git
synced 2025-12-17 17:47:03 +00:00
83 lines
2.1 KiB
Go
83 lines
2.1 KiB
Go
package web
|
|
|
|
import (
|
|
"html/template"
|
|
"net/http"
|
|
"path"
|
|
"path/filepath"
|
|
"sync"
|
|
|
|
"github.com/jhillyerd/inbucket/pkg/log"
|
|
)
|
|
|
|
var cachedMutex sync.Mutex
|
|
var cachedTemplates = map[string]*template.Template{}
|
|
var cachedPartials = map[string]*template.Template{}
|
|
|
|
// RenderTemplate fetches the named template and renders it to the provided
|
|
// ResponseWriter.
|
|
func RenderTemplate(name string, w http.ResponseWriter, data interface{}) error {
|
|
t, err := ParseTemplate(name, false)
|
|
if err != nil {
|
|
log.Errorf("Error in template '%v': %v", name, err)
|
|
return err
|
|
}
|
|
w.Header().Set("Expires", "-1")
|
|
return t.Execute(w, data)
|
|
}
|
|
|
|
// RenderPartial fetches the named template and renders it to the provided
|
|
// ResponseWriter.
|
|
func RenderPartial(name string, w http.ResponseWriter, data interface{}) error {
|
|
t, err := ParseTemplate(name, true)
|
|
if err != nil {
|
|
log.Errorf("Error in template '%v': %v", name, err)
|
|
return err
|
|
}
|
|
w.Header().Set("Expires", "-1")
|
|
return t.Execute(w, data)
|
|
}
|
|
|
|
// ParseTemplate loads the requested template along with _base.html, caching
|
|
// the result (if configured to do so)
|
|
func ParseTemplate(name string, partial bool) (*template.Template, error) {
|
|
cachedMutex.Lock()
|
|
defer cachedMutex.Unlock()
|
|
|
|
if t, ok := cachedTemplates[name]; ok {
|
|
return t, nil
|
|
}
|
|
|
|
tempFile := filepath.Join(rootConfig.Web.UIDir, templateDir, filepath.FromSlash(name))
|
|
log.Tracef("Parsing template %v", tempFile)
|
|
|
|
var err error
|
|
var t *template.Template
|
|
if partial {
|
|
// Need to get basename of file to make it root template w/ funcs
|
|
base := path.Base(name)
|
|
t = template.New(base).Funcs(TemplateFuncs)
|
|
t, err = t.ParseFiles(tempFile)
|
|
} else {
|
|
t = template.New("_base.html").Funcs(TemplateFuncs)
|
|
t, err = t.ParseFiles(
|
|
filepath.Join(rootConfig.Web.UIDir, templateDir, "_base.html"), tempFile)
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Allows us to disable caching for theme development
|
|
if rootConfig.Web.TemplateCache {
|
|
if partial {
|
|
log.Tracef("Caching partial %v", name)
|
|
cachedTemplates[name] = t
|
|
} else {
|
|
log.Tracef("Caching template %v", name)
|
|
cachedTemplates[name] = t
|
|
}
|
|
}
|
|
|
|
return t, nil
|
|
}
|