1 files changed, 8 insertions, 3 deletions

diff --git a/main.go b/main.go
index 6311304..dd24d01 100644
--- a/main.go
+++ b/main.go
@@ -17,12 +17,17 @@ import (
 
         "github.com/julienschmidt/httprouter"
         _ "github.com/lib/pq"
+    "strings"
 )
 
 var app *grilist.Grilist
 
 func index(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
-        http.Redirect(w, r, "/dashboard", 302)
+    if strings.Index(r.Referer(), "127.0.0.1") != -1 {
+        w.WriteHeader(200)
+        return
+    }
+        http.Redirect(w, r, "/dashboard", 301)
 }
 
 func loadModule(mod grilist.Module) error {
@@ -61,6 +66,7 @@ func main() {
                 router,
         }
 
+        router.GET("/", index)
         router.HandlerFunc("GET", "/login", login.DisplayLogin)
         router.HandlerFunc("POST", "/login", login.Login)
         router.HandlerFunc("GET", "/settings", login.DisplayUserSettings)
@@ -68,8 +74,6 @@ func main() {
         router.HandlerFunc("GET", "/logout", login.Logout)
         router.HandlerFunc("GET", "/register", login.DisplayRegistration)
         router.HandlerFunc("POST", "/register", login.Register)
-        router.GET("/", index)
-        router.GET("/dashboard", viewDashboard)
 
         fs := http.FileServer(http.Dir("assets"))
         http.Handle("/assets/", http.StripPrefix("/assets/", fs))
@@ -81,6 +85,7 @@ func main() {
         loadModule(tags.New())
         loadModule(search.New())
         loadModule(user.New())
+        router.GET("/dashboard", viewDashboard)
 
         log.Fatal(http.ListenAndServe(":8080", nil))
 }