woodpecker/pkg/template/template.go

package template

import (
	"errors"
	"html/template"
	"io"
	"crypto/md5"
	"strings"
	"fmt"

	"github.com/GeertJohan/go.rice"
)

// ErrTemplateNotFound indicates the requested template
// does not exists in the TemplateStore.
var ErrTemplateNotFound = errors.New("Template Not Found")

// registry stores a map of Templates where the key
// is the template name and the value is the *template.Template.
var registry = map[string]*template.Template{}

// ExecuteTemplate applies the template associated with t that has
// the given name to the specified data object and writes the output to wr.
func ExecuteTemplate(wr io.Writer, name string, data interface{}) error {
	templ, ok := registry[name]
	if !ok {
		return ErrTemplateNotFound
	}

	return templ.ExecuteTemplate(wr, "_", data)
}

// all template are loaded on initialization.
func init() {
	// location of templates
	box := rice.MustFindBox("pages")

	// these are all the files we need to parse. it is
	// kind of annoying that we can't list files in the
	// box, and have to enumerate each file here, but it is
	// a small price to pay to embed everything and simplify
	// the user installation process :)
	var files = []string{
		// these templates use the form.html
		// shared layout
		"login.html",
		"login_error.html",
		"forgot.html",
		"forgot_sent.html",
		"reset.html",
		"signup.html",
		"register.html",
		"install.html",

		// these templates use the default.html
		// shared layout
		"403.html",
		"404.html",
		"500.html",
		"user_dashboard.html",
		"user_password.html",
		"user_profile.html",
		"user_delete.html",
		"user_teams.html",
		"user_teams_add.html",
		"team_dashboard.html",
		"team_profile.html",
		"team_members.html",
		"team_delete.html",
		"members_add.html",
		"members_edit.html",
		"repo_dashboard.html",
		"repo_settings.html",
		"repo_delete.html",
		"repo_params.html",
		"repo_badges.html",
		"repo_keys.html",
		"repo_commit.html",
		"admin_users.html",
		"admin_users_edit.html",
		"admin_users_add.html",
		"admin_settings.html",
		"github_add.html",
		"github_link.html",
	}

	// extract the base template as a string
	base, err := box.String("base.html")
	if err != nil {
		panic(err)
	}

	assets := rice.MustFindBox("../../cmd/droned/assets")
	mainjs, err := assets.String("js/main.js")
	if err != nil {
		panic(err)
	}

	h := md5.New()
	io.WriteString(h, mainjs)
	jshash := fmt.Sprintf("%x", h.Sum(nil))
	base = strings.Replace(base, "main.js", "main.js?h=" + jshash, 1)

	// extract the base form template as a string
	form, err := box.String("form.html")
	if err != nil {
		panic(err)
	}

	// loop through files and create templates
	for i, file := range files {
		// extract template from box
		page, err := box.String(file)
		if err != nil {
			panic(err)
		}

		// HACK: choose which base template to use FOR THE RECORD I
		// don't really like this, but it works for now.
		var baseTemplate = base
		if i < 8 {
			baseTemplate = form
		}

		// parse the template and then add to the global map
		registry[file] = template.Must(template.Must(template.New("_").Parse(baseTemplate)).Parse(page))
	}

	// location of templates
	box = rice.MustFindBox("emails")

	files = []string{
		"activation.html",
		"failure.html",
		"success.html",
		"invitation.html",
		"reset_password.html",
	}

	// extract the base template as a string
	base, err = box.String("base_email.html")
	if err != nil {
		panic(err)
	}

	// loop through files and create templates
	for _, file := range files {
		// extract template from box
		email, err := box.String(file)
		if err != nil {
			panic(err)
		}

		// parse the template and then add to the global map
		registry[file] = template.Must(template.Must(template.New("_").Parse(base)).Parse(email))
	}
}