From b6e11990cc9bd3c838ad6cdd9e53caafff0fd7cb Mon Sep 17 00:00:00 2001
From: Ariver <shanghai3168@gmail.com>
Date: Thu, 18 Jun 2026 23:24:21 +0800
Subject: [PATCH] Add vertical App Shelf keyboard focus

---
 C1.source/Sources/Aligner/QuickSwitchRootView.swift |  309 +++++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 297 insertions(+), 12 deletions(-)

diff --git a/C1.source/Sources/Aligner/QuickSwitchRootView.swift b/C1.source/Sources/Aligner/QuickSwitchRootView.swift
index 4797b32..9fb63d3 100644
--- a/C1.source/Sources/Aligner/QuickSwitchRootView.swift
+++ b/C1.source/Sources/Aligner/QuickSwitchRootView.swift
@@ -10,6 +10,8 @@
 
 @MainActor
 final class QuickSwitchRootView: NSView {
+    private static let appShelfIndexSymbols = Array("1234567890ABCDEFGHIJKLMNOPQRSTUVWXYZ").map(String.init)
+
     private let backdropBlurView = NSVisualEffectView()
     private let layerHostView = NSView()
     private let backgroundLayer = CAGradientLayer()
@@ -67,8 +69,14 @@
     private var horizontalMasonryScrollOffset: CGFloat = 0
     private var currentSelection: QuickSwitchSelection?
     private var columnSelectionHistory: [Int: Int] = [:]
+    private var keyboardFocusedAppGroupIndex: Int?
     private var keyboardCommandsApplied: [String] = []
     private var lastKeyboardCommand: String?
+    private var lastKeyboardAppIndexCommand: String?
+    private var lastBoundaryBounceAxis: String?
+    private var lastBoundaryBounceDirection: String?
+    private var boundaryBounceCount = 0
+    private var selectionChangedByLastCommand: Bool?
     private var tabIgnoredCount = 0
     private var lastCommittedSelection: QuickSwitchSelection?
     private var lastCommitSource: QuickSwitchCommitSource?
@@ -218,6 +226,7 @@
     private enum HoverTargetSource: String {
         case event
         case debug
+        case keyboard
     }
 
     private struct SpaceLaneSegmentLayers {
@@ -447,6 +456,9 @@
         case TriggerKeyCode.tab:
             performKeyboardCommand("tab")
         default:
+            if handleAppShelfIndexKey(event) {
+                return
+            }
             super.keyDown(with: event)
         }
     }
@@ -799,6 +811,10 @@
             columnSelectionHistory = columnSelectionHistory.filter { appGroupIndex, _ in
                 viewModel?.waterfallColumns.contains { $0.appGroupIndex == appGroupIndex } == true
             }
+            if let keyboardFocusedAppGroupIndex,
+               viewModel?.waterfallColumns.contains(where: { $0.appGroupIndex == keyboardFocusedAppGroupIndex }) != true {
+                self.keyboardFocusedAppGroupIndex = nil
+            }
             if let currentSelection {
                 columnSelectionHistory[currentSelection.appGroupIndex] = currentSelection.windowIndex
             }
@@ -813,8 +829,14 @@
             screenshotSkippedReasonsByWindowID = [:]
             currentSelection = viewModel?.initialSelection
             columnSelectionHistory = [:]
+            keyboardFocusedAppGroupIndex = nil
             keyboardCommandsApplied = []
             lastKeyboardCommand = nil
+            lastKeyboardAppIndexCommand = nil
+            lastBoundaryBounceAxis = nil
+            lastBoundaryBounceDirection = nil
+            boundaryBounceCount = 0
+            selectionChangedByLastCommand = nil
             tabIgnoredCount = 0
             lastCommittedSelection = nil
             lastCommitSource = nil
@@ -1233,8 +1255,14 @@
             "selectedAppGroupIndex": effectiveSelection?.appGroupIndex ?? NSNull(),
             "selectedWindowIndex": effectiveSelection?.windowIndex ?? NSNull(),
             "selectedWindowID": effectiveSelection?.windowID ?? NSNull(),
+            "keyboardFocusedAppGroupIndex": keyboardFocusedAppGroupIndex ?? NSNull(),
             "keyboardCommandsApplied": keyboardCommandsApplied,
             "lastKeyboardCommand": lastKeyboardCommand ?? NSNull(),
+            "lastKeyboardAppIndexCommand": lastKeyboardAppIndexCommand ?? NSNull(),
+            "lastBoundaryBounceAxis": lastBoundaryBounceAxis ?? NSNull(),
+            "lastBoundaryBounceDirection": lastBoundaryBounceDirection ?? NSNull(),
+            "boundaryBounceCount": boundaryBounceCount,
+            "selectionChangedByLastCommand": selectionChangedByLastCommand ?? NSNull(),
             "tabIgnoredCount": tabIgnoredCount,
             "lastCommittedAppGroupIndex": lastCommittedSelection?.appGroupIndex ?? NSNull(),
             "lastCommittedWindowIndex": lastCommittedSelection?.windowIndex ?? NSNull(),
@@ -1307,21 +1335,47 @@
             .lowercased()
         guard !command.isEmpty else { return }
 
+        if let indexSymbol = appShelfIndexSymbol(fromKeyboardCommand: command) {
+            let recordedCommand = "app:\(indexSymbol)"
+            recordKeyboardCommand(recordedCommand)
+            if !focusAppShelfItemForKeyboard(indexSymbol: indexSymbol, commandLabel: recordedCommand) {
+                selectionChangedByLastCommand = false
+            }
+            return
+        }
+
         switch command {
         case "left":
             recordKeyboardCommand("left")
-            moveSelection(.left)
+            if isHorizontalMasonryMode {
+                moveSelection(.left)
+            } else if !moveKeyboardAppFocus(cycleForward: false) {
+                selectionChangedByLastCommand = false
+            }
         case "right":
             recordKeyboardCommand("right")
-            moveSelection(.right)
+            if isHorizontalMasonryMode {
+                moveSelection(.right)
+            } else if !moveKeyboardAppFocus(cycleForward: true) {
+                selectionChangedByLastCommand = false
+            }
         case "up":
             recordKeyboardCommand("up")
-            moveSelection(.up)
+            if isHorizontalMasonryMode {
+                moveSelection(.up)
+            } else {
+                moveVerticalSelectionWithinFocusedColumn(.up)
+            }
         case "down":
             recordKeyboardCommand("down")
-            moveSelection(.down)
+            if isHorizontalMasonryMode {
+                moveSelection(.down)
+            } else {
+                moveVerticalSelectionWithinFocusedColumn(.down)
+            }
         case "enter", "return", "commit":
             recordKeyboardCommand("enter")
+            selectionChangedByLastCommand = false
             commitCurrentSelection()
         case "tab":
             if isHorizontalMasonryMode {
@@ -1330,15 +1384,241 @@
             } else {
                 tabIgnoredCount += 1
                 recordKeyboardCommand("tabIgnored")
+                selectionChangedByLastCommand = false
             }
         default:
             recordKeyboardCommand("unknown:\(command)")
+            selectionChangedByLastCommand = false
         }
     }
 
     private func recordKeyboardCommand(_ command: String) {
         lastKeyboardCommand = command
         keyboardCommandsApplied.append(command)
+    }
+
+    private func handleAppShelfIndexKey(_ event: NSEvent) -> Bool {
+        guard !isHorizontalMasonryMode else { return false }
+        let flags = event.modifierFlags.intersection(.deviceIndependentFlagsMask)
+        guard flags.isDisjoint(with: [.command, .control]) else { return false }
+        guard let characters = event.charactersIgnoringModifiers,
+              let symbol = appShelfIndexSymbol(fromKeyboardCommand: characters)
+        else {
+            return false
+        }
+
+        performKeyboardCommand(symbol)
+        return true
+    }
+
+    private func appShelfIndexSymbol(fromKeyboardCommand command: String) -> String? {
+        var candidate = command.trimmingCharacters(in: .whitespacesAndNewlines).uppercased()
+        for prefix in ["APP:", "APP-", "INDEX:", "INDEX-"] where candidate.hasPrefix(prefix) {
+            candidate.removeFirst(prefix.count)
+            break
+        }
+        guard candidate.count == 1,
+              Self.appShelfIndexSymbols.contains(candidate)
+        else {
+            return nil
+        }
+        return candidate
+    }
+
+    private func focusAppShelfItemForKeyboard(indexSymbol: String, commandLabel: String) -> Bool {
+        guard let ordinal = Self.appShelfIndexSymbols.firstIndex(of: indexSymbol),
+              ordinal < appShelfItems.count
+        else {
+            return false
+        }
+
+        return focusAppShelfItemForKeyboard(
+            appGroupIndex: appShelfItems[ordinal].item.appGroupIndex,
+            appIndexCommand: commandLabel
+        )
+    }
+
+    @discardableResult
+    private func focusAppShelfItemForKeyboard(
+        appGroupIndex: Int,
+        appIndexCommand: String? = nil
+    ) -> Bool {
+        guard appShelfItems.contains(where: { $0.item.appGroupIndex == appGroupIndex }),
+              waterfallColumns.contains(where: { $0.column.appGroupIndex == appGroupIndex })
+        else {
+            return false
+        }
+
+        keyboardFocusedAppGroupIndex = appGroupIndex
+        lastKeyboardAppIndexCommand = appIndexCommand
+        selectionChangedByLastCommand = false
+        clearKeyboardBoundaryFeedback()
+        ensureAppShelfItemVisible(appGroupIndex)
+        layoutSubtreeIfNeeded()
+        hoverAppGroup(appGroupIndex, source: .keyboard)
+        needsLayout = true
+        return true
+    }
+
+    @discardableResult
+    private func moveKeyboardAppFocus(cycleForward: Bool) -> Bool {
+        let orderedAppGroupIndexes = appShelfItems.map(\.item.appGroupIndex)
+        guard !orderedAppGroupIndexes.isEmpty else { return false }
+
+        let currentAppGroupIndex = keyboardFocusedAppGroupIndex
+            ?? appHoverTargetAppGroupIndex
+            ?? effectiveSelection?.appGroupIndex
+        let currentPosition = currentAppGroupIndex.flatMap { orderedAppGroupIndexes.firstIndex(of: $0) }
+        let nextPosition = currentPosition.map { current in
+            if cycleForward {
+                return (current + 1) % orderedAppGroupIndexes.count
+            }
+            return (current - 1 + orderedAppGroupIndexes.count) % orderedAppGroupIndexes.count
+        } ?? (cycleForward ? 0 : orderedAppGroupIndexes.count - 1)
+
+        return focusAppShelfItemForKeyboard(appGroupIndex: orderedAppGroupIndexes[nextPosition])
+    }
+
+    private var appHoverTargetAppGroupIndex: Int? {
+        guard case .app(let appGroupIndex) = hoverTarget else { return nil }
+        return appGroupIndex
+    }
+
+    private func moveVerticalSelectionWithinFocusedColumn(_ direction: QuickSwitchKeyboardDirection) {
+        guard direction == .up || direction == .down,
+              let targetAppGroupIndex = verticalKeyboardNavigationAppGroupIndex(),
+              let column = waterfallColumns.first(where: { $0.column.appGroupIndex == targetAppGroupIndex })
+        else {
+            DevelopmentDiagnostics.log("quickSwitch.view.moveVerticalSelection.blocked", [
+                "direction": keyboardDirectionName(direction),
+                "hasFocusedApp": keyboardFocusedAppGroupIndex != nil,
+                "hasSelection": effectiveSelection != nil
+            ])
+            selectionChangedByLastCommand = false
+            return
+        }
+
+        let cards = column.cards.sorted { $0.item.windowIndex < $1.item.windowIndex }
+        guard !cards.isEmpty else {
+            selectionChangedByLastCommand = false
+            return
+        }
+
+        let previousSelection = effectiveSelection
+        let nextSelection: QuickSwitchSelection
+        if let current = previousSelection,
+           current.appGroupIndex == targetAppGroupIndex,
+           let currentPosition = cards.firstIndex(where: { $0.item.window.id == current.windowID }) {
+            let nextPosition = currentPosition + (direction == .down ? 1 : -1)
+            guard cards.indices.contains(nextPosition) else {
+                keyboardFocusedAppGroupIndex = targetAppGroupIndex
+                hoverAppGroup(targetAppGroupIndex, source: .keyboard)
+                triggerKeyboardBoundaryFeedback(appGroupIndex: targetAppGroupIndex, direction: direction)
+                DevelopmentDiagnostics.log("quickSwitch.view.moveVerticalSelection.boundary", [
+                    "direction": keyboardDirectionName(direction),
+                    "appGroupIndex": targetAppGroupIndex,
+                    "windowIndex": current.windowIndex
+                ])
+                needsLayout = true
+                return
+            }
+            nextSelection = selectionForCard(cards[nextPosition])
+        } else {
+            let rememberedWindowIndex = columnSelectionHistory[targetAppGroupIndex]
+            let rememberedCard = rememberedWindowIndex.flatMap { windowIndex in
+                cards.first { $0.item.windowIndex == windowIndex }
+            }
+            nextSelection = selectionForCard(rememberedCard ?? cards[0])
+        }
+
+        keyboardFocusedAppGroupIndex = targetAppGroupIndex
+        currentSelection = nextSelection
+        columnSelectionHistory[nextSelection.appGroupIndex] = nextSelection.windowIndex
+        selectionChangedByLastCommand = previousSelection != nextSelection
+        clearKeyboardBoundaryFeedback()
+        hoverAppGroup(targetAppGroupIndex, source: .keyboard)
+        ensureCurrentSelectionVisible(horizontalIntent: .alignWithAppShelf(animated: true))
+        DevelopmentDiagnostics.log("quickSwitch.view.moveVerticalSelection", [
+            "direction": keyboardDirectionName(direction),
+            "appGroupIndex": nextSelection.appGroupIndex,
+            "windowIndex": nextSelection.windowIndex,
+            "windowID": nextSelection.windowID
+        ])
+        needsLayout = true
+    }
+
+    private func verticalKeyboardNavigationAppGroupIndex() -> Int? {
+        if let keyboardFocusedAppGroupIndex,
+           waterfallColumns.contains(where: { $0.column.appGroupIndex == keyboardFocusedAppGroupIndex }) {
+            return keyboardFocusedAppGroupIndex
+        }
+        if let appHoverTargetAppGroupIndex,
+           waterfallColumns.contains(where: { $0.column.appGroupIndex == appHoverTargetAppGroupIndex }) {
+            return appHoverTargetAppGroupIndex
+        }
+        if let appGroupIndex = effectiveSelection?.appGroupIndex,
+           waterfallColumns.contains(where: { $0.column.appGroupIndex == appGroupIndex }) {
+            return appGroupIndex
+        }
+        return waterfallColumns.first?.column.appGroupIndex
+    }
+
+    private func clearKeyboardBoundaryFeedback() {
+        lastBoundaryBounceAxis = nil
+        lastBoundaryBounceDirection = nil
+    }
+
+    private func triggerKeyboardBoundaryFeedback(
+        appGroupIndex: Int,
+        direction: QuickSwitchKeyboardDirection
+    ) {
+        selectionChangedByLastCommand = false
+        lastBoundaryBounceAxis = "vertical"
+        lastBoundaryBounceDirection = keyboardDirectionName(direction)
+        boundaryBounceCount += 1
+
+        layoutSubtreeIfNeeded()
+        guard let column = waterfallColumns.first(where: { $0.column.appGroupIndex == appGroupIndex }) else { return }
+
+        let animationKey = "keyboardBoundaryBounce"
+        column.containerLayer.removeAnimation(forKey: animationKey)
+        if NSWorkspace.shared.accessibilityDisplayShouldReduceMotion {
+            let pulse = CABasicAnimation(keyPath: "opacity")
+            pulse.fromValue = 1
+            pulse.toValue = 0.9
+            pulse.duration = 0.12
+            pulse.autoreverses = true
+            pulse.timingFunction = CAMediaTimingFunction(name: .easeInEaseOut)
+            column.containerLayer.add(pulse, forKey: animationKey)
+            return
+        }
+
+        let currentY = column.containerLayer.presentation()?.position.y
+            ?? column.containerLayer.position.y
+        let delta: CGFloat = direction == .up ? 5 : -5
+        let bounce = CAKeyframeAnimation(keyPath: "position.y")
+        bounce.values = [currentY, currentY + delta, currentY]
+        bounce.keyTimes = [0, 0.38, 1]
+        bounce.timingFunctions = [
+            CAMediaTimingFunction(name: .easeOut),
+            CAMediaTimingFunction(name: .easeInEaseOut)
+        ]
+        bounce.duration = 0.23
+        bounce.isRemovedOnCompletion = true
+        column.containerLayer.add(bounce, forKey: animationKey)
+    }
+
+    private func keyboardDirectionName(_ direction: QuickSwitchKeyboardDirection) -> String {
+        switch direction {
+        case .left:
+            return "left"
+        case .right:
+            return "right"
+        case .up:
+            return "up"
+        case .down:
+            return "down"
+        }
     }
 
     private func performMouseCommand(_ rawCommand: String) {
@@ -1491,13 +1771,13 @@
         return command.hasPrefix("hover-")
     }
 
-    private func hoverAppGroup(_ appGroupIndex: Int) {
+    private func hoverAppGroup(_ appGroupIndex: Int, source: HoverTargetSource = .debug) {
         updateHoverState(
             appGroupIndex: appGroupIndex,
             windowID: nil,
             spaceID: nil,
             alignsWaterfallWithAppShelf: true,
-            source: .debug
+            source: source
         )
     }
 
@@ -1778,8 +2058,11 @@
         }
 
         columnSelectionHistory[current.appGroupIndex] = current.windowIndex
+        let previousSelection = effectiveSelection
         currentSelection = next
         columnSelectionHistory[next.appGroupIndex] = next.windowIndex
+        selectionChangedByLastCommand = previousSelection != next
+        clearKeyboardBoundaryFeedback()
         ensureCurrentSelectionVisible(horizontalIntent: .alignWithAppShelf(animated: true))
         DevelopmentDiagnostics.log("quickSwitch.view.moveSelection", [
             "direction": String(describing: direction),
@@ -1802,8 +2085,11 @@
         }
 
         columnSelectionHistory[current.appGroupIndex] = current.windowIndex
+        let previousSelection = effectiveSelection
         currentSelection = next
         columnSelectionHistory[next.appGroupIndex] = next.windowIndex
+        selectionChangedByLastCommand = previousSelection != next
+        clearKeyboardBoundaryFeedback()
         ensureCurrentSelectionVisible(horizontalIntent: .alignWithAppShelf(animated: true))
         DevelopmentDiagnostics.log("quickSwitch.view.moveHorizontalMasonrySelection", [
             "direction": String(describing: direction),
@@ -3543,15 +3829,14 @@
     }
 
     private func appShelfIndexText(forOrdinal ordinal: Int) -> String {
-        let symbols = Array("1234567890ABCDEFGHIJKLMNOPQRSTUVWXYZ")
         guard ordinal >= 0 else { return "" }
-        if ordinal < symbols.count {
-            return String(symbols[ordinal])
+        if ordinal < Self.appShelfIndexSymbols.count {
+            return Self.appShelfIndexSymbols[ordinal]
         }
 
-        let overflowIndex = ordinal - symbols.count
-        let prefix = symbols[10 + (overflowIndex / 26) % 26]
-        let suffix = symbols[10 + overflowIndex % 26]
+        let overflowIndex = ordinal - Self.appShelfIndexSymbols.count
+        let prefix = Self.appShelfIndexSymbols[10 + (overflowIndex / 26) % 26]
+        let suffix = Self.appShelfIndexSymbols[10 + overflowIndex % 26]
         return "\(prefix)\(suffix)"
     }
 

--
Gitblit v1.9.3