fix: fix error when calculating contentMatch within clearNodes command,

fix #1361
ueberdosis · May 28, 2021 · 3709668 · 3709668
1 parent a24e333
commit 3709668
Showing 1 changed file with 22 additions and 16 deletions.
diff --git a/packages/core/src/commands/clearNodes.ts b/packages/core/src/commands/clearNodes.ts
@@ -18,22 +18,28 @@ export const clearNodes: RawCommands['clearNodes'] = () => ({ state, tr, dispatc
 
   ranges.forEach(range => {
     state.doc.nodesBetween(range.$from.pos, range.$to.pos, (node, pos) => {
-      if (!node.type.isText) {
-        const fromPos = tr.doc.resolve(tr.mapping.map(pos + 1))
-        const toPos = tr.doc.resolve(tr.mapping.map(pos + node.nodeSize - 1))
-        const nodeRange = fromPos.blockRange(toPos)
-
-        if (nodeRange) {
-          const targetLiftDepth = liftTarget(nodeRange)
-
-          if (node.type.isTextblock && dispatch) {
-            tr.setNodeMarkup(nodeRange.start, state.doc.type.contentMatch.defaultType)
-          }
-
-          if ((targetLiftDepth || targetLiftDepth === 0) && dispatch) {
-            tr.lift(nodeRange, targetLiftDepth)
-          }
-        }
+      if (node.type.isText) {
+        return
+      }
+
+      const $fromPos = tr.doc.resolve(tr.mapping.map(pos))
+      const $toPos = tr.doc.resolve(tr.mapping.map(pos + node.nodeSize))
+      const nodeRange = $fromPos.blockRange($toPos)
+
+      if (!nodeRange) {
+        return
+      }
+
+      const targetLiftDepth = liftTarget(nodeRange)
+
+      if (node.type.isTextblock && dispatch) {
+        const { defaultType } = $fromPos.parent.contentMatchAt($fromPos.index())
+
+        tr.setNodeMarkup(nodeRange.start, defaultType)
+      }
+
+      if ((targetLiftDepth || targetLiftDepth === 0) && dispatch) {
+        tr.lift(nodeRange, targetLiftDepth)
       }
     })
   })