Add custom docPath flag

1 files changed, 9 insertions, 3 deletions

diff --git a/grimtube.go b/grimtube.go
index 86c102a..e768c12 100644
--- a/grimtube.go
+++ b/grimtube.go
@@ -8,11 +8,13 @@ import (
 	//"io"
 	"log"
 	"strconv"
-	//"path"
+	"path"
 
 	"git.sr.ht/~ark/ytparser"
 )
 
+var docPath string
+
 func getLangs() []string {
 	return []string{"en", "fr", "de", "ja", "ru"}
 }
@@ -22,6 +24,8 @@ func getOrders() []string {
 }
 
 func serve(w http.ResponseWriter, templatePath string, data interface{}) {
+	tmplPath := path.Join(docPath, templatePath)
+	basePath := path.Join(docPath, "templates/base.html")
 	funcMap := template.FuncMap {
 		"inc": func(i int) int {
 			return i + 1
@@ -30,7 +34,7 @@ func serve(w http.ResponseWriter, templatePath string, data interface{}) {
 			return i - 1
 		},
 	}
-	t, err := template.New("base.html").Funcs(funcMap).ParseFiles("templates/base.html", templatePath)
+	t, err := template.New("base.html").Funcs(funcMap).ParseFiles(basePath, tmplPath)
 	if err != nil {
 		panic(err)
 	} else {
@@ -145,9 +149,11 @@ func favicon(w http.ResponseWriter, r *http.Request) {
 
 func main() {
 	port := flag.Int("p", 8080, "The port to bind to.")
+	dir := flag.String("d", "", "Location of the templates and static resources.")
 	flag.Parse()
+	docPath = *dir
 	fs := http.FileServer(http.Dir("static"))
-	http.Handle("/static/", http.StripPrefix("/static/", fs))
+	http.Handle("/static/", http.StripPrefix(path.Join(docPath, "/static/"), fs))
 	http.HandleFunc("/favicon.ico", favicon)
 	http.HandleFunc("/", index)
 	http.HandleFunc("/search", search)