using the logout functionality.

1 files changed, 14 insertions, 3 deletions

diff --git a/main.go b/main.go
index 863dd3a..c3d8ef6 100644
--- a/main.go
+++ b/main.go
@@ -13,9 +13,19 @@ import (
 )
 
 var renderer *frontend.Renderer
+var charakterin *login.Charakterin
 
 func index(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
-        renderer.RenderPage("index", w, make(map[string]interface{}))
+        user, err := charakterin.GetUserFromRequest(r)
+        if err != nil {
+                http.Redirect(w, r, "/login", 302)
+                return
+        }
+
+        data := make(map[string]interface{})
+
+        data["username"] = user.GetName()
+        renderer.RenderPage("index", w, data)
 }
 
 func main() {
@@ -27,16 +37,17 @@ func main() {
         if err := db.Ping(); err != nil {
                 log.Fatal(err)
         }
-        
+
         log.Println("database connection established")
 
         renderer = frontend.New("views")
-        charakterin := login.New(db)
+        charakterin = login.New(db)
         charakterin.UseRenderer(renderer)
 
         router := httprouter.New()
         router.HandlerFunc("GET", "/login", charakterin.DisplayLogin)
         router.HandlerFunc("POST", "/login", charakterin.Login)
+        router.HandlerFunc("GET", "/logout", charakterin.Logout)
         router.GET("/", index)
 
         fs := http.FileServer(http.Dir("assets"))