28 lines
813 B
TypeScript
28 lines
813 B
TypeScript
import { wrapIn as originalWrapIn } from 'prosemirror-commands'
|
|
import { NodeType } from 'prosemirror-model'
|
|
import { RawCommands } from '../types'
|
|
import isNodeActive from '../helpers/isNodeActive'
|
|
import getNodeType from '../helpers/getNodeType'
|
|
|
|
declare module '@tiptap/core' {
|
|
interface Commands<ReturnType> {
|
|
wrapIn: {
|
|
/**
|
|
* Wraps nodes in another node.
|
|
*/
|
|
wrapIn: (typeOrName: string | NodeType, attributes?: Record<string, any>) => ReturnType,
|
|
}
|
|
}
|
|
}
|
|
|
|
export const wrapIn: RawCommands['wrapIn'] = (typeOrName, attributes = {}) => ({ state, dispatch }) => {
|
|
const type = getNodeType(typeOrName, state.schema)
|
|
const isActive = isNodeActive(state, type, attributes)
|
|
|
|
if (isActive) {
|
|
return false
|
|
}
|
|
|
|
return originalWrapIn(type, attributes)(state, dispatch)
|
|
}
|