refactor: move middlewares to corresponding packages

1 files changed, 18 insertions, 0 deletions

diff --git a/internal/auth/middleware.go b/internal/auth/middleware.go
new file mode 100644
index 00000000..263e48b8
--- /dev/null
+++ b/internal/auth/middleware.go
@@ -0,0 +1,18 @@
+package auth
+
+import (
+	"net/http"
+
+	"gitlab.com/gitlab-org/gitlab-pages/internal/source"
+)
+
+// NewMiddleware returns middleware which handles authentication requests
+func NewMiddleware(handler http.Handler, a *Auth, s source.Source) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if a.TryAuthenticate(w, r, s) {
+			return
+		}
+
+		handler.ServeHTTP(w, r)
+	})
+}