import { useCallback, useContext } from 'react';
import { DndContext, useDrag, useDrop } from 'react-dnd';

export const useDndContextAvailable = () => {
    const dndContext = useContext(DndContext);
    const { dragDropManager } = dndContext;
    return !!dragDropManager;
};

export const useDragIfAvailable: typeof useDrag = (...args) => {
    // This is an ugly hack to make `useDrag` not throw if a `DndContext` is not available.
    // See: https://github.com/react-dnd/react-dnd/issues/2212
    const dndContextAvailable = useDndContextAvailable();
    const mockHook = useCallback(() => [{} as any, () => null, () => null], []);
    // @ts-ignore
    const useHook: typeof useDrag = dndContextAvailable ? useDrag : mockHook;
    return useHook(...args);
};

export const useDropIfAvailable: typeof useDrop = (...args) => {
    const dndContextAvailable = useDndContextAvailable();
    const mockHook = useCallback(() => [{} as any, () => null], []);
    // @ts-ignore
    const useHook: typeof useDrop = dndContextAvailable ? useDrop : mockHook;
    return useHook(...args);
};