From 749b77daad21e907e62eebb2b91be622af397950 Mon Sep 17 00:00:00 2001
From: umaranis <mail@umaranis.com>
Date: Mon, 21 Oct 2024 08:52:47 +1100
Subject: [PATCH] feat: add row and columns buttons on table hover

---
 .../Table/TableHoverActionPlugin.svelte       | 221 ++++++++++++++++++
 .../src/routes/RichTextComposer.svelte        |   2 +
 2 files changed, 223 insertions(+)
 create mode 100644 packages/svelte-lexical/src/lib/core/plugins/Table/TableHoverActionPlugin.svelte

diff --git a/packages/svelte-lexical/src/lib/core/plugins/Table/TableHoverActionPlugin.svelte b/packages/svelte-lexical/src/lib/core/plugins/Table/TableHoverActionPlugin.svelte
new file mode 100644
index 0000000..13b3e73
--- /dev/null
+++ b/packages/svelte-lexical/src/lib/core/plugins/Table/TableHoverActionPlugin.svelte
@@ -0,0 +1,221 @@
+<script lang="ts">
+  import {
+    $getTableColumnIndexFromTableCellNode as getTableColumnIndexFromTableCellNode,
+    $getTableRowIndexFromTableCellNode as getTableRowIndexFromTableCellNode,
+    $insertTableColumn__EXPERIMENTAL as insertTableColumn__EXPERIMENTAL,
+    $insertTableRow__EXPERIMENTAL as insertTableRow__EXPERIMENTAL,
+    $isTableCellNode as isTableCellNode,
+    $isTableNode as isTableNode,
+    TableCellNode,
+    TableNode,
+    TableRowNode,
+  } from '@lexical/table';
+  import {
+    $findMatchingParent as findMatchingParent,
+    mergeRegister,
+  } from '@lexical/utils';
+  import {
+    $getNearestNodeFromDOMNode as getNearestNodeFromDOMNode,
+    type NodeKey,
+  } from 'lexical';
+  import {useDebounce} from '../CodeBlock/CodeActionMenuPlugin/utils.js';
+  import {getEditor} from '$lib/core/composerContext.js';
+  import {onDestroy, onMount} from 'svelte';
+  import {writable} from 'svelte/store';
+  import {CAN_USE_DOM} from '$lib/environment/canUseDOM.js';
+
+  const BUTTON_WIDTH_PX = 20;
+
+  const editor = getEditor();
+  let isShownRow = writable(false);
+  let isShownColumn = writable(false);
+  let shouldListenMouseMove = false;
+  let position = writable('');
+  const codeSetRef: Set<NodeKey> = new Set();
+  let tableDOMNodeRef: HTMLElement | null = null;
+
+  export let anchorElem: HTMLElement;
+
+  function getMouseInfo(event: MouseEvent): {
+    tableDOMNode: HTMLElement | null;
+    isOutside: boolean;
+  } {
+    const target = event.target;
+
+    if (target && target instanceof HTMLElement) {
+      const tableDOMNode = target.closest<HTMLElement>(
+        'td.PlaygroundEditorTheme__tableCell, th.PlaygroundEditorTheme__tableCell',
+      );
+
+      const isOutside = !(
+        tableDOMNode ||
+        target.closest<HTMLElement>(
+          'button.PlaygroundEditorTheme__tableAddRows',
+        ) ||
+        target.closest<HTMLElement>(
+          'button.PlaygroundEditorTheme__tableAddColumns',
+        ) ||
+        target.closest<HTMLElement>('div.TableCellResizer__resizer')
+      );
+
+      return {isOutside, tableDOMNode};
+    } else {
+      return {isOutside: true, tableDOMNode: null};
+    }
+  }
+
+  const debouncedOnMouseMove = useDebounce(
+    (event: MouseEvent) => {
+      const {isOutside, tableDOMNode} = getMouseInfo(event);
+
+      if (isOutside) {
+        $isShownRow = false;
+        $isShownColumn = false;
+        return;
+      }
+
+      if (!tableDOMNode) {
+        return;
+      }
+
+      tableDOMNodeRef = tableDOMNode;
+
+      let hoveredRowNode: TableCellNode | null = null;
+      let hoveredColumnNode: TableCellNode | null = null;
+      let tableDOMElement: HTMLElement | null = null;
+
+      editor.update(() => {
+        const maybeTableCell = getNearestNodeFromDOMNode(tableDOMNode);
+
+        if (isTableCellNode(maybeTableCell)) {
+          const table = findMatchingParent(maybeTableCell, (node) =>
+            isTableNode(node),
+          );
+          if (!isTableNode(table)) {
+            return;
+          }
+
+          tableDOMElement = editor.getElementByKey(table?.getKey());
+
+          if (tableDOMElement) {
+            const rowCount = table.getChildrenSize();
+            const colCount = (
+              (table as TableNode).getChildAtIndex(0) as TableRowNode
+            )?.getChildrenSize();
+
+            const rowIndex = getTableRowIndexFromTableCellNode(maybeTableCell);
+            const colIndex =
+              getTableColumnIndexFromTableCellNode(maybeTableCell);
+
+            if (rowIndex === rowCount - 1) {
+              hoveredRowNode = maybeTableCell;
+            } else if (colIndex === colCount - 1) {
+              hoveredColumnNode = maybeTableCell;
+            }
+          }
+        }
+      });
+
+      if (tableDOMElement) {
+        const {
+          width: tableElemWidth,
+          y: tableElemY,
+          x: tableElemX,
+          right: tableElemRight,
+          bottom: tableElemBottom,
+          height: tableElemHeight,
+        } = (tableDOMElement as HTMLTableElement).getBoundingClientRect();
+
+        const {x: editorElemX, y: editorElemY} =
+          anchorElem.getBoundingClientRect();
+
+        if (hoveredRowNode) {
+          $isShownColumn = false;
+          $isShownRow = true;
+          $position = `height: ${BUTTON_WIDTH_PX}px; left: ${tableElemX - editorElemX}px; top: ${tableElemBottom - editorElemY + 5}px; width: ${tableElemWidth}px;`;
+        } else if (hoveredColumnNode) {
+          $isShownColumn = true;
+          $isShownRow = false;
+          $position = `height: ${tableElemHeight}px; left: ${tableElemRight - editorElemX + 5}px; top: ${tableElemY - editorElemY}px; width: ${BUTTON_WIDTH_PX}px;`;
+        }
+      }
+    },
+    50,
+    250,
+  );
+
+  $: if (CAN_USE_DOM && shouldListenMouseMove) {
+    document.addEventListener('mousemove', debouncedOnMouseMove);
+  } else if (CAN_USE_DOM) {
+    $isShownRow = false;
+    $isShownColumn = false;
+    debouncedOnMouseMove.cancel();
+    document.removeEventListener('mousemove', debouncedOnMouseMove);
+  }
+
+  onDestroy(() => {
+    $isShownRow = false;
+    $isShownColumn = false;
+    debouncedOnMouseMove.cancel();
+    if (CAN_USE_DOM) {
+      document.removeEventListener('mousemove', debouncedOnMouseMove);
+    }
+  });
+
+  onMount(() => {
+    return mergeRegister(
+      editor.registerMutationListener(
+        TableNode,
+        (mutations) => {
+          editor.getEditorState().read(() => {
+            for (const [key, type] of mutations) {
+              switch (type) {
+                case 'created':
+                  codeSetRef.add(key);
+                  shouldListenMouseMove = codeSetRef.size > 0;
+                  break;
+
+                case 'destroyed':
+                  codeSetRef.delete(key);
+                  shouldListenMouseMove = codeSetRef.size > 0;
+                  break;
+
+                default:
+                  break;
+              }
+            }
+          });
+        },
+        {skipInitialization: false},
+      ),
+    );
+  });
+  const insertAction = (insertRow: boolean) => {
+    editor.update(() => {
+      if (tableDOMNodeRef) {
+        const maybeTableNode = getNearestNodeFromDOMNode(tableDOMNodeRef);
+        maybeTableNode?.selectEnd();
+        if (insertRow) {
+          insertTableRow__EXPERIMENTAL();
+          $isShownRow = false;
+        } else {
+          insertTableColumn__EXPERIMENTAL();
+          $isShownColumn = false;
+        }
+      }
+    });
+  };
+</script>
+
+{#if $isShownRow}
+  <button
+    class={'PlaygroundEditorTheme__tableAddRows'}
+    style={$position}
+    on:click={() => insertAction(true)} />
+{/if}
+{#if $isShownColumn}
+  <button
+    class={'PlaygroundEditorTheme__tableAddColumns'}
+    style={$position}
+    on:click={() => insertAction(false)} />
+{/if}
diff --git a/packages/svelte-lexical/src/routes/RichTextComposer.svelte b/packages/svelte-lexical/src/routes/RichTextComposer.svelte
index e6ad2ba..5406269 100644
--- a/packages/svelte-lexical/src/routes/RichTextComposer.svelte
+++ b/packages/svelte-lexical/src/routes/RichTextComposer.svelte
@@ -49,6 +49,7 @@
   import {onMount} from 'svelte';
   import TablePlugin from '$lib/core/plugins/Table/TablePlugin.svelte';
   import {TableCellNode, TableNode, TableRowNode} from '@lexical/table';
+  import TableHoverActionPlugin from '$lib/core/plugins/Table/TableHoverActionPlugin.svelte';
 
   let isSmallWidthViewport = true;
   let editorDiv;
@@ -146,6 +147,7 @@
         ]} />
       <ColumnLayoutPlugin />
       <TablePlugin />
+      <TableHoverActionPlugin anchorElem={editorDiv} />
       <ActionBar />
     </div>
   </div>