add core extensions

packages/core/src/extensions/updateMark.ts

@@ -0,0 +1,43 @@
+import { MarkType } from 'prosemirror-model'
+import { Command } from '../Editor'
+import { createExtension } from '../Extension'
+import getMarkType from '../utils/getMarkType'
+import getMarkRange from '../utils/getMarkRange'
+
+export const UpdateMark = createExtension({
+  addCommands() {
+    return {
+      updateMark: (typeOrName: string | MarkType, attrs: {}): Command => ({ tr, state }) => {
+        const { selection, doc } = tr
+        let { from, to } = selection
+        const { $from, empty } = selection
+        const type = getMarkType(typeOrName, state.schema)
+
+        if (empty) {
+          const range = getMarkRange($from, type)
+
+          if (range) {
+            from = range.from
+            to = range.to
+          }
+        }
+
+        const hasMark = doc.rangeHasMark(from, to, type)
+
+        if (hasMark) {
+          tr.removeMark(from, to, type)
+        }
+
+        tr.addMark(from, to, type.create(attrs))
+
+        return true
+      },
+    }
+  },
+})
+
+declare module '../Editor' {
+  interface AllExtensions {
+    UpdateMark: typeof UpdateMark,
+  }
+}