#include <string>
#include <vector>
#include <map>
#include <iomanip>
#include <iostream>
#include <typeinfo>
#include <functional>
#include <Windows.h>
#include <TlHelp32.h>

using namespace std;

#include "Scrapland.h"
#include "Util.h"
#include "Structures.h"
#include "Py_Utils.h"
#include "Hook.h"
#include "D3D8_Hook.h"
#include "REPL.h"
HMODULE hD3D8Dll = 0;

bool initialized = false;
bool running = true;
bool redirect_console = false;
HMODULE mod = 0;

void DllUnload(HMODULE);
int hooked_console(const char *);
void H_Exit();

size_t size_ht(HashTable<EntityList> *ht)
{
    size_t cnt = 0;
    for (size_t i = 0; i < ht->size; ++i)
    {
        HashTableEntry<EntityList> *ent = ht->chains[i];
        if (ent)
        {
            while (ent)
            {
                ++cnt;
                ent = ent->next;
            }
        }
    }
    return cnt;
}

size_t size_ht(HashTable<Entity> *ht)
{
    size_t cnt = 0;
    for (size_t i = 0; i < ht->size; ++i)
    {
        HashTableEntry<Entity> *ent = ht->chains[i];
        if (ent)
        {
            while (ent)
            {
                ++cnt;
                ent = ent->next;
            }
        }
    }
    return cnt;
}

size_t dump_ht(HashTable<EntityList> *ht)
{
    size_t cnt = 0;
    for (size_t i = 0; i < ht->size; ++i)
    {
        HashTableEntry<EntityList> *ent = ht->chains[i];
        if (ent)
        {
            cout << i << ": ";
            while (ent)
            {
                ++cnt;
                cout << "[ " << ent->name << ": " << ent->data << "]";
                if (ent->next)
                {
                    cout << " -> ";
                };
                ent = ent->next;
            }
            cout << endl;
        }
    }
    cout << cnt << " Entries" << endl;
    return cnt;
}

size_t dump_ht(HashTable<Entity> *ht)
{
    size_t cnt = 0;
    for (size_t i = 0; i < ht->size; ++i)
    {
        HashTableEntry<Entity> *ent = ht->chains[i];
        if (ent)
        {
            cout << i << ": ";
            while (ent)
            {
                ++cnt;
                cout << "[ " << ent->name << ": " << ent->data << "]";
                if (ent->next)
                {
                    cout << " -> ";
                };
                ent = ent->next;
            }
            cout << endl;
        }
    }
    cout << cnt << " Entries" << endl;
    return cnt;
}

void MainLoop(HMODULE mod)
{
    Sleep(100);
    Hook::addr(reinterpret_cast<void *>(P_SCRAP_EXIT), H_Exit);
    Hook::addr(reinterpret_cast<void *>(P_D3DCHECK),hook_d3d8);
    Hook::addr(reinterpret_cast<void *>(P_CON_HANDLER), hooked_console);
    overlay=true;
    cout << "[*] Starting main Loop" << endl;
    cout << endl;
    cout << "[F2 ] Redirect game console to ScapHacks console" << endl;
    cout << "[F3 ] Unload ScrapHacks" << endl;
    cout << "[F5 ] Show Overlay" << endl;
    cout << "[F6 ] Show Alarm status" << endl;
    cout << "[F7 ] Set Money to 0x7fffffff" << endl;
    cout << "[F8 ] Dump python modules" << endl;
    cout << "[F9 ] Dump Entity hashtable" << endl;
    cout << "[F10] Enable python tracing" << endl;
    cout << "[ F ] \"Handbrake\" (*Will* crash the game after some time!)" << endl;

    while (running)
    {
        Sleep(100);
        while (key_down('F'))
        {
            scrap_exec("dbg.brake()");
        }
        if (key_down_norepeat(VK_F2))
        {
            redirect_console = !redirect_console;
        }
        if (key_down_norepeat(VK_F3))
        {
            break;
        }

        if (key_down_norepeat(VK_F5))
        {
            overlay = !overlay;
        }
        
        if (key_down_norepeat(VK_F6))
        {

            float *alarm = ptr<float>(P_WORLD, O_ALARM);
            float *alarm_grow = ptr<float>(P_WORLD, O_ALARM_GROW);
            cout << "Alarm: " << alarm[0] << " + " << alarm_grow[0] << endl;
        }
        if (key_down_norepeat(VK_F7))
        {
            int32_t *money = ptr<int32_t>(P_WORLD, O_MONEY);
            *money = 0x7fffffff;
        }
        if (key_down_norepeat(VK_F8))
        {
            for (auto mod : Py)
            {
                for (auto meth : mod.second.methods)
                {
                    cout << mod.first << "." << meth.first << " @ " << meth.second->ml_meth << endl;
                }
            }
        }

        if (key_down_norepeat(VK_F9))
        {
            cout << "Entities:" << endl;
            dump_ht(ptr<HashTable<Entity>>(P_WORLD, O_ENTS));
            cout << "Entity Lists:" << endl;
            dump_ht(ptr<HashTable<EntityList>>(P_WORLD, O_ENTLISTS));
        }
        if (key_down_norepeat(VK_F10))
        {
            scrap_exec("dbg.settrace()");
        }
    }
    FreeLibraryAndExitThread(mod, 0);
}

void InitConsole()
{
    char me[1024];
    GetModuleFileName(mod, me, 1024);
    SetupConsole(me);
}

int hooked_console(const char *cmd)
{
    typedef int(_cdecl * t_func)(const char *);
    if (cmd[0] == '$')
    {
        handle_command(++cmd);
        return 0;
    }
    shared_ptr<Hook> hook = Hook::get(hooked_console);
    int ret = hook->func<t_func>(cmd);
    return ret;
}

void H_Exit()
{
    typedef void(_cdecl * t_func)(void);
    shared_ptr<Hook> hook = Hook::get(H_Exit);
    DllUnload(mod);
    HWND hMainWindow = ptr<HWND>(0x7FA830, 0x7c)[0];
    SendMessage(hMainWindow, WM_CLOSE, 0, 0);
    return;
}


void DllPreInit(HMODULE _mod)
{
    char mfn[1024];
    InitConsole();
    GetModuleFileNameA(0, mfn, 1024);
    Py = get_modules(P_PY_MODS);
    cout << "[+] ScrapHacks v0.1 Loaded in " << mfn << " (PID: " << std::hex << GetCurrentProcessId() << std::dec << ")" << endl;
}

void DllInit(HMODULE _mod)
{
    initialized = true;
    mod = _mod;
    cout << "[*] World: " << ptr<void>(P_WORLD, 0) << endl;
    cout << "[*] Importing python dbg module" << endl;
    scrap_exec("import dbg");
    scrap_log(0xff0000, "ScrapHacks loaded!\n");
    CreateThread(NULL, NULL, (LPTHREAD_START_ROUTINE)MainLoop, mod, 0, 0);
    cout << "[*] Starting message pump" << endl;
    MSG msg;
    while (GetMessage(&msg, NULL, 0, 0))
    {
        TranslateMessage(&msg);
        DispatchMessage(&msg);
    }
    return;
}

void DllUnload(HMODULE _mod)
{
    SetConsoleCtrlHandler(NULL, false);
    unhook_d3d8();
    Hook::clear();
    scrap_log(0xff0000, "ScrapHacks unloaded!\n");
    cout << "[+] ScrapHacks unloaded, you can now close the console!" << endl;
    FreeConsole();
    DestroyWindow(GetConsoleWindow());
    return;
}