61 lines
1.3 KiB
TypeScript
61 lines
1.3 KiB
TypeScript
import { TextSelection } from 'prosemirror-state'
|
|
import { Editor } from '../Editor'
|
|
import { Command, FocusPosition } from '../types'
|
|
import minMax from '../utils/minMax'
|
|
|
|
interface ResolvedSelection {
|
|
from: number,
|
|
to: number,
|
|
}
|
|
|
|
function resolveSelection(editor: Editor, position: FocusPosition = null): ResolvedSelection {
|
|
if (position === null) {
|
|
return editor.selection
|
|
}
|
|
|
|
if (position === 'start' || position === true) {
|
|
return {
|
|
from: 0,
|
|
to: 0,
|
|
}
|
|
}
|
|
|
|
if (position === 'end') {
|
|
const { size } = editor.state.doc.content
|
|
|
|
return {
|
|
from: size,
|
|
to: size - 1, // TODO: -1 only for nodes with content
|
|
}
|
|
}
|
|
|
|
return {
|
|
from: position as number,
|
|
to: position as number,
|
|
}
|
|
}
|
|
|
|
export default (position: FocusPosition = null): Command => ({
|
|
editor,
|
|
view,
|
|
tr,
|
|
dispatch,
|
|
}) => {
|
|
if ((view.hasFocus() && position === null) || position === false) {
|
|
return true
|
|
}
|
|
|
|
const { from, to } = resolveSelection(editor, position)
|
|
const { doc } = tr
|
|
const resolvedFrom = minMax(from, 0, doc.content.size)
|
|
const resolvedEnd = minMax(to, 0, doc.content.size)
|
|
const selection = TextSelection.create(doc, resolvedFrom, resolvedEnd)
|
|
|
|
if (dispatch) {
|
|
tr.setSelection(selection)
|
|
view.focus()
|
|
}
|
|
|
|
return true
|
|
}
|