Added sessions

wybiral · wybiral · commit 430448f95fbe · 2017-11-02T12:30:54.000-05:00
diff --git a/main.go b/main.go
@@ -3,11 +3,13 @@ package main
 import (
 	"net/http"
 	"github.com/gorilla/mux"
+	"github.com/gorilla/sessions"
 	"github.com/go-redis/redis"
 	"html/template"
 )
 
 var client *redis.Client
+var store = sessions.NewCookieStore([]byte("t0p-s3cr3t"))
 var templates *template.Template
 
 func main() {
@@ -18,6 +20,9 @@ func main() {
 	r := mux.NewRouter()
 	r.HandleFunc("/", indexGetHandler).Methods("GET")
 	r.HandleFunc("/", indexPostHandler).Methods("POST")
+	r.HandleFunc("/login", loginGetHandler).Methods("GET")
+	r.HandleFunc("/login", loginPostHandler).Methods("POST")
+	r.HandleFunc("/test", testGetHandler).Methods("GET")
 	fs := http.FileServer(http.Dir("./static/"))
 	r.PathPrefix("/static/").Handler(http.StripPrefix("/static/", fs))
 	http.Handle("/", r)
@@ -38,3 +43,28 @@ func indexPostHandler(w http.ResponseWriter, r *http.Request) {
 	client.LPush("comments", comment)
 	http.Redirect(w, r, "/", 302)
 }
+
+func loginGetHandler(w http.ResponseWriter, r *http.Request) {
+	templates.ExecuteTemplate(w, "login.html", nil)
+}
+
+func loginPostHandler(w http.ResponseWriter, r *http.Request) {
+	r.ParseForm()
+	username := r.PostForm.Get("username")
+	session, _ := store.Get(r, "session")
+	session.Values["username"] = username
+	session.Save(r, w)
+}
+
+func testGetHandler(w http.ResponseWriter, r *http.Request) {
+	session, _ := store.Get(r, "session")
+	untyped, ok := session.Values["username"]
+	if !ok {
+		return
+	}
+	username, ok := untyped.(string)
+	if !ok {
+		return
+	}
+	w.Write([]byte(username))
+}
diff --git a/templates/login.html b/templates/login.html
@@ -0,0 +1,13 @@
+<html>
+    <head>
+        <title>Login</title>
+    </head>
+    <body>
+        <form method="POST">
+            Username: <input name="username">
+            <div>
+                <button type="submit">Login</button>
+            </div>
+        </form>
+    </body>
+</html>
