Update UI while Wireguard key is generated

1 files changed, 12 insertions, 0 deletions

diff --git a/android/src/main/kotlin/net/mullvad/mullvadvpn/ConnectFragment.kt b/android/src/main/kotlin/net/mullvad/mullvadvpn/ConnectFragment.kt
index 4466d22eaf..4a853b2f7b 100644
--- a/android/src/main/kotlin/net/mullvad/mullvadvpn/ConnectFragment.kt
+++ b/android/src/main/kotlin/net/mullvad/mullvadvpn/ConnectFragment.kt
@@ -98,6 +98,7 @@ class ConnectFragment : Fragment() {
     }
 
     private fun connect() {
+        updateViewToPreConnecting()
         activeAction?.cancel()
 
         activeAction = GlobalScope.launch(Dispatchers.Default) {
@@ -114,6 +115,17 @@ class ConnectFragment : Fragment() {
         }
     }
 
+    private fun updateViewToPreConnecting() {
+        val connecting = TunnelStateTransition.Connecting()
+        val disconnected = TunnelStateTransition.Disconnected()
+
+        headerBar.setState(disconnected)
+
+        actionButton.state = connecting
+        notificationBanner.setState(connecting)
+        status.setState(connecting)
+    }
+
     private fun updateView(state: TunnelStateTransition) = GlobalScope.launch(Dispatchers.Main) {
         actionButton.state = state
         headerBar.setState(state)