From 0fe0fecba25b060830c524f50cb9b046273831da Mon Sep 17 00:00:00 2001
From: Vendicated <vendicated@riseup.net>
Date: Fri, 28 Apr 2023 04:23:42 +0200
Subject: MessageLogger: Nicer context menu

---
 src/plugins/messageLogger/index.tsx | 31 +++++++++++++++++--------------
 1 file changed, 17 insertions(+), 14 deletions(-)

(limited to 'src/plugins/messageLogger')

diff --git a/src/plugins/messageLogger/index.tsx b/src/plugins/messageLogger/index.tsx
index d134cd2..484429f 100644
--- a/src/plugins/messageLogger/index.tsx
+++ b/src/plugins/messageLogger/index.tsx
@@ -51,11 +51,28 @@ const patchMessageContextMenu: NavContextMenuPatchCallback = (children, props) =
 
     if (!deleted && !editHistory?.length) return;
 
+    toggle: {
+        if (!deleted) break toggle;
+
+        const domElement = document.getElementById(`chat-messages-${channel_id}-${id}`);
+        if (!domElement) break toggle;
+
+        children.push((
+            <Menu.MenuItem
+                id={TOGGLE_DELETE_STYLE_ID}
+                key={TOGGLE_DELETE_STYLE_ID}
+                label="Toggle Deleted Highlight"
+                action={() => domElement.classList.toggle("messagelogger-deleted")}
+            />
+        ));
+    }
+
     children.push((
         <Menu.MenuItem
             id={REMOVE_HISTORY_ID}
             key={REMOVE_HISTORY_ID}
             label="Remove Message History"
+            color="danger"
             action={() => {
                 if (deleted) {
                     FluxDispatcher.dispatch({
@@ -70,20 +87,6 @@ const patchMessageContextMenu: NavContextMenuPatchCallback = (children, props) =
             }}
         />
     ));
-
-    if (!deleted) return;
-
-    const domElement = document.getElementById(`chat-messages-${channel_id}-${id}`);
-    if (!domElement) return;
-
-    children.push((
-        <Menu.MenuItem
-            id={TOGGLE_DELETE_STYLE_ID}
-            key={TOGGLE_DELETE_STYLE_ID}
-            label="Toggle Deleted Highlight"
-            action={() => domElement.classList.toggle("messagelogger-deleted")}
-        />
-    ));
 };
 
 export default definePlugin({
-- 
cgit