fix: select all with last decorator node

mdx-editor · Sep 9, 2023 · 175c3f8 · 175c3f8
1 parent 137ba12
commit 175c3f8
Showing 1 changed file with 17 additions and 14 deletions.
diff --git a/src/plugins/core/index.ts b/src/plugins/core/index.ts
@@ -8,6 +8,7 @@ import {
   $getRoot,
   $getSelection,
   $insertNodes,
+  $isDecoratorNode,
   $isRangeSelection,
   $isRootOrShadowRoot,
   BLUR_COMMAND,
@@ -288,23 +289,25 @@ export const coreSystem = system((r) => {
       KEY_DOWN_COMMAND,
       (event) => {
         const { keyCode, ctrlKey, metaKey } = event
-        let shouldOverride = false
+        if (keyCode === 65 && controlOrMeta(metaKey, ctrlKey)) {
+          let shouldOverride = false
 
-        theRootEditor.getEditorState().read(() => {
-          shouldOverride = $getRoot().getFirstChild()?.getType() === 'frontmatter'
-        })
+          theRootEditor.getEditorState().read(() => {
+            shouldOverride = $isDecoratorNode($getRoot().getFirstChild()) || $isDecoratorNode($getRoot().getLastChild())
+          })
 
-        if (keyCode === 65 && controlOrMeta(metaKey, ctrlKey) && shouldOverride) {
-          event.preventDefault()
-          event.stopImmediatePropagation()
-          theRootEditor.update(() => {
-            const rootElement = theRootEditor.getRootElement() as HTMLDivElement
-            window.getSelection()?.selectAllChildren(rootElement)
-            rootElement.focus({
-              preventScroll: true
+          if (shouldOverride) {
+            event.preventDefault()
+            event.stopImmediatePropagation()
+            theRootEditor.update(() => {
+              const rootElement = theRootEditor.getRootElement() as HTMLDivElement
+              window.getSelection()?.selectAllChildren(rootElement)
+              rootElement.focus({
+                preventScroll: true
+              })
             })
-          })
-          return true
+            return true
+          }
         }
 
         return false