425 lines
11 KiB
Go
425 lines
11 KiB
Go
//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
|
||
}
|