se-launcher/CringeLauncher/ImGuiHandler.cs

using System.Runtime.CompilerServices;
using System.Runtime.InteropServices;
using System.Runtime.InteropServices.Marshalling;
using System.Runtime.Versioning;
using Windows.Win32;
using Windows.Win32.Foundation;
using Windows.Win32.UI.WindowsAndMessaging;
using CringePlugins.Abstractions;
using CringePlugins.Render;
using ImGuiNET;
using SharpDX.Direct3D11;
using static ImGuiNET.ImGui;
using VRage;

namespace CringeLauncher;

internal class ImGuiHandler : IDisposable
{
    private DeviceContext? _deviceContext;
    private static nint _wndproc;

    public bool BlockMouse { get; private set; }
    public bool BlockKeys { get; private set; }
    public bool DrawMouse { get; private set; }


    public static ImGuiHandler? Instance;
    
    public static RenderTargetView? Rtv;
    
    private readonly IRootRenderComponent _renderHandler = new RenderHandler();
    private static bool _init;

    public unsafe void Init(nint windowHandle, Device1 device, DeviceContext deviceContext)
    {
        _deviceContext = deviceContext;

        CreateContext();
        
        var io = GetIO();
        
        var path = Path.Join(Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData), "CringeLauncher", "imgui.ini");

        io.NativePtr->IniFilename = AnsiStringMarshaller.ConvertToUnmanaged(path);

        io.ConfigWindowsMoveFromTitleBarOnly = true;
        io.ConfigFlags |= ImGuiConfigFlags.DockingEnable | ImGuiConfigFlags.ViewportsEnable;
        
        ImGui_ImplWin32_Init(windowHandle);
        ImGui_ImplDX11_Init(device.NativePointer, deviceContext.NativePointer);
        _init = true;
    }

    public static void HookWindow(HWND windowHandle)
    {
        _wndproc = PInvoke.GetWindowLongPtr(windowHandle, WINDOW_LONG_PTR_INDEX.GWL_WNDPROC);

        unsafe
        {
            delegate* unmanaged[Stdcall]<HWND, int, nint, nint, int> wndProcHook = &WndProcHook;
            
            PInvoke.SetWindowLongPtr(windowHandle, WINDOW_LONG_PTR_INDEX.GWL_WNDPROC, (nint)wndProcHook);
        }
    }

    public void DoRender()
    {
        if (Rtv is null)
            return;
        
        ImGui_ImplDX11_NewFrame();
        ImGui_ImplWin32_NewFrame();
        NewFrame();

        var io = GetIO();
        BlockMouse = io.WantCaptureMouse;
        BlockKeys = io.WantTextInput;
        DrawMouse = io.MouseDrawCursor;
        _renderHandler.OnFrame();
        
        Render();
        
        _deviceContext!.ClearState();
        _deviceContext.OutputMerger.SetRenderTargets(Rtv);
        
        ImGui_ImplDX11_RenderDrawData(GetDrawData());
        
        UpdatePlatformWindows();
        RenderPlatformWindowsDefault();
    }
    
    [UnmanagedCallersOnly(CallConvs = [typeof(CallConvStdcall)])]
    private static unsafe int WndProcHook(HWND hWnd, int msg, nint wParam, nint lParam)
    {
        //ignore input if mouse is hidden
        if (MyVRage.Platform?.Input?.ShowCursor == false && Instance?.DrawMouse != true)
            return CallWindowProc(_wndproc, hWnd, msg, wParam, lParam);

        var hookResult = ImGui_ImplWin32_WndProcHandler(hWnd, msg, wParam, lParam);

        if (hookResult != 0)
            return hookResult;

        var io = GetIO();

        if (!_init)
            return CallWindowProc(_wndproc, hWnd, msg, wParam, lParam);

        var blockMessage = (msg is >= 256 and <= 265 && io.WantTextInput)
            || (msg is >= 512 and <= 526 && io.WantCaptureMouse);

        return blockMessage ? hookResult : CallWindowProc(_wndproc, hWnd, msg, wParam, lParam);
    }

    [DllImport("USER32.dll", ExactSpelling = true, EntryPoint = "CallWindowProcW")]
    [DefaultDllImportSearchPaths(DllImportSearchPath.System32)]
    [SupportedOSPlatform("windows5.0")]
    private static extern int CallWindowProc(nint lpPrevWndFunc, HWND hWnd, int msg, nint wParam, nint lParam);

    public void Dispose()
    {
        _deviceContext?.Dispose();
        _renderHandler.Dispose();
    }
}