#include <windows.h>

#include <assert.h>
#include <stdlib.h>

#include "gfxhook/gfx.h"

#include "hook/table.h"

#include "util/dprintf.h"

/* Hook functions */

static BOOL WINAPI hook_ShowWindow(HWND hWnd, int nCmdShow);
static HWND WINAPI hook_CreateWindowExA(
    DWORD dwExStyle,
    LPCSTR lpClassName,
    LPCSTR lpWindowName,
    DWORD dwStyle,
    int X,
    int Y,
    int nWidth,
    int nHeight,
    HWND hWndParent,
    HMENU hMenu,
    HINSTANCE hInstance,
    LPVOID lpParam
);

/* Link pointers */

static BOOL (WINAPI *next_ShowWindow)(HWND hWnd, int nCmdShow);
static HWND (WINAPI *next_CreateWindowExA)(
    DWORD dwExStyle,
    LPCSTR lpClassName,
    LPCSTR lpWindowName,
    DWORD dwStyle,
    int X,
    int Y,
    int nWidth,
    int nHeight,
    HWND hWndParent,
    HMENU hMenu,
    HINSTANCE hInstance,
    LPVOID lpParam
);

static struct gfx_config gfx_config;

static const struct hook_symbol gfx_hooks[] = {
    {
        .name = "ShowWindow",
        .patch = hook_ShowWindow,
        .link = (void **) &next_ShowWindow,
    }, {
        .name = "CreateWindowExA",
        .patch = hook_CreateWindowExA,
        .link = (void **) &next_CreateWindowExA,
    },
};

void gfx_hook_init(const struct gfx_config *cfg)
{
    assert(cfg != NULL);

    if (!cfg->enable) {
        return;
    }

    if (cfg->dpiAware) {
        if (SetProcessDPIAware()) {
            dprintf("Gfx: Game process set to DPI aware.\n");
        } else {
            dprintf("Gfx: Failed to set process DPI aware\n");
        }
    }

    memcpy(&gfx_config, cfg, sizeof(*cfg));
    hook_table_apply(NULL, "user32.dll", gfx_hooks, _countof(gfx_hooks));
}

static BOOL WINAPI hook_ShowWindow(HWND hWnd, int nCmdShow)
{
    dprintf("Gfx: ShowWindow hook hit\n");

    if (!gfx_config.framed && nCmdShow == SW_RESTORE) {
        nCmdShow = SW_SHOW;
    }

    return next_ShowWindow(hWnd, nCmdShow);
}

static HWND WINAPI hook_CreateWindowExA(
    DWORD dwExStyle,
    LPCSTR lpClassName,
    LPCSTR lpWindowName,
    DWORD dwStyle,
    int X,
    int Y,
    int nWidth,
    int nHeight,
    HWND hWndParent,
    HMENU hMenu,
    HINSTANCE hInstance,
    LPVOID lpParam
)
{
    RECT rect;

    dprintf("Gfx: CreateWindowExA hook hit\n");

    if (gfx_config.windowed)
    {
        if (gfx_config.framed)
            dwStyle |= WS_BORDER | WS_CAPTION | WS_MINIMIZEBOX | WS_SYSMENU;
        else
            dwStyle = WS_POPUP;

        rect.left = ((X == CW_USEDEFAULT) ? 0 : X);
        rect.top = ((Y == CW_USEDEFAULT) ? 0 : Y);
        rect.right = rect.left + nWidth;
        rect.bottom = rect.top + nHeight;

        // Don't care if it's ok or not, since we are creating window and we can't just return a NULL
        AdjustWindowRect(&rect, dwStyle, !!hMenu);

        X = ((X == CW_USEDEFAULT) ? X : rect.left);
        Y = ((Y == CW_USEDEFAULT) ? Y : rect.top);
        nWidth = rect.right - rect.left;
        nHeight = rect.bottom - rect.top;
    }

   return next_CreateWindowExA(
        dwExStyle,
        lpClassName,
        lpWindowName,
        dwStyle,
        X,
        Y,
        nWidth,
        nHeight,
        hWndParent,
        hMenu,
        hInstance,
        lpParam
    );
}