1 files changed, 3 insertions, 3 deletions
diff --git a/pkg/webui/login.go b/pkg/webui/login.go
index d004d82..fda227a 100644
--- a/pkg/webui/login.go
+++ b/pkg/webui/login.go
@@ -88,9 +88,9 @@ func handleLoginPOST(db *database.DB) http.Handler {
 	})
 }
 
-func loginMiddleware(db *database.DB) func(http.Handler) http.Handler {
+func loginMiddleware(db *database.DB, requireSession func(http.Handler) http.Handler) func(http.Handler) http.Handler {
 	return func(next http.Handler) http.Handler {
-		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		return requireSession(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 			w.Header().Set("Cache-Control", "no-store, no-cache")
 			session := r.Context().Value(model.SessionContextKey{})
 			if session == nil {
@@ -110,6 +110,6 @@ func loginMiddleware(db *database.DB) func(http.Handler) http.Handler {
 			}
 			ctx := context.WithValue(r.Context(), model.AccountContextKey{}, account)
 			next.ServeHTTP(w, r.WithContext(ctx))
-		})
+		}))
 	}
 }