From da71eb5bccbdbd22da7cd74b6ddcec031f069482 Mon Sep 17 00:00:00 2001
From: Ariver <shanghai3168@gmail.com>
Date: Thu, 28 May 2026 15:59:53 +0800
Subject: [PATCH] Stabilize empty-drop tag removal confirmation

---
 Apptag/ApptagApp.swift |  216 ++++++++++++++++++++++++++++++++++++++++++++++++++---
 1 files changed, 201 insertions(+), 15 deletions(-)

diff --git a/Apptag/ApptagApp.swift b/Apptag/ApptagApp.swift
index e11a479..81eca50 100644
--- a/Apptag/ApptagApp.swift
+++ b/Apptag/ApptagApp.swift
@@ -43,6 +43,8 @@
     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)
@@ -65,10 +67,16 @@
     private var lastShowDockIcon: Bool?
     private var statusMenuScreenForNextOverlay: NSScreen?
     private var overlayGeneration = 0
+    private var overlayAvoidsSpaceSwitch = false
     private var suppressReopenUntil = Date.distantPast
 
     private struct OverlayPlacementContext {
         let screen: NSScreen
+        let frame: NSRect
+    }
+
+    private struct ForeignWindowFrame {
+        let owner: String
         let frame: NSRect
     }
 
@@ -258,13 +266,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 +286,7 @@
             NSApp.presentationOptions = desiredPresentation
         }
 
-        if activate && requiresForegroundOwnership {
+        if activate && requiresForegroundOwnership && !shouldStayAccessoryForCurrentFullscreenSpace {
             let keyWindow = isSettingsVisible ? settingsWindow : (isOverlayVisible ? overlayWindow : nil)
             claimLauncherForeground(
                 keyWindow: keyWindow,
@@ -585,7 +597,7 @@
             }
 
             self.configureApplicationMenu()
-            self.removeHelpMenu()
+            self.configureHelpMenu()
             if retries > 0 && self.applicationMenuNeedsCleanup() {
                 self.configureApplicationMenuWhenAvailable(retries: retries - 1)
             }
@@ -704,12 +716,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 +730,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() {
@@ -749,10 +781,13 @@
     }
 
     private func showOrFocusOverlay(preferredScreen: NSScreen? = nil) {
+        let placement = overlayPlacementContextForNextOverlay(preferredScreen: preferredScreen)
+        let shouldAvoidSpaceSwitch = placement.map { hasFullscreenWindowOnScreen($0.screen) } ?? false
         if let overlayWindow, overlayWindow.isVisible {
+            overlayAvoidsSpaceSwitch = shouldAvoidSpaceSwitch
             moveOverlayToCurrentPlacement(preferredScreen: preferredScreen)
             overlayWindow.level = currentOverlayLevel
-            refreshLauncherChromeState(activate: true)
+            refreshLauncherChromeState(activate: !shouldAvoidSpaceSwitch, avoidSpaceSwitch: shouldAvoidSpaceSwitch)
             overlayWindow.makeKeyAndOrderFront(nil)
             overlayWindow.orderFrontRegardless()
             if let settingsWindow, settingsWindow.isVisible {
@@ -778,7 +813,8 @@
     ) {
         guard let placement = overlayPlacementContextForNextOverlay(preferredScreen: preferredScreen) else { return }
 
-        let shouldStageAsAccessory = shouldStageOverlayAsAccessory
+        let shouldAvoidSpaceSwitch = hasFullscreenWindowOnScreen(placement.screen)
+        let shouldStageAsAccessory = shouldStageOverlayAsAccessory || shouldAvoidSpaceSwitch
 
         if !stagedForAllSpaces && shouldStageAsAccessory {
             // Let the all-spaces panel attach to the pointer's display before the app reclaims focus.
@@ -804,6 +840,7 @@
             existingWindow.orderOut(nil)
             overlayWindow = nil
         }
+        overlayAvoidsSpaceSwitch = shouldAvoidSpaceSwitch
 
         let window = makeOverlayWindow(
             on: placement.screen,
@@ -831,7 +868,7 @@
         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)
+            self.refreshLauncherChromeState(activate: !shouldAvoidSpaceSwitch, avoidSpaceSwitch: shouldAvoidSpaceSwitch)
             if window.frame != placementFrame {
                 window.setFrame(placementFrame, display: true)
                 window.makeKeyAndOrderFront(nil)
@@ -848,6 +885,100 @@
         } else {
             finishForegroundClaim()
         }
+    }
+
+    private func hasFullscreenWindowOnScreen(_ screen: NSScreen) -> Bool {
+        let screenFrame = screen.frame
+        let windowFrames = foreignLayerZeroWindows(on: screenFrame)
+
+        if windowFrames.contains(where: { isSingleFullscreenWindow($0.frame, on: screenFrame) }) {
+            return true
+        }
+
+        return hasSplitViewFullscreenWindows(windowFrames, on: screenFrame)
+    }
+
+    private func foreignLayerZeroWindows(on screenFrame: NSRect) -> [ForeignWindowFrame] {
+        let windows = CGWindowListCopyWindowInfo([.optionOnScreenOnly, .excludeDesktopElements], kCGNullWindowID) as? [[String: Any]] ?? []
+        return windows.compactMap { info in
+            guard let owner = info[kCGWindowOwnerName as String] as? String,
+                  owner != AppIdentity.displayName,
+                  owner != "Window Server",
+                  owner != "Dock",
+                  owner != "loginwindow",
+                  let layer = info[kCGWindowLayer as String] as? Int,
+                  layer == 0,
+                  let bounds = info[kCGWindowBounds as String] as? NSDictionary
+            else { return nil }
+
+            let frame = NSRect(
+                x: cgWindowDimension(bounds, "X"),
+                y: cgWindowDimension(bounds, "Y"),
+                width: cgWindowDimension(bounds, "Width"),
+                height: cgWindowDimension(bounds, "Height")
+            )
+            guard frame.intersects(screenFrame) else { return nil }
+            return ForeignWindowFrame(owner: owner, frame: frame)
+        }
+    }
+
+    private func isSingleFullscreenWindow(_ windowFrame: NSRect, on screenFrame: NSRect) -> Bool {
+        let widthMatches = abs(windowFrame.width - screenFrame.width) <= 12
+        let heightMatches = windowFrame.height >= screenFrame.height * 0.88
+        let horizontallyAligned = abs(windowFrame.midX - screenFrame.midX) <= 12
+        let verticallyAligned = abs(windowFrame.maxY - screenFrame.maxY) <= 32
+        return widthMatches && heightMatches && horizontallyAligned && verticallyAligned
+    }
+
+    private func hasSplitViewFullscreenWindows(_ windows: [ForeignWindowFrame], on screenFrame: NSRect) -> Bool {
+        let clippedWindows = windows.map {
+            ForeignWindowFrame(owner: $0.owner, frame: $0.frame.intersection(screenFrame))
+        }
+        let tallWindows = clippedWindows
+            .filter { window in
+                let frame = window.frame
+                return frame.height >= screenFrame.height * 0.86
+                    && frame.width >= screenFrame.width * 0.20
+                    && frame.width <= screenFrame.width * 0.86
+                    && abs(frame.maxY - screenFrame.maxY) <= 32
+            }
+            .sorted { $0.frame.minX < $1.frame.minX }
+
+        guard tallWindows.count >= 2 else { return false }
+
+        for startIndex in tallWindows.indices {
+            var union = tallWindows[startIndex].frame
+            var lastMaxX = union.maxX
+
+            for window in tallWindows.dropFirst(startIndex + 1) {
+                let gap = window.frame.minX - lastMaxX
+                if gap < -32 || gap > 48 {
+                    break
+                }
+                union = union.union(window.frame)
+                lastMaxX = max(lastMaxX, window.frame.maxX)
+
+                let touchesLeft = abs(union.minX - screenFrame.minX) <= 32
+                let touchesRight = abs(union.maxX - screenFrame.maxX) <= 32
+                let coversWidth = union.width >= screenFrame.width * 0.92
+                let coversHeight = union.height >= screenFrame.height * 0.86
+                if touchesLeft && touchesRight && coversWidth && coversHeight {
+                    return true
+                }
+            }
+        }
+
+        return false
+    }
+
+    private func cgWindowDimension(_ bounds: NSDictionary, _ key: String) -> CGFloat {
+        if let value = bounds[key] as? CGFloat {
+            return value
+        }
+        if let value = bounds[key] as? NSNumber {
+            return CGFloat(truncating: value)
+        }
+        return 0
     }
 
     private func installOverlayKeyMonitor() {
@@ -966,6 +1097,7 @@
         overlayWindow?.orderOut(nil)
         removeOverlayKeyMonitor()
         removeQuickSearchExternalMouseMonitor()
+        overlayAvoidsSpaceSwitch = false
         refreshLauncherChromeState()
         NotificationCenter.default.post(name: .tagLauncherOverlayDidHide, object: nil)
         if discardWindow {
@@ -1071,7 +1203,10 @@
 
     private func showQuickSearchFromGlobalHotkey() {
         if overlayWindow?.isVisible == true {
-            refreshLauncherChromeState(activate: true)
+            refreshLauncherChromeState(
+                activate: !overlayAvoidsSpaceSwitch,
+                avoidSpaceSwitch: overlayAvoidsSpaceSwitch
+            )
             requestQuickSearch(source: QuickSearchOpenSource.globalVisible)
             return
         }
@@ -1119,6 +1254,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)
@@ -1154,12 +1302,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 +1332,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> {
@@ -1257,7 +1429,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 +1468,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)
@@ -1390,7 +1572,11 @@
 
     @objc private func openPreferences(_ sender: Any? = nil) {
         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)

--
Gitblit v1.9.3