From cc41cd4598a66b40d685672ea23e30c5b1975871 Mon Sep 17 00:00:00 2001
From: Ariver <shanghai3168@gmail.com>
Date: Sat, 30 May 2026 15:39:47 +0800
Subject: [PATCH] Fix quick search focus and hidden overlay routing

---
 Apptag/ApptagApp.swift |  504 +++++++++++++++++++++++++++++++++-----------------------
 1 files changed, 297 insertions(+), 207 deletions(-)

diff --git a/Apptag/ApptagApp.swift b/Apptag/ApptagApp.swift
index e11a479..40f301b 100644
--- a/Apptag/ApptagApp.swift
+++ b/Apptag/ApptagApp.swift
@@ -28,21 +28,14 @@
 
 // MARK: - App Delegate (menubar + overlay window + hotkey)
 
-final class OverlayPanel: NSPanel {
-    override var canBecomeKey: Bool { true }
-    override var canBecomeMain: Bool { true }
-
-    override func constrainFrameRect(_ frameRect: NSRect, to screen: NSScreen?) -> NSRect {
-        frameRect
-    }
-}
-
 final class AppDelegate: NSObject, NSApplicationDelegate, NSMenuDelegate {
     private static let showDockIconKey = "showDockIcon"
     private static let statusItemAutosaveName = AppIdentity.statusItemAutosaveName
     private static let statusItemButtonIdentifier = NSUserInterfaceItemIdentifier("TagLauncherStatusItemButton")
     private static let statusItemAccessibilityLabel = AppIdentity.displayName
     private static let showAppListMenuItemIdentifier = NSUserInterfaceItemIdentifier("TagLauncherShowAppListMenuItem")
+    private static let helpMenuItemIdentifier = NSUserInterfaceItemIdentifier("TagLauncherHelpMenu")
+    private static let downloadHelpMenuItemIdentifier = NSUserInterfaceItemIdentifier("TagLauncherDownloadHelpMenuItem")
     private static let externalActivationNotification = Notification.Name("TagLauncherExternalActivationRequested")
     private static let externalActivationObject = AppIdentity.bundleIdentifier
     private static let launcherOverlayLevel = NSWindow.Level(rawValue: NSWindow.Level.mainMenu.rawValue - 1)
@@ -50,26 +43,104 @@
     private static let overlayTextInputLevel = launcherOverlayLevel
 
     private var statusItem: NSStatusItem?
-    private var overlayWindow: NSWindow?
     private var overlayKeyMonitor: Any?
+    private var quickSearchLocalMouseMonitor: Any?
     private var quickSearchExternalMouseMonitor: Any?
     private var settingsWindow: NSWindow?    // Track Settings window to keep it above overlay
     private var mainHotkeyRef: EventHotKeyRef?
     private var quickSearchHotkeyRef: EventHotKeyRef?
     private var hotkeyEventHandlerInstalled = false
     private var isQuickSearchOpen = false
+    private var quickSearchShouldHideOverlayOnClose = false
+    private var quickSearchOnlyOverlaySession = false
+    private var lastQuickSearchHotkeyAt = Date.distantPast
     private var isModalInteractionActive = false
     private var isInEditMode = false  // Suppress auto-dismiss during editing
     private var isEditingAppNote = false
     private var isConfiguringApplicationMenu = false
     private var lastShowDockIcon: Bool?
     private var statusMenuScreenForNextOverlay: NSScreen?
-    private var overlayGeneration = 0
     private var suppressReopenUntil = Date.distantPast
 
-    private struct OverlayPlacementContext {
-        let screen: NSScreen
-        let frame: NSRect
+    private lazy var overlayController = OverlayWindowController(
+        dependencies: OverlayWindowController.Dependencies(
+            shouldStageAsAccessory: { [weak self] in
+                self?.shouldStageOverlayAsAccessory ?? false
+            },
+            isSettingsVisible: { [weak self] in
+                self?.isSettingsVisible ?? false
+            },
+            settingsWindow: { [weak self] in
+                self?.settingsWindow
+            },
+            canHideOverlay: { [weak self] in
+                self?.isInEditMode == false
+            },
+            currentOverlayLevel: { [weak self] in
+                self?.currentOverlayLevel ?? Self.overlayDefaultLevel
+            },
+            overlayLevel: { [weak self] initialQuickSearchSource in
+                self?.overlayLevel(initialQuickSearchSource: initialQuickSearchSource) ?? Self.overlayDefaultLevel
+            },
+            makeContentView: { [weak self] initialQuickSearchSource in
+                DismissibleHostingView(
+                    rootView: ContentView(
+                        hideOverlay: { [weak self] in
+                            self?.hideOverlay(force: true)
+                        },
+                        initialQuickSearchSource: initialQuickSearchSource
+                    ),
+                    onBackdropTap: { [weak self] in
+                        self?.hideOverlay()
+                    }
+                )
+            },
+            handleOverlayKeyEvent: { [weak self] event in
+                self?.handleOverlayKeyEvent(event) ?? false
+            },
+            installOverlayKeyMonitor: { [weak self] in
+                self?.installOverlayKeyMonitor()
+            },
+            removeOverlayKeyMonitor: { [weak self] in
+                self?.removeOverlayKeyMonitor()
+            },
+            removeQuickSearchMouseMonitor: { [weak self] in
+                self?.removeQuickSearchExternalMouseMonitor()
+                self?.quickSearchShouldHideOverlayOnClose = false
+                self?.quickSearchOnlyOverlaySession = false
+            },
+            detachSettingsWindow: { [weak self] window in
+                self?.detachSettingsWindow(window)
+            },
+            prepareSettingsWindow: { [weak self] window in
+                self?.prepareSettingsWindow(window)
+            },
+            refreshChromeState: { [weak self] activate, avoidSpaceSwitch in
+                self?.refreshLauncherChromeState(activate: activate, avoidSpaceSwitch: avoidSpaceSwitch)
+            },
+            onWillHide: {
+                TagDatabase.flushPendingCategorySchemeBackupBatch()
+            },
+            onDidHide: {
+                NotificationCenter.default.post(name: .tagLauncherOverlayDidHide, object: nil)
+            },
+            onDidShow: {
+                NotificationCenter.default.post(name: .tagLauncherOverlayDidShow, object: nil)
+            }
+        )
+    )
+
+    private var overlayWindow: NSWindow? {
+        overlayController.window
+    }
+
+    private var overlayGeneration: Int {
+        overlayController.generation
+    }
+
+    private var overlayAvoidsSpaceSwitch: Bool {
+        get { overlayController.avoidsSpaceSwitch }
+        set { overlayController.avoidsSpaceSwitch = newValue }
     }
 
     private var isOverlayVisible: Bool {
@@ -258,13 +329,17 @@
         }
     }
 
-    private func refreshLauncherChromeState(activate: Bool = false) {
+    private func refreshLauncherChromeState(activate: Bool = false, avoidSpaceSwitch: Bool = false) {
         let showDock = UserDefaults.standard.bool(forKey: Self.showDockIconKey)
         lastShowDockIcon = showDock
 
-        let desiredPolicy: NSApplication.ActivationPolicy = requiresForegroundOwnership
+        let shouldStayAccessoryForCurrentFullscreenSpace = isOverlayVisible
+            && (avoidSpaceSwitch || overlayAvoidsSpaceSwitch)
+        let desiredPolicy: NSApplication.ActivationPolicy = shouldStayAccessoryForCurrentFullscreenSpace
+            ? .accessory
+            : (requiresForegroundOwnership
             ? .regular
-            : (showDock ? .regular : .accessory)
+            : (showDock ? .regular : .accessory))
         if NSApp.activationPolicy() != desiredPolicy {
             NSApp.setActivationPolicy(desiredPolicy)
         }
@@ -274,7 +349,7 @@
             NSApp.presentationOptions = desiredPresentation
         }
 
-        if activate && requiresForegroundOwnership {
+        if activate && requiresForegroundOwnership && !shouldStayAccessoryForCurrentFullscreenSpace {
             let keyWindow = isSettingsVisible ? settingsWindow : (isOverlayVisible ? overlayWindow : nil)
             claimLauncherForeground(
                 keyWindow: keyWindow,
@@ -474,7 +549,7 @@
     }
 
     func menuWillOpen(_ menu: NSMenu) {
-        statusMenuScreenForNextOverlay = screenContainingCurrentPointer()
+        statusMenuScreenForNextOverlay = overlayController.screenContainingCurrentPointer()
             ?? statusItem?.button?.window?.screen
             ?? NSScreen.main
             ?? NSScreen.screens.first
@@ -585,7 +660,7 @@
             }
 
             self.configureApplicationMenu()
-            self.removeHelpMenu()
+            self.configureHelpMenu()
             if retries > 0 && self.applicationMenuNeedsCleanup() {
                 self.configureApplicationMenuWhenAvailable(retries: retries - 1)
             }
@@ -704,12 +779,12 @@
         }
     }
 
-    private func removeHelpMenu() {
+    private func configureHelpMenu() {
         guard let mainMenu = NSApp.mainMenu else { return }
         NSApp.helpMenu = nil
 
         for item in mainMenu.items.reversed() {
-            let isHelpMenu = item.submenu === NSApp.helpMenu
+            let isHelpMenu = item.identifier == Self.helpMenuItemIdentifier
                 || item.title.localizedCaseInsensitiveContains("help")
                 || item.title == tr("menu.help")
                 || item.submenu?.title.localizedCaseInsensitiveContains("help") == true
@@ -718,6 +793,26 @@
                 mainMenu.removeItem(item)
             }
         }
+
+        let menuItem = NSMenuItem(title: tr("menu.help"), action: nil, keyEquivalent: "")
+        let helpMenu = NSMenu(title: tr("menu.help"))
+        menuItem.identifier = Self.helpMenuItemIdentifier
+        menuItem.submenu = helpMenu
+
+        let downloadItem = NSMenuItem()
+        downloadItem.identifier = Self.downloadHelpMenuItemIdentifier
+        downloadItem.title = tr("help.downloadPDF")
+        downloadItem.action = #selector(openLocalizedHelp(_:))
+        downloadItem.target = self
+        downloadItem.keyEquivalent = ""
+        downloadItem.keyEquivalentModifierMask = []
+        downloadItem.isEnabled = true
+        helpMenu.addItem(downloadItem)
+        mainMenu.addItem(menuItem)
+    }
+
+    @objc private func openLocalizedHelp(_ sender: Any? = nil) {
+        NSWorkspace.shared.open(HelpDocument.currentURL)
     }
 
     private func removeMenuBarItem() {
@@ -741,34 +836,11 @@
     }
 
     private func performToggleOverlay(preferredScreen: NSScreen?) {
-        if overlayWindow?.isVisible == true {
-            hideOverlay(force: true)
-        } else {
-            showOverlay(preferredScreen: preferredScreen)
-        }
+        overlayController.toggle(preferredScreen: preferredScreen)
     }
 
     private func showOrFocusOverlay(preferredScreen: NSScreen? = nil) {
-        if let overlayWindow, overlayWindow.isVisible {
-            moveOverlayToCurrentPlacement(preferredScreen: preferredScreen)
-            overlayWindow.level = currentOverlayLevel
-            refreshLauncherChromeState(activate: true)
-            overlayWindow.makeKeyAndOrderFront(nil)
-            overlayWindow.orderFrontRegardless()
-            if let settingsWindow, settingsWindow.isVisible {
-                prepareSettingsWindow(settingsWindow)
-            }
-            return
-        }
-        showOverlay(preferredScreen: preferredScreen)
-    }
-
-    private func moveOverlayToCurrentPlacement(preferredScreen: NSScreen?) {
-        guard let overlayWindow,
-              let placement = overlayPlacementContextForNextOverlay(preferredScreen: preferredScreen),
-              overlayWindow.frame != placement.frame
-        else { return }
-        overlayWindow.setFrame(placement.frame, display: true)
+        overlayController.showOrFocus(preferredScreen: preferredScreen)
     }
 
     private func showOverlay(
@@ -776,106 +848,80 @@
         preferredScreen: NSScreen? = nil,
         stagedForAllSpaces: Bool = false
     ) {
-        guard let placement = overlayPlacementContextForNextOverlay(preferredScreen: preferredScreen) else { return }
-
-        let shouldStageAsAccessory = shouldStageOverlayAsAccessory
-
-        if !stagedForAllSpaces && shouldStageAsAccessory {
-            // Let the all-spaces panel attach to the pointer's display before the app reclaims focus.
-            if NSApp.activationPolicy() != .accessory {
-                NSApp.setActivationPolicy(.accessory)
-            }
-            if NSApp.isActive {
-                NSApp.deactivate()
-            }
-            DispatchQueue.main.asyncAfter(deadline: .now() + 0.08) { [weak self] in
-                self?.showOverlay(
-                    initialQuickSearchSource: initialQuickSearchSource,
-                    preferredScreen: placement.screen,
-                    stagedForAllSpaces: true
-                )
-            }
-            return
+        if initialQuickSearchSource == QuickSearchOpenSource.globalHidden {
+            isQuickSearchOpen = true
+            quickSearchShouldHideOverlayOnClose = true
+            quickSearchOnlyOverlaySession = true
         }
-
-        if overlayWindow?.isVisible == true {
-            hideOverlay(force: true)
-        } else if let existingWindow = overlayWindow {
-            existingWindow.orderOut(nil)
-            overlayWindow = nil
-        }
-
-        let window = makeOverlayWindow(
-            on: placement.screen,
-            initialQuickSearchSource: initialQuickSearchSource
+        overlayController.show(
+            initialQuickSearchSource: initialQuickSearchSource,
+            preferredScreen: preferredScreen,
+            stagedForAllSpaces: stagedForAllSpaces
         )
-        overlayGeneration &+= 1
-        overlayWindow = window
-        installOverlayKeyMonitor()
-        if shouldStageAsAccessory {
-            if NSApp.isActive {
-                NSApp.deactivate()
-            }
-            if NSApp.activationPolicy() != .accessory {
-                NSApp.setActivationPolicy(.accessory)
-            }
-        }
-
-        let targetLevel = overlayLevel(initialQuickSearchSource: initialQuickSearchSource)
-        window.setFrame(placement.frame, display: true)
-        window.level = targetLevel
-
-        window.makeKeyAndOrderFront(nil)
-        window.orderFrontRegardless()
-
-        let placementFrame = placement.frame
-        let finishForegroundClaim: () -> Void = { [weak self, weak window] in
-            guard let self, let window, self.overlayWindow === window else { return }
-            self.refreshLauncherChromeState(activate: true)
-            if window.frame != placementFrame {
-                window.setFrame(placementFrame, display: true)
-                window.makeKeyAndOrderFront(nil)
-                window.orderFrontRegardless()
-            }
-            if let settingsWindow = self.settingsWindow, settingsWindow.isVisible {
-                self.prepareSettingsWindow(settingsWindow)
-            }
-            NotificationCenter.default.post(name: .tagLauncherOverlayDidShow, object: nil)
-        }
-
-        if !isSettingsVisible {
-            DispatchQueue.main.asyncAfter(deadline: .now() + 0.12, execute: finishForegroundClaim)
-        } else {
-            finishForegroundClaim()
-        }
     }
 
     private func installOverlayKeyMonitor() {
         guard overlayKeyMonitor == nil else { return }
         overlayKeyMonitor = NSEvent.addLocalMonitorForEvents(matching: .keyDown) { [weak self] event in
             guard let self else { return event }
-            guard self.shouldHandleOverlayKeyEvent(event) else { return event }
-            if event.keyCode == 53 { // Escape
-                if self.isQuickSearchOpen {
-                    self.isQuickSearchOpen = false
-                    self.removeQuickSearchExternalMouseMonitor()
-                    self.updateOverlayLevelForTextInput()
-                    NotificationCenter.default.post(name: .tagLauncherQuickSearchDismissRequested, object: nil)
-                    return nil
-                }
-                self.hideOverlay(force: true)
-                return nil
-            }
-            if self.shouldOpenQuickSearch(for: event) {
-                self.requestQuickSearch(source: QuickSearchOpenSource.mainOverlay)
+            if self.handleOverlayKeyEvent(event) {
                 return nil
             }
             return event
         }
     }
 
+    @discardableResult
+    private func handleOverlayKeyEvent(_ event: NSEvent) -> Bool {
+        guard event.type == .keyDown else { return false }
+
+        if event.keyCode == UInt16(kVK_Escape),
+           isQuickSearchOpen || quickSearchOnlyOverlaySession {
+            return handleOverlayEscapeKey()
+        }
+
+        guard shouldHandleOverlayKeyEvent(event) else { return false }
+
+        if event.keyCode == UInt16(kVK_Escape) {
+            return handleOverlayEscapeKey()
+        }
+        if shouldOpenQuickSearch(for: event) {
+            requestQuickSearch(source: QuickSearchOpenSource.mainOverlay)
+            return true
+        }
+        return false
+    }
+
+    private func handleOverlayEscapeKey() -> Bool {
+        if isQuickSearchOpen || quickSearchOnlyOverlaySession {
+            let shouldHideOverlayAfterDismiss = quickSearchShouldHideOverlayOnClose || quickSearchOnlyOverlaySession
+            isQuickSearchOpen = false
+            removeQuickSearchExternalMouseMonitor()
+            updateOverlayLevelForTextInput()
+            NotificationCenter.default.post(name: .tagLauncherQuickSearchDismissRequested, object: nil)
+            if shouldHideOverlayAfterDismiss {
+                quickSearchShouldHideOverlayOnClose = false
+                quickSearchOnlyOverlaySession = false
+                DispatchQueue.main.async { [weak self] in
+                    self?.hideOverlay(force: true)
+                }
+            }
+            return true
+        }
+        guard !isSettingsVisible,
+              !isEditingAppNote,
+              !isModalInteractionActive
+        else { return false }
+        hideOverlay(force: true)
+        return true
+    }
+
     private func requestQuickSearch(source: String) {
         isQuickSearchOpen = true
+        quickSearchShouldHideOverlayOnClose = quickSearchShouldHideOverlayOnClose
+            || source == QuickSearchOpenSource.globalHidden
+        quickSearchOnlyOverlaySession = quickSearchOnlyOverlaySession
+            || source == QuickSearchOpenSource.globalHidden
         promoteOverlayToForegroundInput()
         updateOverlayLevelForTextInput()
         installQuickSearchExternalMouseMonitor()
@@ -918,59 +964,8 @@
         }
     }
 
-    private func makeOverlayWindow(on screen: NSScreen, initialQuickSearchSource: String? = nil) -> NSWindow {
-        let panel = OverlayPanel(
-            contentRect: screen.frame,
-            styleMask: [.borderless, .fullSizeContentView, .nonactivatingPanel],
-            backing: .buffered,
-            defer: false
-        )
-        panel.isFloatingPanel = true
-        panel.hidesOnDeactivate = false
-        // Keep the overlay publishable above fullscreen Spaces without asking
-        // AppKit to migrate an all-spaces window between Spaces.
-        panel.collectionBehavior = [
-            .canJoinAllSpaces,
-            .fullScreenAuxiliary,
-            .stationary,
-            .transient,
-            .ignoresCycle
-        ]
-        panel.isOpaque = false
-        panel.backgroundColor = NSColor.black.withAlphaComponent(0.001)
-        panel.hasShadow = false
-        panel.titlebarAppearsTransparent = true
-        panel.titleVisibility = .hidden
-        panel.isReleasedWhenClosed = false
-        panel.contentView = DismissibleHostingView(
-            rootView: ContentView(
-                hideOverlay: { [weak self] in
-                    self?.hideOverlay(force: true)
-                },
-                initialQuickSearchSource: initialQuickSearchSource
-            ),
-            onBackdropTap: { [weak self] in
-                self?.hideOverlay()
-            }
-        )
-        return panel
-    }
-
     private func hideOverlay(force: Bool = false, discardWindow: Bool = false) {
-        guard force || !isInEditMode else { return }
-        overlayGeneration &+= 1
-        TagDatabase.flushPendingCategorySchemeBackupBatch()
-        if let settingsWindow, settingsWindow.parent == overlayWindow {
-            detachSettingsWindow(settingsWindow)
-        }
-        overlayWindow?.orderOut(nil)
-        removeOverlayKeyMonitor()
-        removeQuickSearchExternalMouseMonitor()
-        refreshLauncherChromeState()
-        NotificationCenter.default.post(name: .tagLauncherOverlayDidHide, object: nil)
-        if discardWindow {
-            overlayWindow = nil
-        }
+        overlayController.hide(force: force, discardWindow: discardWindow)
     }
 
     // MARK: - Global Hotkeys
@@ -1070,11 +1065,30 @@
     }
 
     private func showQuickSearchFromGlobalHotkey() {
+        let now = Date()
+        if now.timeIntervalSince(lastQuickSearchHotkeyAt) < 0.28 {
+            promoteOverlayToForegroundInput()
+            updateOverlayLevelForTextInput()
+            return
+        }
+        lastQuickSearchHotkeyAt = now
+
+        if overlayWindow?.isVisible == true, isQuickSearchOpen || quickSearchOnlyOverlaySession {
+            promoteOverlayToForegroundInput()
+            updateOverlayLevelForTextInput()
+            return
+        }
         if overlayWindow?.isVisible == true {
-            refreshLauncherChromeState(activate: true)
+            refreshLauncherChromeState(
+                activate: !overlayAvoidsSpaceSwitch,
+                avoidSpaceSwitch: overlayAvoidsSpaceSwitch
+            )
             requestQuickSearch(source: QuickSearchOpenSource.globalVisible)
             return
         }
+        isQuickSearchOpen = true
+        quickSearchShouldHideOverlayOnClose = true
+        quickSearchOnlyOverlaySession = true
         showOverlay(initialQuickSearchSource: QuickSearchOpenSource.globalHidden)
     }
 
@@ -1119,6 +1133,19 @@
                 return
             }
 
+            if self.isMenuTrackingWindow(keyWindow) {
+                return
+            }
+
+            if self.isAppOwnedDocumentWindow(keyWindow) {
+                self.prepareSettingsWindow(keyWindow)
+                return
+            }
+
+            if NSApp.windows.contains(keyWindow) {
+                return
+            }
+
             // Settings/Preferences window -> float it above overlay for real-time preview.
             if self.isSettingsWindowCandidate(keyWindow) {
                 self.prepareSettingsWindow(keyWindow)
@@ -1131,6 +1158,7 @@
 
     /// Settings must always appear centered over the current overlay view and float above it.
     private func prepareSettingsWindow(_ window: NSWindow) {
+        dismissQuickSearchIfNeeded()
         let settingsSize = NSSize(width: 880, height: 460)
         window.identifier = NSUserInterfaceItemIdentifier("TagLauncherPreferencesWindow")
         window.minSize = settingsSize
@@ -1154,12 +1182,20 @@
 
         var behavior = window.collectionBehavior
         behavior.remove(.canJoinAllSpaces)
-        behavior.formUnion([.fullScreenAuxiliary, .moveToActiveSpace])
+        if overlayAvoidsSpaceSwitch {
+            behavior.remove(.moveToActiveSpace)
+            behavior.formUnion([.fullScreenAuxiliary, .stationary, .transient, .ignoresCycle])
+        } else {
+            behavior.formUnion([.fullScreenAuxiliary, .moveToActiveSpace])
+        }
         window.collectionBehavior = behavior
         window.makeKeyAndOrderFront(nil)
         window.orderFrontRegardless()
         settingsWindow = window
-        refreshLauncherChromeState(activate: true)
+        refreshLauncherChromeState(
+            activate: !overlayAvoidsSpaceSwitch,
+            avoidSpaceSwitch: overlayAvoidsSpaceSwitch
+        )
     }
 
     private func attachSettingsWindow(_ window: NSWindow, to overlayWindow: NSWindow) {
@@ -1176,12 +1212,28 @@
     private func isSettingsWindowCandidate(_ window: NSWindow) -> Bool {
         if window == settingsWindow { return true }
         if window.identifier?.rawValue == "TagLauncherPreferencesWindow" { return true }
+        if isAppOwnedDocumentWindow(window) { return true }
         guard NSApp.windows.contains(window),
               window != overlayWindow,
               window.isVisible,
               !(window is NSPanel)
         else { return false }
         return settingsWindowTitleCandidates().contains(normalizedWindowTitle(window.title))
+    }
+
+    private func isAppOwnedDocumentWindow(_ window: NSWindow) -> Bool {
+        NSApp.windows.contains(window)
+            && window != overlayWindow
+            && window.isVisible
+            && !(window is NSPanel)
+            && !isMenuTrackingWindow(window)
+            && window.styleMask.contains(.titled)
+    }
+
+    private func isMenuTrackingWindow(_ window: NSWindow) -> Bool {
+        let className = NSStringFromClass(type(of: window))
+        return className.localizedCaseInsensitiveContains("Menu")
+            || className.localizedCaseInsensitiveContains("Popup")
     }
 
     private func settingsWindowTitleCandidates() -> Set<String> {
@@ -1222,24 +1274,7 @@
     }
 
     private func screenUnderMouse() -> NSScreen? {
-        overlayPlacementContextForNextOverlay(preferredScreen: nil)?.screen
-    }
-
-    private func overlayPlacementContextForNextOverlay(preferredScreen: NSScreen?) -> OverlayPlacementContext? {
-        if let screen = preferredScreen {
-            return OverlayPlacementContext(screen: screen, frame: screen.frame)
-        }
-        guard let screen = screenContainingCurrentPointer() ?? NSScreen.main ?? NSScreen.screens.first else {
-            return nil
-        }
-        return OverlayPlacementContext(screen: screen, frame: screen.frame)
-    }
-
-    private func screenContainingCurrentPointer() -> NSScreen? {
-        let mousePoint = NSEvent.mouseLocation
-        return NSScreen.screens.first(where: {
-            NSMouseInRect(mousePoint, $0.frame, false)
-        })
+        overlayController.screenUnderMouse()
     }
 
     /// Clean up settingsWindow reference when the Settings window closes.
@@ -1257,7 +1292,10 @@
             let preferredScreen = self.overlayWindow?.screen
             self.detachSettingsWindow(closingWindow)
             self.settingsWindow = nil
-            self.refreshLauncherChromeState(activate: shouldRefocusOverlay)
+            self.refreshLauncherChromeState(
+                activate: shouldRefocusOverlay && !self.overlayAvoidsSpaceSwitch,
+                avoidSpaceSwitch: self.overlayAvoidsSpaceSwitch
+            )
             guard shouldRefocusOverlay else { return }
             DispatchQueue.main.asyncAfter(deadline: .now() + 0.12) { [weak self] in
                 self?.showOrFocusOverlay(preferredScreen: preferredScreen)
@@ -1293,6 +1331,13 @@
     }
 
     private func promoteOverlayToForegroundInput() {
+        if overlayAvoidsSpaceSwitch {
+            refreshLauncherChromeState(activate: false, avoidSpaceSwitch: true)
+            guard let overlayWindow else { return }
+            overlayWindow.makeKeyAndOrderFront(nil)
+            overlayWindow.orderFrontRegardless()
+            return
+        }
         beginLauncherForegroundOwnership()
         guard let overlayWindow else { return }
         overlayWindow.makeKeyAndOrderFront(nil)
@@ -1306,7 +1351,13 @@
             queue: .main
         ) { [weak self] notification in
             guard let self else { return }
-            self.isQuickSearchOpen = (notification.userInfo?["active"] as? Bool) ?? false
+            let active = (notification.userInfo?["active"] as? Bool) ?? false
+            let hideOverlayOnClose = (notification.userInfo?["hideOverlayOnClose"] as? Bool) ?? false
+            self.isQuickSearchOpen = active
+            if active, hideOverlayOnClose {
+                self.quickSearchShouldHideOverlayOnClose = true
+                self.quickSearchOnlyOverlaySession = true
+            }
             if self.isQuickSearchOpen {
                 self.promoteOverlayToForegroundInput()
             }
@@ -1315,6 +1366,11 @@
                 self.installQuickSearchExternalMouseMonitor()
             } else {
                 self.removeQuickSearchExternalMouseMonitor()
+                if hideOverlayOnClose || self.quickSearchShouldHideOverlayOnClose {
+                    self.quickSearchShouldHideOverlayOnClose = false
+                    self.quickSearchOnlyOverlaySession = false
+                    self.hideOverlay(force: true)
+                }
             }
         }
 
@@ -1336,6 +1392,7 @@
     }
 
     private func installQuickSearchExternalMouseMonitor() {
+        installQuickSearchLocalMouseMonitor()
         guard quickSearchExternalMouseMonitor == nil else { return }
         quickSearchExternalMouseMonitor = NSEvent.addGlobalMonitorForEvents(
             matching: [.leftMouseDown, .rightMouseDown, .otherMouseDown]
@@ -1347,7 +1404,25 @@
         }
     }
 
+    private func installQuickSearchLocalMouseMonitor() {
+        guard quickSearchLocalMouseMonitor == nil else { return }
+        quickSearchLocalMouseMonitor = NSEvent.addLocalMonitorForEvents(
+            matching: [.leftMouseDown, .rightMouseDown, .otherMouseDown]
+        ) { [weak self] event in
+            guard let self, self.isQuickSearchOpen else { return event }
+            if event.window === self.overlayWindow {
+                NotificationCenter.default.post(name: .tagLauncherQuickSearchDismissRequested, object: nil)
+                return nil
+            }
+            return event
+        }
+    }
+
     private func removeQuickSearchExternalMouseMonitor() {
+        if let quickSearchLocalMouseMonitor {
+            NSEvent.removeMonitor(quickSearchLocalMouseMonitor)
+            self.quickSearchLocalMouseMonitor = nil
+        }
         if let quickSearchExternalMouseMonitor {
             NSEvent.removeMonitor(quickSearchExternalMouseMonitor)
             self.quickSearchExternalMouseMonitor = nil
@@ -1389,8 +1464,13 @@
     }
 
     @objc private func openPreferences(_ sender: Any? = nil) {
+        dismissQuickSearchIfNeeded()
         TagDatabase.flushPendingCategorySchemeBackupBatch()
-        beginLauncherForegroundOwnership()
+        if overlayAvoidsSpaceSwitch {
+            refreshLauncherChromeState(activate: false, avoidSpaceSwitch: true)
+        } else {
+            beginLauncherForegroundOwnership()
+        }
         // Don't hide overlay — keep it visible for real-time setting preview.
         if let overlayWindow, overlayWindow.isVisible {
             overlayWindow.makeKeyAndOrderFront(nil)
@@ -1416,6 +1496,16 @@
         prepareSettingsWindow(window)
     }
 
+    private func dismissQuickSearchIfNeeded() {
+        guard isQuickSearchOpen else { return }
+        isQuickSearchOpen = false
+        removeQuickSearchExternalMouseMonitor()
+        updateOverlayLevelForTextInput()
+        quickSearchShouldHideOverlayOnClose = false
+        quickSearchOnlyOverlaySession = false
+        NotificationCenter.default.post(name: .tagLauncherQuickSearchDismissRequested, object: nil)
+    }
+
     @objc private func switchLanguage(_ sender: NSMenuItem) {
         guard let code = sender.representedObject as? String else { return }
         L10n.switchTo(code)

--
Gitblit v1.9.3