✨ Added hotkey service

2025-06-03 13:56:45 +08:00
parent 5d7f6ec6a1
commit 5381e4ba17
20 changed files with 1895 additions and 636 deletions
--- a/internal/services/hotkey_service.go
+++ b/internal/services/hotkey_service.go
@@ -0,0 +1,424 @@
+//go:build windows
+
+package services
+
+/*
+#cgo CFLAGS: -I../lib
+#cgo LDFLAGS: -luser32
+#include "../lib/hotkey_windows.c"
+#include "hotkey_windows.h"
+*/
+import "C"
+
+import (
+	"fmt"
+	"sync"
+	"time"
+	"voidraft/internal/models"
+
+	"github.com/wailsapp/wails/v3/pkg/application"
+	"github.com/wailsapp/wails/v3/pkg/services/log"
+)
+
+// HotkeyService 全局热键服务
+type HotkeyService struct {
+	logger        *log.LoggerService
+	configService *ConfigService
+	app           *application.App
+	mu            sync.RWMutex
+	isRegistered  bool
+	currentHotkey *models.HotkeyCombo
+	stopChan      chan struct{}
+	wg            sync.WaitGroup
+	running       bool
+}
+
+// HotkeyError 热键错误
+type HotkeyError struct {
+	Operation string
+	Err       error
+}
+
+// Error 实现error接口
+func (e *HotkeyError) Error() string {
+	return fmt.Sprintf("hotkey error during %s: %v", e.Operation, e.Err)
+}
+
+// Unwrap 获取原始错误
+func (e *HotkeyError) Unwrap() error {
+	return e.Err
+}
+
+// NewHotkeyService 创建新的热键服务实例
+func NewHotkeyService(configService *ConfigService, logger *log.LoggerService) *HotkeyService {
+	if logger == nil {
+		logger = log.New()
+	}
+
+	return &HotkeyService{
+		logger:        logger,
+		configService: configService,
+		isRegistered:  false,
+		running:       false,
+	}
+}
+
+// Initialize 初始化热键服务
+func (hs *HotkeyService) Initialize(app *application.App) error {
+	hs.app = app
+
+	// 加载并应用当前配置
+	config, err := hs.configService.GetConfig()
+	if err != nil {
+		return &HotkeyError{Operation: "load_config", Err: err}
+	}
+
+	if config.General.EnableGlobalHotkey {
+		err = hs.RegisterHotkey(&config.General.GlobalHotkey)
+		if err != nil {
+			hs.logger.Error("Hotkey: Failed to register hotkey on startup", "error", err)
+		}
+	}
+
+	hs.logger.Info("Hotkey: Service initialized")
+	return nil
+}
+
+// RegisterHotkey 注册全局热键
+func (hs *HotkeyService) RegisterHotkey(hotkey *models.HotkeyCombo) error {
+	hs.mu.Lock()
+	defer hs.mu.Unlock()
+
+	// 先取消注册现有热键
+	if hs.isRegistered {
+		hs.logger.Info("Hotkey: Unregistering existing hotkey before registering new one")
+		err := hs.unregisterHotkeyInternal()
+		if err != nil {
+			return err
+		}
+	}
+
+	// 验证热键组合
+	if !hs.isValidHotkey(hotkey) {
+		return &HotkeyError{Operation: "validate_hotkey", Err: fmt.Errorf("invalid hotkey combination")}
+	}
+
+	hs.logger.Info("Hotkey: Registering global hotkey using Windows API",
+		"ctrl", hotkey.Ctrl,
+		"shift", hotkey.Shift,
+		"alt", hotkey.Alt,
+		"win", hotkey.Win,
+		"key", hotkey.Key)
+
+	// 创建ready channel等待goroutine启动完成
+	readyChan := make(chan struct{})
+
+	// 确保 stopChan 是新的
+	hs.stopChan = make(chan struct{})
+	hs.wg.Add(1)
+	go hs.hotkeyListener(hotkey, readyChan)
+
+	// 等待监听器启动完成，设置超时避免无限等待
+	select {
+	case <-readyChan:
+		// 监听器启动完成
+	case <-time.After(1 * time.Second):
+		// 超时处理
+		hs.logger.Warning("Hotkey: Timeout waiting for listener to start")
+		return &HotkeyError{Operation: "start_listener", Err: fmt.Errorf("timeout waiting for hotkey listener to start")}
+	}
+
+	hs.currentHotkey = hotkey
+	hs.isRegistered = true
+	hs.running = true
+
+	hs.logger.Info("Hotkey: Successfully registered global hotkey")
+	return nil
+}
+
+// UnregisterHotkey 取消注册全局热键
+func (hs *HotkeyService) UnregisterHotkey() error {
+	hs.mu.Lock()
+	defer hs.mu.Unlock()
+
+	return hs.unregisterHotkeyInternal()
+}
+
+// unregisterHotkeyInternal 内部取消注册方法（无锁）
+func (hs *HotkeyService) unregisterHotkeyInternal() error {
+	if !hs.isRegistered {
+		hs.logger.Debug("Hotkey: No hotkey registered, skipping unregister")
+		return nil
+	}
+
+	hs.logger.Info("Hotkey: Unregistering global hotkey")
+
+	// 停止监听
+	if hs.stopChan != nil {
+		close(hs.stopChan)
+		hs.logger.Debug("Hotkey: Waiting for listener goroutine to stop")
+		hs.wg.Wait()
+		hs.logger.Debug("Hotkey: Listener goroutine stopped")
+	}
+
+	// 重置状态
+	hs.currentHotkey = nil
+	hs.isRegistered = false
+	hs.running = false
+	hs.stopChan = nil
+
+	hs.logger.Info("Hotkey: Successfully unregistered global hotkey")
+	return nil
+}
+
+// UpdateHotkey 更新热键配置
+func (hs *HotkeyService) UpdateHotkey(enable bool, hotkey *models.HotkeyCombo) error {
+	hs.logger.Info("Hotkey: === UpdateHotkey called ===",
+		"enable", enable,
+		"ctrl", hotkey.Ctrl,
+		"shift", hotkey.Shift,
+		"alt", hotkey.Alt,
+		"win", hotkey.Win,
+		"key", hotkey.Key)
+
+	// 先获取当前状态
+	hs.mu.RLock()
+	currentRegistered := hs.isRegistered
+	var currentHotkey *models.HotkeyCombo
+	if hs.currentHotkey != nil {
+		currentHotkey = &models.HotkeyCombo{
+			Ctrl:  hs.currentHotkey.Ctrl,
+			Shift: hs.currentHotkey.Shift,
+			Alt:   hs.currentHotkey.Alt,
+			Win:   hs.currentHotkey.Win,
+			Key:   hs.currentHotkey.Key,
+		}
+	}
+	hs.mu.RUnlock()
+
+	hs.logger.Info("Hotkey: Current state",
+		"currentRegistered", currentRegistered,
+		"currentHotkey", currentHotkey)
+
+	// 检查是否需要更新
+	needsUpdate := false
+	if enable != currentRegistered {
+		needsUpdate = true
+		hs.logger.Info("Hotkey: Enable state changed", "old", currentRegistered, "new", enable)
+	} else if enable && currentHotkey != nil {
+		// 如果启用状态，检查热键组合是否变化
+		if hotkey.Ctrl != currentHotkey.Ctrl ||
+			hotkey.Shift != currentHotkey.Shift ||
+			hotkey.Alt != currentHotkey.Alt ||
+			hotkey.Win != currentHotkey.Win ||
+			hotkey.Key != currentHotkey.Key {
+			needsUpdate = true
+			hs.logger.Info("Hotkey: Hotkey combination changed",
+				"old", fmt.Sprintf("Ctrl:%v Shift:%v Alt:%v Win:%v Key:%s",
+					currentHotkey.Ctrl, currentHotkey.Shift, currentHotkey.Alt, currentHotkey.Win, currentHotkey.Key),
+				"new", fmt.Sprintf("Ctrl:%v Shift:%v Alt:%v Win:%v Key:%s",
+					hotkey.Ctrl, hotkey.Shift, hotkey.Alt, hotkey.Win, hotkey.Key))
+		}
+	}
+
+	if !needsUpdate {
+		hs.logger.Info("Hotkey: No changes detected, skipping update")
+		return nil
+	}
+
+	hs.logger.Info("Hotkey: Proceeding with hotkey update", "needsUpdate", needsUpdate)
+
+	if enable {
+		// 启用热键：直接注册新热键（RegisterHotkey 会处理取消旧热键）
+		err := hs.RegisterHotkey(hotkey)
+		if err != nil {
+			hs.logger.Error("Hotkey: Failed to register new hotkey", "error", err)
+			return err
+		}
+		hs.logger.Info("Hotkey: Successfully updated and registered new hotkey")
+		return nil
+	} else {
+		// 禁用热键：取消注册
+		err := hs.UnregisterHotkey()
+		if err != nil {
+			hs.logger.Error("Hotkey: Failed to unregister hotkey", "error", err)
+			return err
+		}
+		hs.logger.Info("Hotkey: Successfully disabled hotkey")
+		return nil
+	}
+}
+
+// ToggleWindow 切换窗口显示/隐藏 - 通过事件通知前端处理
+func (hs *HotkeyService) ToggleWindow() {
+	if hs.app == nil {
+		hs.logger.Warning("Hotkey: App is nil, cannot toggle")
+		return
+	}
+
+	// 发送事件到前端，让前端处理窗口切换
+	hs.app.EmitEvent("hotkey:toggle-window", nil)
+	hs.logger.Debug("Hotkey: Emitted toggle window event")
+}
+
+// hotkeyListener 热键监听器goroutine
+func (hs *HotkeyService) hotkeyListener(hotkey *models.HotkeyCombo, readyChan chan struct{}) {
+	defer hs.wg.Done()
+
+	hs.logger.Debug("Hotkey: Starting Windows API hotkey listener")
+
+	// 将热键转换为虚拟键码
+	mainKeyVK := hs.keyToVirtualKeyCode(hotkey.Key)
+	if mainKeyVK == 0 {
+		hs.logger.Error("Hotkey: Invalid key", "key", hotkey.Key)
+		close(readyChan)
+		return
+	}
+
+	// 检查间隔（100ms，减少CPU使用率）
+	ticker := time.NewTicker(100 * time.Millisecond)
+	defer ticker.Stop()
+
+	// 添加状态跟踪
+	var wasPressed bool = false
+
+	// 标记是否已经发送ready信号，确保只发送一次
+	readySent := false
+
+	for {
+		select {
+		case <-hs.stopChan:
+			hs.logger.Debug("Hotkey: Stopping Windows API hotkey listener")
+			if !readySent {
+				close(readyChan)
+			}
+			return
+		case <-ticker.C:
+			// 第一次循环时发送ready信号，表示监听器已经准备就绪
+			if !readySent {
+				close(readyChan)
+				readySent = true
+				hs.logger.Debug("Hotkey: Listener ready signal sent")
+			}
+
+			// 调用 C 函数检查热键组合
+			ctrl := 0
+			if hotkey.Ctrl {
+				ctrl = 1
+			}
+
+			shift := 0
+			if hotkey.Shift {
+				shift = 1
+			}
+
+			alt := 0
+			if hotkey.Alt {
+				alt = 1
+			}
+
+			win := 0
+			if hotkey.Win {
+				win = 1
+			}
+
+			// 检查热键是否被按下
+			isPressed := int(C.isHotkeyPressed(C.int(ctrl), C.int(shift), C.int(alt), C.int(win), C.int(mainKeyVK))) == 1
+
+			// 只在按键从未按下变为按下时触发（边缘触发）
+			if isPressed && !wasPressed {
+				hs.logger.Debug("Hotkey: Global hotkey triggered via Windows API")
+				hs.ToggleWindow()
+			}
+
+			// 更新状态
+			wasPressed = isPressed
+		}
+	}
+}
+
+// keyToVirtualKeyCode 将键名转换为 Windows 虚拟键码
+func (hs *HotkeyService) keyToVirtualKeyCode(key string) int {
+	keyMap := map[string]int{
+		// 字母键
+		"A": 0x41, "B": 0x42, "C": 0x43, "D": 0x44, "E": 0x45, "F": 0x46, "G": 0x47, "H": 0x48,
+		"I": 0x49, "J": 0x4A, "K": 0x4B, "L": 0x4C, "M": 0x4D, "N": 0x4E, "O": 0x4F, "P": 0x50,
+		"Q": 0x51, "R": 0x52, "S": 0x53, "T": 0x54, "U": 0x55, "V": 0x56, "W": 0x57, "X": 0x58,
+		"Y": 0x59, "Z": 0x5A,
+
+		// 数字键
+		"0": 0x30, "1": 0x31, "2": 0x32, "3": 0x33, "4": 0x34,
+		"5": 0x35, "6": 0x36, "7": 0x37, "8": 0x38, "9": 0x39,
+
+		// 功能键
+		"F1": 0x70, "F2": 0x71, "F3": 0x72, "F4": 0x73, "F5": 0x74, "F6": 0x75,
+		"F7": 0x76, "F8": 0x77, "F9": 0x78, "F10": 0x79, "F11": 0x7A, "F12": 0x7B,
+	}
+
+	if vk, exists := keyMap[key]; exists {
+		return vk
+	}
+	return 0
+}
+
+// isValidHotkey 验证热键组合是否有效
+func (hs *HotkeyService) isValidHotkey(hotkey *models.HotkeyCombo) bool {
+	if hotkey == nil {
+		return false
+	}
+
+	// 必须有主键
+	if hotkey.Key == "" {
+		return false
+	}
+
+	// 必须至少有一个修饰键
+	if !hotkey.Ctrl && !hotkey.Shift && !hotkey.Alt && !hotkey.Win {
+		return false
+	}
+
+	// 验证主键是否在有效范围内
+	return hs.keyToVirtualKeyCode(hotkey.Key) != 0
+}
+
+// GetCurrentHotkey 获取当前注册的热键
+func (hs *HotkeyService) GetCurrentHotkey() *models.HotkeyCombo {
+	hs.mu.RLock()
+	defer hs.mu.RUnlock()
+
+	if hs.currentHotkey == nil {
+		return nil
+	}
+
+	// 返回副本避免并发问题
+	return &models.HotkeyCombo{
+		Ctrl:  hs.currentHotkey.Ctrl,
+		Shift: hs.currentHotkey.Shift,
+		Alt:   hs.currentHotkey.Alt,
+		Win:   hs.currentHotkey.Win,
+		Key:   hs.currentHotkey.Key,
+	}
+}
+
+// IsRegistered 检查是否已注册热键
+func (hs *HotkeyService) IsRegistered() bool {
+	hs.mu.RLock()
+	defer hs.mu.RUnlock()
+	return hs.isRegistered
+}
+
+// ServiceShutdown  关闭热键服务
+func (hs *HotkeyService) ServiceShutdown() error {
+	hs.mu.Lock()
+	defer hs.mu.Unlock()
+
+	if hs.isRegistered {
+		err := hs.unregisterHotkeyInternal()
+		if err != nil {
+			hs.logger.Error("Hotkey: Failed to unregister hotkey on shutdown", "error", err)
+		}
+	}
+
+	hs.logger.Info("Hotkey: Service shutdown completed")
+	return nil
+}