import { MarkType } from 'prosemirror-model'
import { Command } from '../Editor'
import { createExtension } from '../Extension'
import getMarkType from '../utils/getMarkType'
import getMarkRange from '../utils/getMarkRange'

export const RemoveMark = createExtension({
  addCommands() {
    return {
      removeMark: (typeOrName: string | MarkType): Command => ({ tr, state }) => {
        const { selection } = tr
        const type = getMarkType(typeOrName, state.schema)
        let { from, to } = selection
        const { $from, empty } = selection

        if (empty) {
          const range = getMarkRange($from, type)

          if (range) {
            from = range.from
            to = range.to
          }
        }

        tr.removeMark(from, to, type)

        return true
      },
    }
  },
})

declare module '../Editor' {
  interface AllExtensions {
    RemoveMark: typeof RemoveMark,
  }
}