62 lines
1.7 KiB
TypeScript
62 lines
1.7 KiB
TypeScript
import { NodeType, MarkType } from 'prosemirror-model'
|
|
import getNodeType from '../helpers/getNodeType'
|
|
import getMarkType from '../helpers/getMarkType'
|
|
import getSchemaTypeNameByName from '../helpers/getSchemaTypeNameByName'
|
|
import deleteProps from '../utilities/deleteProps'
|
|
import { Command, RawCommands } from '../types'
|
|
|
|
declare module '@tiptap/core' {
|
|
interface Commands {
|
|
resetAttributes: {
|
|
/**
|
|
* Resets some node attributes to the default value.
|
|
*/
|
|
resetAttributes: (typeOrName: string | NodeType | MarkType, attributes: string | string[]) => Command,
|
|
}
|
|
}
|
|
}
|
|
|
|
export const resetAttributes: RawCommands['resetAttributes'] = (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 => {
|
|
state.doc.nodesBetween(range.$from.pos, range.$to.pos, (node, pos) => {
|
|
if (nodeType && nodeType === node.type) {
|
|
tr.setNodeMarkup(pos, undefined, deleteProps(node.attrs, attributes))
|
|
}
|
|
|
|
if (markType && node.marks.length) {
|
|
node.marks.forEach(mark => {
|
|
if (markType === mark.type) {
|
|
tr.addMark(pos, pos + node.nodeSize, markType.create(deleteProps(mark.attrs, attributes)))
|
|
}
|
|
})
|
|
}
|
|
})
|
|
})
|
|
}
|
|
|
|
return true
|
|
}
|