#include "VISIRenderer.hpp"
#include <Windows.h>
#include <WinUser.h>
#include <Shlwapi.h>
#include <strsafe.h>
#include "athena/Global.hpp"
#include "logvisor/logvisor.hpp"
#include <thread>

static logvisor::Module AthenaLog("Athena");
static void AthenaExc(athena::error::Level level, const char* file,
                      const char*, int line, const char* fmt, ...)
{
    va_list ap;
    va_start(ap, fmt);
    AthenaLog.report(logvisor::Level(level), fmt, ap);
    va_end(ap);
}

static float s_Percent = 0.f;
static DWORD s_mainThreadId;
static void UpdatePercent(float percent)
{
    s_Percent = percent;
    PostThreadMessage(s_mainThreadId, WM_USER+1, 0, 0);
}

static const DWORD dwStyle = WS_OVERLAPPED | WS_CAPTION | WS_SYSMENU | WS_MINIMIZEBOX;
static VISIRenderer* s_Renderer;

static LRESULT CALLBACK WindowProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
    if (uMsg == WM_SIZING)
    {
        RECT& dragRect = reinterpret_cast<RECT&>(lParam);
        RECT tmpRect = dragRect;
        tmpRect.bottom = tmpRect.top + 512;
        tmpRect.right = tmpRect.left + 768;
        AdjustWindowRect(&tmpRect, dwStyle, FALSE);
        dragRect = tmpRect;
        return TRUE;
    }
    else if (uMsg == WM_CLOSE)
    {
        s_Renderer->Terminate();
        return 0;
    }
    return DefWindowProc(hwnd, uMsg, wParam, lParam);
}

int wmain(int argc, const hecl::SystemChar** argv)
{
    logvisor::RegisterStandardExceptions();
    logvisor::RegisterConsoleLogger();
    atSetExceptionHandler(AthenaExc);
    VISIRenderer renderer(argc, argv);
    s_Renderer = &renderer;

    int instIdx = -1;
    if (argc > 3)
        instIdx = _wtoi(argv[3]);

    WNDCLASS wndClass =
    {
        CS_NOCLOSE,
        WindowProc,
        0,
        0,
        GetModuleHandle(nullptr),
        0,
        0,
        0,
        0,
        L"VISIGenWindow"
    };
    RegisterClassW(&wndClass);

    RECT clientRect = {0, 0, 768, 512};
    AdjustWindowRect(&clientRect, dwStyle, FALSE);

    int x = 0;
    int y = 0;
    if (instIdx != -1)
    {
        x = (instIdx & 1) != 0;
        y = (instIdx & 2) != 0;
    }

    HWND window = CreateWindowW(L"VISIGenWindow", L"VISIGen", dwStyle,
                                x, y,
                                clientRect.right - clientRect.left,
                                clientRect.bottom - clientRect.top,
                                NULL, NULL, NULL, NULL);

    PIXELFORMATDESCRIPTOR pfd =
    {
        sizeof(PIXELFORMATDESCRIPTOR),
        1,
        PFD_DRAW_TO_WINDOW | PFD_SUPPORT_OPENGL,    //Flags
        PFD_TYPE_RGBA,            //The kind of framebuffer. RGBA or palette.
        32,                        //Colordepth of the framebuffer.
        0, 0, 0, 0, 0, 0,
        0,
        0,
        0,
        0, 0, 0, 0,
        24,                        //Number of bits for the depthbuffer
        8,                        //Number of bits for the stencilbuffer
        0,                        //Number of Aux buffers in the framebuffer.
        PFD_MAIN_PLANE,
        0,
        0, 0, 0
    };

    HDC deviceContext = GetDC(window);
    int pf = ChoosePixelFormat(deviceContext, &pfd);
    SetPixelFormat(deviceContext, pf, &pfd);
    HGLRC glContext = wglCreateContext(deviceContext);
    ShowWindow(window, SW_SHOW);

    s_mainThreadId = GetCurrentThreadId();

    /* Spawn client thread */
    std::thread clientThread([&]()
    {
        wglMakeCurrent(deviceContext, glContext);
        renderer.Run(UpdatePercent);
        PostThreadMessage(s_mainThreadId, WM_USER, 0, 0);
    });

    /* Pump messages */
    MSG msg = {0};
    while (GetMessage(&msg, NULL, 0, 0))
    {
        if (!msg.hwnd)
        {
            /* PostThreadMessage events */
            switch (msg.message)
            {
            case WM_USER:
                /* Quit message from client thread */
                PostQuitMessage(0);
                continue;
            case WM_USER+1:
                /* Update window title from client thread */
                wchar_t title[256];
                StringCbPrintfW(title, 512, L"VISIGen [%g%%]", s_Percent * 100.f);
                SetWindowTextW(window, title);
                continue;
            default: break;
            }
        }
        TranslateMessage(&msg);
        DispatchMessage(&msg);
    }

    renderer.Terminate();
    if (clientThread.joinable())
        clientThread.join();

    wglDeleteContext(glContext);

    return renderer.ReturnVal();
}