Go to login screen when logging out

1 files changed, 13 insertions, 0 deletions

diff --git a/android/src/main/kotlin/net/mullvad/mullvadvpn/AccountFragment.kt b/android/src/main/kotlin/net/mullvad/mullvadvpn/AccountFragment.kt
index 3dda9e2f8b..e85248aca7 100644
--- a/android/src/main/kotlin/net/mullvad/mullvadvpn/AccountFragment.kt
+++ b/android/src/main/kotlin/net/mullvad/mullvadvpn/AccountFragment.kt
@@ -44,6 +44,8 @@ class AccountFragment : Fragment() {
             parentActivity.onBackPressed()
         }
 
+        view.findViewById<View>(R.id.logout).setOnClickListener { logout() }
+
         accountExpiryContainer = view.findViewById<View>(R.id.account_expiry_container)
         accountNumberContainer = view.findViewById<View>(R.id.account_number_container)
 
@@ -78,4 +80,15 @@ class AccountFragment : Fragment() {
 
         return formatter.format(expiryInstant)
     }
+
+    private fun logout() {
+        goToLoginScreen()
+    }
+
+    private fun goToLoginScreen() {
+        fragmentManager?.beginTransaction()?.apply {
+            replace(R.id.main_fragment, LoginFragment())
+            commit()
+        }
+    }
 }