Do not show account revoked banner when on account revoked screen

mullvad · Dec 5, 2023 · 45b552d · 45b552d
1 parent 61921f9
commit 45b552d
Showing 1 changed file with 14 additions and 9 deletions.
diff --git a/ios/MullvadVPN/Notifications/Notification Providers/TunnelStatusNotificationProvider.swift b/ios/MullvadVPN/Notifications/Notification Providers/TunnelStatusNotificationProvider.swift
@@ -56,13 +56,7 @@ final class TunnelStatusNotificationProvider: NotificationProvider, InAppNotific
     // MARK: - Private
 
     private func handleTunnelStatus(_ tunnelStatus: TunnelStatus) {
-        let invalidateForTunnelError: Bool
-        if case let .error(blockStateReason) = tunnelStatus.state, blockStateReason != .accountExpired {
-            invalidateForTunnelError = updateLastTunnelError(blockStateReason)
-        } else {
-            invalidateForTunnelError = updateLastTunnelError(nil)
-        }
-
+        let invalidateForTunnelError = updateLastTunnelError(tunnelStatus.state)
         let invalidateForManagerError = updateTunnelManagerError(tunnelStatus.state)
         let invalidateForConnectivity = updateConnectivity(tunnelStatus.state)
         let invalidateForNetwork = updateNetwork(tunnelStatus.state)
@@ -72,7 +66,9 @@ final class TunnelStatusNotificationProvider: NotificationProvider, InAppNotific
         }
     }
 
-    private func updateLastTunnelError(_ lastTunnelError: BlockedStateReason?) -> Bool {
+    private func updateLastTunnelError(_ tunnelState: TunnelState) -> Bool {
+        let lastTunnelError = tunnelError(from: tunnelState)
+
         if packetTunnelError != lastTunnelError {
             packetTunnelError = lastTunnelError
 
@@ -121,6 +117,15 @@ final class TunnelStatusNotificationProvider: NotificationProvider, InAppNotific
         return false
     }
 
+    private func tunnelError(from tunnelState: TunnelState) -> BlockedStateReason? {
+        if case let .error(blockedStateReason) = tunnelState,
+           ![.accountExpired, .deviceRevoked].contains(blockedStateReason) {
+            return blockedStateReason
+        }
+
+        return nil
+    }
+
     private func notificationDescription(for packetTunnelError: BlockedStateReason) -> InAppNotificationDescriptor {
         InAppNotificationDescriptor(
             identifier: identifier,
@@ -231,7 +236,7 @@ final class TunnelStatusNotificationProvider: NotificationProvider, InAppNotific
             errorString = "No servers match your settings, try changing server or other settings."
         case .invalidAccount:
             errorString = "You are logged in with an invalid account number. Please log out and try another one."
-        case .deviceRevoked, .deviceLoggedOut:
+        case .deviceLoggedOut:
             errorString = "Unable to authenticate account. Please log out and log back in."
         default:
             errorString = "Unable to start tunnel connection. Please send a problem report."