diff --git a/packages/tiptap-commands/src/commands/toggleWrap.js b/packages/tiptap-commands/src/commands/toggleWrap.js index bb71f5f3a..b8b4457c8 100644 --- a/packages/tiptap-commands/src/commands/toggleWrap.js +++ b/packages/tiptap-commands/src/commands/toggleWrap.js @@ -1,14 +1,14 @@ import { wrapIn, lift } from 'prosemirror-commands' import { nodeIsActive } from 'tiptap-utils' -export default function (type) { +export default function (type, attrs = {}) { return (state, dispatch, view) => { - const isActive = nodeIsActive(state, type) + const isActive = nodeIsActive(state, type, attrs) if (isActive) { return lift(state, dispatch) } - return wrapIn(type)(state, dispatch, view) + return wrapIn(type, attrs)(state, dispatch, view) } }