From 25e47604b3442ffabf2ce7732dd70d4be92c686a Mon Sep 17 00:00:00 2001
From: Alois <aloisianer@proton.me>
Date: Mon, 1 Jun 2026 11:13:13 +0200
Subject: [PATCH] (feat): add message context menu

---
 packages/chat/src/components/message.tsx      | 121 ++++-----
 .../src/components/messageContextMenu.tsx     | 242 ++++++++++++++++++
 packages/chat/todo.md                         |   4 +-
 3 files changed, 300 insertions(+), 67 deletions(-)
 create mode 100644 packages/chat/src/components/messageContextMenu.tsx

diff --git a/packages/chat/src/components/message.tsx b/packages/chat/src/components/message.tsx
index 18cf792..60afa75 100644
--- a/packages/chat/src/components/message.tsx
+++ b/packages/chat/src/components/message.tsx
@@ -12,14 +12,9 @@ import {
   TooltipContent,
   TooltipTrigger,
 } from "@tensamin/ui";
-import {
-  ContextMenu,
-  ContextMenuContent,
-  ContextMenuItem,
-  ContextMenuTrigger,
-} from "@tensamin/ui";
 import Wrapper from "@tensamin/user/wrapper";
 import { useChat } from "../context";
+import MessageContextMenu from "./messageContextMenu";
 
 function MessageComponent({
   grouped,
@@ -38,72 +33,66 @@ function MessageComponent({
       // pt-3 is to get a gap between messages
       className={`${grouped ? "" : "pt-3"} w-full flex justify-start transition-opacity duration-150 ${actuallyFailed || message.message_state === "awaiting" ? "opacity-50" : ""}`}
     >
-      <ContextMenu>
-        <ContextMenuTrigger
-          render={
-            <div
-              className={cn(
-                "group hover:bg-muted/50 select-text! justify-start flex gap-1 items-center w-full px-2 whitespace-pre-wrap break-all",
-                {
-                  "bg-(--destructive)/15 text-destructive-foreground":
-                    actuallyFailed,
-                },
-              )}
-            >
-              <Wrapper
-                loading={"State not achievable"}
-                userId={message.sent_by_self ? "own" : userId}
-                component={(user) => (
-                  <>
-                    {grouped ? (
-                      <p className="w-9 text-xs group-hover:visible invisible text-muted-foreground">
+      <MessageContextMenu
+        content={message.content}
+        messageId={message.send_time}
+      >
+        <div
+          className={cn(
+            "group hover:bg-muted/50 select-text! justify-start flex gap-1 items-center w-full px-2 whitespace-pre-wrap break-all",
+            {
+              "bg-(--destructive)/15 text-destructive-foreground":
+                actuallyFailed,
+            },
+          )}
+        >
+          <Wrapper
+            loading={"State not achievable"}
+            userId={message.sent_by_self ? "own" : userId}
+            component={(user) => (
+              <>
+                {grouped ? (
+                  <p className="w-9 text-xs group-hover:visible invisible text-muted-foreground">
+                    {new Date(message.send_time).toLocaleString([], {
+                      hour: "2-digit",
+                      minute: "2-digit",
+                    })}
+                  </p>
+                ) : (
+                  <Avatar className="mr-1 mb-auto mt-1 w-10">
+                    <AvatarImage src={user.avatar} />
+                    <AvatarFallback>
+                      {user.display.slice(0, 2).toUpperCase()}
+                    </AvatarFallback>
+                  </Avatar>
+                )}
+                {message.failed && message.message_state === "awaiting" && (
+                  <Tooltip>
+                    <TooltipContent>
+                      <p>Failed to send message</p>
+                    </TooltipContent>
+                    <TooltipTrigger render={<AlertTriangle size={17} />} />
+                  </Tooltip>
+                )}
+                <div className="flex flex-col">
+                  {!grouped && (
+                    <div className="flex items-center gap-1">
+                      <p className="font-medium">{user.display}</p>
+                      <p className="text-xs text-muted-foreground">
                         {new Date(message.send_time).toLocaleString([], {
                           hour: "2-digit",
                           minute: "2-digit",
                         })}
                       </p>
-                    ) : (
-                      <Avatar className="mr-1 mb-auto mt-1 w-10">
-                        <AvatarImage src={user.avatar} />
-                        <AvatarFallback>
-                          {user.display.slice(0, 2).toUpperCase()}
-                        </AvatarFallback>
-                      </Avatar>
-                    )}
-                    {message.failed && message.message_state === "awaiting" && (
-                      <Tooltip>
-                        <TooltipContent>
-                          <p>Failed to send message</p>
-                        </TooltipContent>
-                        <TooltipTrigger render={<AlertTriangle size={17} />} />
-                      </Tooltip>
-                    )}
-                    <div className="flex flex-col">
-                      {!grouped && (
-                        <div className="flex items-center gap-1">
-                          <p className="font-medium">{user.display}</p>
-                          <p className="text-xs text-muted-foreground">
-                            {new Date(message.send_time).toLocaleString([], {
-                              hour: "2-digit",
-                              minute: "2-digit",
-                            })}
-                          </p>
-                        </div>
-                      )}
-                      <Text value={message.content} />
                     </div>
-                  </>
-                )}
-              />
-            </div>
-          }
-        />
-        <ContextMenuContent>
-          <ContextMenuItem>
-            <Text value={message.content} />
-          </ContextMenuItem>
-        </ContextMenuContent>
-      </ContextMenu>
+                  )}
+                  <Text value={message.content} />
+                </div>
+              </>
+            )}
+          />
+        </div>
+      </MessageContextMenu>
     </div>
   );
 }
diff --git a/packages/chat/src/components/messageContextMenu.tsx b/packages/chat/src/components/messageContextMenu.tsx
new file mode 100644
index 0000000..dcbc707
--- /dev/null
+++ b/packages/chat/src/components/messageContextMenu.tsx
@@ -0,0 +1,242 @@
+import {
+  cn,
+  ContextMenu,
+  ContextMenuContent,
+  ContextMenuGroup,
+  ContextMenuItem,
+  ContextMenuSeparator,
+  ContextMenuSub,
+  ContextMenuSubContent,
+  ContextMenuSubTrigger,
+  ContextMenuTrigger,
+  Drawer,
+  DrawerContent,
+  DrawerTrigger,
+  useIsMobile,
+} from "@tensamin/ui";
+import { Pin, Clipboard, Pen, Reply, Forward, Trash } from "lucide-react";
+import { useMemo, useState } from "react";
+import type { ReactElement, ReactNode } from "react";
+
+async function copyText(text: string) {
+  await navigator.clipboard.writeText(text);
+}
+
+type MenuComponents = {
+  Content: (props: { className?: string; children: ReactNode }) => ReactElement;
+  Group: (props: { children: ReactNode }) => ReactElement;
+  Item: (props: {
+    children: ReactNode;
+    className?: string;
+    disabled?: boolean;
+    onClick?: () => void | Promise<void>;
+    variant?: "default" | "destructive";
+  }) => ReactElement;
+  Separator: (props: { className?: string }) => ReactElement;
+  Sub: (props: { children: ReactNode }) => ReactElement;
+  SubTrigger: (props: {
+    children: ReactNode;
+    onClick?: () => void | Promise<void>;
+  }) => ReactElement;
+  SubContent: (props: { children: ReactNode }) => ReactElement;
+};
+
+const desktopMenuComponents: MenuComponents = {
+  Content: ContextMenuContent,
+  Group: ContextMenuGroup,
+  Item: ContextMenuItem,
+  Separator: ContextMenuSeparator,
+  Sub: ContextMenuSub,
+  SubTrigger: ContextMenuSubTrigger,
+  SubContent: ContextMenuSubContent,
+};
+
+function getMobileMenuComponents(onClose: () => void): MenuComponents {
+  return {
+    Content: ({ className, children }) => (
+      <DrawerContent>
+        <div className={cn("p-3!", className)}>{children}</div>
+      </DrawerContent>
+    ),
+    Group: ({ children }) => <div>{children}</div>,
+    Item: ({ children, className, disabled, onClick, variant = "default" }) => {
+      async function handleClick() {
+        await onClick?.();
+        onClose();
+      }
+
+      return (
+        <button
+          type="button"
+          disabled={disabled}
+          data-variant={variant}
+          className={cn(
+            "relative flex w-full cursor-default items-center gap-1.5 rounded-md px-1.5 py-2 text-left text-sm outline-hidden select-none data-[variant=destructive]:text-destructive disabled:pointer-events-none disabled:opacity-50 [&_svg]:pointer-events-none [&_svg]:shrink-0 [&_svg:not([class*='size-'])]:size-4 data-[variant=destructive]:*:[svg]:text-destructive",
+            className,
+          )}
+          onClick={handleClick}
+        >
+          {children}
+        </button>
+      );
+    },
+    Separator: ({ className }) => (
+      <div className={cn("-mx-1 my-1 h-px bg-border", className)} />
+    ),
+    Sub: ({ children }) => <div>{children}</div>,
+    SubTrigger: ({ children, onClick }) => {
+      async function handleClick() {
+        await onClick?.();
+        onClose();
+      }
+
+      return (
+        <button
+          type="button"
+          className="flex w-full cursor-default items-center gap-1.5 rounded-md px-1.5 py-2 text-left text-sm outline-hidden select-none"
+          onClick={handleClick}
+        >
+          {children}
+        </button>
+      );
+    },
+    SubContent: ({ children }) => <div className="pl-3">{children}</div>,
+  };
+}
+
+function ReactionItems({ Item }: { Item: MenuComponents["Item"] }) {
+  return <Item>Cool item</Item>;
+}
+
+function MessageMenuContent({
+  components,
+  content,
+  devEnabled,
+  messageId,
+  onAddReaction,
+  showReactionItems = true,
+}: {
+  components: MenuComponents;
+  content: string;
+  devEnabled: boolean;
+  messageId: number;
+  onAddReaction?: () => void | Promise<void>;
+  showReactionItems?: boolean;
+}) {
+  const { Content, Group, Item, Separator, Sub, SubContent, SubTrigger } =
+    components;
+
+  return (
+    <Content>
+      <Group>
+        <Sub>
+          <SubTrigger onClick={onAddReaction}>Add Reaction</SubTrigger>
+          {showReactionItems && (
+            <SubContent>
+              <ReactionItems Item={Item} />
+            </SubContent>
+          )}
+        </Sub>
+        <Item disabled className="flex justify-between">
+          <p>Pin Message</p> <Pin />
+        </Item>
+        <Item
+          className="flex justify-between"
+          onClick={() => copyText(content)}
+        >
+          <p>Copy Raw</p> <Clipboard />
+        </Item>
+      </Group>
+      <Separator />
+      <Group>
+        <Item disabled className="flex justify-between">
+          <p>Edit Message</p> <Pen />
+        </Item>
+        <Item disabled className="flex justify-between">
+          <p>Reply</p> <Reply />
+        </Item>
+        <Item disabled className="flex justify-between">
+          <p>Forward</p> <Forward />
+        </Item>
+      </Group>
+      <Separator />
+      <Group>
+        <Item disabled variant="destructive" className="flex justify-between">
+          <p>Delete Message</p> <Trash />
+        </Item>
+      </Group>
+      {devEnabled && (
+        <>
+          <Separator />
+          <Group>
+            <Item
+              className="flex justify-between"
+              onClick={() => copyText(String(messageId))}
+            >
+              <p>Copy ID</p> <Clipboard />
+            </Item>
+          </Group>
+        </>
+      )}
+    </Content>
+  );
+}
+
+export default function MessageContextMenu({
+  children,
+  content,
+  messageId,
+}: {
+  children: ReactElement;
+  content: string;
+  messageId: number;
+}) {
+  const isMobile = useIsMobile();
+  const devEnabled = useMemo(
+    () => Number(localStorage.getItem("log_level")) >= 3,
+    [],
+  );
+  const [mainDrawerOpen, setMainDrawerOpen] = useState(false);
+  const [reactionDrawerOpen, setReactionDrawerOpen] = useState(false);
+  const mainDrawerComponents = getMobileMenuComponents(() =>
+    setMainDrawerOpen(false),
+  );
+  const reactionDrawerComponents = getMobileMenuComponents(() =>
+    setReactionDrawerOpen(false),
+  );
+
+  if (isMobile) {
+    return (
+      <>
+        <Drawer open={mainDrawerOpen} onOpenChange={setMainDrawerOpen}>
+          <DrawerTrigger asChild>{children}</DrawerTrigger>
+          <MessageMenuContent
+            components={mainDrawerComponents}
+            content={content}
+            devEnabled={devEnabled}
+            messageId={messageId}
+            onAddReaction={() => setReactionDrawerOpen(true)}
+            showReactionItems={false}
+          />
+        </Drawer>
+        <Drawer open={reactionDrawerOpen} onOpenChange={setReactionDrawerOpen}>
+          <DrawerContent>
+            <ReactionItems Item={reactionDrawerComponents.Item} />
+          </DrawerContent>
+        </Drawer>
+      </>
+    );
+  }
+
+  return (
+    <ContextMenu>
+      <ContextMenuTrigger render={children} />
+      <MessageMenuContent
+        components={desktopMenuComponents}
+        content={content}
+        devEnabled={devEnabled}
+        messageId={messageId}
+      />
+    </ContextMenu>
+  );
+}
diff --git a/packages/chat/todo.md b/packages/chat/todo.md
index 122f14e..4bd036a 100644
--- a/packages/chat/todo.md
+++ b/packages/chat/todo.md
@@ -1 +1,3 @@
-- Add context menu to messages
+- Implement context menu features
+- Add default-emoji-hotkey in settings
+- On mobile, turn ContextMenu into Drawer