From b4c8dee526c5e95ecc7bc5e37c89ecd9182e9bbc Mon Sep 17 00:00:00 2001
From: Matt Baer <matt@flashstuf.com>
Date: Sun, 11 Jan 2015 17:14:45 -0500
Subject: [PATCH] Ability to specify template

For example, for an entire page which include the log inside it.
---
 tildelog/templates/log.html |  2 ++
 tildelog/tildelog.go        | 15 +++++++++++----
 2 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/tildelog/templates/log.html b/tildelog/templates/log.html
index a0d0bb7..27f31db 100644
--- a/tildelog/templates/log.html
+++ b/tildelog/templates/log.html
@@ -1,3 +1,4 @@
+{{define "log"}}
 <div id="tildelog">
 	{{range .}}
 	<hr />
@@ -5,3 +6,4 @@
 	<p>{{printf "%s" .Body}}</p>
 	{{end}}
 </div>
+{{end}}
diff --git a/tildelog/tildelog.go b/tildelog/tildelog.go
index a54ca68..8daba15 100644
--- a/tildelog/tildelog.go
+++ b/tildelog/tildelog.go
@@ -9,6 +9,7 @@ import (
 	"os"
 	"bufio"
 	"regexp"
+	"flag"
 )
 
 const entriesPath = "./entries/"
@@ -48,6 +49,10 @@ func main() {
 	fmt.Println("    ~log generator v1.0")
 	fmt.Println()
 
+	// Get any arguments
+	templateFilePtr := flag.String("template", "log", "Tildelog template file (defined name).")
+	flag.Parse()
+	
 	entryFiles := getEntries()
 	entries := make([]Entry, len(*entryFiles))
 	i := 0
@@ -62,7 +67,9 @@ func main() {
 		i++
 	}
 	
-	generateLog(entries)
+	fmt.Printf("Using template %s\n", *templateFilePtr)
+
+	generateLog(entries, *templateFilePtr)
 }
 
 var validFileFormat = regexp.MustCompile("^[0-9]{8}$")
@@ -83,7 +90,7 @@ func getEntries() *[]string {
 	return &fileList
 }
 
-func generateLog(entries []Entry) {
+func generateLog(entries []Entry, templateFile string) {
 	file, err := os.Create(outputPath + "log.html")
 	if err != nil {
 		panic(err)
@@ -92,11 +99,11 @@ func generateLog(entries []Entry) {
 	defer file.Close()
 	
 	writer := bufio.NewWriter(file)
-	template, err := template.ParseFiles(templatesPath + "log.html")
+	template, err := template.ParseGlob(templatesPath + "*.html")
 	if err != nil {
 		panic(err)
 	}
-	template.Execute(writer, entries)
+	template.ExecuteTemplate(writer, templateFile, entries)
 	writer.Flush()
 }