mix
15baa1e5d0
test / test (push) Has been cancelled
Co-authored-by: Cursor <cursoragent@cursor.com>
59 lines
1.6 KiB
Go
59 lines
1.6 KiB
Go
package auth
|
|
|
|
import (
|
|
"net/http"
|
|
)
|
|
|
|
// RequireAuth wraps a handler so only requests with a valid session cookie
|
|
// reach it; everyone else is redirected to the login page. The authenticated
|
|
// principal is stashed in the request context for downstream handlers.
|
|
func (m *Module) RequireAuth(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
token, ok := m.sessionToken(r)
|
|
if !ok {
|
|
http.Redirect(w, r, "/login", http.StatusSeeOther)
|
|
return
|
|
}
|
|
username, ok := m.sessions.Lookup(token)
|
|
if !ok {
|
|
http.Redirect(w, r, "/login", http.StatusSeeOther)
|
|
return
|
|
}
|
|
if isSessionActivity(r) && m.sessions.Touch(token) {
|
|
m.setSessionCookie(w, token)
|
|
}
|
|
u, err := m.store.GetUserByUsername(username)
|
|
if err != nil {
|
|
http.Redirect(w, r, "/login", http.StatusSeeOther)
|
|
return
|
|
}
|
|
p := principalFromUser(u)
|
|
ctx := withPrincipal(r.Context(), p)
|
|
next.ServeHTTP(w, r.WithContext(ctx))
|
|
})
|
|
}
|
|
|
|
func isSessionActivity(r *http.Request) bool {
|
|
return !(r.Method == http.MethodGet && r.Header.Get("HX-Request") != "")
|
|
}
|
|
|
|
// CurrentUser returns the authenticated username from the request context.
|
|
func CurrentUser(r *http.Request) string {
|
|
if v, ok := r.Context().Value(usernameKey).(string); ok {
|
|
return v
|
|
}
|
|
return ""
|
|
}
|
|
|
|
// RequireGlobal wraps a handler that only global administrators may reach.
|
|
func RequireGlobal(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
p, ok := CurrentPrincipal(r.Context())
|
|
if !ok || !p.IsGlobal() {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|