From cee139e9601bcd6802c3e754a15643e897ae32dc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?William=20Bouzour=C3=A8ne?= <william@bouzourene.ch>
Date: Sun, 5 Jan 2025 14:45:44 +0100
Subject: [PATCH] Fix redirect after totp enroll

---
 controllers/mfa.go | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/controllers/mfa.go b/controllers/mfa.go
index b40acec..9cd7654 100644
--- a/controllers/mfa.go
+++ b/controllers/mfa.go
@@ -90,10 +90,6 @@ func TotpEnrollPage(c *fiber.Ctx) error {
 			}
 
 			sess.Set("totp-verified", "yes")
-			err = sess.Save()
-			if err != nil {
-				return err
-			}
 
 			redirectId := c.Query("redirect")
 			redirectUrl := "/"
@@ -107,6 +103,11 @@ func TotpEnrollPage(c *fiber.Ctx) error {
 				}
 			}
 
+			err = sess.Save()
+			if err != nil {
+				return err
+			}
+
 			return c.Redirect(redirectUrl)
 		} else {
 			mfaError = "Code temporaire invalide"