From 667f91e9b77e57fe8fbae15baaab2498f361218a Mon Sep 17 00:00:00 2001
From: kirandash <thekirankumardash@gmail.com>
Date: Tue, 18 Feb 2025 02:55:39 +0800
Subject: [PATCH] fix(table): simplify nested table checks per review

- Simplify nested table checks using  helper

Per review feedback, simplified the code checks for nested tables.

Co-authored-by: Bob Ippolito <bob@redivi.com>
---
 .../src/LexicalTablePluginHelpers.ts          | 41 ++++---------------
 1 file changed, 7 insertions(+), 34 deletions(-)

diff --git a/packages/lexical-table/src/LexicalTablePluginHelpers.ts b/packages/lexical-table/src/LexicalTablePluginHelpers.ts
index 84eb2bcdd1e..4d5a1a64cdb 100644
--- a/packages/lexical-table/src/LexicalTablePluginHelpers.ts
+++ b/packages/lexical-table/src/LexicalTablePluginHelpers.ts
@@ -37,6 +37,7 @@ import {$isTableNode, TableNode} from './LexicalTableNode';
 import {$getTableAndElementByKey, TableObserver} from './LexicalTableObserver';
 import {$isTableRowNode, TableRowNode} from './LexicalTableRowNode';
 import {
+  $findTableNode,
   applyTableHandlers,
   getTableElement,
   HTMLTableElementWithWithTableSelectionState,
@@ -58,18 +59,9 @@ function $insertTableCommandListener({
     return false;
   }
 
-  // Check if we're inside a table cell
-  let node = selection.anchor.getNode();
-  while (node !== null) {
-    if ($isTableCellNode(node)) {
-      // We're inside a table cell, don't allow nested tables
-      return false;
-    }
-    const parent = node.getParent();
-    if (parent === null) {
-      break;
-    }
-    node = parent;
+  // Prevent nested tables by checking if we're already inside a table
+  if ($findTableNode(selection.anchor.getNode())) {
+    return false;
   }
 
   const tableNode = $createTableNodeWithDimensions(
@@ -296,28 +288,9 @@ export function registerTablePlugin(editor: LexicalEditor): () => void {
         if (!$isRangeSelection(selection)) {
           return false;
         }
-
-        // Check if we're inside a table cell
-        let isInsideTableCell = false;
-        let currentNode = selection.anchor.getNode();
-        while (currentNode !== null) {
-          if ($isTableCellNode(currentNode)) {
-            isInsideTableCell = true;
-            break;
-          }
-          const parent = currentNode.getParent();
-          if (parent === null) {
-            break;
-          }
-          currentNode = parent;
-        }
-
-        // If inside table cell and clipboard contains a table, prevent the paste operation
-        if (isInsideTableCell && nodes.some($isTableNode)) {
-          return true;
-        }
-
-        return false;
+        const isInsideTableCell =
+          $findTableNode(selection.anchor.getNode()) !== null;
+        return isInsideTableCell && nodes.some($isTableNode);
       },
       COMMAND_PRIORITY_EDITOR,
     ),