react-use/src/useDrop.ts

import * as React from 'react';
import useRefMounted from './useRefMounted';

const {useState, useMemo, useCallback, useEffect} = React;

export interface DropAreaState {
  over: boolean;
}

export interface DropAreaBond {
  onDragOver: React.DragEventHandler;
  onDragEnter: React.DragEventHandler;
  onDragLeave: React.DragEventHandler;
  onDrop: React.DragEventHandler;
  onPaste: React.ClipboardEventHandler;
}

export interface DropAreaOptions {
  onFiles?: (files: File[], event?) => void;
  onText?: (text: string, event?) => void;
  onUri?: (url: string, event?) => void;
}

const noop = () => {};
const defaultState: DropAreaState = {
  over: false,
};

const createProcess = (options: DropAreaOptions, mounted: React.RefObject<boolean>) => (
  dataTransfer: DataTransfer,
  event,
) => {
  const uri = dataTransfer.getData('text/uri-list');

  if (uri) {
    (options.onUri || noop)(uri, event);
    return;
  }

  if (dataTransfer.files && dataTransfer.files.length) {
    (options.onFiles || noop)(Array.from(dataTransfer.files), event);
    return;
  }

  if (dataTransfer.items && dataTransfer.items.length) {
    dataTransfer.items[0].getAsString((text) => {
      if (mounted.current) {
        (options.onText || noop)(text, event);
      }
    });
  }
};

const useDrop = (options: DropAreaOptions = {}, args = []): DropAreaState => {
  const {onFiles, onText, onUri} = options;
  const mounted = useRefMounted();
  const [over, setOverRaw] = useState<boolean>(false);
  const setOver = useCallback(setOverRaw, []);
  const process = useMemo(() => createProcess(options, mounted), [onFiles, onText, onUri]);

  useEffect(() => {
    const onDragOver = (event) => {
      event.preventDefault();
      setOver(true);
    };

    const onDragEnter = (event) => {
      event.preventDefault();
      setOver(true);
    };

    const onDragLeave = () => {
      setOver(false);
    };

    const onDragExit = () => {
      setOver(false);
    };

    const onDrop = (event) => {
      event.preventDefault();
      setOver(false);
      process(event.dataTransfer, event);
    };

    const onPaste = (event) => {
      process(event.clipboardData, event);
    };

    document.addEventListener('dragover', onDragOver);
    document.addEventListener('dragenter', onDragEnter);
    document.addEventListener('dragleave', onDragLeave);
    document.addEventListener('dragexit', onDragExit);
    document.addEventListener('drop', onDrop);
    if (onText)
      document.addEventListener('paste', onPaste);

    return () => {
      document.removeEventListener('dragover', onDragOver);
      document.removeEventListener('dragenter', onDragEnter);
      document.removeEventListener('dragleave', onDragLeave);
      document.removeEventListener('dragexit', onDragExit);
      document.removeEventListener('drop', onDrop);
      document.removeEventListener('paste', onPaste);
    };
  }, [process, ...args]);

  return {over};
};

export default useDrop;