73 lines
1.9 KiB
TypeScript
73 lines
1.9 KiB
TypeScript
import { NodeType, MarkType } from 'prosemirror-model'
|
|
import getNodeType from '../helpers/getNodeType'
|
|
import getMarkType from '../helpers/getMarkType'
|
|
import getSchemaTypeNameByName from '../helpers/getSchemaTypeNameByName'
|
|
import { RawCommands } from '../types'
|
|
|
|
declare module '@tiptap/core' {
|
|
interface Commands<ReturnType> {
|
|
updateAttributes: {
|
|
/**
|
|
* Update attributes of a node or mark.
|
|
*/
|
|
updateAttributes: (typeOrName: string | NodeType | MarkType, attributes: Record<string, any>) => ReturnType,
|
|
}
|
|
}
|
|
}
|
|
|
|
export const updateAttributes: RawCommands['updateAttributes'] = (typeOrName, attributes = {}) => ({ tr, state, dispatch }) => {
|
|
let nodeType: NodeType | null = null
|
|
let markType: MarkType | null = null
|
|
|
|
const schemaType = getSchemaTypeNameByName(
|
|
typeof typeOrName === 'string'
|
|
? typeOrName
|
|
: typeOrName.name,
|
|
state.schema,
|
|
)
|
|
|
|
if (!schemaType) {
|
|
return false
|
|
}
|
|
|
|
if (schemaType === 'node') {
|
|
nodeType = getNodeType(typeOrName as NodeType, state.schema)
|
|
}
|
|
|
|
if (schemaType === 'mark') {
|
|
markType = getMarkType(typeOrName as MarkType, state.schema)
|
|
}
|
|
|
|
if (dispatch) {
|
|
tr.selection.ranges.forEach(range => {
|
|
const from = range.$from.pos
|
|
const to = range.$to.pos
|
|
|
|
state.doc.nodesBetween(from, to, (node, pos) => {
|
|
if (nodeType && nodeType === node.type) {
|
|
tr.setNodeMarkup(pos, undefined, {
|
|
...node.attrs,
|
|
...attributes,
|
|
})
|
|
}
|
|
|
|
if (markType && node.marks.length) {
|
|
node.marks.forEach(mark => {
|
|
if (markType === mark.type) {
|
|
const trimmedFrom = Math.max(pos, from)
|
|
const trimmedTo = Math.min(pos + node.nodeSize, to)
|
|
|
|
tr.addMark(trimmedFrom, trimmedTo, markType.create({
|
|
...mark.attrs,
|
|
...attributes,
|
|
}))
|
|
}
|
|
})
|
|
}
|
|
})
|
|
})
|
|
}
|
|
|
|
return true
|
|
}
|