tiptap/packages/core/src/commands/focus.ts

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
}