Handle cancelling the connection attempt

1 files changed, 13 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 c516c99981..5737e057cc 100644
--- a/android/src/main/kotlin/net/mullvad/mullvadvpn/ConnectFragment.kt
+++ b/android/src/main/kotlin/net/mullvad/mullvadvpn/ConnectFragment.kt
@@ -28,6 +28,7 @@ class ConnectFragment : Fragment() {
 
         actionButton = ConnectActionButton(view)
         actionButton.onConnect = { connect() }
+        actionButton.onCancel = { cancel() }
 
         return view
     }
@@ -43,4 +44,16 @@ class ConnectFragment : Fragment() {
         status.setTextColor(context!!.getColor(R.color.white))
         status.setText(R.string.creating_secure_connection)
     }
+
+    private fun cancel() {
+        actionButton.state = ConnectionState.Disconnected
+
+        connectingSpinner.visibility = View.INVISIBLE
+        notificationBanner.visibility = View.GONE
+
+        headerBar.setBackgroundColor(context!!.getColor(R.color.red))
+
+        status.setTextColor(context!!.getColor(R.color.red))
+        status.setText(R.string.unsecured_connection)
+    }
 }