86 lines
1.8 KiB
TypeScript
86 lines
1.8 KiB
TypeScript
import {
|
||
Node,
|
||
nodeInputRule,
|
||
mergeAttributes,
|
||
} from '@tiptap/core'
|
||
import { TextSelection } from 'prosemirror-state'
|
||
|
||
export interface HorizontalRuleOptions {
|
||
HTMLAttributes: Record<string, any>,
|
||
}
|
||
|
||
declare module '@tiptap/core' {
|
||
interface Commands<ReturnType> {
|
||
horizontalRule: {
|
||
/**
|
||
* Add a horizontal rule
|
||
*/
|
||
setHorizontalRule: () => ReturnType,
|
||
}
|
||
}
|
||
}
|
||
|
||
export const HorizontalRule = Node.create<HorizontalRuleOptions>({
|
||
name: 'horizontalRule',
|
||
|
||
addOptions() {
|
||
return {
|
||
HTMLAttributes: {},
|
||
}
|
||
},
|
||
|
||
group: 'block',
|
||
|
||
parseHTML() {
|
||
return [
|
||
{ tag: 'hr' },
|
||
]
|
||
},
|
||
|
||
renderHTML({ HTMLAttributes }) {
|
||
return ['hr', mergeAttributes(this.options.HTMLAttributes, HTMLAttributes)]
|
||
},
|
||
|
||
addCommands() {
|
||
return {
|
||
setHorizontalRule: () => ({ chain }) => {
|
||
return chain()
|
||
.insertContent({ type: this.name })
|
||
// set cursor after horizontal rule
|
||
.command(({ tr, dispatch }) => {
|
||
if (dispatch) {
|
||
const { $to } = tr.selection
|
||
const posAfter = $to.end()
|
||
|
||
if ($to.nodeAfter) {
|
||
tr.setSelection(TextSelection.create(tr.doc, $to.pos))
|
||
} else {
|
||
// add node after horizontal rule if it’s the end of the document
|
||
const node = $to.parent.type.contentMatch.defaultType?.create()
|
||
|
||
if (node) {
|
||
tr.insert(posAfter, node)
|
||
tr.setSelection(TextSelection.create(tr.doc, posAfter))
|
||
}
|
||
}
|
||
|
||
tr.scrollIntoView()
|
||
}
|
||
|
||
return true
|
||
})
|
||
.run()
|
||
},
|
||
}
|
||
},
|
||
|
||
addInputRules() {
|
||
return [
|
||
nodeInputRule({
|
||
find: /^(?:---|—-|___\s|\*\*\*\s)$/,
|
||
type: this.type,
|
||
}),
|
||
]
|
||
},
|
||
})
|