refactor: conditional focus of flyout menu

dhis2 · Jun 30, 2024 · 841d270 · 841d270
1 parent f31fd08
commit 841d270
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/components/menu/src/flyout-menu/flyout-menu.js b/components/menu/src/flyout-menu/flyout-menu.js
@@ -26,17 +26,20 @@ const FlyoutMenu = ({
 
     const divRef = useRef(null)
 
-    const handleFocus = (event) => {
-        if (event.target === divRef.current) {
-            divRef.current.children[0].focus()
-        }
-    }
-
     useEffect(() => {
         if (!divRef) {
             return
         }
         const div = divRef.current
+
+        const handleFocus = (event) => {
+            if (event.target === div) {
+                if (div.children && div.children.length > 0) {
+                    div.children[0].focus()
+                }
+            }
+        }
+
         div.addEventListener('focus', handleFocus)
 
         return () => {