feat(realtime): sync message edits and deletes instantly

web/src/store/chatStore.ts

@@ -85,6 +85,7 @@ interface ChatState {
     status: DeliveryStatus,
     senderId: number
   ) => void;
+  upsertMessage: (chatId: number, message: Message) => void;
   removeMessage: (chatId: number, messageId: number) => void;
   restoreMessages: (chatId: number, messages: Message[]) => void;
   clearChatMessages: (chatId: number) => void;
@@ -299,6 +300,30 @@ export const useChatStore = create<ChatState>((set, get) => ({
       messagesByChat: { ...state.messagesByChat, [chatId]: next }
     }));
   },
+  upsertMessage: (chatId, message) => {
+    const old = get().messagesByChat[chatId] ?? [];
+    if (!old.length) {
+      set((state) => ({
+        messagesByChat: { ...state.messagesByChat, [chatId]: [message] }
+      }));
+      return;
+    }
+    const idx = old.findIndex((m) => m.id === message.id);
+    if (idx === -1) {
+      const next = [...old, message].sort((a, b) => a.id - b.id);
+      set((state) => ({
+        messagesByChat: { ...state.messagesByChat, [chatId]: next }
+      }));
+      return;
+    }
+    const next = [...old];
+    const existing = next[idx];
+    const deliveryStatus = mergeDeliveryStatus(message.delivery_status, existing.delivery_status);
+    next[idx] = deliveryStatus ? { ...message, delivery_status: deliveryStatus } : message;
+    set((state) => ({
+      messagesByChat: { ...state.messagesByChat, [chatId]: next }
+    }));
+  },
   removeMessage: (chatId, messageId) => {
     const old = get().messagesByChat[chatId] ?? [];
     set((state) => ({