1 files changed, 33 insertions, 0 deletions
diff --git a/android/src/main/kotlin/net/mullvad/mullvadvpn/ui/serviceconnection/KeyStatusListener.kt b/android/src/main/kotlin/net/mullvad/mullvadvpn/ui/serviceconnection/KeyStatusListener.kt
new file mode 100644
index 0000000000..7f7425e9c7
--- /dev/null
+++ b/android/src/main/kotlin/net/mullvad/mullvadvpn/ui/serviceconnection/KeyStatusListener.kt
@@ -0,0 +1,33 @@
+package net.mullvad.mullvadvpn.ui.serviceconnection
+
+import android.os.Messenger
+import net.mullvad.mullvadvpn.ipc.DispatchingHandler
+import net.mullvad.mullvadvpn.ipc.Event
+import net.mullvad.mullvadvpn.ipc.Request
+import net.mullvad.mullvadvpn.model.KeygenEvent
+import net.mullvad.talpid.util.EventNotifier
+
+class KeyStatusListener(val connection: Messenger, val eventDispatcher: DispatchingHandler<Event>) {
+    val onKeyStatusChange = EventNotifier<KeygenEvent?>(null)
+
+    var keyStatus by onKeyStatusChange.notifiable()
+        private set
+
+    init {
+        eventDispatcher.registerHandler(Event.WireGuardKeyStatus::class) { event ->
+            keyStatus = event.keyStatus
+        }
+    }
+
+    fun generateKey() {
+        connection.send(Request.WireGuardGenerateKey.message)
+    }
+
+    fun verifyKey() {
+        connection.send(Request.WireGuardVerifyKey.message)
+    }
+
+    fun onDestroy() {
+        onKeyStatusChange.unsubscribeAll()
+    }
+}