From 7064e0b828283a0125eb7d0140f0211fd448aacf Mon Sep 17 00:00:00 2001
From: Asai Toshiya <to.asai.60@gmail.com>
Date: Fri, 11 Oct 2024 01:04:32 +0900
Subject: [PATCH] make it possible to track relays on publish.

---
 abstract-pool.ts | 12 +++++++++++-
 pool.test.ts     | 30 ++++++++++++++++++++++++++++++
 2 files changed, 41 insertions(+), 1 deletion(-)

diff --git a/abstract-pool.ts b/abstract-pool.ts
index 0655d42..f543cee 100644
--- a/abstract-pool.ts
+++ b/abstract-pool.ts
@@ -205,7 +205,17 @@ export class AbstractSimplePool {
       }
 
       let r = await this.ensureRelay(url)
-      return r.publish(event)
+      return r.publish(event).then(reason => {
+        if (this.trackRelays) {
+          let set = this.seenOn.get(event.id)
+          if (!set) {
+            set = new Set()
+            this.seenOn.set(event.id, set)
+          }
+          set.add(r)
+        }
+        return reason
+      })
     })
   }
 
diff --git a/pool.test.ts b/pool.test.ts
index c90cac8..1c956cc 100644
--- a/pool.test.ts
+++ b/pool.test.ts
@@ -205,3 +205,33 @@ test('get()', async () => {
   expect(event).not.toBeNull()
   expect(event).toHaveProperty('id', ids[0])
 })
+
+test('track relays when publishing', async () => {
+  let event1 = finalizeEvent(
+    {
+      kind: 1,
+      created_at: Math.floor(Date.now() / 1000),
+      tags: [],
+      content: 'hello',
+    },
+    generateSecretKey(),
+  )
+  let event2 = finalizeEvent(
+    {
+      kind: 1,
+      created_at: Math.floor(Date.now() / 1000),
+      tags: [],
+      content: 'hello',
+    },
+    generateSecretKey(),
+  )
+
+  pool.trackRelays = true
+  await Promise.all(pool.publish(relayURLs, event1))
+  expect(pool.seenOn.get(event1.id)).toBeDefined()
+  expect(Array.from(pool.seenOn.get(event1.id)!).map(r => r.url)).toEqual(expect.arrayContaining(relayURLs))
+
+  pool.trackRelays = false
+  await Promise.all(pool.publish(relayURLs, event2))
+  expect(pool.seenOn.get(event2.id)).toBeUndefined()
+})