03-Template-Advance template inheritance

bonfy · Sep 7, 2018 · e5fd3ef · AaronWharton · Dec 1, 2018 · e5fd3ef
1 parent 134267d
commit e5fd3ef
Show file tree

Hide file tree

Showing 4 changed files with 60 additions and 18 deletions.
diff --git a/main.go b/main.go
@@ -2,7 +2,9 @@ package main
 
 import (
 	"html/template"
+	"io/ioutil"
 	"net/http"
+	"os"
 )
 
 // User struct
@@ -23,6 +25,41 @@ type IndexViewModel struct {
 	Posts []Post
 }
 
+// PopulateTemplates func
+// Create map template name to template.Template
+func PopulateTemplates() map[string]*template.Template {
+	const basePath = "templates"
+	result := make(map[string]*template.Template)
+
+	layout := template.Must(template.ParseFiles(basePath + "/_base.html"))
+	dir, err := os.Open(basePath + "/content")
+	if err != nil {
+		panic("Failed to open template blocks directory: " + err.Error())
+	}
+	fis, err := dir.Readdir(-1)
+	if err != nil {
+		panic("Failed to read contents of content directory: " + err.Error())
+	}
+	for _, fi := range fis {
+		f, err := os.Open(basePath + "/content/" + fi.Name())
+		if err != nil {
+			panic("Failed to open template '" + fi.Name() + "'")
+		}
+		content, err := ioutil.ReadAll(f)
+		if err != nil {
+			panic("Failed to read content from file '" + fi.Name() + "'")
+		}
+		f.Close()
+		tmpl := template.Must(layout.Clone())
+		_, err = tmpl.Parse(string(content))
+		if err != nil {
+			panic("Failed to parse contents of '" + fi.Name() + "' as template")
+		}
+		result[fi.Name()] = tmpl
+	}
+	return result
+}
+
 func main() {
 	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
 		u1 := User{Username: "bonfy"}
@@ -34,8 +71,9 @@ func main() {
 		}
 
 		v := IndexViewModel{Title: "Homepage", User: u1, Posts: posts}
-		tpl, _ := template.ParseFiles("templates/index.html")
-		tpl.Execute(w, &v)
+
+		templates := PopulateTemplates()
+		templates["index.html"].Execute(w, &v)
 	})
 	http.ListenAndServe(":8888", nil)
 }
diff --git a/templates/_base.html b/templates/_base.html
@@ -0,0 +1,13 @@
+<html>
+    <head>
+        {{if .Title}}
+        <title>{{.Title}} - blog</title>
+        {{else}}
+        <title>Welcome to blog!</title>
+        {{end}}
+    </head>
+    <body>
+        <div>Blog: <a href="/">Home</a></div>
+        {{template "content" .}}
+    </body>
+</html>
diff --git a/templates/content/index.html b/templates/content/index.html
@@ -0,0 +1,7 @@
+{{define "content"}}
+    <h1>Hello, {{.User.Username}}!</h1>
+
+    {{range .Posts}}
+        <div><p>{{ .User.Username }} says: <b>{{ .Body }}</b></p></div>
+    {{end}}
+{{end}}
diff --git a/templates/index.html b/templates/index.html