🎨 Optimize hotkey service
This commit is contained in:
@@ -1,112 +1,65 @@
|
||||
//go:build windows
|
||||
|
||||
package services
|
||||
|
||||
/*
|
||||
#cgo CFLAGS: -I../lib
|
||||
#cgo LDFLAGS: -luser32
|
||||
#include "../lib/hotkey_windows.c"
|
||||
#include "../lib/hotkey_windows.h"
|
||||
*/
|
||||
import "C"
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
"time"
|
||||
"voidraft/internal/common/hotkey"
|
||||
"voidraft/internal/common/hotkey/mainthread"
|
||||
"voidraft/internal/models"
|
||||
|
||||
"github.com/wailsapp/wails/v3/pkg/application"
|
||||
"github.com/wailsapp/wails/v3/pkg/services/log"
|
||||
)
|
||||
|
||||
// HotkeyService Windows全局热键服务
|
||||
// HotkeyService 全局热键服务
|
||||
type HotkeyService struct {
|
||||
logger *log.LogService
|
||||
configService *ConfigService
|
||||
windowService *WindowService
|
||||
windowHelper *WindowHelper
|
||||
|
||||
mu sync.RWMutex
|
||||
currentHotkey *models.HotkeyCombo
|
||||
isRegistered atomic.Bool
|
||||
app *application.App
|
||||
registered atomic.Bool
|
||||
|
||||
ctx context.Context
|
||||
cancelFunc atomic.Value // 使用atomic.Value存储cancel函数，避免竞态条件
|
||||
wg sync.WaitGroup
|
||||
}
|
||||
|
||||
// HotkeyError 热键错误
|
||||
type HotkeyError struct {
|
||||
Operation string
|
||||
Err error
|
||||
}
|
||||
|
||||
func (e *HotkeyError) Error() string {
|
||||
return fmt.Sprintf("hotkey %s: %v", e.Operation, e.Err)
|
||||
}
|
||||
|
||||
func (e *HotkeyError) Unwrap() error {
|
||||
return e.Err
|
||||
}
|
||||
|
||||
// setCancelFunc 原子地设置cancel函数
|
||||
func (hs *HotkeyService) setCancelFunc(cancel context.CancelFunc) {
|
||||
hs.cancelFunc.Store(cancel)
|
||||
}
|
||||
|
||||
// getCancelFunc 原子地获取cancel函数
|
||||
func (hs *HotkeyService) getCancelFunc() context.CancelFunc {
|
||||
if cancel := hs.cancelFunc.Load(); cancel != nil {
|
||||
if cancelFunc, ok := cancel.(context.CancelFunc); ok {
|
||||
return cancelFunc
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// clearCancelFunc 原子地清除cancel函数
|
||||
func (hs *HotkeyService) clearCancelFunc() {
|
||||
hs.cancelFunc.Store((context.CancelFunc)(nil))
|
||||
hk *hotkey.Hotkey
|
||||
stopChan chan struct{}
|
||||
wg sync.WaitGroup
|
||||
}
|
||||
|
||||
// NewHotkeyService 创建热键服务实例
|
||||
func NewHotkeyService(configService *ConfigService, windowService *WindowService, logger *log.LogService) *HotkeyService {
|
||||
func NewHotkeyService(configService *ConfigService, logger *log.LogService) *HotkeyService {
|
||||
if logger == nil {
|
||||
logger = log.New()
|
||||
}
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
service := &HotkeyService{
|
||||
return &HotkeyService{
|
||||
logger: logger,
|
||||
configService: configService,
|
||||
windowService: windowService,
|
||||
windowHelper: NewWindowHelper(),
|
||||
ctx: ctx,
|
||||
stopChan: make(chan struct{}),
|
||||
}
|
||||
// 初始化时设置cancel函数
|
||||
service.setCancelFunc(cancel)
|
||||
return service
|
||||
}
|
||||
|
||||
// ServiceStartup initializes the service when the application starts
|
||||
func (ds *HotkeyService) ServiceStartup(ctx context.Context, options application.ServiceOptions) error {
|
||||
ds.ctx = ctx
|
||||
return ds.Initialize()
|
||||
// ServiceStartup 服务启动时初始化
|
||||
func (hs *HotkeyService) ServiceStartup(ctx context.Context, options application.ServiceOptions) error {
|
||||
hs.app = application.Get()
|
||||
return hs.Initialize()
|
||||
}
|
||||
|
||||
// Initialize 初始化热键服务
|
||||
func (hs *HotkeyService) Initialize() error {
|
||||
config, err := hs.configService.GetConfig()
|
||||
if err != nil {
|
||||
return &HotkeyError{"load_config", err}
|
||||
return fmt.Errorf("load config: %w", err)
|
||||
}
|
||||
|
||||
if config.General.EnableGlobalHotkey {
|
||||
if err := hs.RegisterHotkey(&config.General.GlobalHotkey); err != nil {
|
||||
hs.logger.Error("failed to register startup hotkey", "error", err)
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
@@ -114,41 +67,42 @@ func (hs *HotkeyService) Initialize() error {
|
||||
}
|
||||
|
||||
// RegisterHotkey 注册全局热键
|
||||
func (hs *HotkeyService) RegisterHotkey(hotkey *models.HotkeyCombo) error {
|
||||
if !hs.isValidHotkey(hotkey) {
|
||||
return &HotkeyError{"validate", fmt.Errorf("invalid hotkey combination")}
|
||||
func (hs *HotkeyService) RegisterHotkey(combo *models.HotkeyCombo) error {
|
||||
if !hs.isValidHotkey(combo) {
|
||||
return errors.New("invalid hotkey combination")
|
||||
}
|
||||
|
||||
hs.mu.Lock()
|
||||
defer hs.mu.Unlock()
|
||||
|
||||
// 取消现有热键
|
||||
if hs.isRegistered.Load() {
|
||||
// 如果已注册，先取消
|
||||
if hs.registered.Load() {
|
||||
hs.unregisterInternal()
|
||||
}
|
||||
|
||||
// 启动监听器
|
||||
ctx, cancel := context.WithCancel(hs.ctx)
|
||||
hs.wg.Add(1)
|
||||
|
||||
ready := make(chan error, 1)
|
||||
go hs.hotkeyListener(ctx, hotkey, ready)
|
||||
|
||||
// 等待启动完成
|
||||
select {
|
||||
case err := <-ready:
|
||||
if err != nil {
|
||||
cancel()
|
||||
return &HotkeyError{"register", err}
|
||||
}
|
||||
case <-time.After(time.Second):
|
||||
cancel()
|
||||
return &HotkeyError{"register", fmt.Errorf("timeout")}
|
||||
// 转换为 hotkey 库的格式
|
||||
key, mods, err := hs.convertHotkey(combo)
|
||||
if err != nil {
|
||||
return fmt.Errorf("convert hotkey: %w", err)
|
||||
}
|
||||
|
||||
hs.currentHotkey = hotkey
|
||||
hs.isRegistered.Store(true)
|
||||
hs.setCancelFunc(cancel)
|
||||
// 在主线程中创建热键
|
||||
var regErr error
|
||||
mainthread.Init(func() {
|
||||
hs.hk = hotkey.New(mods, key)
|
||||
regErr = hs.hk.Register()
|
||||
})
|
||||
|
||||
if regErr != nil {
|
||||
return fmt.Errorf("register hotkey: %w", regErr)
|
||||
}
|
||||
|
||||
hs.registered.Store(true)
|
||||
hs.currentHotkey = combo
|
||||
|
||||
// 启动监听 goroutine
|
||||
hs.wg.Add(1)
|
||||
go hs.listenHotkey()
|
||||
|
||||
return nil
|
||||
}
|
||||
@@ -160,151 +114,177 @@ func (hs *HotkeyService) UnregisterHotkey() error {
|
||||
return hs.unregisterInternal()
|
||||
}
|
||||
|
||||
// unregisterInternal 内部取消注册（无锁）
|
||||
// unregisterInternal 内部取消注册
|
||||
func (hs *HotkeyService) unregisterInternal() error {
|
||||
if !hs.isRegistered.Load() {
|
||||
if !hs.registered.Load() {
|
||||
return nil
|
||||
}
|
||||
|
||||
// 原子地获取并调用cancel函数
|
||||
if cancel := hs.getCancelFunc(); cancel != nil {
|
||||
cancel()
|
||||
hs.wg.Wait()
|
||||
// 停止监听
|
||||
close(hs.stopChan)
|
||||
|
||||
// 取消注册热键
|
||||
if hs.hk != nil {
|
||||
if err := hs.hk.Unregister(); err != nil {
|
||||
hs.logger.Error("failed to unregister hotkey", "error", err)
|
||||
}
|
||||
hs.hk = nil
|
||||
}
|
||||
|
||||
// 等待 goroutine 结束
|
||||
hs.wg.Wait()
|
||||
|
||||
hs.currentHotkey = nil
|
||||
hs.isRegistered.Store(false)
|
||||
hs.clearCancelFunc()
|
||||
hs.registered.Store(false)
|
||||
|
||||
// 重新创建 stopChan
|
||||
hs.stopChan = make(chan struct{})
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// UpdateHotkey 更新热键配置
|
||||
func (hs *HotkeyService) UpdateHotkey(enable bool, hotkey *models.HotkeyCombo) error {
|
||||
func (hs *HotkeyService) UpdateHotkey(enable bool, combo *models.HotkeyCombo) error {
|
||||
if enable {
|
||||
return hs.RegisterHotkey(hotkey)
|
||||
return hs.RegisterHotkey(combo)
|
||||
}
|
||||
return hs.UnregisterHotkey()
|
||||
}
|
||||
|
||||
// hotkeyListener 热键监听器
|
||||
func (hs *HotkeyService) hotkeyListener(ctx context.Context, hotkey *models.HotkeyCombo, ready chan<- error) {
|
||||
// listenHotkey 监听热键事件
|
||||
func (hs *HotkeyService) listenHotkey() {
|
||||
defer hs.wg.Done()
|
||||
|
||||
mainKeyVK := hs.keyToVirtualKeyCode(hotkey.Key)
|
||||
if mainKeyVK == 0 {
|
||||
ready <- fmt.Errorf("invalid key: %s", hotkey.Key)
|
||||
return
|
||||
}
|
||||
|
||||
ticker := time.NewTicker(50 * time.Millisecond)
|
||||
defer ticker.Stop()
|
||||
|
||||
var wasPressed bool
|
||||
ready <- nil // 标记准备就绪
|
||||
|
||||
for {
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
case <-hs.stopChan:
|
||||
return
|
||||
case <-ticker.C:
|
||||
ctrl := cBool(hotkey.Ctrl)
|
||||
shift := cBool(hotkey.Shift)
|
||||
alt := cBool(hotkey.Alt)
|
||||
win := cBool(hotkey.Win)
|
||||
|
||||
isPressed := C.isHotkeyPressed(ctrl, shift, alt, win, C.int(mainKeyVK)) == 1
|
||||
|
||||
if isPressed && !wasPressed {
|
||||
hs.toggleWindow()
|
||||
}
|
||||
wasPressed = isPressed
|
||||
case <-hs.hk.Keydown():
|
||||
hs.toggleWindow()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// cBool 转换Go bool为C int
|
||||
func cBool(b bool) C.int {
|
||||
if b {
|
||||
return 1
|
||||
// convertHotkey 转换热键格式
|
||||
func (hs *HotkeyService) convertHotkey(combo *models.HotkeyCombo) (hotkey.Key, []hotkey.Modifier, error) {
|
||||
// 转换主键
|
||||
key, err := hs.convertKey(combo.Key)
|
||||
if err != nil {
|
||||
return 0, nil, err
|
||||
}
|
||||
return 0
|
||||
|
||||
// 转换修饰键
|
||||
var mods []hotkey.Modifier
|
||||
if combo.Ctrl {
|
||||
mods = append(mods, hotkey.ModCtrl)
|
||||
}
|
||||
if combo.Shift {
|
||||
mods = append(mods, hotkey.ModShift)
|
||||
}
|
||||
if combo.Alt {
|
||||
mods = append(mods, hotkey.ModAlt)
|
||||
}
|
||||
if combo.Win {
|
||||
mods = append(mods, hotkey.ModWin) // Win/Cmd 键
|
||||
}
|
||||
|
||||
return key, mods, nil
|
||||
}
|
||||
|
||||
// convertKey 转换键码
|
||||
func (hs *HotkeyService) convertKey(keyStr string) (hotkey.Key, error) {
|
||||
// 字母键
|
||||
keyMap := map[string]hotkey.Key{
|
||||
"A": hotkey.KeyA, "B": hotkey.KeyB, "C": hotkey.KeyC, "D": hotkey.KeyD,
|
||||
"E": hotkey.KeyE, "F": hotkey.KeyF, "G": hotkey.KeyG, "H": hotkey.KeyH,
|
||||
"I": hotkey.KeyI, "J": hotkey.KeyJ, "K": hotkey.KeyK, "L": hotkey.KeyL,
|
||||
"M": hotkey.KeyM, "N": hotkey.KeyN, "O": hotkey.KeyO, "P": hotkey.KeyP,
|
||||
"Q": hotkey.KeyQ, "R": hotkey.KeyR, "S": hotkey.KeyS, "T": hotkey.KeyT,
|
||||
"U": hotkey.KeyU, "V": hotkey.KeyV, "W": hotkey.KeyW, "X": hotkey.KeyX,
|
||||
"Y": hotkey.KeyY, "Z": hotkey.KeyZ,
|
||||
|
||||
// 数字键
|
||||
"0": hotkey.Key0, "1": hotkey.Key1, "2": hotkey.Key2, "3": hotkey.Key3,
|
||||
"4": hotkey.Key4, "5": hotkey.Key5, "6": hotkey.Key6, "7": hotkey.Key7,
|
||||
"8": hotkey.Key8, "9": hotkey.Key9,
|
||||
|
||||
// 功能键
|
||||
"F1": hotkey.KeyF1, "F2": hotkey.KeyF2, "F3": hotkey.KeyF3, "F4": hotkey.KeyF4,
|
||||
"F5": hotkey.KeyF5, "F6": hotkey.KeyF6, "F7": hotkey.KeyF7, "F8": hotkey.KeyF8,
|
||||
"F9": hotkey.KeyF9, "F10": hotkey.KeyF10, "F11": hotkey.KeyF11, "F12": hotkey.KeyF12,
|
||||
|
||||
// 特殊键
|
||||
"Space": hotkey.KeySpace,
|
||||
"Tab": hotkey.KeyTab,
|
||||
"Enter": hotkey.KeyReturn,
|
||||
"Escape": hotkey.KeyEscape,
|
||||
"Delete": hotkey.KeyDelete,
|
||||
"ArrowUp": hotkey.KeyUp,
|
||||
"ArrowDown": hotkey.KeyDown,
|
||||
"ArrowLeft": hotkey.KeyLeft,
|
||||
"ArrowRight": hotkey.KeyRight,
|
||||
}
|
||||
|
||||
if key, ok := keyMap[keyStr]; ok {
|
||||
return key, nil
|
||||
}
|
||||
|
||||
return 0, fmt.Errorf("unsupported key: %s", keyStr)
|
||||
}
|
||||
|
||||
// toggleWindow 切换窗口显示状态
|
||||
func (hs *HotkeyService) toggleWindow() {
|
||||
mainWindow := hs.windowHelper.MustGetMainWindow()
|
||||
if mainWindow == nil {
|
||||
hs.logger.Error("main window not found")
|
||||
return
|
||||
}
|
||||
|
||||
// 检查主窗口是否可见
|
||||
if mainWindow.IsVisible() {
|
||||
// 如果主窗口可见，隐藏所有窗口
|
||||
// 隐藏所有窗口
|
||||
hs.hideAllWindows()
|
||||
} else {
|
||||
// 如果主窗口不可见，显示所有窗口
|
||||
// 显示所有窗口
|
||||
hs.showAllWindows()
|
||||
}
|
||||
}
|
||||
|
||||
// isWindowVisible 检查窗口是否可见
|
||||
func (hs *HotkeyService) isWindowVisible(window *application.WebviewWindow) bool {
|
||||
return window.IsVisible()
|
||||
}
|
||||
|
||||
// hideAllWindows 隐藏所有窗口
|
||||
func (hs *HotkeyService) hideAllWindows() {
|
||||
// 隐藏所有子窗口
|
||||
if hs.windowService != nil {
|
||||
openWindows := hs.windowService.GetOpenWindows()
|
||||
for _, windowInfo := range openWindows {
|
||||
windowInfo.Hide()
|
||||
}
|
||||
if hs.app == nil {
|
||||
return
|
||||
}
|
||||
|
||||
openWindows := hs.app.Window.GetAll()
|
||||
for _, window := range openWindows {
|
||||
window.Hide()
|
||||
}
|
||||
}
|
||||
|
||||
// showAllWindows 显示所有窗口
|
||||
func (hs *HotkeyService) showAllWindows() {
|
||||
// 显示所有子窗口
|
||||
if hs.windowService != nil {
|
||||
openWindows := hs.windowService.GetOpenWindows()
|
||||
for _, windowInfo := range openWindows {
|
||||
windowInfo.Show()
|
||||
windowInfo.Restore()
|
||||
}
|
||||
if hs.app == nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
// keyToVirtualKeyCode 键名转虚拟键码
|
||||
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,
|
||||
openWindows := hs.app.Window.GetAll()
|
||||
for _, window := range openWindows {
|
||||
window.Show()
|
||||
window.Restore()
|
||||
window.Focus()
|
||||
}
|
||||
return keyMap[key]
|
||||
}
|
||||
|
||||
// isValidHotkey 验证热键组合
|
||||
func (hs *HotkeyService) isValidHotkey(hotkey *models.HotkeyCombo) bool {
|
||||
if hotkey == nil || hotkey.Key == "" {
|
||||
func (hs *HotkeyService) isValidHotkey(combo *models.HotkeyCombo) bool {
|
||||
if combo == nil || combo.Key == "" {
|
||||
return false
|
||||
}
|
||||
// 至少需要一个修饰键
|
||||
if !hotkey.Ctrl && !hotkey.Shift && !hotkey.Alt && !hotkey.Win {
|
||||
if !combo.Ctrl && !combo.Shift && !combo.Alt && !combo.Win {
|
||||
return false
|
||||
}
|
||||
return hs.keyToVirtualKeyCode(hotkey.Key) != 0
|
||||
return true
|
||||
}
|
||||
|
||||
// GetCurrentHotkey 获取当前热键
|
||||
@@ -327,15 +307,10 @@ func (hs *HotkeyService) GetCurrentHotkey() *models.HotkeyCombo {
|
||||
|
||||
// IsRegistered 检查是否已注册
|
||||
func (hs *HotkeyService) IsRegistered() bool {
|
||||
return hs.isRegistered.Load()
|
||||
return hs.registered.Load()
|
||||
}
|
||||
|
||||
// ServiceShutdown 关闭服务
|
||||
func (hs *HotkeyService) ServiceShutdown() error {
|
||||
// 原子地获取并调用cancel函数
|
||||
if cancel := hs.getCancelFunc(); cancel != nil {
|
||||
cancel()
|
||||
}
|
||||
hs.wg.Wait()
|
||||
return nil
|
||||
return hs.UnregisterHotkey()
|
||||
}
|
||||
|
||||
@@ -1,373 +0,0 @@
|
||||
//go:build darwin
|
||||
|
||||
package services
|
||||
|
||||
/*
|
||||
#cgo CFLAGS: -x objective-c
|
||||
#cgo LDFLAGS: -framework Cocoa -framework Carbon
|
||||
#include <ApplicationServices/ApplicationServices.h>
|
||||
#include <Carbon/Carbon.h>
|
||||
|
||||
// 全局变量用于存储热键ID和回调
|
||||
static EventHotKeyRef g_hotKeyRef = NULL;
|
||||
static int g_hotkeyRegistered = 0;
|
||||
|
||||
// 热键事件处理函数
|
||||
OSStatus hotKeyHandler(EventHandlerCallRef nextHandler, EventRef theEvent, void *userData) {
|
||||
// 通知Go代码热键被触发
|
||||
extern void hotkeyTriggered();
|
||||
hotkeyTriggered();
|
||||
return noErr;
|
||||
}
|
||||
|
||||
// 注册全局热键
|
||||
int registerGlobalHotkey(int keyCode, int modifiers) {
|
||||
if (g_hotkeyRegistered) {
|
||||
// 先取消注册现有热键
|
||||
UnregisterEventHotKey(g_hotKeyRef);
|
||||
g_hotkeyRegistered = 0;
|
||||
}
|
||||
|
||||
EventHotKeyID hotKeyID;
|
||||
hotKeyID.signature = 'htk1';
|
||||
hotKeyID.id = 1;
|
||||
|
||||
EventTypeSpec eventType;
|
||||
eventType.eventClass = kEventClassKeyboard;
|
||||
eventType.eventKind = kEventHotKeyPressed;
|
||||
|
||||
InstallApplicationEventHandler(&hotKeyHandler, 1, &eventType, NULL, NULL);
|
||||
|
||||
OSStatus status = RegisterEventHotKey(keyCode, modifiers, hotKeyID, GetApplicationEventTarget(), 0, &g_hotKeyRef);
|
||||
|
||||
if (status == noErr) {
|
||||
g_hotkeyRegistered = 1;
|
||||
return 1;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
// 取消注册全局热键
|
||||
int unregisterGlobalHotkey() {
|
||||
if (g_hotkeyRegistered && g_hotKeyRef != NULL) {
|
||||
OSStatus status = UnregisterEventHotKey(g_hotKeyRef);
|
||||
g_hotkeyRegistered = 0;
|
||||
g_hotKeyRef = NULL;
|
||||
return (status == noErr) ? 1 : 0;
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
|
||||
// 检查是否已注册热键
|
||||
int isHotkeyRegistered() {
|
||||
return g_hotkeyRegistered;
|
||||
}
|
||||
*/
|
||||
import "C"
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
"voidraft/internal/models"
|
||||
|
||||
"github.com/wailsapp/wails/v3/pkg/application"
|
||||
"github.com/wailsapp/wails/v3/pkg/services/log"
|
||||
)
|
||||
|
||||
// 全局服务实例，用于C回调
|
||||
var globalHotkeyService *HotkeyService
|
||||
|
||||
// HotkeyService macOS全局热键服务
|
||||
type HotkeyService struct {
|
||||
logger *log.LogService
|
||||
configService *ConfigService
|
||||
windowService *WindowService
|
||||
windowHelper *WindowHelper
|
||||
mu sync.RWMutex
|
||||
isRegistered atomic.Bool
|
||||
currentHotkey *models.HotkeyCombo
|
||||
cancelFunc atomic.Value // 使用atomic.Value存储cancel函数，避免竞态条件
|
||||
ctx context.Context
|
||||
}
|
||||
|
||||
// HotkeyError 热键错误
|
||||
type HotkeyError struct {
|
||||
Operation string
|
||||
Err error
|
||||
}
|
||||
|
||||
// Error 实现error接口
|
||||
func (e *HotkeyError) Error() string {
|
||||
return fmt.Sprintf("hotkey %s: %v", e.Operation, e.Err)
|
||||
}
|
||||
|
||||
// Unwrap 获取原始错误
|
||||
func (e *HotkeyError) Unwrap() error {
|
||||
return e.Err
|
||||
}
|
||||
|
||||
// setCancelFunc 原子地设置cancel函数
|
||||
func (hs *HotkeyService) setCancelFunc(cancel context.CancelFunc) {
|
||||
hs.cancelFunc.Store(cancel)
|
||||
}
|
||||
|
||||
// getCancelFunc 原子地获取cancel函数
|
||||
func (hs *HotkeyService) getCancelFunc() context.CancelFunc {
|
||||
if cancel := hs.cancelFunc.Load(); cancel != nil {
|
||||
if cancelFunc, ok := cancel.(context.CancelFunc); ok {
|
||||
return cancelFunc
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// clearCancelFunc 原子地清除cancel函数
|
||||
func (hs *HotkeyService) clearCancelFunc() {
|
||||
hs.cancelFunc.Store((context.CancelFunc)(nil))
|
||||
}
|
||||
|
||||
// NewHotkeyService 创建新的热键服务实例
|
||||
func NewHotkeyService(configService *ConfigService, windowService *WindowService, logger *log.LogService) *HotkeyService {
|
||||
if logger == nil {
|
||||
logger = log.New()
|
||||
}
|
||||
|
||||
service := &HotkeyService{
|
||||
logger: logger,
|
||||
configService: configService,
|
||||
windowService: windowService,
|
||||
windowHelper: NewWindowHelper(),
|
||||
}
|
||||
|
||||
// 设置全局实例
|
||||
globalHotkeyService = service
|
||||
|
||||
return service
|
||||
}
|
||||
|
||||
// ServiceStartup initializes the service when the application starts
|
||||
func (ds *HotkeyService) ServiceStartup(ctx context.Context, options application.ServiceOptions) error {
|
||||
ds.ctx = ctx
|
||||
return ds.Initialize()
|
||||
}
|
||||
|
||||
// Initialize 初始化热键服务
|
||||
func (hs *HotkeyService) Initialize() error {
|
||||
config, err := hs.configService.GetConfig()
|
||||
if err != nil {
|
||||
return &HotkeyError{"load_config", err}
|
||||
}
|
||||
|
||||
if config.General.EnableGlobalHotkey {
|
||||
if err := hs.RegisterHotkey(&config.General.GlobalHotkey); err != nil {
|
||||
hs.logger.Error("failed to register startup hotkey", "error", err)
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// RegisterHotkey 注册全局热键
|
||||
func (hs *HotkeyService) RegisterHotkey(hotkey *models.HotkeyCombo) error {
|
||||
if !hs.isValidHotkey(hotkey) {
|
||||
return &HotkeyError{"validate", fmt.Errorf("invalid hotkey combination")}
|
||||
}
|
||||
|
||||
hs.mu.Lock()
|
||||
defer hs.mu.Unlock()
|
||||
|
||||
// 转换键码和修饰符
|
||||
keyCode := hs.keyToMacKeyCode(hotkey.Key)
|
||||
if keyCode == 0 {
|
||||
return &HotkeyError{"convert_key", fmt.Errorf("unsupported key: %s", hotkey.Key)}
|
||||
}
|
||||
|
||||
modifiers := hs.buildMacModifiers(hotkey)
|
||||
|
||||
// 调用C函数注册热键
|
||||
result := int(C.registerGlobalHotkey(C.int(keyCode), C.int(modifiers)))
|
||||
if result == 0 {
|
||||
return &HotkeyError{"register", fmt.Errorf("failed to register hotkey")}
|
||||
}
|
||||
|
||||
hs.currentHotkey = hotkey
|
||||
hs.isRegistered.Store(true)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// UnregisterHotkey 取消注册全局热键
|
||||
func (hs *HotkeyService) UnregisterHotkey() error {
|
||||
hs.mu.Lock()
|
||||
defer hs.mu.Unlock()
|
||||
|
||||
if !hs.isRegistered.Load() {
|
||||
return nil
|
||||
}
|
||||
|
||||
// 调用C函数取消注册热键
|
||||
result := int(C.unregisterGlobalHotkey())
|
||||
if result == 0 {
|
||||
return &HotkeyError{"unregister", fmt.Errorf("failed to unregister hotkey")}
|
||||
}
|
||||
|
||||
hs.currentHotkey = nil
|
||||
hs.isRegistered.Store(false)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// UpdateHotkey 更新热键配置
|
||||
func (hs *HotkeyService) UpdateHotkey(enable bool, hotkey *models.HotkeyCombo) error {
|
||||
if enable {
|
||||
return hs.RegisterHotkey(hotkey)
|
||||
}
|
||||
return hs.UnregisterHotkey()
|
||||
}
|
||||
|
||||
// keyToMacKeyCode 将键名转换为macOS虚拟键码
|
||||
func (hs *HotkeyService) keyToMacKeyCode(key string) int {
|
||||
keyMap := map[string]int{
|
||||
// 字母键
|
||||
"A": 0, "S": 1, "D": 2, "F": 3, "H": 4, "G": 5, "Z": 6, "X": 7,
|
||||
"C": 8, "V": 9, "B": 11, "Q": 12, "W": 13, "E": 14, "R": 15,
|
||||
"Y": 16, "T": 17, "1": 18, "2": 19, "3": 20, "4": 21, "6": 22,
|
||||
"5": 23, "9": 25, "7": 26, "8": 28, "0": 29, "O": 31, "U": 32,
|
||||
"I": 34, "P": 35, "L": 37, "J": 38, "K": 40, "N": 45, "M": 46,
|
||||
|
||||
// 功能键
|
||||
"F1": 122, "F2": 120, "F3": 99, "F4": 118, "F5": 96, "F6": 97,
|
||||
"F7": 98, "F8": 100, "F9": 101, "F10": 109, "F11": 103, "F12": 111,
|
||||
}
|
||||
|
||||
if keyCode, exists := keyMap[key]; exists {
|
||||
return keyCode
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
// buildMacModifiers 构建macOS修饰符
|
||||
func (hs *HotkeyService) buildMacModifiers(hotkey *models.HotkeyCombo) int {
|
||||
var modifiers int = 0
|
||||
|
||||
if hotkey.Ctrl {
|
||||
modifiers |= 0x1000 // controlKey
|
||||
}
|
||||
if hotkey.Shift {
|
||||
modifiers |= 0x200 // shiftKey
|
||||
}
|
||||
if hotkey.Alt {
|
||||
modifiers |= 0x800 // optionKey
|
||||
}
|
||||
if hotkey.Win {
|
||||
modifiers |= 0x100 // cmdKey
|
||||
}
|
||||
|
||||
return modifiers
|
||||
}
|
||||
|
||||
// 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.keyToMacKeyCode(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 {
|
||||
return hs.isRegistered.Load()
|
||||
}
|
||||
|
||||
// ToggleWindow 切换窗口显示状态
|
||||
func (hs *HotkeyService) ToggleWindow() {
|
||||
mainWindow := hs.windowHelper.MustGetMainWindow()
|
||||
if mainWindow == nil {
|
||||
hs.logger.Error("main window not found")
|
||||
return
|
||||
}
|
||||
|
||||
// 检查主窗口是否可见
|
||||
if mainWindow.IsVisible() {
|
||||
// 如果主窗口可见，隐藏所有窗口
|
||||
hs.hideAllWindows()
|
||||
} else {
|
||||
// 如果主窗口不可见，显示所有窗口
|
||||
hs.showAllWindows()
|
||||
}
|
||||
}
|
||||
|
||||
// isWindowVisible 检查窗口是否可见
|
||||
func (hs *HotkeyService) isWindowVisible(window *application.WebviewWindow) bool {
|
||||
return window.IsVisible()
|
||||
}
|
||||
|
||||
// hideAllWindows 隐藏所有窗口
|
||||
func (hs *HotkeyService) hideAllWindows() {
|
||||
// 隐藏所有子窗口
|
||||
if hs.windowService != nil {
|
||||
openWindows := hs.windowService.GetOpenWindows()
|
||||
for _, windowInfo := range openWindows {
|
||||
windowInfo.Hide()
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
// showAllWindows 显示所有窗口
|
||||
func (hs *HotkeyService) showAllWindows() {
|
||||
// 显示所有子窗口
|
||||
if hs.windowService != nil {
|
||||
openWindows := hs.windowService.GetOpenWindows()
|
||||
for _, windowInfo := range openWindows {
|
||||
windowInfo.Show()
|
||||
windowInfo.Restore()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// ServiceShutdown 关闭热键服务
|
||||
func (hs *HotkeyService) ServiceShutdown() error {
|
||||
return hs.UnregisterHotkey()
|
||||
}
|
||||
|
||||
//export hotkeyTriggered
|
||||
func hotkeyTriggered() {
|
||||
// 通过全局实例调用ToggleWindow
|
||||
if globalHotkeyService != nil {
|
||||
globalHotkeyService.ToggleWindow()
|
||||
}
|
||||
}
|
||||
@@ -1,468 +0,0 @@
|
||||
//go:build linux
|
||||
|
||||
package services
|
||||
|
||||
/*
|
||||
#cgo pkg-config: x11
|
||||
#include <X11/Xlib.h>
|
||||
#include <X11/keysym.h>
|
||||
#include <X11/extensions/XTest.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
static Display *display = NULL;
|
||||
static Window root_window;
|
||||
static int hotkey_registered = 0;
|
||||
static int registered_keycode = 0;
|
||||
static unsigned int registered_modifiers = 0;
|
||||
|
||||
// 初始化X11显示
|
||||
int initX11Display() {
|
||||
display = XOpenDisplay(NULL);
|
||||
if (display == NULL) {
|
||||
return 0;
|
||||
}
|
||||
root_window = DefaultRootWindow(display);
|
||||
return 1;
|
||||
}
|
||||
|
||||
// 关闭X11显示
|
||||
void closeX11Display() {
|
||||
if (display != NULL) {
|
||||
XCloseDisplay(display);
|
||||
display = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
// 注册全局热键
|
||||
int registerGlobalHotkey(int keycode, unsigned int modifiers) {
|
||||
if (display == NULL && !initX11Display()) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
// 如果已经注册了热键，先取消注册
|
||||
if (hotkey_registered) {
|
||||
XUngrabKey(display, registered_keycode, registered_modifiers, root_window);
|
||||
hotkey_registered = 0;
|
||||
}
|
||||
|
||||
// 注册新热键
|
||||
XGrabKey(display, keycode, modifiers, root_window, False, GrabModeAsync, GrabModeAsync);
|
||||
XGrabKey(display, keycode, modifiers | LockMask, root_window, False, GrabModeAsync, GrabModeAsync);
|
||||
XGrabKey(display, keycode, modifiers | Mod2Mask, root_window, False, GrabModeAsync, GrabModeAsync);
|
||||
XGrabKey(display, keycode, modifiers | LockMask | Mod2Mask, root_window, False, GrabModeAsync, GrabModeAsync);
|
||||
|
||||
XSync(display, False);
|
||||
|
||||
registered_keycode = keycode;
|
||||
registered_modifiers = modifiers;
|
||||
hotkey_registered = 1;
|
||||
|
||||
return 1;
|
||||
}
|
||||
|
||||
// 取消注册全局热键
|
||||
int unregisterGlobalHotkey() {
|
||||
if (display == NULL || !hotkey_registered) {
|
||||
return 1;
|
||||
}
|
||||
|
||||
XUngrabKey(display, registered_keycode, registered_modifiers, root_window);
|
||||
XUngrabKey(display, registered_keycode, registered_modifiers | LockMask, root_window);
|
||||
XUngrabKey(display, registered_keycode, registered_modifiers | Mod2Mask, root_window);
|
||||
XUngrabKey(display, registered_keycode, registered_modifiers | LockMask | Mod2Mask, root_window);
|
||||
|
||||
XSync(display, False);
|
||||
|
||||
hotkey_registered = 0;
|
||||
registered_keycode = 0;
|
||||
registered_modifiers = 0;
|
||||
|
||||
return 1;
|
||||
}
|
||||
|
||||
// 检查热键事件
|
||||
int checkHotkeyEvent() {
|
||||
if (display == NULL || !hotkey_registered) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
XEvent event;
|
||||
while (XPending(display)) {
|
||||
XNextEvent(display, &event);
|
||||
if (event.type == KeyPress) {
|
||||
XKeyEvent *key_event = (XKeyEvent*)&event;
|
||||
if (key_event->keycode == registered_keycode) {
|
||||
// 检查修饰符匹配（忽略Caps Lock和Num Lock）
|
||||
unsigned int clean_modifiers = key_event->state & ~(LockMask | Mod2Mask);
|
||||
if (clean_modifiers == registered_modifiers) {
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
// 将键名转换为X11键码
|
||||
int getX11Keycode(const char* keyname) {
|
||||
if (display == NULL) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
KeySym keysym = XStringToKeysym(keyname);
|
||||
if (keysym == NoSymbol) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
return XKeysymToKeycode(display, keysym);
|
||||
}
|
||||
|
||||
// 检查是否已注册热键
|
||||
int isHotkeyRegistered() {
|
||||
return hotkey_registered;
|
||||
}
|
||||
*/
|
||||
import "C"
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
"time"
|
||||
"unsafe"
|
||||
"voidraft/internal/models"
|
||||
|
||||
"github.com/wailsapp/wails/v3/pkg/application"
|
||||
"github.com/wailsapp/wails/v3/pkg/services/log"
|
||||
)
|
||||
|
||||
// HotkeyService Linux全局热键服务
|
||||
type HotkeyService struct {
|
||||
logger *log.LogService
|
||||
configService *ConfigService
|
||||
windowService *WindowService
|
||||
windowHelper *WindowHelper
|
||||
|
||||
mu sync.RWMutex
|
||||
currentHotkey *models.HotkeyCombo
|
||||
isRegistered atomic.Bool
|
||||
|
||||
ctx context.Context
|
||||
cancelFunc atomic.Value // 使用atomic.Value存储cancel函数，避免竞态条件
|
||||
wg sync.WaitGroup
|
||||
}
|
||||
|
||||
// HotkeyError 热键错误
|
||||
type HotkeyError struct {
|
||||
Operation string
|
||||
Err error
|
||||
}
|
||||
|
||||
// Error 实现error接口
|
||||
func (e *HotkeyError) Error() string {
|
||||
return fmt.Sprintf("hotkey %s: %v", e.Operation, e.Err)
|
||||
}
|
||||
|
||||
func (e *HotkeyError) Unwrap() error {
|
||||
return e.Err
|
||||
}
|
||||
|
||||
// setCancelFunc 原子地设置cancel函数
|
||||
func (hs *HotkeyService) setCancelFunc(cancel context.CancelFunc) {
|
||||
hs.cancelFunc.Store(cancel)
|
||||
}
|
||||
|
||||
// getCancelFunc 原子地获取cancel函数
|
||||
func (hs *HotkeyService) getCancelFunc() context.CancelFunc {
|
||||
if cancel := hs.cancelFunc.Load(); cancel != nil {
|
||||
if cancelFunc, ok := cancel.(context.CancelFunc); ok {
|
||||
return cancelFunc
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// clearCancelFunc 原子地清除cancel函数
|
||||
func (hs *HotkeyService) clearCancelFunc() {
|
||||
hs.cancelFunc.Store((context.CancelFunc)(nil))
|
||||
}
|
||||
|
||||
// NewHotkeyService 创建热键服务实例
|
||||
func NewHotkeyService(configService *ConfigService, windowService *WindowService, logger *log.LogService) *HotkeyService {
|
||||
if logger == nil {
|
||||
logger = log.New()
|
||||
}
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
service := &HotkeyService{
|
||||
logger: logger,
|
||||
configService: configService,
|
||||
windowService: windowService,
|
||||
windowHelper: NewWindowHelper(),
|
||||
ctx: ctx,
|
||||
}
|
||||
// 初始化时设置cancel函数
|
||||
service.setCancelFunc(cancel)
|
||||
return service
|
||||
}
|
||||
|
||||
// ServiceStartup initializes the service when the application starts
|
||||
func (ds *HotkeyService) ServiceStartup(ctx context.Context, options application.ServiceOptions) error {
|
||||
ds.ctx = ctx
|
||||
return ds.Initialize()
|
||||
}
|
||||
|
||||
// Initialize 初始化热键服务
|
||||
func (hs *HotkeyService) Initialize() error {
|
||||
if int(C.initX11Display()) == 0 {
|
||||
return &HotkeyError{"init_x11", fmt.Errorf("failed to initialize X11 display")}
|
||||
}
|
||||
|
||||
config, err := hs.configService.GetConfig()
|
||||
if err != nil {
|
||||
return &HotkeyError{"load_config", err}
|
||||
}
|
||||
|
||||
if config.General.EnableGlobalHotkey {
|
||||
if err := hs.RegisterHotkey(&config.General.GlobalHotkey); err != nil {
|
||||
hs.logger.Error("failed to register startup hotkey", "error", err)
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// RegisterHotkey 注册全局热键
|
||||
func (hs *HotkeyService) RegisterHotkey(hotkey *models.HotkeyCombo) error {
|
||||
if !hs.isValidHotkey(hotkey) {
|
||||
return &HotkeyError{"validate", fmt.Errorf("invalid hotkey combination")}
|
||||
}
|
||||
|
||||
hs.mu.Lock()
|
||||
defer hs.mu.Unlock()
|
||||
|
||||
// 取消现有热键
|
||||
if hs.isRegistered.Load() {
|
||||
hs.unregisterInternal()
|
||||
}
|
||||
|
||||
keyCode := hs.keyToX11KeyCode(hotkey.Key)
|
||||
if keyCode == 0 {
|
||||
return &HotkeyError{"convert_key", fmt.Errorf("unsupported key: %s", hotkey.Key)}
|
||||
}
|
||||
|
||||
modifiers := hs.buildX11Modifiers(hotkey)
|
||||
|
||||
result := int(C.registerGlobalHotkey(C.int(keyCode), C.uint(modifiers)))
|
||||
if result == 0 {
|
||||
return &HotkeyError{"register", fmt.Errorf("failed to register hotkey")}
|
||||
}
|
||||
|
||||
// 启动监听器
|
||||
ctx, cancel := context.WithCancel(hs.ctx)
|
||||
hs.wg.Add(1)
|
||||
|
||||
ready := make(chan error, 1)
|
||||
go hs.hotkeyListener(ctx, ready)
|
||||
|
||||
// 等待启动完成
|
||||
select {
|
||||
case err := <-ready:
|
||||
if err != nil {
|
||||
cancel()
|
||||
return &HotkeyError{"start_listener", err}
|
||||
}
|
||||
case <-time.After(time.Second):
|
||||
cancel()
|
||||
return &HotkeyError{"start_listener", fmt.Errorf("timeout")}
|
||||
}
|
||||
|
||||
hs.currentHotkey = hotkey
|
||||
hs.isRegistered.Store(true)
|
||||
hs.setCancelFunc(cancel)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// UnregisterHotkey 取消注册全局热键
|
||||
func (hs *HotkeyService) UnregisterHotkey() error {
|
||||
hs.mu.Lock()
|
||||
defer hs.mu.Unlock()
|
||||
return hs.unregisterInternal()
|
||||
}
|
||||
|
||||
// unregisterInternal 内部取消注册（无锁）
|
||||
func (hs *HotkeyService) unregisterInternal() error {
|
||||
if !hs.isRegistered.Load() {
|
||||
return nil
|
||||
}
|
||||
|
||||
// 原子地获取并调用cancel函数
|
||||
if cancel := hs.getCancelFunc(); cancel != nil {
|
||||
cancel()
|
||||
hs.wg.Wait()
|
||||
}
|
||||
|
||||
result := int(C.unregisterGlobalHotkey())
|
||||
if result == 0 {
|
||||
return &HotkeyError{"unregister", fmt.Errorf("failed to unregister hotkey")}
|
||||
}
|
||||
|
||||
hs.currentHotkey = nil
|
||||
hs.isRegistered.Store(false)
|
||||
hs.clearCancelFunc()
|
||||
return nil
|
||||
}
|
||||
|
||||
// UpdateHotkey 更新热键配置
|
||||
func (hs *HotkeyService) UpdateHotkey(enable bool, hotkey *models.HotkeyCombo) error {
|
||||
if enable {
|
||||
return hs.RegisterHotkey(hotkey)
|
||||
}
|
||||
return hs.UnregisterHotkey()
|
||||
}
|
||||
|
||||
// hotkeyListener 热键监听器
|
||||
func (hs *HotkeyService) hotkeyListener(ctx context.Context, ready chan<- error) {
|
||||
defer hs.wg.Done()
|
||||
|
||||
// 优化轮询频率从100ms改为50ms，提高响应性
|
||||
ticker := time.NewTicker(50 * time.Millisecond)
|
||||
defer ticker.Stop()
|
||||
|
||||
ready <- nil // 标记准备就绪
|
||||
|
||||
for {
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
return
|
||||
case <-ticker.C:
|
||||
if int(C.checkHotkeyEvent()) == 1 {
|
||||
hs.toggleWindow()
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// toggleWindow 切换窗口显示状态
|
||||
func (hs *HotkeyService) toggleWindow() {
|
||||
mainWindow := hs.windowHelper.MustGetMainWindow()
|
||||
if mainWindow == nil {
|
||||
hs.logger.Error("main window not found")
|
||||
return
|
||||
}
|
||||
|
||||
// 检查主窗口是否可见
|
||||
if mainWindow.IsVisible() {
|
||||
// 如果主窗口可见，隐藏所有窗口
|
||||
hs.hideAllWindows()
|
||||
} else {
|
||||
// 如果主窗口不可见，显示所有窗口
|
||||
hs.showAllWindows()
|
||||
}
|
||||
}
|
||||
|
||||
// isWindowVisible 检查窗口是否可见
|
||||
func (hs *HotkeyService) isWindowVisible(window *application.WebviewWindow) bool {
|
||||
return window.IsVisible()
|
||||
}
|
||||
|
||||
// hideAllWindows 隐藏所有窗口
|
||||
func (hs *HotkeyService) hideAllWindows() {
|
||||
// 隐藏所有子窗口
|
||||
if hs.windowService != nil {
|
||||
openWindows := hs.windowService.GetOpenWindows()
|
||||
for _, windowInfo := range openWindows {
|
||||
windowInfo.Hide()
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
// showAllWindows 显示所有窗口
|
||||
func (hs *HotkeyService) showAllWindows() {
|
||||
// 显示所有子窗口
|
||||
if hs.windowService != nil {
|
||||
openWindows := hs.windowService.GetOpenWindows()
|
||||
for _, windowInfo := range openWindows {
|
||||
windowInfo.Show()
|
||||
windowInfo.Restore()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// keyToX11KeyCode 键名转X11键码
|
||||
func (hs *HotkeyService) keyToX11KeyCode(key string) int {
|
||||
cKey := C.CString(key)
|
||||
defer C.free(unsafe.Pointer(cKey))
|
||||
return int(C.getX11Keycode(cKey))
|
||||
}
|
||||
|
||||
// buildX11Modifiers 构建X11修饰符
|
||||
func (hs *HotkeyService) buildX11Modifiers(hotkey *models.HotkeyCombo) uint {
|
||||
var modifiers uint = 0
|
||||
|
||||
if hotkey.Ctrl {
|
||||
modifiers |= 0x04 // ControlMask
|
||||
}
|
||||
if hotkey.Shift {
|
||||
modifiers |= 0x01 // ShiftMask
|
||||
}
|
||||
if hotkey.Alt {
|
||||
modifiers |= 0x08 // Mod1Mask (Alt)
|
||||
}
|
||||
if hotkey.Win {
|
||||
modifiers |= 0x40 // Mod4Mask (Super/Win)
|
||||
}
|
||||
|
||||
return modifiers
|
||||
}
|
||||
|
||||
// isValidHotkey 验证热键组合
|
||||
func (hs *HotkeyService) isValidHotkey(hotkey *models.HotkeyCombo) bool {
|
||||
if hotkey == nil || hotkey.Key == "" {
|
||||
return false
|
||||
}
|
||||
// 至少需要一个修饰键
|
||||
if !hotkey.Ctrl && !hotkey.Shift && !hotkey.Alt && !hotkey.Win {
|
||||
return false
|
||||
}
|
||||
return hs.keyToX11KeyCode(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 {
|
||||
return hs.isRegistered.Load()
|
||||
}
|
||||
|
||||
// ServiceShutdown 关闭服务
|
||||
func (hs *HotkeyService) ServiceShutdown() error {
|
||||
// 原子地获取并调用cancel函数
|
||||
if cancel := hs.getCancelFunc(); cancel != nil {
|
||||
cancel()
|
||||
}
|
||||
hs.wg.Wait()
|
||||
C.closeX11Display()
|
||||
return nil
|
||||
}
|
||||
@@ -68,7 +68,7 @@ func NewServiceManager() *ServiceManager {
|
||||
systemService := NewSystemService(logger)
|
||||
|
||||
// 初始化热键服务
|
||||
hotkeyService := NewHotkeyService(configService, windowService, logger)
|
||||
hotkeyService := NewHotkeyService(configService, logger)
|
||||
|
||||
// 初始化对话服务
|
||||
dialogService := NewDialogService(logger)
|
||||
|
||||
@@ -80,14 +80,14 @@ func (wh *WindowHelper) AutoShowMainWindow() {
|
||||
}
|
||||
}
|
||||
|
||||
// GetDocumentWindow 根据文档ID获取窗口（利用 Wails3 的 WindowManager）
|
||||
// GetDocumentWindow 根据文档ID获取窗口
|
||||
func (wh *WindowHelper) GetDocumentWindow(documentID int64) (application.Window, bool) {
|
||||
app := application.Get()
|
||||
windowName := strconv.FormatInt(documentID, 10)
|
||||
return app.Window.GetByName(windowName)
|
||||
}
|
||||
|
||||
// GetAllDocumentWindows 获取所有文档窗口（排除主窗口）
|
||||
// GetAllDocumentWindows 获取所有文档窗口
|
||||
func (wh *WindowHelper) GetAllDocumentWindows() []application.Window {
|
||||
app := application.Get()
|
||||
allWindows := app.Window.GetAll()
|
||||
|
||||
Reference in New Issue
Block a user