Refactor command client

2023-08-15 18:13:45 +08:00
parent c2c3db6835
commit 8147ec71da
4 changed files with 148 additions and 111 deletions
--- a/app/src/main/java/io/nekohasekai/sfa/utils/CommandClient.kt
+++ b/app/src/main/java/io/nekohasekai/sfa/utils/CommandClient.kt
@@ -0,0 +1,120 @@
+package io.nekohasekai.sfa.utils
+
+import go.Seq
+import io.nekohasekai.libbox.CommandClient
+import io.nekohasekai.libbox.CommandClientHandler
+import io.nekohasekai.libbox.CommandClientOptions
+import io.nekohasekai.libbox.Libbox
+import io.nekohasekai.libbox.OutboundGroup
+import io.nekohasekai.libbox.OutboundGroupIterator
+import io.nekohasekai.libbox.StatusMessage
+import kotlinx.coroutines.CoroutineScope
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.delay
+import kotlinx.coroutines.isActive
+import kotlinx.coroutines.launch
+
+class CommandClient(
+    private val scope: CoroutineScope,
+    private val connectionType: ConnectionType,
+    private val handler: Handler
+) {
+
+    enum class ConnectionType {
+        Status, Groups, Log
+    }
+
+    interface Handler {
+
+        fun onConnected() {}
+        fun onDisconnected() {}
+        fun updateStatus(status: StatusMessage) {}
+        fun updateGroups(groups: List<OutboundGroup>) {}
+        fun appendLog(message: String) {}
+
+    }
+
+
+    private var commandClient: CommandClient? = null
+    private val clientHandler = ClientHandler()
+    fun connect() {
+        disconnect()
+        val options = CommandClientOptions()
+        options.command = when (connectionType) {
+            ConnectionType.Status -> Libbox.CommandStatus
+            ConnectionType.Groups -> Libbox.CommandGroup
+            ConnectionType.Log -> Libbox.CommandLog
+        }
+        options.statusInterval = 2 * 1000 * 1000 * 1000
+        val commandClient = CommandClient(clientHandler, options)
+        scope.launch(Dispatchers.IO) {
+            for (i in 1..10) {
+                delay(100 + i.toLong() * 50)
+                try {
+                    commandClient.connect()
+                } catch (ignored: Exception) {
+                    continue
+                }
+                if (!isActive) {
+                    runCatching {
+                        commandClient.disconnect()
+                    }
+                    return@launch
+                }
+                this@CommandClient.commandClient = commandClient
+                return@launch
+            }
+            runCatching {
+                commandClient.disconnect()
+            }
+        }
+    }
+
+    fun disconnect() {
+        commandClient?.apply {
+            runCatching {
+                disconnect()
+            }
+            Seq.destroyRef(refnum)
+        }
+
+    }
+
+    private inner class ClientHandler : CommandClientHandler {
+
+        override fun connected() {
+            handler.onConnected()
+        }
+
+        override fun disconnected(message: String?) {
+            handler.onDisconnected()
+        }
+
+        override fun writeGroups(message: OutboundGroupIterator?) {
+            if (message == null) {
+                return
+            }
+            val groups = mutableListOf<OutboundGroup>()
+            while (message.hasNext()) {
+                groups.add(message.next())
+            }
+            handler.updateGroups(groups)
+        }
+
+        override fun writeLog(message: String?) {
+            if (message == null) {
+                return
+            }
+            handler.appendLog(message)
+        }
+
+        override fun writeStatus(message: StatusMessage?) {
+            if (message == null) {
+                return
+            }
+            handler.updateStatus(message)
+        }
+
+    }
+
+}