Create `ConnectivityListener` helper class

1 files changed, 59 insertions, 0 deletions

diff --git a/android/src/main/kotlin/net/mullvad/mullvadvpn/ConnectivityListener.kt b/android/src/main/kotlin/net/mullvad/mullvadvpn/ConnectivityListener.kt
new file mode 100644
index 0000000000..cf212762aa
--- /dev/null
+++ b/android/src/main/kotlin/net/mullvad/mullvadvpn/ConnectivityListener.kt
@@ -0,0 +1,59 @@
+package net.mullvad.mullvadvpn
+
+import kotlinx.coroutines.CompletableDeferred
+
+import android.content.BroadcastReceiver
+import android.content.Context
+import android.content.Intent
+import android.content.IntentFilter
+import android.net.ConnectivityManager
+import android.net.NetworkInfo
+import android.net.NetworkInfo.DetailedState
+
+class ConnectivityListener : BroadcastReceiver() {
+    var vpnDisconnected = CompletableDeferred<Unit>()
+        private set
+
+    fun register(context: Context) {
+        val intentFilter = IntentFilter()
+
+        intentFilter.addAction(ConnectivityManager.CONNECTIVITY_ACTION)
+        context.registerReceiver(this, intentFilter)
+
+        checkInitialState(context)
+    }
+
+    fun unregister(context: Context) {
+        context.unregisterReceiver(this)
+    }
+
+    override fun onReceive(context: Context, intent: Intent) {
+        val networkInfo =
+            intent.getParcelableExtra<NetworkInfo>(ConnectivityManager.EXTRA_NETWORK_INFO)
+
+        if (networkInfo.type == ConnectivityManager.TYPE_VPN) {
+            if (networkInfo.detailedState == DetailedState.DISCONNECTED) {
+                vpnDisconnected.complete(Unit)
+            } else if (networkInfo.detailedState == DetailedState.CONNECTED) {
+                vpnDisconnected.cancel()
+                vpnDisconnected = CompletableDeferred<Unit>()
+            }
+        }
+    }
+
+    private fun checkInitialState(context: Context) {
+        val connectivityManager =
+            context.getSystemService(Context.CONNECTIVITY_SERVICE) as ConnectivityManager
+
+        val isVpnConnected = connectivityManager.allNetworks
+            .map({ network -> connectivityManager.getNetworkInfo(network) })
+            .any({ networkInfo ->
+                networkInfo.type == ConnectivityManager.TYPE_VPN
+                    && networkInfo.detailedState == DetailedState.CONNECTED
+            })
+
+        if (!isVpnConnected) {
+            vpnDisconnected.complete(Unit)
+        }
+    }
+}