Allow adding an extra action in the `UrlButton`

1 files changed, 7 insertions, 1 deletions

diff --git a/android/src/main/kotlin/net/mullvad/mullvadvpn/ui/widget/UrlButton.kt b/android/src/main/kotlin/net/mullvad/mullvadvpn/ui/widget/UrlButton.kt
index 3215e6f616..c2a7839748 100644
--- a/android/src/main/kotlin/net/mullvad/mullvadvpn/ui/widget/UrlButton.kt
+++ b/android/src/main/kotlin/net/mullvad/mullvadvpn/ui/widget/UrlButton.kt
@@ -45,7 +45,12 @@ class UrlButton : Button {
         super.showSpinner = true
     }
 
-    fun prepare(daemon: MullvadDaemon, jobTracker: JobTracker, jobName: String = "fetchUrl") {
+    fun prepare(
+        daemon: MullvadDaemon,
+        jobTracker: JobTracker,
+        jobName: String = "fetchUrl",
+        extraOnClickAction: (suspend () -> Unit)? = null
+    ) {
         synchronized(this) {
             super.setEnabled(shouldEnable)
 
@@ -55,6 +60,7 @@ class UrlButton : Button {
                 super.setEnabled(false)
 
                 context.startActivity(buildIntent(jobTracker))
+                extraOnClickAction?.invoke()
 
                 super.setEnabled(true)
             }