1

2026-05-25 17:08:18 +08:00
parent df501f6151
commit f1259b71b5
9178 changed files with 1626125 additions and 0 deletions
@@ -0,0 +1,66 @@
+import KeyCode from "@rc-component/util/es/KeyCode";
+import raf from "@rc-component/util/es/raf";
+import * as React from 'react';
+const {
+  ESC,
+  TAB
+} = KeyCode;
+export default function useAccessibility({
+  visible,
+  triggerRef,
+  onVisibleChange,
+  autoFocus,
+  overlayRef
+}) {
+  const focusMenuRef = React.useRef(false);
+  const handleCloseMenuAndReturnFocus = () => {
+    if (visible) {
+      triggerRef.current?.focus?.();
+      onVisibleChange?.(false);
+    }
+  };
+  const focusMenu = () => {
+    if (overlayRef.current?.focus) {
+      overlayRef.current.focus();
+      focusMenuRef.current = true;
+      return true;
+    }
+    return false;
+  };
+  const handleKeyDown = event => {
+    switch (event.keyCode) {
+      case ESC:
+        handleCloseMenuAndReturnFocus();
+        break;
+      case TAB:
+        {
+          let focusResult = false;
+          if (!focusMenuRef.current) {
+            focusResult = focusMenu();
+          }
+          if (focusResult) {
+            event.preventDefault();
+          } else {
+            handleCloseMenuAndReturnFocus();
+          }
+          break;
+        }
+    }
+  };
+  React.useEffect(() => {
+    if (visible) {
+      window.addEventListener('keydown', handleKeyDown);
+      if (autoFocus) {
+        // FIXME: hack with raf
+        raf(focusMenu, 3);
+      }
+      return () => {
+        window.removeEventListener('keydown', handleKeyDown);
+        focusMenuRef.current = false;
+      };
+    }
+    return () => {
+      focusMenuRef.current = false;
+    };
+  }, [visible]); // eslint-disable-line react-hooks/exhaustive-deps
+}