fix

2026-05-09 09:07:27 +07:00
parent 5c31f7a0d6
commit e824aa7e16
8 changed files with 222 additions and 81 deletions
@@ -0,0 +1,75 @@
+package su.reya.coop
+
+import io.ktor.client.HttpClient
+import io.ktor.client.plugins.websocket.DefaultClientWebSocketSession
+import io.ktor.client.plugins.websocket.webSocketSession
+import io.ktor.client.request.url
+import io.ktor.websocket.Frame
+import io.ktor.websocket.close
+import io.ktor.websocket.readBytes
+import io.ktor.websocket.readText
+import kotlinx.coroutines.cancel
+import kotlinx.coroutines.channels.ClosedReceiveChannelException
+import rust.nostr.sdk.ConnectionMode
+import rust.nostr.sdk.CustomWebSocketTransport
+import rust.nostr.sdk.WebSocketAdapter
+import rust.nostr.sdk.WebSocketAdapterWrapper
+import rust.nostr.sdk.WebSocketMessage
+
+class KtorWebSocketAdapter(
+    private val client: HttpClient,
+    private val session: DefaultClientWebSocketSession
+) : WebSocketAdapter {
+
+    override suspend fun send(msg: WebSocketMessage) {
+        try {
+            when (msg) {
+                is WebSocketMessage.Text -> session.send(Frame.Text(msg.text))
+                is WebSocketMessage.Binary -> session.send(Frame.Binary(true, msg.bytes))
+                is WebSocketMessage.Ping -> session.send(Frame.Ping(msg.bytes))
+                is WebSocketMessage.Pong -> session.send(Frame.Pong(msg.bytes))
+                else -> {}
+            }
+        } catch (e: Exception) {
+            println("Attempted to send on a closed WebSocket: ${e.message}")
+            throw e
+        }
+    }
+
+    override suspend fun recv(): WebSocketMessage? {
+        return try {
+            when (val frame = session.incoming.receive()) {
+                is Frame.Text -> WebSocketMessage.Text(frame.readText())
+                is Frame.Binary -> WebSocketMessage.Binary(frame.readBytes())
+                is Frame.Ping -> WebSocketMessage.Ping(frame.readBytes())
+                is Frame.Pong -> WebSocketMessage.Pong(frame.readBytes())
+                else -> null
+            }
+        } catch (e: ClosedReceiveChannelException) {
+            null
+        } catch (e: Exception) {
+            throw e
+        }
+    }
+
+    override suspend fun closeConnection() {
+        session.cancel()
+        session.close()
+    }
+}
+
+class CoopWebSocketClient(private val httpClient: HttpClient) : CustomWebSocketTransport {
+    override fun supportPing(): Boolean = false
+
+    override suspend fun connect(url: String, mode: ConnectionMode): WebSocketAdapterWrapper {
+        try {
+            val session = httpClient.webSocketSession {
+                url(url)
+            }
+            val adapter = KtorWebSocketAdapter(httpClient, session)
+            return WebSocketAdapterWrapper(adapter)
+        } catch (e: Exception) {
+            throw e
+        }
+    }
+}