diff options
Diffstat (limited to 'app/templates.go')
| -rw-r--r-- | app/templates.go | 53 |
1 files changed, 53 insertions, 0 deletions
diff --git a/app/templates.go b/app/templates.go new file mode 100644 index 0000000..4583254 --- /dev/null +++ b/app/templates.go @@ -0,0 +1,53 @@ +package app + +import ( + "bytes" + "fmt" + "html/template" + "io" + "log" + "sync" +) + +type templateMap map[string]*template.Template + +type templateStore struct { + sync.Mutex + + base string + templates templateMap +} + +func newTemplateStore(base string) *templateStore { + return &templateStore{ + base: base, + templates: make(templateMap), + } +} + +func (t *templateStore) Add(name string, template *template.Template) { + t.Lock() + defer t.Unlock() + + t.templates[name] = template +} + +func (t *templateStore) 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 +} |
