From c3edd71bf31e0745995a09cbe4e1ad52e1c2dbb1 Mon Sep 17 00:00:00 2001
From: Alex Gleason <alex@alexgleason.me>
Date: Wed, 21 Sep 2022 17:52:20 -0500
Subject: [PATCH] Chats: WIP streaming mutations

---
 app/soapbox/actions/streaming.ts | 56 ++++++++++++++++++++++----------
 app/soapbox/normalizers/chat.ts  |  2 +-
 2 files changed, 39 insertions(+), 19 deletions(-)

diff --git a/app/soapbox/actions/streaming.ts b/app/soapbox/actions/streaming.ts
index 9c9f9ad39..da5f52dc3 100644
--- a/app/soapbox/actions/streaming.ts
+++ b/app/soapbox/actions/streaming.ts
@@ -1,5 +1,6 @@
 import { getSettings } from 'soapbox/actions/settings';
 import messages from 'soapbox/locales/messages';
+import { normalizeChat } from 'soapbox/normalizers';
 import { queryClient } from 'soapbox/queries/client';
 import { play, soundCache } from 'soapbox/utils/sounds';
 
@@ -24,8 +25,9 @@ import {
   processTimelineUpdate,
 } from './timelines';
 
+import type { InfiniteData, UseInfiniteQueryResult } from '@tanstack/react-query';
 import type { AppDispatch, RootState } from 'soapbox/store';
-import type { APIEntity } from 'soapbox/types/entities';
+import type { APIEntity, Chat, ChatMessage } from 'soapbox/types/entities';
 
 const STREAMING_CHAT_UPDATE = 'STREAMING_CHAT_UPDATE';
 const STREAMING_FOLLOW_RELATIONSHIPS_UPDATE = 'STREAMING_FOLLOW_RELATIONSHIPS_UPDATE';
@@ -47,6 +49,38 @@ const updateFollowRelationships = (relationships: APIEntity) =>
     });
   };
 
+interface ChatPayload extends Omit<Chat, 'last_message'> {
+  last_message: ChatMessage | null,
+}
+
+interface ChatPage {
+  result: Chat[],
+  hasMore: boolean,
+  link?: string,
+}
+
+const updateChat = (payload: ChatPayload) => {
+  queryClient.setQueriesData<InfiniteData<ChatPage>>(['chats', 'search'], (data) => {
+    if (data) {
+      const pages = data.pages.map(page => {
+        const result = page.result.map(chat => chat.id === payload.id ? payload as any : chat);
+        return { ...page, result };
+      });
+      return { ...data, pages };
+    }
+  });
+
+  // if (payload.last_message) {
+  //   queryClient.setQueryData<UseInfiniteQueryResult<ChatMessage[]>>(['chats', 'messages', payload.id], (data) => {
+  //     if (data) {
+  //       const pages = [...data.pages];
+  //       pages[0]?.push(payload.last_message);
+  //       return { ...result, pages };
+  //     }
+  //   });
+  // }
+};
+
 const connectTimelineStream = (
   timelineId: string,
   path: string,
@@ -91,24 +125,10 @@ const connectTimelineStream = (
         case 'filters_changed':
           dispatch(fetchFilters());
           break;
-        case 'chat_message':
-          queryClient.invalidateQueries(['chats', 'messages', JSON.parse(data.payload).chat_id]);
-          play(soundCache.chat);
-          break;
         case 'pleroma:chat_update':
-          dispatch((dispatch: AppDispatch, getState: () => RootState) => {
-            const chat = JSON.parse(data.payload);
-            const me = getState().me;
-            const messageOwned = !(chat.last_message && chat.last_message.account_id !== me);
-
-            dispatch({
-              type: STREAMING_CHAT_UPDATE,
-              chat,
-              me,
-              // Only play sounds for recipient messages
-              meta: !messageOwned && getSettings(getState()).getIn(['chats', 'sound']) && { sound: 'chat' },
-            });
-          });
+        case 'chat_message': // TruthSocial
+          updateChat(JSON.parse(data.payload));
+          play(soundCache.chat);
           break;
         case 'pleroma:follow_relationships_update':
           dispatch(updateFollowRelationships(JSON.parse(data.payload)));
diff --git a/app/soapbox/normalizers/chat.ts b/app/soapbox/normalizers/chat.ts
index 04149c682..3c0d3a6e1 100644
--- a/app/soapbox/normalizers/chat.ts
+++ b/app/soapbox/normalizers/chat.ts
@@ -8,7 +8,7 @@ export const ChatRecord = ImmutableRecord({
   id: '',
   unread: 0,
   last_message: '' as string || null,
-  updated_at: new Date(),
+  updated_at: '',
 });
 
 export const normalizeChat = (chat: Record<string, any>) => {