支持多个快捷键的注册

Author: JiuLing-zhang

src/ComputerLock/Enums/HotkeyType.cs

@@ -0,0 +1,7 @@
+namespace ComputerLock.Enums;
+
+public enum HotkeyType
+{
+    Lock = 90,
+    Unlock = 90,
+}

src/ComputerLock/Pages/Index.razor.cs

@@ -48,7 +48,7 @@ protected override async Task OnInitializedAsync()
             RegisterHotkey();
         }
 
-        HotkeyHook.HotkeyPressed += () =>
+        HotkeyHook.HotkeyPressed += (id) =>
         {
             if (GlobalLockService.IsLocked)
             {
@@ -172,7 +172,7 @@ public void RegisterHotkey()
             if (AppSettings.LockHotkey != null)
             {
                 Logger.Write("注册锁屏热键");
-                HotkeyHook.Register(AppSettings.LockHotkey);
+                HotkeyHook.Register((int)HotkeyType.Lock, AppSettings.LockHotkey);
             }
         }
         catch (Exception ex)
@@ -187,7 +187,7 @@ public void UnregisterHotkey()
         try
         {
             Logger.Write("释放锁屏热键");
-            HotkeyHook.Unregister();
+            HotkeyHook.Unregister((int)HotkeyType.Lock);
         }
         catch (Exception ex)
         {

src/ComputerLock/Platforms/HotkeyHook.cs

@@ -7,20 +7,19 @@ namespace ComputerLock.Platforms;
 /// </summary>
 public class HotkeyHook : IDisposable
 {
+    private List<int> ids = new List<int>();
+
     [DllImport("user32.dll")]
     private static extern bool RegisterHotKey(IntPtr hWnd, int id, uint fsModifiers, uint vk);
 
     [DllImport("user32.dll")]
     private static extern bool UnregisterHotKey(IntPtr hWnd, int id);
 
-    private const int HotkeyId = 90;
-    private bool _isRegistered;
-
-    public event Action? HotkeyPressed;
+    public event Action<int>? HotkeyPressed;
 
     private sealed class HotkeyNativeWindow : NativeWindow
     {
-        public event Action? OnHotkeyPressed;
+        public event Action<int>? OnHotkeyPressed;
 
         public HotkeyNativeWindow()
         {
@@ -31,10 +30,7 @@ protected override void WndProc(ref Message m)
         {
             if (m.Msg == 0x0312) // WM_HOTKEY
             {
-                if (m.WParam.ToInt32() == HotkeyId)
-                {
-                    OnHotkeyPressed?.Invoke();
-                }
+                OnHotkeyPressed?.Invoke(m.WParam.ToInt32());
             }
             else
             {
@@ -48,38 +44,32 @@ protected override void WndProc(ref Message m)
     public HotkeyHook()
     {
         _nativeWindow = new HotkeyNativeWindow();
-        _nativeWindow.OnHotkeyPressed += () => HotkeyPressed?.Invoke();
+        _nativeWindow.OnHotkeyPressed += (id) => HotkeyPressed?.Invoke(id);
     }
 
     /// <summary>
     /// 注册快捷键
     /// </summary>
-    public void Register(Hotkey hotKey)
+    public void Register(int id, Hotkey hotKey)
     {
-        if (_isRegistered)
+        var success = RegisterHotKey(_nativeWindow.Handle, id, (uint)hotKey.Modifiers, (uint)hotKey.Key);
+        if (!success)
         {
-            Unregister();
+            throw new Exception($"注册快捷键失败。id({id})");
         }
 
-        var success = RegisterHotKey(_nativeWindow.Handle, HotkeyId, (uint)hotKey.Modifiers, (uint)hotKey.Key);
-        if (!success)
+        if (!ids.Contains(id))
         {
-            throw new Exception("注册快捷键失败");
+            ids.Add(id);
         }
-        _isRegistered = success;
     }
 
     /// <summary>
     /// 注销快捷键
     /// </summary>
-    public void Unregister()
+    public void Unregister(int id)
     {
-        if (!_isRegistered)
-        {
-            return;
-        }
-        UnregisterHotKey(_nativeWindow.Handle, HotkeyId);
-        _isRegistered = false;
+        UnregisterHotKey(_nativeWindow.Handle, id);
     }
 
     public void Dispose()
@@ -95,7 +85,10 @@ protected virtual void Dispose(bool disposing)
             return;
         }
 
-        Unregister();
+        foreach (var id in ids)
+        {
+            Unregister(id);
+        }
         _nativeWindow.DestroyHandle();
     }