mirror of
https://github.com/jhillyerd/inbucket.git
synced 2025-12-17 09:37:02 +00:00
85 lines
2.3 KiB
Go
85 lines
2.3 KiB
Go
package web
|
|
|
|
import (
|
|
"html/template"
|
|
"net/http"
|
|
"path"
|
|
"path/filepath"
|
|
"sync"
|
|
|
|
"github.com/rs/zerolog/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.Error().Str("module", "web").Str("path", name).Err(err).
|
|
Msg("Error in template")
|
|
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.Error().Str("module", "web").Str("path", name).Err(err).
|
|
Msg("Error in template")
|
|
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.Debug().Str("module", "web").Str("path", name).Msg("Parsing template")
|
|
|
|
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.Debug().Str("module", "web").Str("path", name).Msg("Caching partial")
|
|
cachedTemplates[name] = t
|
|
} else {
|
|
log.Debug().Str("module", "web").Str("path", name).Msg("Caching template")
|
|
cachedTemplates[name] = t
|
|
}
|
|
}
|
|
|
|
return t, nil
|
|
}
|