Add external overlay as fallback

Author: PicoShot

UnityInspector/UnityInspector.vcxproj

@@ -178,6 +178,9 @@
     <ClInclude Include="src\pch\pch.h" />
     <ClInclude Include="src\game\sdk\sdk.h" />
     <ClInclude Include="src\gui\window\window.h" />
+    <ClInclude Include="src\gui\window\window_finder.h" />
+    <ClInclude Include="src\gui\window\external_overlay.h" />
+    <ClInclude Include="src\gui\window\input_forwarder.h" />
   </ItemGroup>
   <ItemGroup>
     <ClCompile Include="src\dllmain.cpp" />
@@ -228,6 +231,9 @@
     <ClCompile Include="src\game\hooks\hooks.cpp" />
     <ClCompile Include="src\gui\menu\menu.cpp" />
     <ClCompile Include="src\gui\window\window.cpp" />
+    <ClCompile Include="src\gui\window\window_finder.cpp" />
+    <ClCompile Include="src\gui\window\external_overlay.cpp" />
+    <ClCompile Include="src\gui\window\input_forwarder.cpp" />
     <ClCompile Include="src\pch\pch.cpp">
       <PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">Create</PrecompiledHeader>
       <PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|x64'">Create</PrecompiledHeader>

UnityInspector/src/dllmain.cpp

@@ -2,6 +2,9 @@
 
 #include "game/core/core.h"
 #include "gui/window/window.h"
+#include "gui/window/window_finder.h"
+#include "gui/window/external_overlay.h"
+#include "gui/window/input_forwarder.h"
 
 static void Init(HMODULE hMod)
 {
@@ -42,23 +45,56 @@ static void Init(HMODULE hMod)
 
     LOG_INFO(X("initializing Overlay Hook."));
 
-    for (const auto& [method, name] : hookMethods) {
-        LOG_DEBUG(X("Trying {} DirectX Hook."), name);
-        if (dx_hook::Hk11::Build(Window::OnPresent, method)) {
+    for (const auto& [method, name] : hookMethods) 
+    {
+        if (dx_hook::Hk11::Build(Window::OnPresent, method)) 
+        {
             LOG_INFO(X("Successfully initialized {} DirectX Hook."), name);
             hookSuccess = true;
             break;
         }
-
-        LOG_ERROR(X("Failed to initialize {} DirectX Hook."), name);
     }
 
-    if (!hookSuccess) 
+    if (!hookSuccess)
     {
-        LOG_ERROR(X("All hooking methods failed!"));
-        goto exit;
+        HWND gameHwnd = nullptr;
+        for (uint8_t i = 0; i < 10 && !gameHwnd; i++) 
+        {
+            gameHwnd = WindowFinder::FindGameWindow();
+            if (!gameHwnd) std::this_thread::sleep_for(500ms);
+        }
+
+        if (!gameHwnd) {
+            LOG_ERROR(X("Failed to find game window for external overlay!"));
+            goto exit;
+        }
+
+        LOG_INFO(X("Found game window: {}"), reinterpret_cast<uint64_t>(gameHwnd));
+
+        const auto externalOverlay = std::make_unique<ExternalOverlay>();
+        if (!externalOverlay->Create(gameHwnd)) 
+        {
+            LOG_ERROR(X("Failed to create external overlay!"));
+            goto exit;
+        }
+
+        config->externalOverlay = externalOverlay.get();
+        InputForwarder::Initialize(externalOverlay->GetOverlayHwnd(), gameHwnd);
+        externalOverlay->SetInputCapture(config->ShowImGui);
+
+        LOG_INFO(X("External overlay initialized successfully."));
+        externalOverlay->RunRenderLoop();
+
+        config->externalOverlay = nullptr;
+        InputForwarder::Shutdown();
+        externalOverlay->Destroy();
+        LOG_INFO(X("External overlay shut down."));
+    }
+    else 
+    {
+        dx_hook::Hk11::SetWndProc(Window::MyWndProc);
     }
-    dx_hook::Hk11::SetWndProc(Window::MyWndProc);
+
     return;
 
 exit:

UnityInspector/src/game/config/config.h

@@ -1,6 +1,8 @@
 #pragma once
 #include "pch.h"
 
+struct ExternalOverlay;
+
 struct Config {
 private:
     struct Inspector {
@@ -14,5 +16,6 @@ struct Config {
     bool ShowImGui = true;
 	HMODULE gameAssembly = nullptr;
 	UR::Mode gameMode;
+    ExternalOverlay* externalOverlay = nullptr;
     Inspector inspector;
 };