From 7934f209858f789357a49cd40adaf7a287e98917 Mon Sep 17 00:00:00 2001
From: Klesti Selimaj <klesti.selimay@icloud.com>
Date: Thu, 20 Aug 2026 18:21:28 +0200
Subject: [PATCH] Messages

---
 src/app/app.ts                              |  7 ++++---
 src/app/server.ts                           |  2 +-
 src/components/page/channel/TextChannel.tsx | 15 ++++++++++-----
 3 files changed, 15 insertions(+), 9 deletions(-)

diff --git a/src/app/app.ts b/src/app/app.ts
index 169196e..b79a940 100644
--- a/src/app/app.ts
+++ b/src/app/app.ts
@@ -165,17 +165,18 @@ export default class Enclave<P = Page> {
         return;
 
       case "Messages":
-        console.log("msg", msg.messages);
         Object.entries(msg.messages).forEach(([channelId, messages]) => {
           if (!server.messages[channelId]) {
-            server.messages[channelId] = new Set();
+            server.messages[channelId] = {};
           }
 
           messages.forEach((message) => {
-            server.messages[channelId].add(message);
+            server.messages[channelId][message.id] = message;
           });
         });
 
+        this.forceRender();
+
         return;
     }
   }
diff --git a/src/app/server.ts b/src/app/server.ts
index 7d23595..be0d8fe 100644
--- a/src/app/server.ts
+++ b/src/app/server.ts
@@ -29,7 +29,7 @@ export default class EnclaveServer {
   public isSecure: boolean;
   public websocket?: EnclaveWebSocket;
   public meta?: ServerMeta;
-  public messages: Record<string, Set<StoredMessage>>;
+  public messages: Record<string, Record<string, StoredMessage>>;
 
   public constructor(hostname: string, isSecure: boolean) {
     this.hostname = hostname;
diff --git a/src/components/page/channel/TextChannel.tsx b/src/components/page/channel/TextChannel.tsx
index be60580..65fc5c0 100644
--- a/src/components/page/channel/TextChannel.tsx
+++ b/src/components/page/channel/TextChannel.tsx
@@ -1,5 +1,6 @@
 import Enclave from "@/app/app";
 import { ChannelPageProps } from "@/components/page/PageView";
+import { Avatar, AvatarFallback } from "@/components/ui/avatar";
 import { Button } from "@/components/ui/button";
 import { Input } from "@/components/ui/input";
 import { StoredMessage } from "@/lib/types";
@@ -40,8 +41,10 @@ export default function TextChannel({
 
       <div className="h-full flex flex-col gap-2.5 px-3 pt-4 overflow-y-scroll">
         {appRef.current?.server?.messages[channel.id] &&
-          Array.from(appRef.current?.server?.messages[channel.id]).map(
-            (message) => <TextMessage appRef={appRef} message={message} />,
+          Object.entries(appRef.current?.server?.messages[channel.id]).map(
+            ([_, message]) => (
+              <TextMessage key={message.id} appRef={appRef} message={message} />
+            ),
           )}
       </div>
 
@@ -79,9 +82,11 @@ export function TextMessage({
 
   return (
     <div className="rounded-lg flex gap-3 px-3 py-3 hover:bg-muted/40">
-      <div className="flex h-9 w-9 shrink-0 items-center justify-center rounded-full bg-muted text-xs font-medium text-muted-foreground">
-        {message.author.slice(0, 2).toUpperCase()}
-      </div>
+      <Avatar className="h-10 w-10">
+        <AvatarFallback>
+          {message.author.slice(0, 2).toUpperCase()}
+        </AvatarFallback>
+      </Avatar>
 
       <div className="min-w-0 flex-1">
         <div className="flex items-baseline gap-2">