Function useFocusReturn

web/hooks/useFocusReturn.ts:22–40

()

 * };
 */
export function useFocusReturn() {
  const savedRef = useRef<HTMLElement | null>(null);

  const save = useCallback(() => {
    savedRef.current = document.activeElement as HTMLElement | null;
  }, []);

  const restore = useCallback(() => {
    if (savedRef.current && typeof savedRef.current.focus === "function") {
      savedRef.current.focus();
      savedRef.current = null;
    }
  }, []);

  // Safety cleanup on unmount
  useEffect(() => () => { savedRef.current = null; }, []);

  return { save, restore };
}