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

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

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