Merge pull request #3740 from udecode/focus/selection

udecode · Nov 7, 2024 · 0b427c2 · 0b427c2
2 parents be2d222 + 1f2ec22
commit 0b427c2
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 3 deletions.
diff --git a/.changeset/soft-dots-train.md b/.changeset/soft-dots-train.md
@@ -0,0 +1,5 @@
+---
+'@udecode/plate-selection': patch
+---
+
+Fix enter can't unselect.
diff --git a/packages/selection/src/react/BlockSelectionPlugin.tsx b/packages/selection/src/react/BlockSelectionPlugin.tsx
@@ -258,6 +258,22 @@ export const BlockSelectionAfterEditable: EditableSiblingComponent = () => {
 
 export const BlockSelectionPlugin = createTPlatePlugin<BlockSelectionConfig>({
   key: 'blockSelection',
+  extendEditor: ({ api, editor, getOptions }) => {
+    const { setSelection } = editor;
+
+    editor.setSelection = (...args) => {
+      if (
+        getOptions().selectedIds!.size > 0 &&
+        !editor.getOption(BlockMenuPlugin, 'openId')
+      ) {
+        api.blockSelection.unselect();
+      }
+
+      setSelection(...args);
+    };
+
+    return editor;
+  },
   options: {
     areaOptions: {
       behaviour: {
@@ -301,9 +317,6 @@ export const BlockSelectionPlugin = createTPlatePlugin<BlockSelectionConfig>({
       const target = event.target as HTMLElement;
 
       if (target.dataset.platePreventUnselect) return;
-
-      console.log(editor.getOption(BlockMenuPlugin, 'openId'), 'fj');
-
       if (
         event.button === 0 &&
         getOptions().selectedIds!.size > 0 &&