import { NodeType } from 'prosemirror-model'
import { Command, RawCommands, AnyObject } from '../types'

declare module '@tiptap/core' {
  interface AllCommands {
    replace: {
      /**
       * Replaces text with a node.
       */
      replace: (typeOrName: string | NodeType, attributes?: AnyObject) => Command,
    }
  }
}

export const replace: RawCommands['replace'] = (typeOrName, attributes = {}) => ({ state, commands }) => {
  const { from, to } = state.selection
  const range = { from, to }

  return commands.replaceRange(range, typeOrName, attributes)
}