From 3d8d37306e39db5c8eb41fafd95a13112d573f62 Mon Sep 17 00:00:00 2001
From: Ariver <shanghai3168@gmail.com>
Date: Thu, 11 Jun 2026 22:24:39 +0800
Subject: [PATCH] Center filtered Waterfall columns

---
 C1.source/Sources/Aligner/QuickSwitchSessionController.swift |  179 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 176 insertions(+), 3 deletions(-)

diff --git a/C1.source/Sources/Aligner/QuickSwitchSessionController.swift b/C1.source/Sources/Aligner/QuickSwitchSessionController.swift
index 5a36765..03e9501 100644
--- a/C1.source/Sources/Aligner/QuickSwitchSessionController.swift
+++ b/C1.source/Sources/Aligner/QuickSwitchSessionController.swift
@@ -15,6 +15,7 @@
     private let snapshotLoader: any QuickSwitchSnapshotLoading
     private let screenshotProvider: any ScreenshotProviderProtocol
     private let windowActivationService: any WindowActivationServiceProtocol
+    private let spaceActivationService: any SpaceActivationServiceProtocol
     private let windowCloseService: any WindowCloseServiceProtocol
     private let systemCriticalWindowDetector: any SystemCriticalWindowDetecting
     private let disableScreenshotRefresh: Bool
@@ -35,6 +36,8 @@
     private var lockedSpaceFilterID: UInt64?
     private var selectionBeforeSpaceFilter: QuickSwitchSelection?
     private var lastSpaceFilterAction: String?
+    private var pendingLockedSpaceUnlockSpaceID: UInt64?
+    private var pendingLockedSpaceUnlockTask: Task<Void, Never>?
     private var lastSnapshotError: String?
     private var snapshotTask: Task<Void, Never>?
     private var screenshotTask: Task<Void, Never>?
@@ -52,6 +55,11 @@
     private var lastActivationWindowID: UInt32?
     private var lastActivationResult: WindowActivationResult?
     private var lastActivationError: String?
+    private var lastSpaceActivationSpaceID: UInt64?
+    private var lastSpaceActivationDidRequestFocus: Bool?
+    private var lastSpaceActivationDisplayIdentifier: String?
+    private var lastSpaceActivationPreviousCurrentSpaceID: UInt64?
+    private var lastSpaceActivationError: String?
     private var lastCloseTargetKind: String?
     private var lastCloseAppGroupIndex: Int?
     private var lastCloseAppName: String?
@@ -106,6 +114,7 @@
         windowActivationService: any WindowActivationServiceProtocol = CGWindowAXWindowService(
             spaceIDsByWindowIDProvider: { _ in [:] }
         ),
+        spaceActivationService: any SpaceActivationServiceProtocol = PrivateSpaceActivationService(),
         windowCloseService: any WindowCloseServiceProtocol = CGWindowAXWindowService(
             spaceIDsByWindowIDProvider: { _ in [:] }
         ),
@@ -122,6 +131,7 @@
         self.snapshotLoader = snapshotLoader
         self.screenshotProvider = screenshotProvider
         self.windowActivationService = windowActivationService
+        self.spaceActivationService = spaceActivationService
         self.windowCloseService = windowCloseService
         self.systemCriticalWindowDetector = systemCriticalWindowDetector
         self.disableScreenshotRefresh = disableScreenshotRefresh
@@ -155,6 +165,11 @@
         lastActivationWindowID = nil
         lastActivationResult = nil
         lastActivationError = nil
+        lastSpaceActivationSpaceID = nil
+        lastSpaceActivationDidRequestFocus = nil
+        lastSpaceActivationDisplayIdentifier = nil
+        lastSpaceActivationPreviousCurrentSpaceID = nil
+        lastSpaceActivationError = nil
         lastCloseTargetKind = nil
         lastCloseAppGroupIndex = nil
         lastCloseAppName = nil
@@ -163,6 +178,7 @@
         lastCloseError = nil
         pendingCloseVerification = nil
         suppressedCloseTargets = []
+        cancelPendingLockedSpaceUnlock()
         lockedSpaceFilterID = nil
         selectionBeforeSpaceFilter = nil
         lastSpaceFilterAction = nil
@@ -187,9 +203,13 @@
         view.onCommitSelection = { [weak self] selection, source in
             self?.commitSelection(selection, source: source)
         }
-        view.onSpaceLaneClick = { [weak self] spaceID in
-            self?.handleSpaceLaneClick(spaceID)
+        view.onSpaceLaneClick = { [weak self] spaceID, clickCount in
+            self?.handleSpaceLaneClick(spaceID, clickCount: clickCount)
         }
+        view.onBackgroundClick = { [weak self] in
+            self?.handleBackgroundClick() ?? false
+        }
+        view.setSuppressEventInput(!debugMouseSequence.isEmpty || !debugKeySequence.isEmpty)
         view.onRequestClose = { [weak self] request in
             self?.requestClose(request)
         }
@@ -228,6 +248,7 @@
         snapshotTask = nil
         screenshotTask?.cancel()
         screenshotTask = nil
+        cancelPendingLockedSpaceUnlock()
         pendingCloseVerification = nil
         view.clearCloseFeedback()
         invalidateSystemCriticalMonitoring()
@@ -288,6 +309,11 @@
             "lastActivationWindowID": lastActivationWindowID ?? NSNull(),
             "lastActivationResult": lastActivationResult.map(Self.activationResultString) ?? NSNull(),
             "lastActivationError": lastActivationError ?? NSNull(),
+            "lastSpaceActivationSpaceID": lastSpaceActivationSpaceID ?? NSNull(),
+            "lastSpaceActivationDidRequestFocus": lastSpaceActivationDidRequestFocus ?? NSNull(),
+            "lastSpaceActivationDisplayIdentifier": lastSpaceActivationDisplayIdentifier ?? NSNull(),
+            "lastSpaceActivationPreviousCurrentSpaceID": lastSpaceActivationPreviousCurrentSpaceID ?? NSNull(),
+            "lastSpaceActivationError": lastSpaceActivationError ?? NSNull(),
             "lastCloseTargetKind": lastCloseTargetKind ?? NSNull(),
             "lastCloseAppGroupIndex": lastCloseAppGroupIndex ?? NSNull(),
             "lastCloseAppName": lastCloseAppName ?? NSNull(),
@@ -534,7 +560,31 @@
         onSnapshotUpdated?()
     }
 
-    private func handleSpaceLaneClick(_ spaceID: UInt64) {
+    private func handleSpaceLaneClick(_ spaceID: UInt64, clickCount: Int) {
+        if clickCount >= 2 {
+            if pendingLockedSpaceUnlockSpaceID == spaceID {
+                cancelPendingLockedSpaceUnlock()
+                activateLockedSpace(spaceID)
+            } else {
+                DevelopmentDiagnostics.log("quickSwitch.spaceFilter.doubleClick.ignored", [
+                    "spaceID": spaceID,
+                    "lockedSpaceID": lockedSpaceFilterID ?? NSNull(),
+                    "pendingUnlockSpaceID": pendingLockedSpaceUnlockSpaceID ?? NSNull()
+                ])
+            }
+            return
+        }
+
+        cancelPendingLockedSpaceUnlock()
+        if lockedSpaceFilterID == spaceID {
+            scheduleLockedSpaceUnlock(spaceID)
+            return
+        }
+
+        applySpaceLaneSingleClick(spaceID)
+    }
+
+    private func applySpaceLaneSingleClick(_ spaceID: UInt64) {
         guard let sourceViewModel else {
             lastSpaceFilterAction = "blockedNoSourceViewModel"
             DevelopmentDiagnostics.log("quickSwitch.spaceFilter.click.blocked", [
@@ -605,6 +655,129 @@
         onSnapshotUpdated?()
     }
 
+    @discardableResult
+    private func handleBackgroundClick() -> Bool {
+        guard lockedSpaceFilterID != nil else { return false }
+
+        cancelPendingLockedSpaceUnlock()
+        let unlocked = unlockSpaceFilter(action: "unlockBackground")
+        DevelopmentDiagnostics.log("quickSwitch.spaceFilter.backgroundUnlock", [
+            "unlocked": unlocked
+        ])
+        return unlocked
+    }
+
+    private func scheduleLockedSpaceUnlock(_ spaceID: UInt64) {
+        pendingLockedSpaceUnlockSpaceID = spaceID
+        let delay = max(0.12, NSEvent.doubleClickInterval + 0.03)
+        DevelopmentDiagnostics.log("quickSwitch.spaceFilter.unlock.pending", [
+            "spaceID": spaceID,
+            "delay": delay
+        ])
+
+        pendingLockedSpaceUnlockTask = Task { @MainActor [weak self] in
+            try? await Task.sleep(nanoseconds: UInt64(delay * 1_000_000_000))
+            guard !Task.isCancelled,
+                  let self,
+                  self.pendingLockedSpaceUnlockSpaceID == spaceID
+            else {
+                return
+            }
+
+            self.pendingLockedSpaceUnlockSpaceID = nil
+            self.pendingLockedSpaceUnlockTask = nil
+            self.applySpaceLaneSingleClick(spaceID)
+        }
+    }
+
+    private func cancelPendingLockedSpaceUnlock() {
+        pendingLockedSpaceUnlockTask?.cancel()
+        pendingLockedSpaceUnlockTask = nil
+        pendingLockedSpaceUnlockSpaceID = nil
+    }
+
+    @discardableResult
+    private func unlockSpaceFilter(action: String) -> Bool {
+        guard let sourceViewModel, lockedSpaceFilterID != nil else {
+            return false
+        }
+
+        lockedSpaceFilterID = nil
+        let preferredSelection = selectionBeforeSpaceFilter
+        selectionBeforeSpaceFilter = nil
+        lastSpaceFilterAction = action
+
+        let visibleViewModel = projectedViewModel(
+            from: sourceViewModel,
+            preferredSelection: preferredSelection
+        )
+        currentViewModel = visibleViewModel
+        view.applyProjected(viewModel: visibleViewModel)
+
+        if disableScreenshotRefresh {
+            view.markScreenshotsNotRequested(for: visibleViewModel, reason: "disabledByLaunchOption")
+            screenshotTask?.cancel()
+            screenshotTask = nil
+        } else {
+            scheduleScreenshotRefresh(for: visibleViewModel, generation: sessionGeneration)
+        }
+
+        onSnapshotUpdated?()
+        return true
+    }
+
+    private func activateLockedSpace(_ spaceID: UInt64) {
+        guard lockedSpaceFilterID == spaceID else {
+            DevelopmentDiagnostics.log("quickSwitch.spaceFilter.activateSpace.ignored", [
+                "spaceID": spaceID,
+                "lockedSpaceID": lockedSpaceFilterID ?? NSNull()
+            ])
+            return
+        }
+
+        guard let sourceViewModel else {
+            lastSpaceFilterAction = "blockedNoSourceViewModel"
+            DevelopmentDiagnostics.log("quickSwitch.spaceFilter.activateSpace.blocked", [
+                "spaceID": spaceID,
+                "reason": "noSourceViewModel"
+            ])
+            onSnapshotUpdated?()
+            return
+        }
+
+        lockedSpaceFilterID = nil
+        selectionBeforeSpaceFilter = nil
+        lastSpaceFilterAction = "activateSpace"
+        lastSpaceActivationSpaceID = spaceID
+        lastSpaceActivationDidRequestFocus = nil
+        lastSpaceActivationDisplayIdentifier = nil
+        lastSpaceActivationPreviousCurrentSpaceID = nil
+        lastSpaceActivationError = nil
+
+        let visibleViewModel = projectedViewModel(from: sourceViewModel)
+        currentViewModel = visibleViewModel
+        view.applyProjected(viewModel: visibleViewModel)
+
+        DevelopmentDiagnostics.log("quickSwitch.spaceFilter.activateSpace.start", [
+            "spaceID": spaceID
+        ])
+        hide(reason: .userClosed)
+        let outcome = spaceActivationService.activate(spaceID: spaceID)
+        lastSpaceActivationSpaceID = outcome.targetSpaceID
+        lastSpaceActivationDidRequestFocus = outcome.didRequestFocus
+        lastSpaceActivationDisplayIdentifier = outcome.displayIdentifier
+        lastSpaceActivationPreviousCurrentSpaceID = outcome.previousCurrentSpaceID
+        lastSpaceActivationError = outcome.error
+        DevelopmentDiagnostics.log("quickSwitch.spaceFilter.activateSpace.result", [
+            "spaceID": outcome.targetSpaceID,
+            "displayIdentifier": outcome.displayIdentifier ?? NSNull(),
+            "previousCurrentSpaceID": outcome.previousCurrentSpaceID ?? NSNull(),
+            "didRequestFocus": outcome.didRequestFocus,
+            "error": outcome.error ?? NSNull()
+        ])
+        onSnapshotUpdated?()
+    }
+
     private func requestClose(_ request: QuickSwitchCloseRequest) {
         DevelopmentDiagnostics.log("quickSwitch.close.request.start", [
             "targetKind": request.kindDescription,

--
Gitblit v1.9.3