UNPKG

@wulperstudio/cms

Version:
31 lines (29 loc) 1.11 kB
/* eslint-disable no-undef */ /* eslint-disable consistent-return */ /* eslint-disable no-redeclare */ import { useEffect, useRef } from 'react'; import { useIsomorphicLayoutEffect } from './useIsomorphicLayoutEffect'; function useEventListener(eventName, handler, element) { // Create a ref that stores handler var savedHandler = useRef(handler); useIsomorphicLayoutEffect(function () { savedHandler.current = handler; }, [handler]); useEffect(function () { // Define the listening target var targetElement = (element == null ? void 0 : element.current) || window; if (!(targetElement && targetElement.addEventListener)) { return; } // Create event listener that calls handler function stored in ref var eventListener = function eventListener(event) { return savedHandler.current(event); }; targetElement.addEventListener(eventName, eventListener); // Remove event listener on cleanup return function () { targetElement.removeEventListener(eventName, eventListener); }; }, [eventName, element]); } export { useEventListener };