import Mousetrap from 'mousetrap';
import { useEffect, useRef } from 'react';

/**
 * Use mousetrap hook
 *
 * @param handlerKey - A key, key combo or array of combos according to Mousetrap documentation.
 * @param  handlerCallback - A function that is triggered on key combo catch.
 */
export function useMousetrap(
  handlerKey: string | string[],
  handlerCallback: (e: ExtendedKeyboardEvent, combo: string) => void,
  elementOrElementRef: typeof document | React.RefObject<Element | null>,
) {
  const actionRef = useRef<typeof handlerCallback | null>(null);
  actionRef.current = handlerCallback;
  const element =
    'current' in elementOrElementRef ? elementOrElementRef.current : document;

  useEffect(() => {
    const instance = new Mousetrap(element as Element);

    instance.bind(handlerKey, (e, combo) => {
      typeof actionRef.current === 'function' && actionRef.current(e, combo);
    });

    return () => {
      instance.unbind(handlerKey);
    };
  }, [handlerKey, element]);
}