Fixed a focus issue

Author: 0x5bfa

src/Files.App.Controls/Omnibar/Omnibar.cs

@@ -30,6 +30,7 @@ public partial class Omnibar : Control
 		private Popup? _autoSuggestPopup;
 		private Border? _autoSuggestBoxBorder;
 		private bool _isFocused;
+		private bool _stillHasFocus;
 
 		public Omnibar()
 		{
@@ -87,6 +88,7 @@ protected override void OnApplyTemplate()
 
 			GotFocus += Omnibar_GotFocus;
 			LostFocus += Omnibar_LostFocus;
+			LosingFocus += Omnibar_LosingFocus;
 
 			UpdateVisualStates();
 
@@ -147,14 +149,25 @@ private void Omnibar_GotFocus(object sender, RoutedEventArgs e)
 			UpdateVisualStates();
 		}
 
-		private void Omnibar_LostFocus(object sender, RoutedEventArgs e)
+		private void Omnibar_LosingFocus(UIElement sender, LosingFocusEventArgs args)
 		{
 			// Ignore when user clicks on the TextBox or the button area of an OmnibarMode, Omnibar still has focus anyway
-			if (e.OriginalSource.GetType() is not { } originalSourceType ||
-				originalSourceType == typeof(TextBox) ||
-				originalSourceType == typeof(OmnibarMode) ||
-				originalSourceType == typeof(Omnibar))
+			if (args.NewFocusedElement?.GetType() is not { } focusedType ||
+				focusedType == typeof(TextBox) ||
+				focusedType == typeof(OmnibarMode) ||
+				focusedType == typeof(Omnibar))
+			{
+				_stillHasFocus = true;
+			}
+		}
+
+		private void Omnibar_LostFocus(object sender, RoutedEventArgs e)
+		{
+			if (_stillHasFocus)
+			{
+				_stillHasFocus = false;
 				return;
+			}
 
 			_isFocused = false;
 			UpdateVisualStates();