adyxax/tfstated
1
0
Fork 0
Code Issues 48 Pull requests Releases Packages Wiki Activity Actions

chore(webui): refactor session routing

Browse source
This commit is contained in:
Julien Dessaux 2025-04-23 00:15:04 +02:00
parent 4722efb73d
commit 342e1d6328
Signed by: adyxax
GPG key ID: F92E51B86E07177E
2 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

4
pkg/webui/login.go
View file

@ -97,9 +97,9 @@ func handleLoginPOST(db *database.DB) http.Handler {
})
}
func loginMiddleware(db *database.DB, requireSession func(http.Handler) http.Handler) func(http.Handler) http.Handler {
func loginMiddleware(db *database.DB, processSession func(http.Handler) http.Handler) func(http.Handler) http.Handler {
return func(next http.Handler) http.Handler {
return requireSession(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
return processSession(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 {

8
pkg/webui/routes.go
View file

@ -10,8 +10,8 @@ func addRoutes(
mux *http.ServeMux,
db *database.DB,
) {
requireSession := sessionsMiddleware(db)
requireLogin := loginMiddleware(db, requireSession)
processSession := sessionsMiddleware(db)
requireLogin := loginMiddleware(db, processSession)
requireAdmin := adminMiddleware(db, requireLogin)
mux.Handle("GET /accounts", requireLogin(handleAccountsGET(db)))
mux.Handle("GET /accounts/{id}", requireLogin(handleAccountsIdGET(db)))
@ -19,8 +19,8 @@ func addRoutes(
mux.Handle("POST /accounts/{id}/reset/{token}", handleAccountsIdResetPasswordPOST(db))
mux.Handle("POST /accounts", requireAdmin(handleAccountsPOST(db)))
mux.Handle("GET /healthz", handleHealthz())
mux.Handle("GET /login", requireSession(handleLoginGET()))
mux.Handle("POST /login", requireSession(handleLoginPOST(db)))
mux.Handle("GET /login", processSession(handleLoginGET()))
mux.Handle("POST /login", processSession(handleLoginPOST(db)))
mux.Handle("GET /logout", requireLogin(handleLogoutGET(db)))
mux.Handle("GET /settings", requireLogin(handleSettingsGET(db)))
mux.Handle("POST /settings", requireLogin(handleSettingsPOST(db)))