From 18c90acda54cb46815a081ecfc36ea8f2b9e93fd Mon Sep 17 00:00:00 2001
From: Ariver <shanghai3168@gmail.com>
Date: Mon, 29 Jun 2026 00:12:04 +0800
Subject: [PATCH] Fix custom hotkey settings UI for 8.2.1

---
 src/Apptag/PreferencesView.swift |  288 +++++++++++++++++++++++++++++++++++++++++++++++++++------
 1 files changed, 256 insertions(+), 32 deletions(-)

diff --git a/src/Apptag/PreferencesView.swift b/src/Apptag/PreferencesView.swift
index 53d9dee..8838d3d 100644
--- a/src/Apptag/PreferencesView.swift
+++ b/src/Apptag/PreferencesView.swift
@@ -1,5 +1,6 @@
 import SwiftUI
 import AppKit
+import Carbon
 
 // MARK: - Preferences View
 
@@ -86,6 +87,9 @@
     @State private var isDataFilePanelPresented = false
     @State private var hotkeyStatusToast: String? = nil
     @State private var hotkeyStatusToastToken: UUID? = nil
+    @State private var recordingHotkeyKind: LauncherHotkeyKind? = nil
+    @State private var hotkeyRecorderMonitor: Any? = nil
+    @State private var hotkeySettingsRefreshToken = UUID()
     @State private var selectedTab: SettingsTab = .general
     @State private var selectedInitialLayoutMode: InitialLayoutMode = .smart
     @State private var settingsProPromptFeature: ProFeature? = nil
@@ -305,8 +309,78 @@
         let status = LauncherHotkeyRegistrationStore.failureCode(for: kind)
             .map(String.init) ?? "-"
         return tr(key)
-            .replacingOccurrences(of: "%shortcut%", with: kind.hotkey.displayString)
+            .replacingOccurrences(of: "%shortcut%", with: LauncherHotkeySettings.effectiveHotkey(for: kind).displayString)
             .replacingOccurrences(of: "%status%", with: status)
+    }
+
+    private func startHotkeyRecording(for kind: LauncherHotkeyKind) {
+        guard ProEntitlementPolicy.isUnlocked(.customHotkeys) else {
+            presentSettingsProPrompt(for: .customHotkeys)
+            return
+        }
+        stopHotkeyRecording(showCancelledToast: false)
+        recordingHotkeyKind = kind
+        showHotkeyStatusToast(tr("hotkeys.recording"))
+        hotkeyRecorderMonitor = NSEvent.addLocalMonitorForEvents(matching: .keyDown) { event in
+            DispatchQueue.main.async {
+                self.captureHotkeyEvent(event)
+            }
+            return nil
+        }
+    }
+
+    private func captureHotkeyEvent(_ event: NSEvent) {
+        guard let kind = recordingHotkeyKind else { return }
+        if Int(event.keyCode) == kVK_Escape {
+            stopHotkeyRecording(showCancelledToast: true)
+            return
+        }
+        let candidate = LauncherHotkeySettings.candidate(from: event)
+        let result = (NSApp.delegate as? AppDelegate)?
+            .applyCustomHotkey(candidate, for: kind)
+            ?? .registrationFailed(-1)
+        stopHotkeyRecording(showCancelledToast: false)
+        handleHotkeyCustomizationResult(result)
+    }
+
+    private func restoreDefaultHotkey(for kind: LauncherHotkeyKind) {
+        guard ProEntitlementPolicy.isUnlocked(.customHotkeys) else {
+            presentSettingsProPrompt(for: .customHotkeys)
+            return
+        }
+        let result = (NSApp.delegate as? AppDelegate)?
+            .restoreDefaultHotkey(for: kind)
+            ?? .registrationFailed(-1)
+        handleHotkeyCustomizationResult(result)
+    }
+
+    private func stopHotkeyRecording(showCancelledToast: Bool = false) {
+        if let hotkeyRecorderMonitor {
+            NSEvent.removeMonitor(hotkeyRecorderMonitor)
+        }
+        hotkeyRecorderMonitor = nil
+        recordingHotkeyKind = nil
+        if showCancelledToast {
+            showHotkeyStatusToast(tr("hotkeys.recordingCancelled"))
+        }
+    }
+
+    private func handleHotkeyCustomizationResult(_ result: LauncherHotkeyCustomizationResult) {
+        if case .locked = result {
+            presentSettingsProPrompt(for: .customHotkeys)
+            return
+        }
+        hotkeySettingsRefreshToken = UUID()
+        showHotkeyStatusToast(hotkeyCustomizationMessage(for: result))
+    }
+
+    private func hotkeyCustomizationMessage(for result: LauncherHotkeyCustomizationResult) -> String {
+        switch result {
+        case .registrationFailed(let status):
+            return tr(result.messageKey).replacingOccurrences(of: "%status%", with: String(status))
+        default:
+            return tr(result.messageKey)
+        }
     }
 
     private func showHotkeyStatusToast(_ message: String) {
@@ -861,12 +935,19 @@
                 .font(.system(size: 13, weight: .semibold))
                 .foregroundStyle(.secondary)
 
-            StaticHotkeyInfoRow(
+            CustomizableHotkeyInfoRow(
                 title: tr("quickSearch.mainHotkey"),
                 description: tr("quickSearch.mainHotkeyDesc"),
-                displayText: LauncherHotkey.main.displayString,
+                displayText: LauncherHotkeySettings.effectiveHotkey(for: .main).displayString,
                 statusText: hotkeyStatusText(for: .main),
-                statusTone: hotkeyStatusTone(for: .main)
+                statusTone: hotkeyStatusTone(for: .main),
+                customizeTitle: proEntitlement.isUnlocked ? tr("hotkeys.customize") : tr("pro.card.badge"),
+                restoreTitle: tr("hotkeys.restoreDefault"),
+                isRecording: recordingHotkeyKind == .main,
+                isProLocked: !proEntitlement.isUnlocked,
+                canRestore: proEntitlement.isUnlocked && LauncherHotkeySettings.hasCustomHotkey(for: .main),
+                onCustomize: { startHotkeyRecording(for: .main) },
+                onRestore: { restoreDefaultHotkey(for: .main) }
             )
 
             StaticHotkeyInfoRow(
@@ -877,12 +958,19 @@
                 statusTone: .neutral
             )
 
-            StaticHotkeyInfoRow(
+            CustomizableHotkeyInfoRow(
                 title: tr("quickSearch.globalHotkey"),
                 description: tr("quickSearch.globalHotkeyDesc"),
-                displayText: LauncherHotkey.quickSearch.displayString,
+                displayText: LauncherHotkeySettings.effectiveHotkey(for: .quickSearch).displayString,
                 statusText: hotkeyStatusText(for: .quickSearch),
-                statusTone: hotkeyStatusTone(for: .quickSearch)
+                statusTone: hotkeyStatusTone(for: .quickSearch),
+                customizeTitle: proEntitlement.isUnlocked ? tr("hotkeys.customize") : tr("pro.card.badge"),
+                restoreTitle: tr("hotkeys.restoreDefault"),
+                isRecording: recordingHotkeyKind == .quickSearch,
+                isProLocked: !proEntitlement.isUnlocked,
+                canRestore: proEntitlement.isUnlocked && LauncherHotkeySettings.hasCustomHotkey(for: .quickSearch),
+                onCustomize: { startHotkeyRecording(for: .quickSearch) },
+                onRestore: { restoreDefaultHotkey(for: .quickSearch) }
             )
 
             StaticHotkeyInfoRow(
@@ -893,6 +981,7 @@
                 statusTone: .neutral
             )
         }
+        .id(hotkeySettingsRefreshToken)
     }
 
     private var themeProStatusRow: some View {
@@ -1590,10 +1679,14 @@
             showPendingHotkeyWarningIfNeeded()
             refreshThemePreviewCountdown()
         }
+        .onDisappear {
+            stopHotkeyRecording()
+        }
         .onReceive(themePreviewCountdownTimer) { now in
             refreshThemePreviewCountdown(now: now)
         }
         .onReceive(NotificationCenter.default.publisher(for: .tagLauncherHotkeyRegistrationChanged)) { _ in
+            hotkeySettingsRefreshToken = UUID()
             showPendingHotkeyWarningIfNeeded()
         }
         .onReceive(NotificationCenter.default.publisher(for: .tagLauncherPreferencesTabRequested)) { notification in
@@ -1637,32 +1730,27 @@
     let statusTone: HotkeyStatusTone
 
     var body: some View {
-        VStack(alignment: .leading, spacing: 12) {
-            HStack(alignment: .top, spacing: 12) {
-                VStack(alignment: .leading, spacing: 3) {
-                    Text(title)
-                        .font(.system(size: 13, weight: .semibold))
-                    Text(description)
-                        .font(.caption)
-                        .foregroundStyle(.secondary)
-                        .fixedSize(horizontal: false, vertical: true)
-                }
-                .frame(maxWidth: .infinity, alignment: .leading)
-
-                VStack(alignment: .leading, spacing: 4) {
-                    Text(displayText)
-                        .font(.system(size: 15, weight: .semibold))
-                        .lineLimit(1)
-                        .minimumScaleFactor(0.85)
-                    Text(statusText)
-                        .font(.system(size: 11, weight: .semibold))
-                        .foregroundStyle(statusTone.foreground)
-                        .padding(.horizontal, 8)
-                        .padding(.vertical, 3)
-                        .background(Capsule().fill(statusTone.background))
-                }
-                .frame(width: 180, alignment: .leading)
+        HStack(alignment: .center, spacing: 18) {
+            VStack(alignment: .leading, spacing: 3) {
+                Text(title)
+                    .font(.system(size: 13, weight: .semibold))
+                Text(description)
+                    .font(.caption)
+                    .foregroundStyle(.secondary)
+                    .fixedSize(horizontal: false, vertical: true)
             }
+            .frame(maxWidth: .infinity, alignment: .leading)
+
+            HStack(alignment: .center, spacing: 12) {
+                Text(displayText)
+                    .font(.system(size: 15, weight: .semibold))
+                    .lineLimit(1)
+                    .minimumScaleFactor(0.85)
+                    .frame(width: 156, alignment: .leading)
+
+                HotkeyStatusPill(text: statusText, tone: statusTone)
+            }
+            .frame(width: 310, alignment: .leading)
         }
         .padding(12)
         .background(
@@ -1676,6 +1764,142 @@
     }
 }
 
+private struct CustomizableHotkeyInfoRow: View {
+    let title: String
+    let description: String
+    let displayText: String
+    let statusText: String
+    let statusTone: HotkeyStatusTone
+    let customizeTitle: String
+    let restoreTitle: String
+    let isRecording: Bool
+    let isProLocked: Bool
+    let canRestore: Bool
+    let onCustomize: () -> Void
+    let onRestore: () -> Void
+
+    var body: some View {
+        HStack(alignment: .center, spacing: 18) {
+            VStack(alignment: .leading, spacing: 3) {
+                Text(title)
+                    .font(.system(size: 13, weight: .semibold))
+                Text(description)
+                    .font(.caption)
+                    .foregroundStyle(.secondary)
+                    .fixedSize(horizontal: false, vertical: true)
+            }
+            .frame(maxWidth: .infinity, alignment: .leading)
+
+            HStack(alignment: .center, spacing: 12) {
+                VStack(alignment: .leading, spacing: 6) {
+                    Text(isRecording ? "…" : displayText)
+                        .font(.system(size: 15, weight: .semibold))
+                        .lineLimit(1)
+                        .minimumScaleFactor(0.85)
+
+                    HStack(spacing: 6) {
+                        HotkeyStatusPill(text: statusText, tone: statusTone)
+
+                        if isRecording {
+                            Text(tr("hotkeys.recordingInline"))
+                                .font(.system(size: 11, weight: .semibold))
+                                .foregroundStyle(.blue)
+                        }
+                    }
+                }
+                .frame(width: 156, alignment: .leading)
+
+                HotkeyPrimaryActionButton(
+                    title: customizeTitle,
+                    isProLocked: isProLocked,
+                    action: onCustomize
+                )
+
+                if canRestore {
+                    HotkeySecondaryActionButton(
+                        title: restoreTitle,
+                        action: onRestore
+                    )
+                }
+            }
+            .frame(width: 310, alignment: .leading)
+        }
+        .padding(12)
+        .background(
+            RoundedRectangle(cornerRadius: 8, style: .continuous)
+                .fill(Color(nsColor: .controlBackgroundColor))
+        )
+        .overlay(
+            RoundedRectangle(cornerRadius: 8, style: .continuous)
+                .stroke(Color.secondary.opacity(0.14), lineWidth: 1)
+        )
+    }
+}
+
+private struct HotkeyStatusPill: View {
+    let text: String
+    let tone: HotkeyStatusTone
+
+    var body: some View {
+        Text(text)
+            .font(.system(size: 11, weight: .semibold))
+            .foregroundStyle(tone.foreground)
+            .lineLimit(1)
+            .padding(.horizontal, 8)
+            .padding(.vertical, 3)
+            .background(Capsule().fill(tone.background))
+    }
+}
+
+private struct HotkeyPrimaryActionButton: View {
+    let title: String
+    let isProLocked: Bool
+    let action: () -> Void
+
+    var body: some View {
+        Button(action: action) {
+            if isProLocked {
+                ProStatusPill(text: title, style: .locked, compact: true)
+            } else {
+                Text(title)
+                    .font(.system(size: 11, weight: .semibold))
+                    .foregroundStyle(Color.white)
+                    .lineLimit(1)
+                    .padding(.horizontal, 12)
+                    .padding(.vertical, 6)
+                    .background(
+                        Capsule(style: .continuous)
+                            .fill(Color.accentColor)
+                    )
+            }
+        }
+        .buttonStyle(.plain)
+        .contentShape(Capsule(style: .continuous))
+    }
+}
+
+private struct HotkeySecondaryActionButton: View {
+    let title: String
+    let action: () -> Void
+
+    var body: some View {
+        Button(action: action) {
+            Text(title)
+                .font(.system(size: 11, weight: .semibold))
+                .foregroundStyle(.secondary)
+                .lineLimit(1)
+                .padding(.horizontal, 12)
+                .padding(.vertical, 6)
+                .background(
+                    Capsule(style: .continuous)
+                        .fill(Color.secondary.opacity(0.12))
+                )
+        }
+        .buttonStyle(.plain)
+        .contentShape(Capsule(style: .continuous))
+    }
+}
+
 private struct SettingsConfirmationView: View {
     let title: String
     let message: String

--
Gitblit v1.9.3