Create a quick settings tile

1 files changed, 76 insertions, 0 deletions

diff --git a/android/src/main/kotlin/net/mullvad/mullvadvpn/service/MullvadTileService.kt b/android/src/main/kotlin/net/mullvad/mullvadvpn/service/MullvadTileService.kt
new file mode 100644
index 0000000000..01a14db174
--- /dev/null
+++ b/android/src/main/kotlin/net/mullvad/mullvadvpn/service/MullvadTileService.kt
@@ -0,0 +1,76 @@
+package net.mullvad.mullvadvpn.service
+
+import android.content.Intent
+import android.service.quicksettings.Tile
+import android.service.quicksettings.TileService
+import net.mullvad.mullvadvpn.model.TunnelState
+import net.mullvad.mullvadvpn.service.tunnelstate.TunnelStateListener
+import net.mullvad.talpid.tunnel.ActionAfterDisconnect
+
+class MullvadTileService : TileService() {
+    private var secured = false
+        set(value) {
+            if (field != value) {
+                field = value
+                updateTileState()
+            }
+        }
+
+    private lateinit var listener: TunnelStateListener
+
+    override fun onCreate() {
+        super.onCreate()
+
+        listener = TunnelStateListener(this)
+    }
+
+    override fun onStartListening() {
+        super.onStartListening()
+
+        listener.onStateChange = { state ->
+            secured = when (state) {
+                is TunnelState.Disconnected -> false
+                is TunnelState.Connecting -> true
+                is TunnelState.Connected -> true
+                is TunnelState.Disconnecting -> {
+                    state.actionAfterDisconnect == ActionAfterDisconnect.Reconnect
+                }
+                is TunnelState.Error -> {
+                    state.errorState.isBlocking
+                }
+            }
+        }
+    }
+
+    override fun onClick() {
+        super.onClick()
+
+        val tunnelActionKey = if (secured) {
+            KEY_DISCONNECT_ACTION
+        } else {
+            KEY_CONNECT_ACTION
+        }
+
+        val intent = Intent(tunnelActionKey).setPackage("net.mullvad.mullvadvpn")
+
+        sendBroadcast(intent)
+    }
+
+    override fun onStopListening() {
+        super.onStartListening()
+
+        listener.onStateChange = null
+    }
+
+    private fun updateTileState() {
+        qsTile.apply {
+            if (secured) {
+                state = Tile.STATE_ACTIVE
+            } else {
+                state = Tile.STATE_INACTIVE
+            }
+
+            updateTile()
+        }
+    }
+}