logout button

TODO.md

@@ -30,7 +30,7 @@
 - [x] Authentication System
   - [x] Persistent logins across server restarts
   - [x] User login
-  - [ ] User logout
+  - [x] User logout
   - [ ] Role-based access (admin, staff)
   - [x] Password hashing
 - [ ] CSRF Protection

internal/auth/auth.go

@@ -43,6 +43,7 @@ func (a *Authenticator) Middleware(in http.Handler) http.Handler {
 	handler := http.NewServeMux()
 	handler.HandleFunc("GET /login", getLogin)
 	handler.HandleFunc("POST /login", a.login)
+	handler.HandleFunc("GET /logout", a.logout)
 	handler.Handle("/", a.ensureAuth(in))
 	return handler
 }

internal/auth/session.go

@@ -50,3 +50,26 @@ func (a *Authenticator) login(w http.ResponseWriter, r *http.Request) {
 	http.SetCookie(w, &cookie)
 	http.Redirect(w, r, "/", http.StatusFound)
 }
+
+func (a *Authenticator) logout(w http.ResponseWriter, r *http.Request) {
+	var cookie http.Cookie
+	cookie.Name = "token"
+	cookie.Value = "nil"
+	cookie.HttpOnly = true
+	cookie.SameSite = http.SameSiteStrictMode
+	cookie.Path = "/"
+	http.SetCookie(w, &cookie)
+	http.Redirect(w, r, "/", http.StatusFound)
+	c, err := r.Cookie("token")
+	if err != nil {
+		return
+	}
+	_, ok := a.sessions[c.Value]
+	if !ok {
+		return
+	}
+	fmt.Println("removing session")
+	delete(a.sessions, c.Value)
+	err = a.s.RevokeSession(c.Value)
+	fmt.Println(err)
+}

internal/server/templates/layout/nav.gohtml

@@ -3,4 +3,5 @@
 <a href="/order">New order</a>
 <a href="/cakes">All Cakes</a>
 <a href="/cake">New Cake</a>
+<a href="/logout">Logout</a>
 {{end}}

internal/store/session.go

@@ -12,6 +12,12 @@ func (s *Store) AddSession(token, userName string, expiration time.Time) error {
 	return err
 }
 
+func (s *Store) RevokeSession(token string) error {
+	query := "delete from session where token = ?;"
+	_, err := s.db.Exec(query, token)
+	return err
+}
+
 func (s *Store) CleanSessions() error {
 	now := time.Now().Format("2006-01-02 15:04")
 	query := "delete from session where expiration < ?;"