use custom splitblock command for text align

2020-10-27 12:37:20 +01:00
parent 2c150f5192
commit 8d0c67cc66
4 changed files with 82 additions and 68 deletions
--- a/packages/core/src/extensions/splitBlock.ts
+++ b/packages/core/src/extensions/splitBlock.ts
@@ -0,0 +1,71 @@
+// import {
+//   baseKeymap, chainCommands, newlineInCode, createParagraphNear, liftEmptyBlock, splitBlock,
+// } from 'prosemirror-commands'
+import { canSplit } from 'prosemirror-transform'
+import { ContentMatch, Fragment } from 'prosemirror-model'
+import { NodeSelection, TextSelection } from 'prosemirror-state'
+import { Command } from '../Editor'
+import { createExtension } from '../Extension'
+// import getNodeType from '../utils/getNodeType'
+
+function defaultBlockAt(match: ContentMatch) {
+  for (let i = 0; i < match.edgeCount; i + 1) {
+    const { type } = match.edge(i)
+    // @ts-ignore
+    if (type.isTextblock && !type.hasRequiredAttrs()) return type
+  }
+  return null
+}
+
+export const SplitBlock = createExtension({
+  addCommands() {
+    return {
+      splitBlock: (copyAttributes = false): Command => ({ state, dispatch }) => {
+        // const type = getNodeType(typeOrName, state.schema)
+
+        const { $from, $to } = state.selection
+        if (state.selection instanceof NodeSelection && state.selection.node.isBlock) {
+          if (!$from.parentOffset || !canSplit(state.doc, $from.pos)) return false
+          if (dispatch) dispatch(state.tr.split($from.pos).scrollIntoView())
+          return true
+        }
+
+        if (!$from.parent.isBlock) return false
+
+        if (dispatch) {
+          const atEnd = $to.parentOffset === $to.parent.content.size
+          const { tr } = state
+          if (state.selection instanceof TextSelection) tr.deleteSelection()
+          const deflt = $from.depth === 0 ? null : defaultBlockAt($from.node(-1).contentMatchAt($from.indexAfter(-1)))
+          let types = atEnd && deflt ? [{ type: deflt, attrs: copyAttributes ? $from.node().attrs : {} }] : null
+          // let types = atEnd && deflt ? [{ type: deflt }] : null
+          // @ts-ignore
+          let can = canSplit(tr.doc, tr.mapping.map($from.pos), 1, types)
+          // @ts-ignore
+          if (!types && !can && canSplit(tr.doc, tr.mapping.map($from.pos), 1, deflt && [{ type: deflt }])) {
+            // @ts-ignore
+            types = [{ type: deflt, attrs: copyAttributes ? $from.node().attrs : {} }]
+            // types = [{ type: deflt }]
+            can = true
+          }
+          if (can) {
+            // @ts-ignore
+            tr.split(tr.mapping.map($from.pos), 1, types)
+            if (!atEnd && !$from.parentOffset && $from.parent.type !== deflt
+            // @ts-ignore
+                && $from.node(-1).canReplace($from.index(-1), $from.indexAfter(-1), Fragment.from(deflt.create(), $from.parent))) { tr.setNodeMarkup(tr.mapping.map($from.before()), deflt) }
+          }
+          dispatch(tr.scrollIntoView())
+        }
+
+        return true
+      },
+    }
+  },
+})
+
+declare module '../Editor' {
+  interface AllExtensions {
+    SplitBlock: typeof SplitBlock,
+  }
+}