From 2b3ca6ed7d3076d4af1c906786018dcd55242825 Mon Sep 17 00:00:00 2001
From: SimonLaux <mobile.info@simonlaux.de>
Date: Thu, 28 Sep 2023 22:54:46 +0200
Subject: [PATCH] update previous/next on incoming message

---
 .../components/dialogs/FullscreenMedia.tsx    | 43 ++++++++++++-------
 1 file changed, 27 insertions(+), 16 deletions(-)

diff --git a/src/renderer/components/dialogs/FullscreenMedia.tsx b/src/renderer/components/dialogs/FullscreenMedia.tsx
index ac88b744f0..41c70b7727 100644
--- a/src/renderer/components/dialogs/FullscreenMedia.tsx
+++ b/src/renderer/components/dialogs/FullscreenMedia.tsx
@@ -11,8 +11,9 @@ import { preventDefault } from '../../../shared/util'
 import { TransformWrapper, TransformComponent } from 'react-zoom-pan-pinch'
 import { useContextMenu } from '../ContextMenu'
 import { jumpToMessage } from '../helpers/ChatMethods'
-import { BackendRemote, Type } from '../../backend-com'
+import { BackendRemote, onDCEvent, Type } from '../../backend-com'
 import { selectedAccountId } from '../../ScreenController'
+import debounce from 'debounce'
 
 const log = getLogger('renderer/fullscreen_media')
 
@@ -46,23 +47,33 @@ export default function FullscreenMedia(props: {
 
   const [mediaMessageList, setMediaMessageList] = useState<number[]>([])
   useEffect(() => {
-    if (props.neighboringMedia === NeighboringMediaMode.Off) {
-      setMediaMessageList([])
-    } else {
-      const { viewType, chatId } = props.msg
-      // workaround to get gifs and images into the same media list
-      let additionalViewType: Type.Viewtype | null = null
-      if (props.msg.viewType === 'Image') {
-        additionalViewType = 'Gif'
-      } else if (props.msg.viewType === 'Gif') {
-        additionalViewType = 'Image'
+    const update = () => {
+      if (props.neighboringMedia === NeighboringMediaMode.Off) {
+        setMediaMessageList([])
+      } else {
+        const { viewType, chatId } = props.msg
+        // workaround to get gifs and images into the same media list
+        let additionalViewType: Type.Viewtype | null = null
+        if (props.msg.viewType === 'Image') {
+          additionalViewType = 'Gif'
+        } else if (props.msg.viewType === 'Gif') {
+          additionalViewType = 'Image'
+        }
+        const scope =
+          props.neighboringMedia === NeighboringMediaMode.Global ? null : chatId
+        BackendRemote.rpc
+          .getChatMedia(accountId, scope, viewType, additionalViewType, null)
+          .then(setMediaMessageList)
       }
-      const scope =
-        props.neighboringMedia === NeighboringMediaMode.Global ? null : chatId
-      BackendRemote.rpc
-        .getChatMedia(accountId, scope, viewType, additionalViewType, null)
-        .then(setMediaMessageList)
     }
+    update()
+    const debouncedUpdate = debounce(update, 400)
+    const listeners = [
+      onDCEvent(accountId, 'MsgsChanged', debouncedUpdate),
+      onDCEvent(accountId, 'IncomingMsgBunch', debouncedUpdate),
+      onDCEvent(accountId, 'MsgDeleted', debouncedUpdate),
+    ]
+    return () => listeners.every(cleanup => cleanup())
   }, [props.msg, props.neighboringMedia, accountId])
 
   const { file, fileMime } = msg