Toggle > Handle deleteForward before a non-selectable and deleteBackward after a non-selectable

Author: Liboul

packages/toggle/src/transforms/index.ts

@@ -2,4 +2,6 @@
  * @file Automatically generated by barrelsby.
  */
 
+export * from './moveCurrentBlockAfterPreviousSelectable';
+export * from './moveNextSelectableAfterCurrentBlock';
 export * from './openNextToggles';

packages/toggle/src/transforms/moveCurrentBlockAfterPreviousSelectable.ts

@@ -0,0 +1,37 @@
+import {
+  getBlockAbove,
+  getPointBefore,
+  getPreviousNode,
+  isElement,
+  isSelectionAtBlockStart,
+  moveNodes,
+  PlateEditor,
+} from '@udecode/plate-common';
+
+import { isInClosedToggle } from '../queries';
+
+// Return false only if the all previous blocks are not selectable
+export const moveCurrentBlockAfterPreviousSelectable = (
+  editor: PlateEditor
+): boolean | undefined => {
+  const { selection } = editor;
+  if (!selection) return;
+  const aboveBlock = getBlockAbove(editor);
+  if (!aboveBlock) return;
+  if (!isSelectionAtBlockStart(editor)) return;
+  const beforePoint = getPointBefore(editor, selection);
+  if (!beforePoint) return;
+  const blockBefore = getBlockAbove(editor, { at: beforePoint });
+  if (!blockBefore) return;
+  if (!isInClosedToggle(editor, blockBefore[0].id)) return; // We're already after a selectable then
+  const previousSelectableBlock = getPreviousNode(editor, {
+    match: (node) =>
+      isElement(node) && !isInClosedToggle(editor, node.id as string),
+  });
+  if (!previousSelectableBlock) return false;
+  const afterSelectableBlock = [previousSelectableBlock[1][0] + 1];
+  moveNodes(editor, {
+    at: aboveBlock[1],
+    to: afterSelectableBlock,
+  });
+};

packages/toggle/src/transforms/moveNextSelectableAfterCurrentBlock.ts

@@ -0,0 +1,35 @@
+import {
+  getBlockAbove,
+  getNextNode,
+  getPointAfter,
+  isElement,
+  isSelectionAtBlockEnd,
+  moveNodes,
+  PlateEditor,
+} from '@udecode/plate-common';
+
+import { isInClosedToggle } from '../queries';
+
+// Return false only if all next blocks are not selectable
+export const moveNextSelectableAfterCurrentBlock = (editor: PlateEditor) => {
+  const { selection } = editor;
+  if (!selection) return;
+  const aboveBlock = getBlockAbove(editor);
+  if (!aboveBlock) return;
+  if (!isSelectionAtBlockEnd(editor)) return;
+  const afterPoint = getPointAfter(editor, selection);
+  if (!afterPoint) return;
+  const blockAfter = getBlockAbove(editor, { at: afterPoint });
+  if (!blockAfter) return;
+  if (!isInClosedToggle(editor, blockAfter[0].id)) return; // We're already before a selectable then
+  const nextSelectableBlock = getNextNode(editor, {
+    match: (node) =>
+      isElement(node) && !isInClosedToggle(editor, node.id as string),
+  });
+  if (!nextSelectableBlock) return false;
+  const afterCurrentBlock = [aboveBlock[1][0] + 1];
+  moveNodes(editor, {
+    at: nextSelectableBlock[1],
+    to: afterCurrentBlock,
+  });
+};

packages/toggle/src/withToggle.ts

@@ -3,13 +3,18 @@ import {
   isNode,
   moveNodes,
   PlateEditor,
+  select,
   toggleNodeType,
   Value,
 } from '@udecode/plate-common';
 import { indent, TIndentElement } from '@udecode/plate-indent';
 
 import { getLastEntryEnclosedInToggle, isInClosedToggle } from './queries';
 import { isToggleOpen } from './toggle-controller-store';
+import {
+  moveCurrentBlockAfterPreviousSelectable,
+  moveNextSelectableAfterCurrentBlock,
+} from './transforms';
 import { ELEMENT_TOGGLE } from './types';
 
 export const withToggle = <
@@ -18,14 +23,28 @@ export const withToggle = <
 >(
   editor: E
 ) => {
-  const { insertBreak, isSelectable } = editor;
+  const { insertBreak, isSelectable, deleteBackward, deleteForward } = editor;
 
   editor.isSelectable = (element) => {
     if (isNode(element) && isInClosedToggle<V, E>(editor, element.id as string))
       return false;
     return isSelectable(element);
   };
 
+  editor.deleteBackward = (unit) => {
+    if (
+      moveCurrentBlockAfterPreviousSelectable(editor as PlateEditor) === false
+    )
+      return;
+    deleteBackward(unit);
+  };
+
+  editor.deleteForward = (unit) => {
+    if (moveNextSelectableAfterCurrentBlock(editor as PlateEditor) === false)
+      return;
+    deleteForward(unit);
+  };
+
   editor.insertBreak = () => {
     // If we are inserting a break in a toggle:
     //   If the toggle is open
@@ -57,10 +76,15 @@ export const withToggle = <
         insertBreak();
 
         if (lastEntryEnclosedInToggle) {
+          const newlyInsertedTogglePath = [currentBlockEntry[1][0] + 1];
+          const afterLastEntryEncloseInToggle = [
+            lastEntryEnclosedInToggle[1][0] + 1,
+          ];
           moveNodes(editor, {
-            at: [currentBlockEntry[1][0] + 1], // Newly inserted toggle
-            to: [lastEntryEnclosedInToggle[1][0] + 1], // Right after the last enclosed element
+            at: newlyInsertedTogglePath,
+            to: afterLastEntryEncloseInToggle,
           });
+          select(editor, afterLastEntryEncloseInToggle);
         }
       }
     });