From 7cd49fce0da2296f0eef73ab0195dcd16415c900 Mon Sep 17 00:00:00 2001
From: MrMelon54 <github@mrmelon54.com>
Date: Tue, 13 Feb 2024 18:14:52 +0000
Subject: [PATCH] Move setLoginDataCookie before redirecting to /login/otp

---
 server/login.go | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/server/login.go b/server/login.go
index fe33b5c..1251949 100644
--- a/server/login.go
+++ b/server/login.go
@@ -126,6 +126,10 @@ func (h *HttpServer) LoginPost(rw http.ResponseWriter, req *http.Request, _ http
 		NeedOtp: hasOtp,
 	}
 
+	if h.setLoginDataCookie(rw, auth) {
+		return
+	}
+
 	if hasOtp {
 		originUrl, err := url.Parse(req.FormValue("redirect"))
 		if err != nil {
@@ -137,9 +141,6 @@ func (h *HttpServer) LoginPost(rw http.ResponseWriter, req *http.Request, _ http
 		return
 	}
 
-	if h.setLoginDataCookie(rw, auth) {
-		return
-	}
 	h.SafeRedirect(rw, req)
 }