Dont lose table selection when focus changes (facebook#1766)

* Add tests for special Table selection behavior * Dont lose table selection when focus changes * Update packages/lexical-table/src/LexicalTableSelectionHelpers.js Co-authored-by: Strek <harish.sethuraman@freshworks.com> Co-authored-by: Strek <harish.sethuraman@freshworks.com>
yuanyan · Apr 15, 2022 · 69997ae · 69997ae
1 parent 7aa228d
commit 69997ae
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/packages/lexical-table/src/LexicalTableSelectionHelpers.js b/packages/lexical-table/src/LexicalTableSelectionHelpers.js
@@ -29,6 +29,7 @@ import {
   $isRangeSelection,
   $setSelection,
   DELETE_CHARACTER_COMMAND,
+  FOCUS_COMMAND,
   FORMAT_TEXT_COMMAND,
   INSERT_TEXT_COMMAND,
   KEY_ARROW_DOWN_COMMAND,
@@ -778,6 +779,16 @@ export function applyTableHandlers(
     ),
   );
 
+  tableSelection.listenersToRemove.add(
+    editor.registerCommand(
+      FOCUS_COMMAND,
+      (payload) => {
+        return tableNode.isSelected();
+      },
+      CriticalPriority,
+    ),
+  );
+
   tableSelection.listenersToRemove.add(
     editor.registerCommand(
       SELECTION_CHANGE_COMMAND,