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

#include <windows.h>

#include "XCEngine/RHI/OpenGL/OpenGLDevice.h"
#include "XCEngine/RHI/OpenGL/OpenGLSwapChain.h"
#include "XCEngine/RHI/OpenGL/OpenGLCommandList.h"
#include "XCEngine/RHI/OpenGL/OpenGLBuffer.h"
#include "XCEngine/RHI/OpenGL/OpenGLVertexArray.h"
#include "XCEngine/RHI/OpenGL/OpenGLShader.h"
#include "XCEngine/RHI/OpenGL/OpenGLPipelineState.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) {
    Logger::Get().Initialize();
    Logger::Get().AddSink(std::make_unique<ConsoleLogSink>());
    Logger::Get().SetMinimumLevel(LogLevel::Debug);

    Log("[INFO] OpenGL Triangle 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_Triangle";

    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_Triangle",
        L"OpenGL Triangle 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(".\\triangle_frame30");

    OpenGLDevice device;
    RHIDeviceDesc desc = {};
    desc.windowHandle = hwnd;
    desc.width = gWidth;
    desc.height = gHeight;
    desc.appName = L"OpenGL_Triangle_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;

    struct Vertex {
        float pos[4];
        float col[4];
    };

    Vertex vertices[] = {
        { {  0.0f,  0.5f, 0.0f, 1.0f }, { 1.0f, 0.0f, 0.0f, 1.0f } },
        { { -0.5f, -0.5f, 0.0f, 1.0f }, { 0.0f, 1.0f, 0.0f, 1.0f } },
        { {  0.5f, -0.5f, 0.0f, 1.0f }, { 0.0f, 0.0f, 1.0f, 1.0f } },
    };

    OpenGLBuffer vertexBuffer;
    if (!vertexBuffer.InitializeVertexBuffer(vertices, sizeof(vertices))) {
        Log("[ERROR] Failed to initialize vertex buffer");
        return -1;
    }
    vertexBuffer.SetStride(sizeof(Vertex));

    OpenGLVertexArray vertexArray;
    vertexArray.Initialize();

    VertexAttribute posAttr = {};
    posAttr.index = 0;
    posAttr.count = 4;
    posAttr.type = VertexAttributeType::Float;
    posAttr.normalized = VertexAttributeNormalized::False;
    posAttr.stride = sizeof(Vertex);
    posAttr.offset = 0;
    vertexArray.AddVertexBuffer(vertexBuffer.GetID(), posAttr);

    VertexAttribute colAttr = {};
    colAttr.index = 1;
    colAttr.count = 4;
    colAttr.type = VertexAttributeType::Float;
    colAttr.normalized = VertexAttributeNormalized::False;
    colAttr.stride = sizeof(Vertex);
    colAttr.offset = sizeof(float) * 4;
    vertexArray.AddVertexBuffer(vertexBuffer.GetID(), colAttr);

    OpenGLShader shader;
    if (!shader.CompileFromFile("Res/Shader/triangle.vert", "Res/Shader/triangle.frag")) {
        Log("[ERROR] Failed to compile shaders");
        return -1;
    }
    Log("[INFO] Shaders compiled successfully");

    OpenGLPipelineState pipelineState;
    OpenGLRasterizerState rasterizerState;
    rasterizerState.cullFaceEnable = false;
    pipelineState.SetRasterizerState(rasterizerState);

    OpenGLDepthStencilState depthStencilState;
    depthStencilState.depthTestEnable = false;
    depthStencilState.depthWriteEnable = false;
    pipelineState.SetDepthStencilState(depthStencilState);

    ViewportState viewportState = { 0.0f, 0.0f, (float)gWidth, (float)gHeight, 0.0f, 1.0f };
    pipelineState.SetViewport(viewportState);

    pipelineState.AttachShader(shader.GetID());
    pipelineState.Apply();

    MSG msg = {};
    int frameCount = 0;
    const int captureStartFrame = 25;
    const int captureEndFrame = 35;

    while (frameCount < captureEndFrame) {
        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(0.0f, 0.0f, 1.0f, 1.0f, 1);

            pipelineState.Bind();
            vertexArray.Bind();
            commandList.Draw(PrimitiveType::Triangles, 3, 0);

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

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

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

    Log("[INFO] Capture complete - taking screenshot!");
    OpenGLScreenshot::Capture(device, swapChain, "triangle.ppm");

    vertexArray.Shutdown();
    vertexBuffer.Shutdown();
    shader.Shutdown();
    pipelineState.Shutdown();

    swapChain.Shutdown();
    device.Shutdown();

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

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