#include <stdio.h>
#include <stdarg.h>
#include <stdlib.h>
#include <string.h>

#include <windows.h>
#include <GL/gl.h>

#include "XCEngine/RHI/OpenGL/OpenGLDevice.h"
#include "XCEngine/RHI/OpenGL/OpenGLSwapChain.h"
#include "XCEngine/RHI/OpenGL/OpenGLCommandList.h"
#include "XCEngine/RHI/OpenGL/OpenGLScreenshot.h"
#include "XCEngine/Debug/Logger.h"
#include "XCEngine/Debug/ConsoleLogSink.h"
#include "XCEngine/Debug/RenderDocCapture.h"
#include "XCEngine/Containers/String.h"

#pragma comment(lib, "opengl32.lib")

using namespace XCEngine::RHI;
using namespace XCEngine::Debug;
using namespace XCEngine::Containers;

static const int gWidth = 1280;
static const int gHeight = 720;

void Log(const char* format, ...) {
    char buffer[1024];
    va_list args;
    va_start(args, format);
    vsnprintf(buffer, sizeof(buffer), format, args);
    va_end(args);
    Logger::Get().Debug(LogCategory::Rendering, String(buffer));
}

LRESULT CALLBACK WindowProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) {
    switch (msg) {
    case WM_CLOSE:
        PostQuitMessage(0);
        break;
    }
    return DefWindowProc(hwnd, msg, wParam, lParam);
}

int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nShowCmd) {
    _putenv_s("RENDERDOC_CAPTUREINTERACTIVE", "0");
    _putenv_s("RENDERDOC_CAPTUREFRAMESTART", "0");
    
    Logger::Get().Initialize();
    Logger::Get().AddSink(std::make_unique<ConsoleLogSink>());
    Logger::Get().SetMinimumLevel(LogLevel::Debug);

    Log("[INFO] OpenGL Integration Test Starting");

    WNDCLASSEXW wc = {};
    wc.cbSize = sizeof(WNDCLASSEXW);
    wc.style = CS_HREDRAW | CS_VREDRAW;
    wc.lpfnWndProc = WindowProc;
    wc.hInstance = hInstance;
    wc.lpszClassName = L"XCEngine_OpenGL_Test";

    if (!RegisterClassExW(&wc)) {
        Log("[ERROR] Failed to register window class");
        return -1;
    }

    RECT rect = { 0, 0, gWidth, gHeight };
    AdjustWindowRect(&rect, WS_OVERLAPPEDWINDOW, FALSE);

    HWND hwnd = CreateWindowExW(
        0,
        L"XCEngine_OpenGL_Test",
        L"OpenGL Integration Test",
        WS_OVERLAPPEDWINDOW,
        CW_USEDEFAULT, CW_USEDEFAULT,
        rect.right - rect.left, rect.bottom - rect.top,
        NULL, NULL, hInstance, NULL
    );

    if (!hwnd) {
        Log("[ERROR] Failed to create window");
        return -1;
    }

    RenderDocCapture::Get().Initialize(nullptr, hwnd);
    RenderDocCapture::Get().SetCaptureFilePath(".\\minimal_frame30");

    OpenGLDevice device;
    RHIDeviceDesc desc = {};
    desc.windowHandle = hwnd;
    desc.width = gWidth;
    desc.height = gHeight;
    desc.appName = L"OpenGL_Minimal_Test";
    desc.enableDebugLayer = true;
    if (!device.Initialize(desc)) {
        Log("[ERROR] Failed to initialize OpenGL device");
        return -1;
    }

    RenderDocCapture::Get().SetDevice(device.GetGLContext());

    ShowWindow(hwnd, nShowCmd);
    UpdateWindow(hwnd);

    Log("[INFO] OpenGL Device: %S", device.GetDeviceInfo().renderer.c_str());
    Log("[INFO] OpenGL Version: %S", device.GetDeviceInfo().version.c_str());

    OpenGLSwapChain swapChain;
    swapChain.Initialize(&device, hwnd, gWidth, gHeight);

    OpenGLCommandList commandList;

    MSG msg = {};
    int frameCount = 0;
    const int captureStartFrame = 29;
    const int captureEndFrame = 30;

    while (true) {
        if (PeekMessageW(&msg, NULL, 0, 0, PM_REMOVE)) {
            if (msg.message == WM_QUIT) {
                break;
            }
            TranslateMessage(&msg);
            DispatchMessageW(&msg);
        } else {
            wglMakeCurrent(device.GetPresentationDC(), device.GetGLContext());

            commandList.SetViewport(0, 0, gWidth, gHeight);
            commandList.Clear(1.0f, 0.0f, 0.0f, 1.0f, 1 | 2);

            swapChain.Present(0, 0);
            frameCount++;

            if (frameCount >= captureEndFrame) {
                RenderDocCapture::Get().EndCapture();
                Log("[INFO] RenderDoc capture ended at frame %d", frameCount);
                break;
            }

            if (frameCount == captureStartFrame) {
                RenderDocCapture::Get().BeginCapture("OpenGL_Minimal_Test");
                Log("[INFO] RenderDoc capture started at frame %d", frameCount);
            }
        }
    }

    Log("[INFO] Taking screenshot!");
    OpenGLScreenshot::Capture(device, swapChain, "minimal.ppm");

    RenderDocCapture::Get().Shutdown();
    swapChain.Shutdown();
    device.Shutdown();
    Logger::Get().Shutdown();

    Log("[INFO] OpenGL Integration Test Finished");
    return 0;
}