Move https middleware to internal/handlers

1 files changed, 27 insertions, 0 deletions

diff --git a/internal/handlers/https.go b/internal/handlers/https.go
new file mode 100644
index 00000000..84c7e159
--- /dev/null
+++ b/internal/handlers/https.go
@@ -0,0 +1,27 @@
+package handlers
+
+import (
+	"net/http"
+
+	"gitlab.com/gitlab-org/gitlab-pages/internal/request"
+)
+
+func HTTPSRedirectMiddleware(handler http.Handler, redirect bool) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if redirect && !request.IsHTTPS(r) {
+			redirectToHTTPS(w, r, http.StatusTemporaryRedirect)
+			return
+		}
+
+		handler.ServeHTTP(w, r)
+	})
+}
+
+func redirectToHTTPS(w http.ResponseWriter, r *http.Request, statusCode int) {
+	u := *r.URL
+	u.Scheme = request.SchemeHTTPS
+	u.Host = r.Host
+	u.User = nil
+
+	http.Redirect(w, r, u.String(), statusCode)
+}