From b0091ccfe165890e3a50735d31d28a20ec55bd7f Mon Sep 17 00:00:00 2001
From: Bojan Mojsilovic <bmojsilovic@itekako.com>
Date: Tue, 23 Apr 2024 13:48:48 +0200
Subject: [PATCH] Instant refresh top zaps

---
 src/components/CustomZap/CustomZap.tsx        |  4 +-
 src/components/Note/Note.tsx                  | 46 +++++++++++++++----
 src/components/Note/NoteFooter/NoteFooter.tsx | 12 ++---
 src/contexts/ThreadContext.tsx                | 18 ++++++++
 4 files changed, 61 insertions(+), 19 deletions(-)

diff --git a/src/components/CustomZap/CustomZap.tsx b/src/components/CustomZap/CustomZap.tsx
index 34e873c..0e89d2d 100644
--- a/src/components/CustomZap/CustomZap.tsx
+++ b/src/components/CustomZap/CustomZap.tsx
@@ -130,9 +130,7 @@ const CustomZap: Component<{
 
   const handleZap = (success = false) => {
     if (success) {
-      setTimeout(() => {
-        props.onSuccess(selectedValue());
-      }, 2_000)
+      props.onSuccess(selectedValue());
       return;
     }
 
diff --git a/src/components/Note/Note.tsx b/src/components/Note/Note.tsx
index a30b227..1eb0f64 100644
--- a/src/components/Note/Note.tsx
+++ b/src/components/Note/Note.tsx
@@ -18,7 +18,9 @@ import { CustomZapInfo, useAppContext } from '../../contexts/AppContext';
 import NoteContextTrigger from './NoteContextTrigger';
 import { date, longDate, veryLongDate } from '../../lib/dates';
 import { hexToNpub } from '../../lib/keys';
-import { zapCustomOption } from '../../translations';
+import { thread, zapCustomOption } from '../../translations';
+import { useAccountContext } from '../../contexts/AccountContext';
+import { uuidv4 } from '../../utils';
 
 export type NoteFooterState = {
   likes: number,
@@ -37,6 +39,7 @@ export type NoteFooterState = {
   hideZapIcon: boolean,
   moreZapsAvailable: boolean,
   isRepostMenuVisible: boolean,
+  topZaps: TopZap[],
 };
 
 const Note: Component<{
@@ -49,6 +52,7 @@ const Note: Component<{
 
   const threadContext = useThreadContext();
   const app = useAppContext();
+  const account = useAccountContext();
   const intl = useIntl();
 
   const noteType = () => props.noteType || 'feed';
@@ -76,6 +80,7 @@ const Note: Component<{
     hideZapIcon: false,
     moreZapsAvailable: false,
     isRepostMenuVisible: false,
+    topZaps: [],
   });
 
   let noteContextMenu: HTMLDivElement | undefined;
@@ -94,6 +99,27 @@ const Note: Component<{
   const onSuccessZap = (zapOption: ZapOption) => {
     app?.actions.closeCustomZapModal();
     app?.actions.resetCustomZap();
+
+    const pubkey = account?.publicKey;
+
+    if (!pubkey) return;
+
+    const oldZaps = [ ...reactionsState.topZaps ];
+
+    const newZap = {
+      amount: zapOption.amount || 0,
+      message: zapOption.message || '',
+      pubkey,
+      eventId: props.note.post.id,
+      id: uuidv4() as string,
+    };
+
+    if (!threadContext?.users.find((u) => u.pubkey === pubkey)) {
+      threadContext?.actions.fetchUsers([pubkey])
+    }
+
+    const zaps = [ ...oldZaps, { ...newZap }].sort((a, b) => b.amount - a.amount);
+
     batch(() => {
       updateReactionsState('zapCount', (z) => z + 1);
       // updateFooterState('satsZapped', (z) => z + (zapOption.amount || 0));
@@ -102,8 +128,8 @@ const Note: Component<{
       updateReactionsState('showZapAnim', () => false);
       updateReactionsState('hideZapIcon', () => false);
       updateReactionsState('zapped', () => true);
+      updateReactionsState('topZaps', () => [...zaps]);
     });
-    threadContext?.actions.fetchTopZaps(props.note.post.id);
   };
 
   const onFailZap = (zapOption: ZapOption) => {
@@ -169,12 +195,10 @@ const Note: Component<{
     return (likes || 0) + (zapCount || 0) + (reposts || 0);
   };
 
-  const topZaps = () => threadContext?.topZaps[props.note.post.id] || [];
-
-  const firstZap = createMemo(() => topZaps()[0]);
+  const firstZap = createMemo(() => reactionsState.topZaps[0]);
 
   const restZaps = createMemo(() => {
-    const zaps = topZaps().slice(1);
+    const zaps = reactionsState.topZaps.slice(1);
 
     let limit = 0;
     let digits = 0;
@@ -200,11 +224,15 @@ const Note: Component<{
     }
 
     return rest;
-  })
+  });
 
   const zapSender = (zap: TopZap) => {
     return threadContext?.users.find(u => u.pubkey === zap.pubkey);
-  }
+  };
+
+  createEffect(() => {
+    updateReactionsState('topZaps', () =>  [ ...(threadContext?.topZaps[props.note.post.id] || []) ]);
+  });
 
   return (
     <Switch>
@@ -260,7 +288,7 @@ const Note: Component<{
               <ParsedNote note={props.note} width={Math.min(574, window.innerWidth)} />
             </div>
 
-            <div class={`${styles.zapHighlights} ${topZaps().length < 4 ? styles.onlyFew : ''}`}>
+            <div class={`${styles.zapHighlights} ${reactionsState.topZaps.length < 4 ? styles.onlyFew : ''}`}>
               <Show when={firstZap()}>
                 <button
                   class={styles.firstZap}
diff --git a/src/components/Note/NoteFooter/NoteFooter.tsx b/src/components/Note/NoteFooter/NoteFooter.tsx
index f7a2f2b..b697ef6 100644
--- a/src/components/Note/NoteFooter/NoteFooter.tsx
+++ b/src/components/Note/NoteFooter/NoteFooter.tsx
@@ -279,13 +279,11 @@ const NoteFooter: Component<{
       props.updateState('isZapping', () => false);
 
       if (success) {
-        setTimeout(() => {
-          props.customZapInfo.onSuccess({
-            emoji,
-            amount,
-            message,
-          });
-        }, 2_000);
+        props.customZapInfo.onSuccess({
+          emoji,
+          amount,
+          message,
+        });
 
         return;
       }
diff --git a/src/contexts/ThreadContext.tsx b/src/contexts/ThreadContext.tsx
index 9ac0d53..f266b55 100644
--- a/src/contexts/ThreadContext.tsx
+++ b/src/contexts/ThreadContext.tsx
@@ -40,6 +40,7 @@ import { APP_ID } from "../App";
 import { useAccountContext } from "./AccountContext";
 import { getEventZaps, setLinkPreviews } from "../lib/notes";
 import { parseBolt11 } from "../utils";
+import { getUserProfiles } from "../lib/profile";
 
 export type TopZap = {
   id: string,
@@ -69,6 +70,7 @@ export type ThreadContextStore = {
     savePage: (page: FeedPage) => void,
     setPrimaryNote: (context: PrimalNote | undefined) => void,
     fetchTopZaps: (noteId: string) => void,
+    fetchUsers: (pubkeys: string[]) => void,
   }
 }
 
@@ -292,6 +294,10 @@ export const ThreadProvider = (props: { children: ContextChildren }) => {
     getEventZaps(noteId, account?.publicKey, `thread_zapps_${APP_ID}`, 10, 0);
   };
 
+  const fetchUsers = (pubkeys: string[]) => {
+    getUserProfiles(pubkeys, `thread_pk_${APP_ID}`);
+  };
+
 // SOCKET HANDLERS ------------------------------
 
   const onMessage = (event: MessageEvent) => {
@@ -351,6 +357,17 @@ export const ThreadProvider = (props: { children: ContextChildren }) => {
         return;
       }
     }
+
+    if (subId === `thread_pk_${APP_ID}`) {
+      if (type === 'EOSE') {
+        savePage(store.page);
+      }
+
+      if (type === 'EVENT') {
+        updatePage(content);
+        return;
+      }
+    }
   };
 
   const onSocketClose = (closeEvent: CloseEvent) => {
@@ -420,6 +437,7 @@ export const ThreadProvider = (props: { children: ContextChildren }) => {
       savePage,
       setPrimaryNote,
       fetchTopZaps,
+      fetchUsers,
     },
   });