Stub log in and log out handling

Author: zainhoda

main.go

@@ -1,10 +1,10 @@
 package main
 
 import (
-	"fmt"
 	"html/template"
 	"log"
 	"net/http"
+	"time"
 
 	"github.com/julienschmidt/httprouter"
 )
@@ -15,34 +15,51 @@ func LoginPage(w http.ResponseWriter, req *http.Request, _ httprouter.Params) {
 
 	baseT.ExecuteTemplate(w, "base", map[string]string{
 		"PageName": "login",
+		"User":     getUser(w, req),
 	})
 }
 
+func LoginHandler(w http.ResponseWriter, req *http.Request, p httprouter.Params) {
+	username := p.ByName("email")
+	password := p.ByName("password")
+
+	if verifyUser(w, req, username, password) {
+		http.Redirect(w, req, "/admin/", http.StatusFound)
+	}
+}
+
+func LogoutHandler(w http.ResponseWriter, req *http.Request, p httprouter.Params) {
+	http.Redirect(w, req, "/", http.StatusFound)
+}
+
 func MainPage(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
 	baseT := template.Must(template.New("base").Parse(base))
 	baseT = template.Must(baseT.Parse(mainPage))
 
 	baseT.ExecuteTemplate(w, "base", map[string]string{
 		"PageName": "main",
+		"User":     getUser(w, r),
 	})
 }
 
-func SignupPage(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
-	username := ps.ByName("username")
-	password := ps.ByName("password")
-
+func SignupPage(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
 	baseT := template.Must(template.New("base").Parse(base))
 	baseT = template.Must(baseT.Parse(signup))
 
 	baseT.ExecuteTemplate(w, "base", map[string]string{
 		"PageName": "signup",
+		"User":     getUser(w, r),
 	})
+}
+
+func SignupHandler(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
+	username := ps.ByName("email")
+	password := ps.ByName("password")
 
-	// create new user
-	if len(username)+len(password) < 11 {
-		fmt.Fprintf(w, "New user"+username+" with pass:"+password+"has signed up")
+	if addUser(username, password) {
+		http.Redirect(w, r, "/admin/", http.StatusFound)
 	} else {
-		fmt.Fprintf(w, "Invalid arguments!")
+		http.Redirect(w, r, "/signup/", http.StatusFound)
 	}
 }
 
@@ -59,14 +76,45 @@ func AdminPage(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
 
 	baseT.ExecuteTemplate(w, "base", map[string]string{
 		"PageName": "admin",
+		"User":     getUser(w, r),
 	})
 }
 
+func verifyUser(w http.ResponseWriter, r *http.Request, username string, password string) bool {
+	cookie := http.Cookie{Name: "goblog", Value: "blah", Expires: time.Now().Add(time.Hour * 24 * 7 * 52), HttpOnly: true, MaxAge: 50000, Path: "/"}
+	http.SetCookie(w, &cookie)
+	return true
+}
+
+func addUser(username string, password string) bool {
+	// Insert into database
+
+	return true
+}
+
+func getUser(w http.ResponseWriter, r *http.Request) string {
+	cookie, err := r.Cookie("goblog")
+
+	if err == nil {
+		if cookie.Value != "" {
+			return cookieToUsername(cookie.Value)
+		}
+	}
+	return ""
+}
+
+func cookieToUsername(cookie string) string {
+	return "[email protected]"
+}
+
 func main() {
 	router := httprouter.New()
 	router.GET("/", MainPage)
-	router.GET("/login/", LoginPage)
+	//router.GET("/login/", LoginPage)
+	router.POST("/login/", LoginHandler)
 	router.GET("/signup/", SignupPage)
+	router.POST("/signup/", SignupHandler)
 	router.GET("/admin/", AdminPage)
+	router.GET("/logout/", LogoutHandler)
 	log.Fatal(http.ListenAndServe(":1337", router))
 }

pages.go

@@ -46,8 +46,10 @@ var base = `
         <button type="submit" class="btn btn-default">Submit</button>
       </form>
       <ul class="nav navbar-nav navbar-right">
-		<li {{if eq .PageName "signup" }}class="active"{{end}}><a href="/signup">Sign Up</a></li>
-	<li class="dropdown">
+		<li {{if not .User}}class="hidden"{{end}}> <a href="/admin">{{if .User}}{{.User}}{{end}}</a> </li>
+		<li {{if not .User}}class="hidden"{{end}}> <a href="/logout">Log Out</a> </li>
+		<li {{if .User}}class="hidden"{{end}} {{if eq .PageName "signup" }}class="active"{{end}}><a href="/signup">Sign Up</a></li>
+	<li {{if .User}}class="hidden"{{end}} class="dropdown">
 		<a href="#" class="dropdown-toggle" data-toggle="dropdown">Sign in <b class="caret"></b></a>
 		<ul class="dropdown-menu" style="padding: 15px;min-width: 250px;">
 			<li>