Add app root options for heroku can found template file

Procfile

@@ -1,2 +1,2 @@
-web: go-plurk-robot serve
+web: go-plurk-robot serve -r /app
 worker: go-plurk-robot robot

cmd/go-plurk-robot/commands.go

@@ -20,6 +20,7 @@ func setupCommandFlags() {
 
 	// Add Server
 	cmdServe.Flags().StringVarP(&Port, "port", "p", "5000", "Specify web server port")
+	cmdServe.Flags().StringP("root", "r", "", "Specify app root if cannot found tempalte file")
 }
 
 // A shortcut for Add Plurk, can use with cronjob
@@ -86,7 +87,8 @@ var cmdServe = &cobra.Command{
 	Short: "Start a Web UI for robot",
 	Long:  `Start a Web UI with analytics and some useful tool for robot and plurk`,
 	Run: func(cmd *cobra.Command, args []string) {
-		server.Serve(Port)
+		appRoot, _ := cmd.Flags().GetString("root")
+		server.Serve(Port, appRoot)
 	},
 }
 

server/server.go

@@ -14,7 +14,8 @@ import (
 )
 
 var (
-	cookie *sessions.CookieStore
+	cookie  *sessions.CookieStore
+	appRoot string
 )
 
 type Renderer struct {
@@ -32,13 +33,15 @@ func getDatabase() (mdb *mgo.Database, err error) {
 	return
 }
 
-func Serve(Port string) {
+func Serve(Port string, AppRoot string) {
 
 	// Fallback port to 5000 for local test
 	if len(Port) <= 0 {
 		Port = "5000"
 	}
 
+	appRoot = AppRoot
+
 	cookie = NewCookieStore()
 
 	server := echo.New()
@@ -64,9 +67,14 @@ func Serve(Port string) {
 }
 
 func getRenderer() *Renderer {
-	// Get correctly path to template file
-	_, filename, _, _ := runtime.Caller(1)
-	tmplPath := filepath.Dir(filename)
+	var tmplPath string
+	if len(appRoot) <= 0 {
+		// Get correctly path to template file
+		_, filename, _, _ := runtime.Caller(1)
+		tmplPath = filepath.Dir(filename)
+	} else {
+		tmplPath = appRoot + "/server"
+	}
 	templates := template.Must(template.ParseGlob(tmplPath + "/template/*.tmpl"))
 	templates.ParseGlob(tmplPath + "/template/*/*.tmpl")
 	return &Renderer{
@@ -75,8 +83,13 @@ func getRenderer() *Renderer {
 }
 
 func setupStatic(server *echo.Echo) {
-	_, filename, _, _ := runtime.Caller(1)
-	packagePath := filepath.Dir(filename)
+	var packagePath string
+	if len(appRoot) <= 0 {
+		_, filename, _, _ := runtime.Caller(1)
+		packagePath = filepath.Dir(filename)
+	} else {
+		packagePath = appRoot + "/server"
+	}
 	server.Static("/js", packagePath+"/static/js")
 	server.Static("/css", packagePath+"/static/css")
 	server.Static("/vendor", packagePath+"/static/vendor")