huly-platform/plugins/text-editor-resources/src/components/extension/focus.ts

import { type Editor, Extension } from '@tiptap/core'

const canBlur = (editor: Editor, options: FocusOptions): boolean => {
  return (
    options.canBlur?.(editor) ??
    (!editor.isActive('bulletList') &&
      !editor.isActive('orderedList') &&
      !editor.isActive('taskList') &&
      !editor.isActive('todoList') &&
      !editor.isActive('table') &&
      !editor.isActive('code') &&
      !editor.isActive('codeBlock'))
  )
}

export interface FocusOptions {
  canBlur?: (editor: Editor) => boolean
  onCanBlur?: (canBlur: boolean) => void
  onFocus?: (focused: boolean) => void
}

export interface FocusStorage {
  canBlur: boolean
}

export const FocusExtension = Extension.create<FocusOptions, FocusStorage>({
  name: 'focus',
  addStorage () {
    return { canBlur: true }
  },
  onCreate () {
    this.options.onFocus?.(this.editor.isFocused)

    this.storage.canBlur = canBlur(this.editor, this.options)
    this.options.onCanBlur?.(this.storage.canBlur)
  },
  onBlur () {
    this.options.onFocus?.(false)
  },
  onFocus () {
    this.options.onFocus?.(true)
  },
  onSelectionUpdate () {
    const canBlurNow = canBlur(this.editor, this.options)

    if (this.storage.canBlur !== canBlurNow) {
      this.storage.canBlur = canBlurNow
      this.options.onCanBlur?.(this.storage.canBlur)
    }
  }
})