From bd46f1230b7c813efebc64e6c3685b0e19240c82 Mon Sep 17 00:00:00 2001
From: Alex Gleason <alex@alexgleason.me>
Date: Wed, 27 Mar 2024 15:19:00 -0500
Subject: [PATCH] useNostrReq: deduplicate events with NSet

---
 src/features/nostr/hooks/useNostrReq.ts | 12 ++++++++----
 src/hooks/useForceUpdate.ts             | 11 +++++++++++
 2 files changed, 19 insertions(+), 4 deletions(-)
 create mode 100644 src/hooks/useForceUpdate.ts

diff --git a/src/features/nostr/hooks/useNostrReq.ts b/src/features/nostr/hooks/useNostrReq.ts
index 654c814f3..72e1187ac 100644
--- a/src/features/nostr/hooks/useNostrReq.ts
+++ b/src/features/nostr/hooks/useNostrReq.ts
@@ -1,14 +1,17 @@
-import { NostrEvent, NostrFilter } from '@soapbox/nspec';
+import { NSet, NostrEvent, NostrFilter } from '@soapbox/nspec';
 import isEqual from 'lodash/isEqual';
 import { useEffect, useRef, useState } from 'react';
 
 import { useNostr } from 'soapbox/contexts/nostr-context';
+import { useForceUpdate } from 'soapbox/hooks/useForceUpdate';
 
 /** Streams events from the relay for the given filters. */
 export function useNostrReq(filters: NostrFilter[]): { events: NostrEvent[]; eose: boolean; closed: boolean } {
   const { relay } = useNostr();
 
-  const [events, setEvents] = useState<NostrEvent[]>([]);
+  const nset = useRef<NSet>(new NSet());
+  const forceUpdate = useForceUpdate();
+
   const [closed, setClosed] = useState(false);
   const [eose, setEose] = useState(false);
 
@@ -21,7 +24,8 @@ export function useNostrReq(filters: NostrFilter[]): { events: NostrEvent[]; eos
       (async () => {
         for await (const msg of relay.req(value, { signal })) {
           if (msg[0] === 'EVENT') {
-            setEvents((prev) => [msg[2], ...prev]);
+            nset.current.add(msg[2]);
+            forceUpdate();
           } else if (msg[0] === 'EOSE') {
             setEose(true);
           } else if (msg[0] === 'CLOSED') {
@@ -41,7 +45,7 @@ export function useNostrReq(filters: NostrFilter[]): { events: NostrEvent[]; eos
   }, [relay, value]);
 
   return {
-    events,
+    events: [...nset.current],
     eose,
     closed,
   };
diff --git a/src/hooks/useForceUpdate.ts b/src/hooks/useForceUpdate.ts
new file mode 100644
index 000000000..50a84a468
--- /dev/null
+++ b/src/hooks/useForceUpdate.ts
@@ -0,0 +1,11 @@
+import { useState, useCallback } from 'react';
+
+export function useForceUpdate(): () => void {
+  const [, setState] = useState(false);
+
+  const forceUpdate = useCallback(() => {
+    setState(prevState => !prevState);
+  }, []);
+
+  return forceUpdate;
+}