From 6320ceeb6b4dbfd001140ceb582b8826080a1a02 Mon Sep 17 00:00:00 2001
From: thewhitetulip <sapatil@live.com>
Date: Tue, 17 Nov 2015 22:02:14 +0530
Subject: [PATCH] now reads all .gtpl files in ./templates folder

---
 main.go | 34 ++++++++++++++++++----------------
 1 file changed, 18 insertions(+), 16 deletions(-)

diff --git a/main.go b/main.go
index c756210..0037875 100644
--- a/main.go
+++ b/main.go
@@ -8,9 +8,11 @@ import (
 	"fmt"
 	"github.com/julienschmidt/httprouter"
 	"github.com/thewhitetulip/Tasks/viewmodels"
+	"io/ioutil"
 	"log"
 	"net/http"
 	"strconv"
+	"strings"
 	"text/template"
 )
 
@@ -19,31 +21,31 @@ var deletedTemplate *template.Template
 var completedTemplate *template.Template
 var editTemplate *template.Template
 var searchTemplate *template.Template
+var templates *template.Template
 var err error
 
 func main() {
 	defer viewmodels.Close()
-	homeTemplate, err = template.ParseFiles("./templates/home.gtpl")
-	if err != nil {
-		fmt.Println(err)
-	}
-	deletedTemplate, err = template.ParseFiles("./templates/deleted.gtpl")
-	if err != nil {
-		fmt.Println(err)
+	var allFiles []string
+	files, err := ioutil.ReadDir("./templates")
+	for _, file := range files {
+		filename := file.Name()
+		if strings.HasSuffix(filename, ".gtpl") {
+			allFiles = append(allFiles, "./templates/"+filename)
+		}
 	}
 
-	editTemplate, err = template.ParseFiles("./templates/edit.gtpl")
-	if err != nil {
-		fmt.Println(err)
-	}
-	searchTemplate, err = template.ParseFiles("./templates/search.gtpl")
-	if err != nil {
-		fmt.Println(err)
-	}
-	completedTemplate, err = template.ParseFiles("./templates/completed.gtpl")
 	if err != nil {
 		fmt.Println(err)
 	}
+	templates, err = template.ParseFiles(allFiles...)
+	fmt.Println(templates)
+	homeTemplate = templates.Lookup("home.gtpl")
+	deletedTemplate = templates.Lookup("deleted.gtpl")
+
+	editTemplate = templates.Lookup("edit.gtpl")
+	searchTemplate = templates.Lookup("search.gtpl")
+	completedTemplate = templates.Lookup("completed.gtpl")
 
 	router := httprouter.New()
 	router.GET("/", ShowAllTasks)