template.go

package main

import (
	"errors"
	"html/template"
	"io"
	"path/filepath"
	"time"
)

var templates map[string]*template.Template

type page struct {
	base     string
	template string
	data     map[string]interface{}
}

func (p page) toHTML(t string) string {
	return t + ".html"
}

func initTemplates() error {
	if templates == nil {
		templates = make(map[string]*template.Template)
	}

	templatesDir := "./templates/"

	layouts, err := filepath.Glob(templatesDir + "layouts/*.html")

	if err != nil {
		return err
	}

	pages, err := filepath.Glob(templatesDir + "pages/*.html")

	if err != nil {
		return err
	}

	for _, page := range pages {
		files := append(layouts, page)
		filename := filepath.Base(page)

		var err error

		templates[filename], err = template.New(filename).Funcs(template.FuncMap{
			"osIcon": osIcon,
			"trustHTML": func(s string) template.HTML {
				return template.HTML(s)
			},
		}).ParseFiles(files...)

		if err != nil {
			return err
		}
	}

	return nil
}

func renderTemplate(w io.Writer, name string, base string, data map[string]interface{}) error {
	if templates == nil {
		err := initTemplates()

		if err != nil {
			return err
		}
	}

	t, ok := templates[name]

	if !ok {
		return errors.New("unable to find template: " + name)
	}

	if data == nil {
		data = make(map[string]interface{})
	}

	data["BuildDate"] = time.Now()

	return t.ExecuteTemplate(w, base, data)
}

func osIcon(platform string) string {
	if platform == "Windows" {
		return "windows"
	} else if platform == "macOS" {
		return "apple"
	} else if platform == "iOS" {
		return "mobile"
	} else if platform == "Linux" {
		return "linux"
	} else {
		return "question"
	}
}