重构钩子服务

Author: JiuLing-zhang

src/ComputerLock/Platforms/BaseHook.cs

@@ -0,0 +1,40 @@
+using System.Diagnostics;
+
+namespace ComputerLock.Platforms;
+
+/// <summary>
+/// Windows 系统输入钩子基类
+/// </summary>
+internal abstract class WindowsInputHook : IDisposable
+{
+    protected int _hookId;
+    protected readonly WinApi.HookDelegate _hookCallback;
+
+    protected WindowsInputHook()
+    {
+        _hookCallback = HookCallback;
+    }
+
+    protected abstract int HookType { get; }
+
+    protected abstract int HookCallback(int nCode, int wParam, IntPtr lParam);
+
+    /// <summary>
+    /// 安装系统钩子
+    /// </summary>
+    public void InstallHook()
+    {
+        using Process curProcess = Process.GetCurrentProcess();
+        string? moduleName = curProcess.MainModule?.ModuleName;
+        if (moduleName == null)
+        {
+            return;
+        }
+        _hookId = WinApi.SetWindowsHookEx(HookType, _hookCallback, WinApi.GetModuleHandle(moduleName), 0);
+    }
+
+    public void Dispose()
+    {
+        WinApi.UnhookWindowsHookEx(_hookId);
+    }
+}

src/ComputerLock/Platforms/MouseHook.cs

@@ -1,8 +1,23 @@
-namespace ComputerLock.Platforms;
-internal class MouseHook
+using System.Runtime.InteropServices;
+
+namespace ComputerLock.Platforms;
+internal class MouseHook : WindowsInputHook
 {
     private int _cursorCount = 0;
     private Random _random = new Random();
+    private bool _isAutoInput;
+    protected override int HookType => WinApi.WH_MOUSE_LL;
+
+    public event EventHandler? OnUserInput;
+    protected override int HookCallback(int nCode, int wParam, IntPtr lParam)
+    {
+        if (nCode >= 0)
+        {
+            // 处理鼠标事件
+            HandleMouseEvent(wParam);
+        }
+        return WinApi.CallNextHookEx(_hookId, nCode, wParam, lParam);
+    }
 
     /// <summary>
     /// 隐藏鼠标光标。
@@ -49,11 +64,31 @@ public void ResetCursorState()
     /// </summary>
     public void MoveAndClick()
     {
+        _isAutoInput = true;
         var x = _random.Next(0, 100);
         var y = _random.Next(0, 100);
 
         var p = new Point(x, y);
         WinApi.SetCursorPos((int)p.X, (int)p.Y);
         WinApi.mouse_event(WinApi.MOUSEEVENTF_RIGHTDOWN | WinApi.MOUSEEVENTF_RIGHTUP, x, y, 0, 0);
+        _isAutoInput = false;
+    }
+
+    /// <summary>
+    /// 处理鼠标事件
+    /// </summary>
+    private void HandleMouseEvent(int wParam)
+    {
+        if (_isAutoInput)
+        {
+            return;
+        }
+
+        // 只处理鼠标按下和释放事件
+        if (wParam == WinApi.WM_LBUTTONDOWN || wParam == WinApi.WM_LBUTTONUP ||
+            wParam == WinApi.WM_RBUTTONDOWN || wParam == WinApi.WM_RBUTTONUP)
+        {
+            OnUserInput?.Invoke(this, EventArgs.Empty);
+        }
     }
 }

src/ComputerLock/Platforms/SystemKeyHook.cs

@@ -1,34 +1,23 @@
-using System.Diagnostics;
-using System.Runtime.InteropServices;
+using System.Runtime.InteropServices;
 
 namespace ComputerLock.Platforms;
-internal class SystemKeyHook : IDisposable
+/// <summary>
+/// 系统按键钩子，用于禁用系统按键
+/// </summary>
+internal class SystemKeyHook : WindowsInputHook
 {
-    private int _hookId;
-    private readonly WinApi.HookDelegate _hookCallback;
     private Hotkey? _ignoreHotkey; // 使用单个Hotkey变量来存储需要忽略的快捷键
 
-    public SystemKeyHook()
-    {
-        _hookCallback = KeyboardHookCallback;
-    }
+    protected override int HookType => WinApi.WH_KEYBOARD_LL;
 
-    public void DisableSystemKey()
-    {
-        using Process curProcess = Process.GetCurrentProcess();
-        string? moduleName = curProcess.MainModule?.ModuleName;
-        if (moduleName == null)
-        {
-            return;
-        }
-        _hookId = WinApi.SetWindowsHookEx(WinApi.WH_KEYBOARD_LL, _hookCallback, WinApi.GetModuleHandle(moduleName), 0);
-    }
+    public event EventHandler? OnUserInput;
 
     public void SetIgnoreHotkey(Hotkey hotKey)
     {
         _ignoreHotkey = hotKey;
     }
-    private int KeyboardHookCallback(int nCode, int wParam, IntPtr lParam)
+
+    protected override int HookCallback(int nCode, int wParam, IntPtr lParam)
     {
         if (nCode >= 0)
         {
@@ -38,6 +27,7 @@ private int KeyboardHookCallback(int nCode, int wParam, IntPtr lParam)
             {
                 if (IsSystemKey(vkCode) && (wParam == WinApi.WM_KEYDOWN || wParam == WinApi.WM_SYSKEYDOWN))
                 {
+                    OnUserInput?.Invoke(this, EventArgs.Empty);
                     return 1; // 阻止事件传递
                 }
                 return WinApi.CallNextHookEx(_hookId, nCode, wParam, lParam); // 其他按键放行
@@ -112,10 +102,4 @@ private bool IsModifierRequired(int vkCode)
         }
         return false;
     }
-
-    public void Dispose()
-    {
-        _ignoreHotkey = null;
-        WinApi.UnhookWindowsHookEx(_hookId);
-    }
 }

src/ComputerLock/Platforms/WinApi.cs

@@ -24,6 +24,13 @@ internal static class WinApi
     public const int MOUSEEVENTF_RIGHTDOWN = 0x0008;
     public const int MOUSEEVENTF_RIGHTUP = 0x0010;
 
+    public const int WH_MOUSE_LL = 14;
+    public const int WM_MOUSEMOVE = 0x0200;
+    public const int WM_LBUTTONDOWN = 0x0201;
+    public const int WM_LBUTTONUP = 0x0202;
+    public const int WM_RBUTTONDOWN = 0x0204;
+    public const int WM_RBUTTONUP = 0x0205;
+
     [DllImport("user32.dll")]
     public static extern int ShowCursor(bool bShow);