Fix combobox visible focus styling when opening the menu via keyboard (#4299)

* Make sure selection manager isFocused returns true when menu is opened

a selectable item is only considered isFocused if "manager.isFocused" returns true and the item key matches the tracked key. Usually we update "manager.isFocused" when the collection or something in it is focused but virtual focused collections wont do this

* adding to one other area for correctness

* simplify logic

also moved clearing of setFocusedKey to onOpenChange so it is colocated with its relevant logic

* forgot to remove old code

Author: LFDanLu

packages/@react-stately/combobox/src/useComboBoxState.ts

@@ -102,6 +102,11 @@ export function useComboBoxState<T extends object>(props: ComboBoxStateOptions<T
     if (props.onOpenChange) {
       props.onOpenChange(open, open ? menuOpenTrigger.current : undefined);
     }
+
+    selectionManager.setFocused(open);
+    if (!open) {
+      selectionManager.setFocusedKey(null);
+    }
   };
 
   let triggerState = useMenuTriggerState({...props, onOpenChange, isOpen: undefined, defaultOpen: undefined});
@@ -248,13 +253,6 @@ export function useComboBoxState<T extends object>(props: ComboBoxStateOptions<T
     lastSelectedKeyText.current = selectedItemText;
   });
 
-  useEffect(() => {
-    // Reset focused key when the menu closes
-    if (!triggerState.isOpen) {
-      selectionManager.setFocusedKey(null);
-    }
-  }, [triggerState.isOpen, selectionManager]);
-
   // Revert input value and close menu
   let revert = () => {
     if (allowsCustomValue && selectedKey == null) {