templates.go

package main

import (
	"bytes"
	"fmt"
	"html/template"
	"io"
	"log"
	"sync"
)

const OpenSearchTemplate string = `<?xml version="1.0" encoding="UTF-8"?>
<OpenSearchDescription xmlns="http://a9.com/-/spec/opensearch/1.1/">
  <ShortName>%s</ShortName>
  <Description>Smart bookmarks</Description>
  <Tags>search</Tags>
  <Contact>admin@localhost</Contact>
  <Url type="text/html" method="get" template="http://%s/?q={searchTerms}"/>
  <Url type="application/x-suggestions+json" method="get" template="http://%s/suggest?q={searchTerms}"/>
</OpenSearchDescription>
`

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
}