module dllmain;

private import core.sys.windows.windows;
private import core.sys.windows.dll;
private import core.memory;
private import std.stdio;
private import IndentedStreamWriter;
private import ScriptClasses;
private import ScriptHooks;
private import SigScans;
private import Win32API;
private import HeaderGenerator;
private import Profiler;
private import TestMod;

__gshared HINSTANCE g_hInst = NULL;

static void Init()
{
    try
    {
        while (!g_hInst) { }

        MODULEINFO moduleInfo = GetModuleInfo("TribesAscend.exe");

        ScriptName.NameArray = *cast(ScriptArray!(ScriptNameEntry*)**)(FindPattern(moduleInfo.lpBaseOfDll, moduleInfo.SizeOfImage, kSigNames, kMaskNames) + 2);
        ScriptObject.SetObjectArray(*cast(ScriptArray!(ScriptObject)**)(FindPattern(moduleInfo.lpBaseOfDll, moduleInfo.SizeOfImage, kSigObjects, kMaskObjects) + 1));

        ScriptHooks.NativeArray = *cast(NativeFunction**)(FindPattern(moduleInfo.lpBaseOfDll, moduleInfo.SizeOfImage, kSigNatives, kMaskNatives) + 3);
        ScriptHooks.CleanupStackPtr = cast(CleanupStack)FindPattern(moduleInfo.lpBaseOfDll, moduleInfo.SizeOfImage, kSigCleanupStack, kMaskCleanupStack);
        ScriptHooks.CallFunctionPtr = cast(CallFunction)FindPattern(moduleInfo.lpBaseOfDll, moduleInfo.SizeOfImage, kSigCallFunction, kMaskCallFunction);

        ModInit();
        //HeaderGenerator.Generate();
    }
    catch (Throwable e)
    {
        writef("ERROR occurred: %s:%u '%s'", e.file, e.line, e.msg);
    }
}

static void ThreadTrampoline()
{
    Init();
}

extern(Windows) BOOL DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID pvReserved)
{
    switch (dwReason)
    {
        case DLL_PROCESS_ATTACH:
            dll_process_attach(hInstance, false);
            g_hInst = hInstance;
            Thread th = new Thread(&ThreadTrampoline);
            th.start();
            break;
        default:
            return false;
        case DLL_PROCESS_DETACH:
            dll_process_detach(hInstance, true);
            break;
        case DLL_THREAD_ATTACH:
            dll_thread_attach(true, true);
            break;
        case DLL_THREAD_DETACH:
            dll_thread_detach(true, true);
            break;
    }
    return true;
}