From 1551e46fed1562c20ebc1523c706bf29a3b8cc8e Mon Sep 17 00:00:00 2001
From: Victor Vrantchan <groob@users.noreply.github.com>
Date: Wed, 11 Jan 2017 14:05:07 -0500
Subject: [PATCH] add middleware to redirect setup to login if the app has an
 admin (#900)

user.
---
 cli/serve.go              |  2 ++
 server/service/handler.go | 16 +++++++++++++++-
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/cli/serve.go b/cli/serve.go
index cb8bb39efe..6ace56f3d8 100644
--- a/cli/serve.go
+++ b/cli/serve.go
@@ -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)
 				}
 			}
 
diff --git a/server/service/handler.go b/server/service/handler.go
index 54a2730d27..75a5b8a585 100644
--- a/server/service/handler.go
+++ b/server/service/handler.go
@@ -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)
+	}
+}