@carbon/react/es/internal/useEvent.js

React components for the Carbon Design System

/**
 * Copyright IBM Corp. 2016, 2023
 *
 * This source code is licensed under the Apache-2.0 license found in the
 * LICENSE file in the root directory of this source tree.
 */

import { useRef, useEffect } from 'react';

const useEvent = (elementOrRef, eventName, callback) => {
  const savedCallback = useRef(null);
  useEffect(() => {
    savedCallback.current = callback;
  }, [callback]);
  useEffect(() => {
    const element = 'current' in elementOrRef ? elementOrRef.current : elementOrRef;
    if (!element) return;
    const handler = event => {
      if (savedCallback.current) {
        savedCallback.current(event);
      }
    };
    element.addEventListener(eventName, handler);
    return () => {
      element.removeEventListener(eventName, handler);
    };
  }, [elementOrRef, eventName]);
};
const useWindowEvent = (eventName, callback) => {
  const savedCallback = useRef(null);
  useEffect(() => {
    savedCallback.current = callback;
  }, [callback]);
  useEffect(() => {
    const handler = event => {
      if (savedCallback.current) {
        savedCallback.current(event);
      }
    };
    window.addEventListener(eventName, handler);
    return () => {
      window.removeEventListener(eventName, handler);
    };
  }, [eventName]);
};

export { useEvent, useWindowEvent };