Files
tiptap/packages/tiptap-commands/src/commands/markInputRule.js
2019-10-23 17:17:28 +08:00

57 lines
1.5 KiB
JavaScript

import { InputRule } from 'prosemirror-inputrules'
function getMarksBetween(start, end, state) {
let marks = []
state.doc.nodesBetween(start, end, (node, pos) => {
marks = [...marks, ...node.marks.map(mark => ({
start: pos,
end: pos + node.nodeSize,
mark,
}))]
})
return marks
}
export default function (regexp, markType, getAttrs) {
return new InputRule(regexp, (state, match, start, end) => {
const attrs = getAttrs instanceof Function ? getAttrs(match) : getAttrs
const { tr } = state
const m = match.length - 1
let markEnd = end
let markStart = start
if (match[m]) {
const matchStart = start + match[0].indexOf(match[m - 1])
const matchEnd = matchStart + match[m - 1].length - 1
const textStart = matchStart + match[m - 1].lastIndexOf(match[m])
const textEnd = textStart + match[m].length
const excludedMarks = getMarksBetween(start, end, state)
.filter(item => {
const { excluded } = item.mark.type
return excluded.find(type => type.name === markType.name)
})
.filter(item => item.end > matchStart)
if (excludedMarks.length) {
return false
}
if (textEnd < matchEnd) {
tr.delete(textEnd, matchEnd)
}
if (textStart > matchStart) {
tr.delete(matchStart, textStart)
}
markStart = matchStart
markEnd = markStart + match[m].length
}
tr.addMark(markStart, markEnd, markType.create(attrs))
tr.removeStoredMark(markType)
return tr
})
}