import {useEffect} from "react";

export default function useKey(keyName, callback) {
  useEffect(() => {
    const keys = Array.isArray(keyName) ? keyName : [keyName];

    const handler = function (ev) {
      if (keys.includes(ev.key)) {
        if (callback()) {
          ev.preventDefault();
        }
      }
    };

    window.addEventListener("keydown", handler);

    return () => {
      window.removeEventListener("keydown", handler);
    }
  })
}