From 1f75a4fce4e71e7c87da56f4e00fba3b3551b1f6 Mon Sep 17 00:00:00 2001
From: Tanish Manku <tanish2k09.dev@gmail.com>
Date: Tue, 25 Feb 2025 02:43:47 -0800
Subject: [PATCH] Restyle context menu

---
 .../features/ToolbarNode/ToolbarNode.tsx      | 26 ++++++++++++-------
 .../components/ui/TextEditor/ColorPicker.tsx  |  5 +++-
 .../src/signals/uiAccess/toolbarNode.ts       |  3 ++-
 3 files changed, 22 insertions(+), 12 deletions(-)

diff --git a/frontend/realtime/src/components/features/ToolbarNode/ToolbarNode.tsx b/frontend/realtime/src/components/features/ToolbarNode/ToolbarNode.tsx
index 6080456108..09747f017a 100644
--- a/frontend/realtime/src/components/features/ToolbarNode/ToolbarNode.tsx
+++ b/frontend/realtime/src/components/features/ToolbarNode/ToolbarNode.tsx
@@ -14,6 +14,7 @@ import { paperWrapperStyles, toolTipStyles } from "~/components/styles";
 import { ToolbarNodeButtonNames } from "./enums";
 import { ToolbarNodeButtonData } from "./data";
 import { ColorPicker } from "~/components/ui/TextEditor/ColorPicker";
+import { DEFAULT_PAINT_COLOR } from "~/signals/uiEvents/toolbarMain/paintMode";
 
 export interface ToolbarNodeProps {
   disabled?: boolean;
@@ -35,7 +36,7 @@ export const ToolbarNode = ({
   lockDisabled,
   onLockClicked,
   buttonsProps,
-  color = "#000000",
+  color = DEFAULT_PAINT_COLOR,
   onColorChange = () => { },
 }: ToolbarNodeProps) => {
   const handleOnLockClicked: React.MouseEventHandler<HTMLButtonElement> = (
@@ -46,12 +47,13 @@ export const ToolbarNode = ({
     }
   };
   const lockDisabledOrUnknown = lockDisabled || locked === "unknown";
+  const lockText = locked === "unknown" ? "Unavailable" : locked ? "Unlock" : "Lock"
   return (
     <div
       className={twMerge(
         paperWrapperStyles,
         disabled && "pointer-events-none cursor-default bg-ui-border shadow-md",
-        "flex flex-col gap-2 transition",
+        "flex flex-col gap-1 transition",
       )}
     >
       <ToolbarButton
@@ -60,12 +62,10 @@ export const ToolbarNode = ({
             locked && "text-primary hover:bg-primary hover:text-white",
             lockDisabledOrUnknown &&
             "text-secondary-300 hover:text-secondary-300",
+            "w-full p-2"
           ),
           disabled: lockDisabledOrUnknown,
         }}
-        tooltip={
-          locked === "unknown" ? "Unavailable" : locked ? "Unlock" : "Lock"
-        }
         icon={
           locked === "unknown"
             ? faUnlockKeyhole
@@ -74,7 +74,9 @@ export const ToolbarNode = ({
               : faLockKeyholeOpen
         }
         onClick={handleOnLockClicked}
-      />
+      >
+        <span className="text-[16px]">{lockText}</span>
+      </ToolbarButton>
       <span className="border-r border-r-ui-border" />
       {ToolbarNodeButtonData.map((buttonDatum, idx) => {
         const buttonProps = buttonsProps?.[buttonDatum.name];
@@ -87,14 +89,18 @@ export const ToolbarNode = ({
           return (
             <ToolbarButton
               icon={buttonDatum.icon}
-              tooltip={buttonDatum.tooltip}
               key={idx}
-              buttonProps={buttonProps}
-            />
+              buttonProps={{ ...buttonProps, className: "w-full p-2" }}
+            >
+              <span className="text-[16px]">{buttonDatum.tooltip}</span>
+            </ToolbarButton>
+
           );
         } else {
           return (
-            <ColorPicker color={color} onChange={onColorChange} faIcon={faPalette} borderStyle="" key={idx} />
+            <ColorPicker color={color} onChange={onColorChange} faIcon={faPalette} borderStyle="w-full justify-start h-auto gap-2.5 text-white/80" showBar={false} key={idx}>
+              <span className="text-[16px]">{buttonDatum.tooltip}</span>
+            </ColorPicker>
           )
         }
       })}
diff --git a/frontend/realtime/src/components/ui/TextEditor/ColorPicker.tsx b/frontend/realtime/src/components/ui/TextEditor/ColorPicker.tsx
index 74c0cff399..807b06baf1 100644
--- a/frontend/realtime/src/components/ui/TextEditor/ColorPicker.tsx
+++ b/frontend/realtime/src/components/ui/TextEditor/ColorPicker.tsx
@@ -26,6 +26,7 @@ export const ColorPicker = ({
   streamChanges = false,
   defaultOpen = false,
   closeOnMouseLeave = false,
+  children,
 }: {
   color: string;
   onChange: (newColor: string) => void;
@@ -39,6 +40,7 @@ export const ColorPicker = ({
   streamChanges?: boolean;
   defaultOpen?: boolean;
   closeOnMouseLeave?: boolean;
+  children?: React.ReactNode;
 }) => {
   const [isOpen, setIsOpen] = useState(defaultOpen);
   const [{ currColor, textInput }, setStates] = useState<{
@@ -87,8 +89,8 @@ export const ColorPicker = ({
           <PopoverButton
             className={twMerge(
               "flex size-10 items-center gap-1 rounded-lg bg-ui-controls p-2",
-              borderStyle,
               showBar ? "flex-col" : "justify-center",
+              borderStyle,
             )}
             style={fillBg ? { backgroundColor: prevColor } : {}}
             onMouseEnter={() => defaultOpen && setIsOpen(true)}
@@ -97,6 +99,7 @@ export const ColorPicker = ({
               icon={faIcon}
               color={staticIconColor ?? (fillBg ? undefined : prevColor)}
             />
+            {children}
             {showBar && (
               <span
                 className="h-1 w-full"
diff --git a/frontend/realtime/src/signals/uiAccess/toolbarNode.ts b/frontend/realtime/src/signals/uiAccess/toolbarNode.ts
index 49308fcff1..9b9dcc88b1 100644
--- a/frontend/realtime/src/signals/uiAccess/toolbarNode.ts
+++ b/frontend/realtime/src/signals/uiAccess/toolbarNode.ts
@@ -1,6 +1,7 @@
 import { signal } from "@preact/signals-react";
 import { ContextualUi } from "./type";
 import { ToolbarNodeButtonNames as ButtonNames } from "~/components/features/ToolbarNode/enums";
+import { DEFAULT_PAINT_COLOR } from "../uiEvents/toolbarMain/paintMode";
 
 type ButtonStates = {
   [key in ButtonNames]: {
@@ -33,7 +34,7 @@ const toolbarNodeSignal = signal<ContextualToolbarProps>({
   locked: false,
   lockDisabled: false,
   buttonStates: initButtonStates(),
-  color: "#000000",
+  color: DEFAULT_PAINT_COLOR,
 });
 
 export const toolbarNode = {