Add requireMainActivity extension

1 files changed, 12 insertions, 0 deletions

diff --git a/android/app/src/main/kotlin/net/mullvad/mullvadvpn/ui/extension/ContextExtensions.kt b/android/app/src/main/kotlin/net/mullvad/mullvadvpn/ui/extension/ContextExtensions.kt
index bedfff3a73..6314c3eaef 100644
--- a/android/app/src/main/kotlin/net/mullvad/mullvadvpn/ui/extension/ContextExtensions.kt
+++ b/android/app/src/main/kotlin/net/mullvad/mullvadvpn/ui/extension/ContextExtensions.kt
@@ -3,7 +3,9 @@ package net.mullvad.mullvadvpn.ui.extension
 import android.content.Context
 import android.content.Intent
 import android.net.Uri
+import androidx.fragment.app.Fragment
 import net.mullvad.mullvadvpn.R
+import net.mullvad.mullvadvpn.ui.MainActivity
 
 fun Context.openAccountPageInBrowser(authToken: String) {
     startActivity(
@@ -13,3 +15,13 @@ fun Context.openAccountPageInBrowser(authToken: String) {
         )
     )
 }
+
+fun Fragment.requireMainActivity(): MainActivity {
+    return if (this.activity is MainActivity) {
+        this.activity as MainActivity
+    } else {
+        throw IllegalStateException(
+            "Fragment $this not attached to ${MainActivity::class.simpleName}."
+        )
+    }
+}