add core extensions

packages/core/src/extensions/toggleWrap.ts

@@ -0,0 +1,29 @@
+import { wrapIn, lift } from 'prosemirror-commands'
+import { NodeType } from 'prosemirror-model'
+import { Command } from '../Editor'
+import { createExtension } from '../Extension'
+import nodeIsActive from '../utils/nodeIsActive'
+import getNodeType from '../utils/getNodeType'
+
+export const ToggleWrap = createExtension({
+  addCommands() {
+    return {
+      toggleWrap: (typeOrName: string | NodeType, attrs = {}): Command => ({ state, dispatch }) => {
+        const type = getNodeType(typeOrName, state.schema)
+        const isActive = nodeIsActive(state, type, attrs)
+
+        if (isActive) {
+          return lift(state, dispatch)
+        }
+
+        return wrapIn(type, attrs)(state, dispatch)
+      },
+    }
+  },
+})
+
+declare module '../Editor' {
+  interface AllExtensions {
+    ToggleWrap: typeof ToggleWrap,
+  }
+}