adds simple logout link functionality

ivarprudnikov · Apr 22, 2024 · 6e0a3c2 · 6e0a3c2
1 parent 91820aa
commit 6e0a3c2
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 0 deletions.
diff --git a/cypress/e2e/auth.cy.js b/cypress/e2e/auth.cy.js
@@ -7,6 +7,20 @@ describe('auth spec', () => {
     cy.contains('footer', 'User: joe').should('be.visible')
     cy.clearCookies()
   })
+  it('can logout after logging in', () => {
+    cy.visit('/accounts/login')
+    cy.get('#username').type('joe')
+    cy.get('#password').type('joe')
+    cy.get('.btn-primary').click()
+    cy.contains('footer', 'User: joe').should('be.visible')
+
+    cy.contains('header a', 'Logout').should('be.visible')
+    cy.get('.logout-link').click()
+
+    cy.contains('footer', 'User: joe').should('not.exist')
+
+    cy.clearCookies()
+  })
   it('creates account and can login', () => {
     Cypress.Cookies.debug(true)
     cy.visit('/accounts/new')

diff --git a/routes.go b/routes.go
@@ -48,6 +48,7 @@ func AddRoutes(
 	preReq := newAppMiddleware(sessions, users)
 	mux.Handle("GET /accounts/login", preReq(loginPageHandler(sessions)))
 	mux.Handle("POST /accounts/login", preReq(loginAccountHandler(sessions, users)))
+	mux.Handle("GET /accounts/logout", preReq(logoutAccountHandler(sessions)))
 	mux.Handle("GET /accounts/new", preReq(createAccountPageHandler(sessions)))
 	mux.Handle("POST /accounts", preReq(createAccountHandler(sessions, users)))
 	mux.Handle("GET /messages", preReq(hasAuth(listMsgHandler(sessions, messages))))
@@ -128,6 +129,19 @@ func loginAccountHandler(sessions *sessions.CookieStore, store storage.UserStore
 	}
 }
 
+func logoutAccountHandler(sessions *sessions.CookieStore) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		sess, _ := sessions.Get(r, SESS_COOKIE)
+		sess.Values[SESS_USER_KEY] = nil
+		err := sess.Save(r, w)
+		if err != nil {
+			sendError(r.Context(), sess, w, "failed to save session", err)
+			return
+		}
+		http.Redirect(w, r, "/", http.StatusSeeOther)
+	}
+}
+
 func createAccountPageHandler(sessions *sessions.CookieStore) http.HandlerFunc {
 	return func(w http.ResponseWriter, r *http.Request) {
 		sess, _ := sessions.Get(r, SESS_COOKIE)

diff --git a/web/nav.tmpl b/web/nav.tmpl
@@ -21,6 +21,7 @@
         {{if .session.user }}
             <li class="nav-item"><a href="/messages" class="nav-link">Messages</a></li>
             <li class="nav-item"><a href="/messages/new" class="nav-link">Create new</a></li>
+            <li class="nav-item"><a href="/accounts/logout" class="nav-link logout-link">Logout</a></li>
         {{else}}
             <li class="nav-item"><a href="/accounts/login" class="nav-link">Login</a></li>
             <li class="nav-item"><a href="/accounts/new" class="nav-link">Sign up</a></li>