zent/src/copy-button/toggleSelection.ts

一套前端设计语言和基于React的实现

import noop from '../utils/noop';

export default function toggleSelection() {
  const selection = document.getSelection();
  if (!selection.rangeCount) {
    return noop;
  }

  let active = document.activeElement;

  const ranges = [];

  for (let i = 0; i < selection.rangeCount; i++) {
    ranges.push(selection.getRangeAt(i));
  }

  switch (
    active.tagName.toUpperCase() // .toUpperCase handles XHTML
  ) {
    case 'INPUT':
    case 'TEXTAREA':
      (active as HTMLElement).blur();
      break;

    default:
      active = null;
      break;
  }

  selection.removeAllRanges();
  return function restoreSelection() {
    selection.type === 'Caret' && selection.removeAllRanges();

    if (!selection.rangeCount) {
      ranges.forEach(range => {
        selection.addRange(range);
      });
    }

    active && (active as HTMLElement).focus();
  };
}