#define LOG_MODULE "path"

#include <stdint.h>
#include <string.h>
#include <windows.h>

#include "hook/table.h"

#include "iidxhook4-cn/path.h"

#include "util/log.h"
#include "util/str.h"

#define PATH_A "D:/JDZ-001/contents/"
#define PATH_W L"D:/JDZ-001/contents/"

static HANDLE WINAPI my_CreateFileA(
    LPCSTR lpFileName,
    DWORD dwDesiredAccess,
    DWORD dwShareMode,
    LPSECURITY_ATTRIBUTES lpSecurityAttributes,
    DWORD dwCreationDisposition,
    DWORD dwFlagsAndAttributes,
    HANDLE hTemplateFile);
static HANDLE(WINAPI *real_CreateFileA)(
    LPCSTR lpFileName,
    DWORD dwDesiredAccess,
    DWORD dwShareMode,
    LPSECURITY_ATTRIBUTES lpSecurityAttributes,
    DWORD dwCreationDisposition,
    DWORD dwFlagsAndAttributes,
    HANDLE hTemplateFile);

static HANDLE WINAPI my_CreateFileW(
    LPCWSTR lpFileName,
    DWORD dwDesiredAccess,
    DWORD dwShareMode,
    LPSECURITY_ATTRIBUTES lpSecurityAttributes,
    DWORD dwCreationDisposition,
    DWORD dwFlagsAndAttributes,
    HANDLE hTemplateFile);
static HANDLE(WINAPI *real_CreateFileW)(
    LPCWSTR lpFileName,
    DWORD dwDesiredAccess,
    DWORD dwShareMode,
    LPSECURITY_ATTRIBUTES lpSecurityAttributes,
    DWORD dwCreationDisposition,
    DWORD dwFlagsAndAttributes,
    HANDLE hTemplateFile);

static HANDLE WINAPI
my_FindFirstFileA(LPCSTR lpFileName, LPWIN32_FIND_DATAA lpFindFileData);
static HANDLE(WINAPI *real_FindFirstFileA)(
    LPCSTR lpFileName, LPWIN32_FIND_DATAA lpFindFileData);

static const struct hook_symbol iidxhook4_cn_path_hook_syms[] = {
    {.name = "CreateFileA",
     .patch = my_CreateFileA,
     .link = (void **) &real_CreateFileA},
    {.name = "CreateFileW",
     .patch = my_CreateFileW,
     .link = (void **) &real_CreateFileW},
    {.name = "FindFirstFileA",
     .patch = my_FindFirstFileA,
     .link = (void **) &real_FindFirstFileA},
};

static HANDLE WINAPI my_CreateFileA(
    LPCSTR lpFileName,
    DWORD dwDesiredAccess,
    DWORD dwShareMode,
    LPSECURITY_ATTRIBUTES lpSecurityAttributes,
    DWORD dwCreationDisposition,
    DWORD dwFlagsAndAttributes,
    HANDLE hTemplateFile)
{
    if (lpFileName != NULL && strstr(lpFileName, PATH_A) == lpFileName) {
        char relative_path[MAX_PATH] = ".";
        str_cat(relative_path, MAX_PATH, lpFileName + strlen(PATH_A) - 1);
        return real_CreateFileA(
            relative_path,
            dwDesiredAccess,
            dwShareMode,
            lpSecurityAttributes,
            dwCreationDisposition,
            dwFlagsAndAttributes,
            hTemplateFile);
    }

    return real_CreateFileA(
        lpFileName,
        dwDesiredAccess,
        dwShareMode,
        lpSecurityAttributes,
        dwCreationDisposition,
        dwFlagsAndAttributes,
        hTemplateFile);
}

static HANDLE WINAPI my_CreateFileW(
    LPCWSTR lpFileName,
    DWORD dwDesiredAccess,
    DWORD dwShareMode,
    LPSECURITY_ATTRIBUTES lpSecurityAttributes,
    DWORD dwCreationDisposition,
    DWORD dwFlagsAndAttributes,
    HANDLE hTemplateFile)
{
    if (lpFileName != NULL && wcsstr(lpFileName, PATH_W) == lpFileName) {
        wchar_t relative_path[MAX_PATH] = L".";
        wstr_cat(relative_path, MAX_PATH, lpFileName + wcslen(PATH_W) - 1);
        return real_CreateFileW(
            relative_path,
            dwDesiredAccess,
            dwShareMode,
            lpSecurityAttributes,
            dwCreationDisposition,
            dwFlagsAndAttributes,
            hTemplateFile);
    }

    return real_CreateFileW(
        lpFileName,
        dwDesiredAccess,
        dwShareMode,
        lpSecurityAttributes,
        dwCreationDisposition,
        dwFlagsAndAttributes,
        hTemplateFile);
}

static HANDLE WINAPI
my_FindFirstFileA(LPCSTR lpFileName, LPWIN32_FIND_DATAA lpFindFileData)
{
    if (lpFileName != NULL && strstr(lpFileName, PATH_A) == lpFileName) {
        char relative_path[MAX_PATH] = ".";
        str_cat(relative_path, MAX_PATH, lpFileName + strlen(PATH_A) - 1);
        return real_FindFirstFileA(relative_path, lpFindFileData);
    }

    return real_FindFirstFileA(lpFileName, lpFindFileData);
}

void iidxhook4_cn_path_init()
{
    hook_table_apply(
        NULL,
        "kernel32.dll",
        iidxhook4_cn_path_hook_syms,
        lengthof(iidxhook4_cn_path_hook_syms));

    log_info("Inserted path hooks");
}