import { EditorState, RichUtils } from 'draft-js';

const unsetLink = (editorState: EditorState): EditorState => {
  const containsLink = RichUtils.currentBlockContainsLink(editorState);
  if (!containsLink) return editorState;

  const selectionState = editorState.getSelection();
  return RichUtils.toggleLink(editorState, selectionState, null);
};

export default unsetLink;