// 全局变量
VTableHook g_dx11Hook;
ID3D11Device* g_pD3DDevice = nullptr;
ID3D11DeviceContext* g_pD3DContext = nullptr;
bool g_bInit = false;

// 初始化DX11 Hook
bool DX11HookInit(IDXGISwapChain* pSwapChain) {
    // 1. 初始化虚表Hook（挂钩交换链）
    if (!g_dx11Hook.Init(pSwapChain)) return false;

    // 2. 绑定Present函数（索引8）
    g_dx11Hook.Bind(8, &Hooked_Present);
    // 3. 绑定Reset函数（索引13，解决窗口缩放异常）
    g_dx11Hook.Bind(13, &Hooked_Reset);

    // 4. 获取D3D设备和上下文（用于渲染）
    pSwapChain->GetDevice(__uuidof(ID3D11Device), (void**)&g_pD3DDevice);
    g_pD3DDevice->GetImmediateContext(&g_pD3DContext);

    // 5. 初始化MGUI
    MGUI_Init(g_pD3DDevice, g_pD3DContext);
    return true;
}

// 原始Present函数指针
HRESULT(__stdcall* Original_Present)(IDXGISwapChain* pSwapChain, UINT SyncInterval, UINT Flags);

// 挂钩后的Present函数
HRESULT __stdcall Hooked_Present(IDXGISwapChain* pSwapChain, UINT SyncInterval, UINT Flags) {
    // 1. 首次调用初始化MGUI渲染目标
    if (!g_bInit) {
        MGUI_CreateRenderTarget(pSwapChain);
        g_bInit = true;
    }

    // 2. 开始MGUI渲染（在游戏渲染前插入）
    MGUI_BeginRender();
    // 自定义渲染内容（如绘制菜单、文本）
    MGUI_DrawText(100, 100, "D3D11内部绘制测试", 255, 0, 0);
    MGUI_DrawRect(200, 200, 300, 200, 0, 255, 0, 180);
    MGUI_EndRender();

    // 3. 调用原始Present，提交游戏渲染
    return Original_Present(pSwapChain, SyncInterval, Flags);
}

// MGUI初始化
void IMGUI_Init(ID3D11Device* pDevice, ID3D11DeviceContext* pContext) {
    // 初始化MGUI环境
    ImGui::CreateContext();
    ImGuiIO& io = ImGui::GetIO();
    io.IniFilename = nullptr; // 不保存配置文件

    // 初始化D3D渲染器
    ImGui_ImplWin32_Init(GetGameWindowHandle()); // 获取游戏窗口句柄
    ImGui_ImplDX11_Init(pDevice, pContext);
}

// 创建渲染目标（兼容游戏窗口）
void MGUI_CreateRenderTarget(IDXGISwapChain* pSwapChain) {
    ID3D11Texture2D* pBackBuffer = nullptr;
    pSwapChain->GetBuffer(0, __uuidof(ID3D11Texture2D), (void**)&pBackBuffer);
    g_pD3DDevice->CreateRenderTargetView(pBackBuffer, nullptr, &g_pRenderTargetView);
    pBackBuffer->Release();
}

// 原始Reset函数指针
HRESULT(__stdcall* Original_Reset)(IDXGISwapChain* pSwapChain, const DXGI_SWAP_CHAIN_DESC* pDesc);

// 挂钩后的Reset函数
HRESULT __stdcall Hooked_Reset(IDXGISwapChain* pSwapChain, const DXGI_SWAP_CHAIN_DESC* pDesc) {
    // 1. 释放旧渲染资源
    ImGui_ImplDX11_Shutdown();
    g_pRenderTargetView->Release();

    // 2. 调用原始Reset重置交换链
    HRESULT hr = Original_Reset(pSwapChain, pDesc);

    // 3. 重新创建渲染目标和MGUI资源
    MGUI_CreateRenderTarget(pSwapChain);
    ImGui_ImplDX11_Init(g_pD3DDevice, g_pD3DContext);
    return hr;
}

// 原始窗口过程函数
WNDPROC g_pOriginalWndProc = nullptr;

// 挂钩后的窗口过程
LRESULT CALLBACK Hooked_WndProc(HWND hWnd, UINT Msg, WPARAM wParam, LPARAM lParam) {
    // 1. 将消息传递给MGUI（支持鼠标/键盘交互）
    ImGui_ImplWin32_WndProcHandler(hWnd, Msg, wParam, lParam);

    // 2. 调用原始窗口过程，保证游戏正常消息处理
    return CallWindowProc(g_pOriginalWndProc, hWnd, Msg, wParam, lParam);
}

// 初始化时设置窗口过程Hook
void HookWndProc(HWND hGameWnd) {
    g_pOriginalWndProc = (WNDPROC)SetWindowLongPtrA(hGameWnd, GWLP_WNDPROC, (LONG_PTR)Hooked_WndProc);
}