修正HotKeyAction

Author: nahida-mouo

src/Snap.Hutao.Remastered.Native/CustomImplements.h

@@ -30,12 +30,12 @@ template<typename T>
 struct IsIUnknown : std::false_type {};
 
 template<>
-struct IsIUnknown<IUnknown> : std::true_type {};
+struct IsIUnknown<::IUnknown> : std::true_type {};
 
 // Helper to get IUnknown pointer from interface
 template<typename Interface>
-inline IUnknown* GetIUnknownFromInterface(Interface* pInterface) {
-    return static_cast<IUnknown*>(static_cast<void*>(pInterface));
+inline ::IUnknown* GetIUnknownFromInterface(Interface* pInterface) {
+    return static_cast<::IUnknown*>(static_cast<void*>(pInterface));
 }
 
 // CustomImplements class template
@@ -55,9 +55,9 @@ class CustomImplements : public Interfaces... {
         *ppvObject = nullptr;
 
         // Check for IUnknown
-        if (IsEqualIID(riid, __uuidof(IUnknown))) {
+        if (IsEqualIID(riid, __uuidof(::IUnknown))) {
             // 使用reinterpret_cast避免多重继承导致的转换不明确
-            *ppvObject = reinterpret_cast<IUnknown*>(this);
+            *ppvObject = reinterpret_cast<::IUnknown*>(this);
             AddRef();
             return S_OK;
         }

src/Snap.Hutao.Remastered.Native/HutaoNative.cpp

@@ -19,6 +19,7 @@
 #include "types.h"
 #include "CustomImplements.h"
 #include "Error.h"
+#include "HutaoNativeHotKeyActionCallback.h"
 #include <Windows.h>
 #include <hstring.h>
 #include <winternl.h>
@@ -151,11 +152,11 @@ HRESULT STDMETHODCALLTYPE HutaoNative::MakeHotKeyAction(HutaoNativeHotKeyActionK
 {
     AssertNonNullAndReturn(ppv);
 
-    // Convert nint to appropriate types
-    WNDPROC wndProc = reinterpret_cast<WNDPROC>(callback);
-    LONG_PTR userDataPtr = static_cast<LONG_PTR>(userData);
+    HutaoNativeHotKeyActionCallback cb = {};
+    cb.value = reinterpret_cast<HutaoNativeHotKeyActionCallbackFunc>(callback);
+    GCHandle userDataHandle = static_cast<GCHandle>(userData);
 
-    hutao::com_ptr<IHutaoNativeHotKeyAction> hotKeyAction = hutao::make_com_ptr<HutaoNativeHotKeyAction>(kind, wndProc, userDataPtr);
+    hutao::com_ptr<IHutaoNativeHotKeyAction> hotKeyAction = hutao::make_com_ptr<HutaoNativeHotKeyAction>(kind, cb, userDataHandle);
     *ppv = hotKeyAction.detach();
 
     return S_OK;

src/Snap.Hutao.Remastered.Native/HutaoNativeHotKeyAction.cpp

@@ -1,39 +1,38 @@
 #include "HutaoNativeHotKeyAction.h"
 #include "Error.h"
 #include <Windows.h>
+#include <chrono>
 
 const wchar_t* WINDOW_CLASS_NAME = L"HutaoNativeHotKeyActionWindowClass";
 const UINT WM_HOTKEY_MESSAGE = WM_APP + 2;
 
 UINT HutaoNativeHotKeyAction::s_nextHotKeyId = 0x4000; // 从0x4000开始，避免与系统热键冲突
 
-HutaoNativeHotKeyAction::HutaoNativeHotKeyAction()
-    : m_callback(nullptr)
-    , m_modifiers(0)
-    , m_vk(0)
-    , m_enabled(false)
-    , m_hotKeyId(0)
-    , m_hWnd(nullptr)
-{
-    m_hotKeyId = static_cast<int>(++s_nextHotKeyId);
-}
-
-HutaoNativeHotKeyAction::HutaoNativeHotKeyAction(HutaoNativeHotKeyActionKind kind, WNDPROC callback, LONG_PTR userData)
+HutaoNativeHotKeyAction::HutaoNativeHotKeyAction(HutaoNativeHotKeyActionKind kind, HutaoNativeHotKeyActionCallback callback, GCHandle userData)
     : m_callback(callback)
+    , m_userData(userData)
+    , m_kind(kind)
     , m_modifiers(0)
     , m_vk(0)
     , m_enabled(false)
     , m_hotKeyId(0)
     , m_hWnd(nullptr)
+    , m_clicking(false)
+    , m_clickIntervalMs(50) // default 50ms interval
+    , m_keyPressing(false)
+    , m_keyIntervalMs(100) // default 100ms interval for key press
+    , m_active(false)
 {
     m_hotKeyId = static_cast<int>(++s_nextHotKeyId);
-    (void)kind;
-    (void)userData;
 }
 
 HutaoNativeHotKeyAction::~HutaoNativeHotKeyAction()
 {
     SetIsEnabled(FALSE);
+
+    StopAutoClick();
+    StopAutoKeyPress();
+
     if (m_hWnd != nullptr)
     {
         DestroyWindow(m_hWnd);
@@ -71,7 +70,7 @@ HWND HutaoNativeHotKeyAction::CreateMessageWindow()
         nullptr,
         nullptr,
         GetModuleHandle(nullptr),
-        this // 将this指针传递给窗口
+        reinterpret_cast<LPVOID>(this)
     );
 
     return hWnd;
@@ -89,9 +88,29 @@ LRESULT CALLBACK HutaoNativeHotKeyAction::WndProc(HWND hWnd, UINT message, WPARA
     HutaoNativeHotKeyAction* pThis = reinterpret_cast<HutaoNativeHotKeyAction*>(GetWindowLongPtrW(hWnd, GWLP_USERDATA));
     if (pThis != nullptr)
     {
-        if (message == WM_HOTKEY && pThis->m_callback != nullptr)
+        if (message == WM_HOTKEY)
         {
-            pThis->m_callback(hWnd, message, wParam, lParam);
+            // Toggle active state on hotkey press
+            bool newState = !pThis->m_active.load();
+            pThis->m_active.store(newState);
+
+            // Notify managed side about toggled state
+            if (pThis->m_callback.value != nullptr)
+            {
+                BOOL isOn = newState ? TRUE : FALSE;
+                pThis->m_callback.value(isOn, pThis->m_userData);
+            }
+
+            // Start/stop auto actions based on kind
+            if (pThis->m_kind == HutaoNativeHotKeyActionKind::MouseClickRepeatForever)
+            {
+                if (pThis->m_active.load()) pThis->StartAutoClick(); else pThis->StopAutoClick();
+            }
+            else if (pThis->m_kind == HutaoNativeHotKeyActionKind::KeyPressRepeatForever)
+            {
+                if (pThis->m_active.load()) pThis->StartAutoKeyPress(); else pThis->StopAutoKeyPress();
+            }
+
             return 0;
         }
     }
@@ -117,6 +136,87 @@ void HutaoNativeHotKeyAction::RegisterHotKey()
     ::RegisterHotKey(m_hWnd, m_hotKeyId, m_modifiers, m_vk);
 }
 
+// Auto-click implementation
+void HutaoNativeHotKeyAction::StartAutoClick()
+{
+    bool expected = false;
+    if (!m_clicking.compare_exchange_strong(expected, true))
+        return; // already clicking
+
+    // Launch thread
+    m_clickThread = std::thread([this]() {
+        // Continue clicking until stopped
+        while (m_clicking.load())
+        {
+            // Simulate left mouse down and up
+            INPUT inputs[2];
+            ZeroMemory(inputs, sizeof(inputs));
+
+            inputs[0].type = INPUT_MOUSE;
+            inputs[0].mi.dwFlags = MOUSEEVENTF_LEFTDOWN;
+
+            inputs[1].type = INPUT_MOUSE;
+            inputs[1].mi.dwFlags = MOUSEEVENTF_LEFTUP;
+
+            SendInput(2, inputs, sizeof(INPUT));
+
+            std::this_thread::sleep_for(std::chrono::milliseconds(m_clickIntervalMs));
+        }
+    });
+}
+
+void HutaoNativeHotKeyAction::StopAutoClick()
+{
+    bool expected = true;
+    if (!m_clicking.compare_exchange_strong(expected, false))
+        return; // not clicking
+
+    if (m_clickThread.joinable())
+    {
+        m_clickThread.join();
+    }
+}
+
+// Auto-keypress implementation
+void HutaoNativeHotKeyAction::StartAutoKeyPress()
+{
+    bool expected = false;
+    if (!m_keyPressing.compare_exchange_strong(expected, true))
+        return; // already pressing
+
+    m_keyThread = std::thread([this]() {
+        // Repeatedly send configured vk or 'F'
+        WORD vkToSend = (WORD)m_vk ? (WORD)m_vk : 0x46;
+        INPUT inputs[2];
+        ZeroMemory(inputs, sizeof(inputs));
+        inputs[0].type = INPUT_KEYBOARD;
+        inputs[0].ki.wVk = vkToSend;
+        inputs[0].ki.dwFlags = 0; // key down
+
+        inputs[1].type = INPUT_KEYBOARD;
+        inputs[1].ki.wVk = vkToSend;
+        inputs[1].ki.dwFlags = KEYEVENTF_KEYUP; // key up
+
+        while (m_keyPressing.load())
+        {
+            SendInput(2, inputs, sizeof(INPUT));
+            std::this_thread::sleep_for(std::chrono::milliseconds(m_keyIntervalMs));
+        }
+    });
+}
+
+void HutaoNativeHotKeyAction::StopAutoKeyPress()
+{
+    bool expected = true;
+    if (!m_keyPressing.compare_exchange_strong(expected, false))
+        return; // not pressing
+
+    if (m_keyThread.joinable())
+    {
+        m_keyThread.join();
+    }
+}
+
 HRESULT STDMETHODCALLTYPE HutaoNativeHotKeyAction::Update(int modifiers, uint vk)
 {
     bool wasEnabled = m_enabled;
@@ -180,5 +280,6 @@ HRESULT STDMETHODCALLTYPE HutaoNativeHotKeyAction::SetIsEnabled(BOOL isEnabled)
     }
 
     m_enabled = newEnabled;
+
     return S_OK;
 }

src/Snap.Hutao.Remastered.Native/HutaoNativeHotKeyAction.h

@@ -3,15 +3,17 @@
 #include "IHutaoNativeHotKeyAction_h.h"
 #include "Types_h.h"
 #include "CustomImplements.h"
+#include "HutaoNativeHotKeyActionCallback.h"
 #include <Windows.h>
+#include <thread>
+#include <atomic>
 
 class HutaoNativeHotKeyAction : public hutao::CustomImplements<
     HutaoNativeHotKeyAction,
     IHutaoNativeHotKeyAction>
 {
 public:
-    HutaoNativeHotKeyAction();
-	HutaoNativeHotKeyAction(HutaoNativeHotKeyActionKind kind, WNDPROC callback, LONG_PTR userData);
+	HutaoNativeHotKeyAction(HutaoNativeHotKeyActionKind kind, HutaoNativeHotKeyActionCallback callback, GCHandle userData);
     ~HutaoNativeHotKeyAction();
 
     // IHutaoNativeHotKeyAction methods
@@ -20,17 +22,40 @@ class HutaoNativeHotKeyAction : public hutao::CustomImplements<
     virtual HRESULT STDMETHODCALLTYPE SetIsEnabled(BOOL isEnabled) override;
 
 private:
-	WNDPROC m_callback;
+	HutaoNativeHotKeyActionCallback m_callback;
+	GCHandle m_userData;
+	HutaoNativeHotKeyActionKind m_kind;
     int m_modifiers;
     uint m_vk;
     bool m_enabled;
     int m_hotKeyId;
     HWND m_hWnd;
     static UINT s_nextHotKeyId;
 
-    static LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam);
-    static ATOM RegisterWindowClass();
-    HWND CreateMessageWindow();
+    // running state toggled by WM_HOTKEY
+    std::atomic<bool> m_active;
+
     void UnregisterHotKey();
     void RegisterHotKey();
+
+    // Window helpers
+    static ATOM RegisterWindowClass();
+    HWND CreateMessageWindow();
+    static LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam);
+
+    // Auto-click support
+    std::thread m_clickThread;
+    std::atomic<bool> m_clicking;
+    int m_clickIntervalMs;
+
+    void StartAutoClick();
+    void StopAutoClick();
+
+    // Auto-keypress support (for KeyPressRepeatForever)
+    std::thread m_keyThread;
+    std::atomic<bool> m_keyPressing;
+    int m_keyIntervalMs;
+
+    void StartAutoKeyPress();
+    void StopAutoKeyPress();
 };

src/Snap.Hutao.Remastered.Native/HutaoNativeHotKeyActionCallback.h

@@ -0,0 +1,11 @@
+#pragma once
+
+#include "Types.h"
+#include <Windows.h>
+
+typedef void (__stdcall *HutaoNativeHotKeyActionCallbackFunc)(BOOL, GCHandle);
+
+struct HutaoNativeHotKeyActionCallback
+{
+    HutaoNativeHotKeyActionCallbackFunc value;
+};

src/Snap.Hutao.Remastered.Native/HutaoNativeNotifyIconCallback.h

@@ -3,7 +3,6 @@
 #include "Types.h"
 #include <Windows.h>
 
-// 通知图标回调种类枚举
 enum class HutaoNativeNotifyIconCallbackKind : int
 {
     TaskbarCreated = 0,
@@ -12,10 +11,8 @@ enum class HutaoNativeNotifyIconCallbackKind : int
     LeftButtonDoubleClick = 3,
 };
 
-// 通知图标回调函数指针类型
 typedef void (*HutaoNativeNotifyIconCallbackFunc)(HutaoNativeNotifyIconCallbackKind kind, RECT rect, POINT point, nint userData);
 
-// 通知图标回调结构体（模仿项目风格）
 struct HutaoNativeNotifyIconCallback
 {
     HutaoNativeNotifyIconCallbackFunc value;

src/Snap.Hutao.Remastered.Native/Snap.Hutao.Remastered.Native.vcxproj

@@ -70,6 +70,12 @@
     <Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
   </ImportGroup>
   <PropertyGroup Label="UserMacros" />
+  <PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
+    <LibraryPath>$(VC_LibraryPath_x64);$(WindowsSDK_LibraryPath_x64);$(OutDir);$(LibraryPath)</LibraryPath>
+  </PropertyGroup>
+  <PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">
+    <LibraryPath>$(VC_LibraryPath_x64);$(WindowsSDK_LibraryPath_x64);$(OutDir);$(LibraryPath)</LibraryPath>
+  </PropertyGroup>
   <ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
     <ClCompile>
       <WarningLevel>Level3</WarningLevel>
@@ -172,6 +178,7 @@
   <ItemGroup>
     <ClInclude Include="CustomImplements.h" />
     <ClInclude Include="Error.h" />
+    <ClInclude Include="HutaoNativeHotKeyActionCallback.h" />
     <ClInclude Include="HutaoNativeNotifyIconCallback.h" />
     <ClInclude Include="WindowUtils.h" />
     <ClInclude Include="DllInjectionUtils.h" />

src/Snap.Hutao.Remastered.Native/Snap.Hutao.Remastered.Native.vcxproj.filters

@@ -302,6 +302,9 @@
     <ClInclude Include="Error.h">
       <Filter>头文件</Filter>
     </ClInclude>
+    <ClInclude Include="HutaoNativeHotKeyActionCallback.h">
+      <Filter>头文件\core\struct</Filter>
+    </ClInclude>
   </ItemGroup>
   <ItemGroup>
     <Midl Include="IHutaoString.idl">