initial commit (fmstatic -> snkt)

1 files changed, 139 insertions, 0 deletions

diff --git a/render/render.go b/render/render.go
new file mode 100644
index 0000000..3661607
--- /dev/null
+++ b/render/render.go
@@ -0,0 +1,139 @@
+package render
+
+import (
+	"bytes"
+	"snkt/config"
+	"io/ioutil"
+	"log"
+	"os"
+	"path"
+	"path/filepath"
+	"regexp"
+	"text/template"
+)
+
+var templates map[string]*template.Template
+var BASE_TEMPLATE = "base"
+var rel_href *regexp.Regexp
+var rel_src *regexp.Regexp
+
+/*
+Renderable interface - objects that render themeslves to a []byte and
+know where they should end up in the filesystem
+*/
+type Renderable interface {
+	Render() []byte
+	Target() string
+}
+
+func Write(a Renderable) {
+	if config.Config.Verbose {
+		log.Printf("Writing to %s\n", a.Target())
+	}
+	os.MkdirAll(path.Dir(a.Target()), 0755)
+	err := ioutil.WriteFile(a.Target(), a.Render(), 0755)
+	if err != nil {
+		log.Println(err)
+	}
+}
+
+/*
+Initializes templates from config.TmplDir
+Templates are mapped by filename
+*/
+func Init() {
+	templates = make(map[string]*template.Template)
+	ts, err := filepath.Glob(config.Config.TmplDir + "/*")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	tmplFuncs := template.FuncMap{
+		"ResolveURLs": ResolveURLs,
+		"SiteTitle": SiteTitle,
+		"SiteURL": SiteURL,
+	}
+
+	
+	base := path.Join(config.Config.TmplDir, BASE_TEMPLATE)
+	for _, t := range ts {
+		tf := filepath.Base(t)
+
+		// Duly Noted: set funcs before parsefiles or you get no funcs
+		// templates[tf] = template.Must(template.ParseFiles(t, base))
+
+		tx := template.New("t").Funcs(tmplFuncs)
+		templates[tf], err = tx.ParseFiles(base, t)
+		if err != nil {
+			panic(err)
+		}
+	}
+	rel_href = regexp.MustCompile(`href="/(.+)"`)
+	rel_src = regexp.MustCompile(`src="/(.+)"`)
+}
+
+/*
+Render fills the template "name" using data via the BASE_TEMPLATE
+*/
+func Render(name string, data interface{}) []byte {
+	return RenderNameVia(name, BASE_TEMPLATE, data)
+}
+
+/*
+Render fills the template "name" using data
+*/
+func RenderOnly(name string, data interface{}) []byte {
+	return RenderNameVia(name, name, data)
+}
+
+/*
+Render fills the template "name" using data through via (ex: BASE_TEMPLATE)
+*/
+func RenderNameVia(name string, via string, data interface{}) []byte {
+	t, ok := templates[name]
+	if !ok {
+		log.Printf("can not find template named %s\n", name)
+	}
+
+	var buf bytes.Buffer
+	err := t.ExecuteTemplate(&buf, via, data)
+	if err != nil {
+		log.Println(err)
+	}
+	return buf.Bytes()
+}
+
+/*
+Finds any relative links/images in html and resolves by adding prefix
+*/
+func ResolveURLs(html, prefix string) string {
+	bts := []byte(html)
+	bts = rel_href.ReplaceAll(bts, []byte(`href="`+prefix+`/$1"`))
+	bts = rel_src.ReplaceAll(bts, []byte(`src="`+prefix+`/$1"`))
+	return string(bts)
+}
+
+/*
+Runs all regex filters specified in config.Config.Filters
+*/
+func Filter(txt []byte) []byte {
+	for _, f := range config.Config.Filters {
+		// TODO: only compile these once at init
+		re := regexp.MustCompile(f.S)
+		txt = re.ReplaceAll(txt, []byte(f.R))
+	}
+	return txt
+}
+
+func TmplExists(t string) bool {
+	_, ok := templates[t]
+	return ok
+}
+
+func SiteTitle() string {
+	return config.Config.SiteTitle
+}
+
+func SiteURL() string {
+	return config.Config.SiteURL
+}