Handle after disconnect in `NotificationBanner`

1 files changed, 11 insertions, 2 deletions

diff --git a/android/src/main/kotlin/net/mullvad/mullvadvpn/NotificationBanner.kt b/android/src/main/kotlin/net/mullvad/mullvadvpn/NotificationBanner.kt
index b90b3aa251..f6e50fd94e 100644
--- a/android/src/main/kotlin/net/mullvad/mullvadvpn/NotificationBanner.kt
+++ b/android/src/main/kotlin/net/mullvad/mullvadvpn/NotificationBanner.kt
@@ -3,6 +3,7 @@ package net.mullvad.mullvadvpn
 import android.widget.TextView
 import android.view.View
 
+import net.mullvad.mullvadvpn.model.ActionAfterDisconnect
 import net.mullvad.mullvadvpn.model.KeygenEvent
 import net.mullvad.mullvadvpn.model.TunnelState
 
@@ -40,8 +41,16 @@ class NotificationBanner(val parentView: View) {
     }
 
     private fun updateBasedOnTunnelState(): Boolean {
-        when (tunnelState) {
-            is TunnelState.Disconnecting -> hide()
+        val state = tunnelState
+
+        when (state) {
+            is TunnelState.Disconnecting -> {
+                when (state.actionAfterDisconnect) {
+                    is ActionAfterDisconnect.Nothing -> hide()
+                    is ActionAfterDisconnect.Block -> show(R.string.blocking_internet)
+                    is ActionAfterDisconnect.Reconnect -> show(R.string.blocking_internet)
+                }
+            }
             is TunnelState.Disconnected -> hide()
             is TunnelState.Connecting -> show(R.string.blocking_internet)
             is TunnelState.Connected -> hide()