From 277e22ad0dc1c867925724a1f2957f97e662be5c Mon Sep 17 00:00:00 2001
From: Oliver Walters <oliver.henry.walters@gmail.com>
Date: Tue, 26 May 2026 10:47:26 +0000
Subject: [PATCH] Adjust header actions

---
 .../src/components/editors/NotesEditor.tsx    | 21 ++++++++-----------
 1 file changed, 9 insertions(+), 12 deletions(-)

diff --git a/src/frontend/src/components/editors/NotesEditor.tsx b/src/frontend/src/components/editors/NotesEditor.tsx
index 4d43d68fa1..ce19c1ae97 100644
--- a/src/frontend/src/components/editors/NotesEditor.tsx
+++ b/src/frontend/src/components/editors/NotesEditor.tsx
@@ -504,13 +504,8 @@ export default function NotesEditor({
                         >
                           <IconTablePlus size='0.9rem' />
                         </RichTextEditor.Control>
-                      </RichTextEditor.ControlsGroup>
-                    </RichTextEditor.Toolbar>
-                  )}
-                  {canEdit && isInTable && (
-                    <RichTextEditor.Toolbar>
-                      <RichTextEditor.ControlsGroup>
                         <RichTextEditor.Control
+                          disabled={!isInTable}
                           onClick={() =>
                             editor?.chain().focus().addColumnBefore().run()
                           }
@@ -520,6 +515,7 @@ export default function NotesEditor({
                           <IconColumnInsertLeft size='0.9rem' />
                         </RichTextEditor.Control>
                         <RichTextEditor.Control
+                          disabled={!isInTable}
                           onClick={() =>
                             editor?.chain().focus().addColumnAfter().run()
                           }
@@ -529,6 +525,7 @@ export default function NotesEditor({
                           <IconColumnInsertRight size='0.9rem' />
                         </RichTextEditor.Control>
                         <RichTextEditor.Control
+                          disabled={!isInTable}
                           onClick={() =>
                             editor?.chain().focus().deleteColumn().run()
                           }
@@ -537,9 +534,8 @@ export default function NotesEditor({
                         >
                           <IconColumnRemove size='0.9rem' />
                         </RichTextEditor.Control>
-                      </RichTextEditor.ControlsGroup>
-                      <RichTextEditor.ControlsGroup>
                         <RichTextEditor.Control
+                          disabled={!isInTable}
                           onClick={() =>
                             editor?.chain().focus().addRowBefore().run()
                           }
@@ -549,6 +545,7 @@ export default function NotesEditor({
                           <IconRowInsertTop size='0.9rem' />
                         </RichTextEditor.Control>
                         <RichTextEditor.Control
+                          disabled={!isInTable}
                           onClick={() =>
                             editor?.chain().focus().addRowAfter().run()
                           }
@@ -558,6 +555,7 @@ export default function NotesEditor({
                           <IconRowInsertBottom size='0.9rem' />
                         </RichTextEditor.Control>
                         <RichTextEditor.Control
+                          disabled={!isInTable}
                           onClick={() =>
                             editor?.chain().focus().deleteRow().run()
                           }
@@ -566,9 +564,8 @@ export default function NotesEditor({
                         >
                           <IconRowRemove size='0.9rem' />
                         </RichTextEditor.Control>
-                      </RichTextEditor.ControlsGroup>
-                      <RichTextEditor.ControlsGroup>
                         <RichTextEditor.Control
+                          disabled={!isInTable}
                           onClick={() =>
                             editor?.chain().focus().mergeOrSplit().run()
                           }
@@ -578,6 +575,7 @@ export default function NotesEditor({
                           <IconArrowMerge size='0.9rem' />
                         </RichTextEditor.Control>
                         <RichTextEditor.Control
+                          disabled={!isInTable}
                           onClick={() =>
                             editor?.chain().focus().toggleHeaderRow().run()
                           }
@@ -586,9 +584,8 @@ export default function NotesEditor({
                         >
                           <IconTableRow size='0.9rem' />
                         </RichTextEditor.Control>
-                      </RichTextEditor.ControlsGroup>
-                      <RichTextEditor.ControlsGroup>
                         <RichTextEditor.Control
+                          disabled={!isInTable}
                           onClick={() =>
                             editor?.chain().focus().deleteTable().run()
                           }