mirror of
https://github.com/fleetdm/fleet
synced 2026-05-24 09:28:54 +00:00
add middleware to redirect setup to login if the app has an admin (#900)
user.
This commit is contained in:
parent
671255b105
commit
1551e46fed
2 changed files with 17 additions and 1 deletions
|
|
@ -143,6 +143,8 @@ the way that the kolide server works.
|
|||
if service.RequireSetup(svc, logger) {
|
||||
apiHandler = service.WithSetup(svc, logger, apiHandler)
|
||||
frontendHandler = service.RedirectLoginToSetup(svc, logger, frontendHandler)
|
||||
} else {
|
||||
frontendHandler = service.RedirectSetupToLogin(svc, logger, frontendHandler)
|
||||
}
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -405,7 +405,7 @@ func RedirectLoginToSetup(svc kolide.Service, logger kitlog.Logger, next http.Ha
|
|||
if RequireSetup(svc, logger) {
|
||||
redirect.ServeHTTP(w, r)
|
||||
} else {
|
||||
next.ServeHTTP(w, r)
|
||||
RedirectSetupToLogin(svc, logger, next).ServeHTTP(w, r)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
@ -419,3 +419,17 @@ func RequireSetup(svc kolide.Service, logger kitlog.Logger) bool {
|
|||
}
|
||||
return len(users) == 0
|
||||
}
|
||||
|
||||
// RedirectSetupToLogin forces the /setup path to be redirected to login. This middleware is used after
|
||||
// the app has been setup.
|
||||
func RedirectSetupToLogin(svc kolide.Service, logger kitlog.Logger, next http.Handler) http.HandlerFunc {
|
||||
return func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.URL.Path == "/setup" {
|
||||
newURL := r.URL
|
||||
newURL.Path = "/login"
|
||||
http.Redirect(w, r, newURL.String(), http.StatusTemporaryRedirect)
|
||||
return
|
||||
}
|
||||
next.ServeHTTP(w, r)
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in a new issue