import {EditorType} from '../global/interface'

export const withMentions = (editor: EditorType) => {
  const {isInline, isVoid} = editor

  editor.isInline = (element: any) => {
    return element.type === 'mention' ? true : isInline(element)
  }
  editor.isVoid = (element: any) => {
    return element.type === 'mention' ? true : isVoid(element)
  }
  return editor
}