fix - ensure login step catches incorrectly entered 2FA (#599)

mlemgroup · Sep 14, 2023 · 5e95c0b · 5e95c0b
1 parent 124e914
commit 5e95c0b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/Mlem/API/Models/APIErrorResponse.swift b/Mlem/API/Models/APIErrorResponse.swift
@@ -18,8 +18,13 @@ private let possibleCredentialErrors = [
     "couldnt_find_that_username_or_email"
 ]
 
+private let possible2FAErrors = [
+    "missing_totp_token",
+    "incorrect_totp_token"
+]
+
 extension APIErrorResponse {
     var isIncorrectLogin: Bool { possibleCredentialErrors.contains(error) }
-    var requires2FA: Bool { error == "missing_totp_token" }
+    var requires2FA: Bool { possible2FAErrors.contains(error) }
     var isNotLoggedIn: Bool { error == "not_logged_in" }
 }