Improved structure and added caching for server messages

src/hooks/use-messages.ts

@@ -9,6 +9,13 @@ interface MessageStore {
   clearMessages: () => void;
 }
 
+type ServerMessageStore = {
+  messages: Record<string, Message[]>; // server_id → messages
+  setMessages: (serverId: string, msgs: Message[]) => void;
+  addMessage: (serverId: string, msg: Message) => void;
+  clearMessages: (serverId?: string) => void;
+};
+
 export const useMessages = create<MessageStore>()(
   persist(
     (set) => ({
@@ -25,3 +32,44 @@ export const useMessages = create<MessageStore>()(
     { name: "chat-messages" }
   )
 );
+
+export const useServerMessages = create<ServerMessageStore>()(
+  persist(
+    (set) => ({
+      messages: {},
+
+      setMessages: (serverId, msgs) =>
+        set((state) => ({
+          messages: { ...state.messages, [serverId]: msgs },
+        })),
+
+      addMessage: (serverId, msg) =>
+        set((state) => {
+          const serverMsgs = state.messages[serverId] || [];
+
+          // avoid duplicates
+          if (serverMsgs.some((m) => m.id === msg.id)) return state;
+
+          return {
+            messages: {
+              ...state.messages,
+              [serverId]: [...serverMsgs, msg],
+            },
+          };
+        }),
+
+      clearMessages: (serverId) =>
+        set((state) =>
+          serverId
+            ? {
+                messages: {
+                  ...state.messages,
+                  [serverId]: [],
+                },
+              }
+            : { messages: {} }
+        ),
+    }),
+    { name: "server-messages" }
+  )
+);