pastebin/templates.go
2017-07-09 01:57:21 -07:00

54 lines
915 B
Go

package main
import (
"bytes"
"fmt"
"html/template"
"io"
"log"
"sync"
)
type TemplateMap map[string]*template.Template
type Templates struct {
sync.Mutex
base string
templates TemplateMap
}
func NewTemplates(base string) *Templates {
return &Templates{
base: base,
templates: make(TemplateMap),
}
}
func (t *Templates) Add(name string, template *template.Template) {
t.Lock()
defer t.Unlock()
t.templates[name] = template
}
func (t *Templates) Exec(name string, ctx interface{}) (io.WriterTo, error) {
t.Lock()
defer t.Unlock()
template, ok := t.templates[name]
if !ok {
log.Printf("template %s not found", name)
return nil, fmt.Errorf("no such template: %s", name)
}
buf := bytes.NewBuffer([]byte{})
err := template.ExecuteTemplate(buf, t.base, ctx)
if err != nil {
log.Printf("error parsing template %s: %s", name, err)
return nil, err
}
return buf, nil
}