rsuite/esm/internals/hooks/useFocus.js

A suite of react components

'use client';
import { useCallback, useRef } from 'react';
export function useFocus(elementRef) {
  // When grabbing focus, keep track of previous activeElement
  // so that we can return focus later
  const previousActiveElementRef = useRef(null);

  // Focus the element itself
  const grab = useCallback(() => {
    requestAnimationFrame(() => {
      if (document.activeElement !== elementRef.current) {
        previousActiveElementRef.current = document.activeElement;
        elementRef.current?.focus();
      }
    });
  }, [elementRef]);

  // Return focus to previous active element
  const release = useCallback(options => {
    requestAnimationFrame(() => {
      previousActiveElementRef.current?.focus(options);
    });
  }, []);
  return {
    grab,
    release
  };
}
export default useFocus;