25 lines
731 B
TypeScript
25 lines
731 B
TypeScript
"use client";
|
||
|
||
import { useEffect, useRef, type RefObject } from "react";
|
||
|
||
// 弹层（Drawer / ContextDrawer）关闭后把焦点还给触发元素（a11y 焦点管理，CR-H11）。
|
||
// wasOpenRef 门闩：仅在「曾打开过」后的关闭才还原，避免从未打开时误抢焦点。
|
||
// null-safe：triggerRef 或其 current 为空时静默跳过。
|
||
export function useRestoreFocus(
|
||
open: boolean,
|
||
triggerRef?: RefObject<HTMLElement | null>,
|
||
): void {
|
||
const wasOpenRef = useRef(false);
|
||
|
||
useEffect(() => {
|
||
if (!open) {
|
||
if (wasOpenRef.current) {
|
||
wasOpenRef.current = false;
|
||
triggerRef?.current?.focus();
|
||
}
|
||
return;
|
||
}
|
||
wasOpenRef.current = true;
|
||
}, [open, triggerRef]);
|
||
}
|