Notify controller when notification changes

1 files changed, 6 insertions, 1 deletions

diff --git a/android/src/main/kotlin/net/mullvad/mullvadvpn/ui/notification/InAppNotification.kt b/android/src/main/kotlin/net/mullvad/mullvadvpn/ui/notification/InAppNotification.kt
index 8fb0a64b7c..aa58b0bbf5 100644
--- a/android/src/main/kotlin/net/mullvad/mullvadvpn/ui/notification/InAppNotification.kt
+++ b/android/src/main/kotlin/net/mullvad/mullvadvpn/ui/notification/InAppNotification.kt
@@ -7,6 +7,8 @@ abstract class InAppNotification {
     private val changeMonitor = ChangeMonitor()
     protected val jobTracker = JobTracker()
 
+    var controller: InAppNotificationController? = null
+
     var status by changeMonitor.monitor(StatusLevel.Error)
         protected set
 
@@ -33,7 +35,10 @@ abstract class InAppNotification {
     }
 
     protected fun update() {
-        if (changeMonitor.changed) {
+        val controller = this.controller
+
+        if (controller != null && changeMonitor.changed) {
+            controller.notificationChanged(this@InAppNotification)
             changeMonitor.reset()
         }
     }