From 71a520840eb3debe46173c1f73bc85bbea0b1aa6 Mon Sep 17 00:00:00 2001
From: Ariver <shanghai3168@gmail.com>
Date: Sun, 28 Jun 2026 23:42:38 +0800
Subject: [PATCH] Implement Pro custom hotkeys for 8.2.0

---
 src/Apptag/PreferencesView.swift |  301 +++++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 274 insertions(+), 27 deletions(-)

diff --git a/src/Apptag/PreferencesView.swift b/src/Apptag/PreferencesView.swift
index 97c2152..41fec66 100644
--- a/src/Apptag/PreferencesView.swift
+++ b/src/Apptag/PreferencesView.swift
@@ -1,5 +1,6 @@
 import SwiftUI
 import AppKit
+import Carbon
 
 // MARK: - Preferences View
 
@@ -69,6 +70,8 @@
     @AppStorage("showUncommonAppBubbles") private var showUncommonAppBubbles = AppDefaults.showUncommonAppBubbles
     @AppStorage("hideUsageTips") private var hideUsageTips = AppDefaults.hideUsageTips
     @AppStorage(AppGridTheme.storageKey) private var appGridThemeID = AppDefaults.appGridThemeID
+    @AppStorage(AppGridTheme.deepBlueTuningStorageKey) private var deepBlueThemeTuning = AppGridTheme.deepBlue.defaultColorTuning
+    @AppStorage(AppGridTheme.colorfulTuningStorageKey) private var colorfulThemeTuning = AppGridTheme.colorful.defaultColorTuning
     @AppStorage("mainHotkeyRegistrationState") private var mainHotkeyRegistrationState = LauncherHotkeyRegistrationState.active.rawValue
     @AppStorage("quickSearchHotkeyRegistrationState") private var quickSearchHotkeyRegistrationState = LauncherHotkeyRegistrationState.active.rawValue
     @ObservedObject private var proEntitlement = ProEntitlementCenter.shared
@@ -84,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
@@ -303,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) {
@@ -637,7 +713,59 @@
             appGridThemeID = theme.rawValue
             return
         }
-        proEntitlement.startThemePreview(for: theme)
+        proEntitlement.startThemePreview(for: theme, tuning: currentThemeTuning(for: theme))
+    }
+
+    private var activeColorTuningTheme: AppGridTheme? {
+        let theme = effectiveAppGridTheme
+        return theme.supportsColorTuning ? theme : nil
+    }
+
+    private func currentThemeTuning(for theme: AppGridTheme) -> Double {
+        if let preview = proEntitlement.themePreviewState,
+           preview.theme == theme {
+            return theme.normalizedTuning(preview.themeTuning)
+        }
+        switch theme {
+        case .deepBlue:
+            return theme.normalizedTuning(deepBlueThemeTuning)
+        case .colorful:
+            return theme.normalizedTuning(colorfulThemeTuning)
+        case .defaultLight, .black, .pink, .purple, .green, .blue:
+            return theme.defaultColorTuning
+        }
+    }
+
+    private func persistThemeTuning(_ tuning: Double?, for theme: AppGridTheme) {
+        let normalized = theme.normalizedTuning(tuning)
+        switch theme {
+        case .deepBlue:
+            deepBlueThemeTuning = normalized
+        case .colorful:
+            colorfulThemeTuning = normalized
+        case .defaultLight, .black, .pink, .purple, .green, .blue:
+            break
+        }
+    }
+
+    private func updateThemeTuning(_ value: Double, for theme: AppGridTheme) {
+        let normalized = theme.normalizedTuning(value)
+        if proEntitlement.isUnlocked || theme.isFreeTheme {
+            persistThemeTuning(normalized, for: theme)
+            return
+        }
+        if proEntitlement.themePreviewState?.theme == theme {
+            proEntitlement.updateThemePreviewTuning(normalized, for: theme)
+        } else {
+            proEntitlement.startThemePreview(for: theme, tuning: normalized)
+        }
+    }
+
+    private func themeTuningBinding(for theme: AppGridTheme) -> Binding<Double> {
+        Binding(
+            get: { currentThemeTuning(for: theme) },
+            set: { updateThemeTuning($0, for: theme) }
+        )
     }
 
     private func themeOptionAccessory(for theme: AppGridTheme) -> ThemeOptionAccessory {
@@ -686,10 +814,53 @@
         guard newState.isUnlocked else { return }
         if let preview = proEntitlement.themePreviewState {
             appGridThemeID = preview.theme.rawValue
+            persistThemeTuning(preview.themeTuning, for: preview.theme)
             proEntitlement.stopThemePreview()
         }
         if settingsProPromptFeature != nil {
             dismissSettingsProPrompt()
+        }
+    }
+
+    @ViewBuilder
+    private var themeColorTuningSection: some View {
+        if let theme = activeColorTuningTheme {
+            VStack(alignment: .leading, spacing: 8) {
+                HStack(alignment: .firstTextBaseline, spacing: 10) {
+                    Label(
+                        tr(theme == .deepBlue ? "theme.tuning.dark" : "theme.tuning.light"),
+                        systemImage: "slider.horizontal.3"
+                    )
+                    .font(.system(size: 13, weight: .semibold))
+                    .foregroundStyle(.primary)
+
+                    Spacer(minLength: 12)
+
+                    Text("\(Int(round(currentThemeTuning(for: theme) * 100)))")
+                        .font(.system(size: 12, weight: .semibold, design: .monospaced))
+                        .foregroundStyle(.secondary)
+                }
+
+                Slider(value: themeTuningBinding(for: theme), in: 0...1)
+                    .controlSize(.regular)
+                    .accessibilityLabel(tr(theme == .deepBlue ? "theme.tuning.dark" : "theme.tuning.light"))
+
+                Text(tr("theme.tuning.desc"))
+                    .font(.caption)
+                    .foregroundStyle(.secondary)
+                    .fixedSize(horizontal: false, vertical: true)
+            }
+            .padding(14)
+            .frame(maxWidth: .infinity, alignment: .leading)
+            .background(
+                RoundedRectangle(cornerRadius: 10, style: .continuous)
+                    .fill(Color(nsColor: .controlBackgroundColor).opacity(0.82))
+            )
+            .overlay(
+                RoundedRectangle(cornerRadius: 10, style: .continuous)
+                    .stroke(Color.secondary.opacity(0.18), lineWidth: 1)
+            )
+            .transition(.opacity.combined(with: .move(edge: .top)))
         }
     }
 
@@ -764,12 +935,18 @@
                 .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,
+                canRestore: proEntitlement.isUnlocked && LauncherHotkeySettings.hasCustomHotkey(for: .main),
+                onCustomize: { startHotkeyRecording(for: .main) },
+                onRestore: { restoreDefaultHotkey(for: .main) }
             )
 
             StaticHotkeyInfoRow(
@@ -780,12 +957,18 @@
                 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,
+                canRestore: proEntitlement.isUnlocked && LauncherHotkeySettings.hasCustomHotkey(for: .quickSearch),
+                onCustomize: { startHotkeyRecording(for: .quickSearch) },
+                onRestore: { restoreDefaultHotkey(for: .quickSearch) }
             )
 
             StaticHotkeyInfoRow(
@@ -796,6 +979,7 @@
                 statusTone: .neutral
             )
         }
+        .id(hotkeySettingsRefreshToken)
     }
 
     private var themeProStatusRow: some View {
@@ -854,11 +1038,6 @@
                     .lineSpacing(2)
                     .fixedSize(horizontal: false, vertical: true)
 
-                if let price = proEntitlement.priceDisplayText, !proEntitlement.isUnlocked {
-                    Text(price)
-                        .font(.system(size: 12, weight: .semibold))
-                        .foregroundStyle(Color.accentColor)
-                }
             }
 
             Spacer(minLength: 16)
@@ -1110,22 +1289,15 @@
                     case .theme:
             ScrollView(.vertical, showsIndicators: true) {
                 VStack(alignment: .leading, spacing: 18) {
-                    VStack(alignment: .leading, spacing: 6) {
-                        Text(tr("settings.theme"))
-                            .font(.headline)
-                        Text(tr("settings.themeDesc"))
-                            .font(.caption)
-                            .foregroundStyle(.secondary)
-                    }
-
                     themeProStatusRow
 
                     LazyVGrid(columns: themeColumns, alignment: .leading, spacing: 12) {
-                        ForEach(AppGridTheme.allCases) { theme in
+                        ForEach(AppGridTheme.settingsDisplayOrder) { theme in
                             ThemeOptionButton(
                                 theme: theme,
                                 title: tr(theme.titleKey),
                                 isSelected: effectiveAppGridTheme == theme,
+                                previewTuning: currentThemeTuning(for: theme),
                                 accessory: themeOptionAccessory(for: theme)
                             ) {
                                 handleThemeSelection(theme)
@@ -1133,10 +1305,7 @@
                         }
                     }
 
-                    Text(tr("settings.themeScopeDesc"))
-                        .font(.caption)
-                        .foregroundStyle(.secondary)
-                        .fixedSize(horizontal: false, vertical: true)
+                    themeColorTuningSection
                 }
                 .frame(maxWidth: generalContentWidth, alignment: .leading)
                 .padding(.horizontal)
@@ -1508,10 +1677,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
@@ -1580,6 +1753,79 @@
                         .background(Capsule().fill(statusTone.background))
                 }
                 .frame(width: 180, 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 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 canRestore: Bool
+    let onCustomize: () -> Void
+    let onRestore: () -> Void
+
+    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: 7) {
+                    Text(isRecording ? "…" : displayText)
+                        .font(.system(size: 15, weight: .semibold))
+                        .lineLimit(1)
+                        .minimumScaleFactor(0.85)
+
+                    HStack(spacing: 6) {
+                        Text(statusText)
+                            .font(.system(size: 11, weight: .semibold))
+                            .foregroundStyle(statusTone.foreground)
+                            .padding(.horizontal, 8)
+                            .padding(.vertical, 3)
+                            .background(Capsule().fill(statusTone.background))
+
+                        if isRecording {
+                            Text(tr("hotkeys.recordingInline"))
+                                .font(.system(size: 11, weight: .semibold))
+                                .foregroundStyle(.blue)
+                        }
+                    }
+
+                    HStack(spacing: 8) {
+                        Button(customizeTitle, action: onCustomize)
+                            .buttonStyle(.borderedProminent)
+                            .controlSize(.small)
+
+                        Button(restoreTitle, action: onRestore)
+                            .buttonStyle(.bordered)
+                            .controlSize(.small)
+                            .disabled(!canRestore)
+                    }
+                }
+                .frame(width: 230, alignment: .leading)
             }
         }
         .padding(12)
@@ -1707,6 +1953,7 @@
     let theme: AppGridTheme
     let title: String
     let isSelected: Bool
+    let previewTuning: Double
     let accessory: ThemeOptionAccessory
     let action: () -> Void
 
@@ -1716,7 +1963,7 @@
                 RoundedRectangle(cornerRadius: 8, style: .continuous)
                     .fill(
                         LinearGradient(
-                            colors: theme.previewColors,
+                            colors: theme.previewColors(tuning: previewTuning),
                             startPoint: .topLeading,
                             endPoint: .bottomTrailing
                         )

--
Gitblit v1.9.3