From 59b3ff1b940a30683b902b5773027e2598b9667c Mon Sep 17 00:00:00 2001
From: greenart7c3 <greenart7c3@proton.me>
Date: Fri, 19 Jul 2024 07:58:12 -0300
Subject: [PATCH] Added sendAndWaitForResponse to Client.kt

---
 .../vitorpamplona/ammolite/relays/Client.kt   | 64 +++++++++++++++++++
 1 file changed, 64 insertions(+)

diff --git a/ammolite/src/main/java/com/vitorpamplona/ammolite/relays/Client.kt b/ammolite/src/main/java/com/vitorpamplona/ammolite/relays/Client.kt
index 12bc0a62f..f9d5bb1c3 100644
--- a/ammolite/src/main/java/com/vitorpamplona/ammolite/relays/Client.kt
+++ b/ammolite/src/main/java/com/vitorpamplona/ammolite/relays/Client.kt
@@ -29,6 +29,8 @@ import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.GlobalScope
 import kotlinx.coroutines.launch
 import java.util.UUID
+import java.util.concurrent.CountDownLatch
+import java.util.concurrent.TimeUnit
 
 /**
  * The Nostr Client manages multiple personae the user may switch between. Events are received and
@@ -130,6 +132,68 @@ object Client : RelayPool.Listener {
         RelayPool.sendFilter(subscriptionId, filters)
     }
 
+    fun sendAndWaitForResponse(
+        signedEvent: EventInterface,
+        relay: String? = null,
+        feedTypes: Set<FeedType>? = null,
+        relayList: List<RelaySetupInfo>? = null,
+        onDone: (() -> Unit)? = null,
+        timeoutInSeconds: Long = 30,
+    ) {
+        checkNotInMainThread()
+
+        val size = if (relay != null) 1 else relayList?.size ?: RelayPool.availableRelays()
+        val latch = CountDownLatch(size)
+
+        Log.d("Relay", "Waiting for $size responses")
+
+        subscribe(
+            object : Listener() {
+                override fun onEvent(
+                    event: Event,
+                    subscriptionId: String,
+                    relay: Relay,
+                    afterEOSE: Boolean,
+                ) {
+                    if (event.id() == signedEvent.id()) {
+                        unsubscribe(this)
+                        latch.countDown()
+                        Log.d("Relay", "Received response for ${event.id()} relay ${relay.url} count: ${latch.count}")
+                    }
+                }
+
+                override fun onError(error: Error, subscriptionId: String, relay: Relay) {
+                    unsubscribe(this)
+                    latch.countDown()
+                    Log.d("Relay", "Error from relay ${relay.url} count: ${latch.count} error: $error")
+                }
+
+                override fun onRelayStateChange(
+                    type: Relay.StateType,
+                    relay: Relay,
+                    subscriptionId: String?,
+                ) {
+                    if (type == Relay.StateType.DISCONNECT) {
+                        unsubscribe(this)
+                        latch.countDown()
+                        Log.d("Relay", "Disconnected from relay ${relay.url} count: ${latch.count}")
+                    }
+                }
+
+                override fun onSendResponse(eventId: String, success: Boolean, message: String, relay: Relay) {
+                    if (eventId == signedEvent.id()) {
+                        unsubscribe(this)
+                        latch.countDown()
+                        Log.d("Relay", "Received response for $eventId from relay ${relay.url} count: ${latch.count}")
+                    }
+                }
+            },
+        )
+
+        send(signedEvent, relay, feedTypes, relayList, onDone)
+        latch.await(timeoutInSeconds, TimeUnit.SECONDS)
+    }
+
     fun sendFilterOnlyIfDisconnected(
         subscriptionId: String = UUID.randomUUID().toString().substring(0..10),
         filters: List<TypedFilter> = listOf(),