Message delete

src/hooks/use-messages.ts

@@ -6,6 +6,8 @@ interface MessageStore {
   messages: Message[];
   setMessages: (msgs: Message[]) => void;
   addMessage: (msg: Message) => void;
+  deleteMessage: (id: number) => void;
+  editMessage: (id: number, contents: string) => void;
   clearMessages: () => void;
 }
 
@@ -13,6 +15,8 @@ type ServerMessageStore = {
   messages: Record<string, Message[]>; // server_id → messages
   setMessages: (serverId: string, msgs: Message[]) => void;
   addMessage: (serverId: string, msg: Message) => void;
+  deleteMessage: (serverId: string, id: number) => void;
+  editMessage: (serverId: string, id: number, contents: string) => void;
   clearMessages: (serverId?: string) => void;
 };
 
@@ -27,6 +31,16 @@ export const useMessages = create<MessageStore>()(
             ? state
             : { messages: [...state.messages, msg] }
         ),
+      deleteMessage: (id: number) =>
+        set((state) => ({
+          messages: state.messages.filter((m) => m.id !== id),
+        })),
+      editMessage: (id: number, contents) =>
+        set((state) => ({
+          messages: state.messages.map((m) =>
+            m.id === id ? { ...m, contents } : m
+          ),
+        })),
       clearMessages: () => set({ messages: [] }),
     }),
     { name: "chat-messages" }
@@ -47,7 +61,6 @@ export const useServerMessages = create<ServerMessageStore>()(
         set((state) => {
           const serverMsgs = state.messages[serverId] || [];
 
-          // avoid duplicates
           if (serverMsgs.some((m) => m.id === msg.id)) return state;
 
           return {
@@ -58,6 +71,26 @@ export const useServerMessages = create<ServerMessageStore>()(
           };
         }),
 
+      deleteMessage: (serverId, id) =>
+        set((state) => ({
+          messages: {
+            ...state.messages,
+            [serverId]: (state.messages[serverId] || []).filter(
+              (m) => m.id !== id
+            ),
+          },
+        })),
+
+      editMessage: (serverId, id, contents) =>
+        set((state) => ({
+          messages: {
+            ...state.messages,
+            [serverId]: (state.messages[serverId] || []).map((m) =>
+              m.id === id ? { ...m, contents } : m
+            ),
+          },
+        })),
+
       clearMessages: (serverId) =>
         set((state) =>
           serverId