add getNodeType

packages/core/src/commands/toggleBlockType.ts

@@ -2,10 +2,11 @@ import { NodeType } from 'prosemirror-model'
 import { setBlockType } from 'prosemirror-commands'
 import { Editor } from '../Editor'
 import nodeIsActive from '../utils/nodeIsActive'
+import getNodeType from '../utils/getNodeType'
 
 type ToggleBlockType = (
-  type: NodeType,
+  type: string | NodeType,
-  toggleType: NodeType,
+  toggleType: string | NodeType,
   attrs?: {}
 ) => any
 
@@ -15,8 +16,10 @@ declare module '../Editor' {
   }
 }
 
-export default (next: Function, editor: Editor): ToggleBlockType => (type, toggleType, attrs) => {
+export default (next: Function, editor: Editor): ToggleBlockType => (typeOrName, toggleTypeOrName, attrs) => {
-  const { view, state } = editor
+  const { view, state, schema } = editor
+  const type = getNodeType(typeOrName, schema)
+  const toggleType = getNodeType(toggleTypeOrName, schema)
   const isActive = nodeIsActive(state, type, attrs)
 
   if (isActive) {

packages/core/src/utils/getNodeType.ts

@@ -0,0 +1,9 @@
+import { NodeType, Schema } from 'prosemirror-model'
+
+export default function getNodeType(nameOrType: string | NodeType, schema: Schema): NodeType {
+  if (typeof nameOrType === 'string') {
+    return schema.nodes[nameOrType]
+  }
+
+  return nameOrType
+}

packages/extension-heading/index.ts

@@ -48,7 +48,7 @@ export default class Heading extends Node {
   commands(): CommandSpec {
     return {
       heading: next => attrs => {
-        this.editor.toggleBlockType(this.type, this.editor.schema.nodes.paragraph, attrs)
+        this.editor.toggleBlockType(this.name, 'paragraph', attrs)
         next()
       },
     }