fix: replies

src/providers/ReplyProvider.tsx

@@ -0,0 +1,70 @@
+import { getParentEventTag, getRootEventTag } from '@/lib/event'
+import { Event } from 'nostr-tools'
+import { createContext, useCallback, useContext, useState } from 'react'
+
+type TReplyContext = {
+  repliesMap: Map<string, { events: Event[]; eventIdSet: Set<string> }>
+  addReplies: (replies: Event[]) => void
+}
+
+const ReplyContext = createContext<TReplyContext | undefined>(undefined)
+
+export const useReply = () => {
+  const context = useContext(ReplyContext)
+  if (!context) {
+    throw new Error('useReply must be used within a ReplyProvider')
+  }
+  return context
+}
+
+export function ReplyProvider({ children }: { children: React.ReactNode }) {
+  const [repliesMap, setRepliesMap] = useState<
+    Map<string, { events: Event[]; eventIdSet: Set<string> }>
+  >(new Map())
+
+  const addReplies = useCallback((replies: Event[]) => {
+    const newReplyIdSet = new Set<string>()
+    const newReplyEventMap = new Map<string, Event[]>()
+    replies.forEach((reply) => {
+      if (newReplyIdSet.has(reply.id)) return
+      newReplyIdSet.add(reply.id)
+
+      const rootETag = getRootEventTag(reply)
+      if (!rootETag) return
+      const rootId = rootETag[1]
+      newReplyEventMap.set(rootId, [...(newReplyEventMap.get(rootId) || []), reply])
+
+      const parentETag = getParentEventTag(reply)
+      if (!parentETag) return
+      const parentId = parentETag[1]
+      newReplyEventMap.set(parentId, [...(newReplyEventMap.get(parentId) || []), reply])
+    })
+    if (newReplyEventMap.size === 0) return
+
+    setRepliesMap((prev) => {
+      for (const [id, newReplyEvents] of newReplyEventMap.entries()) {
+        const replies = prev.get(id) || { events: [], eventIdSet: new Set() }
+        newReplyEvents.forEach((reply) => {
+          if (!replies.eventIdSet.has(reply.id)) {
+            replies.events.push(reply)
+            replies.eventIdSet.add(reply.id)
+          }
+        })
+        replies.events.sort((a, b) => a.created_at - b.created_at)
+        prev.set(id, replies)
+      }
+      return new Map(prev)
+    })
+  }, [])
+
+  return (
+    <ReplyContext.Provider
+      value={{
+        repliesMap,
+        addReplies
+      }}
+    >
+      {children}
+    </ReplyContext.Provider>
+  )
+}