#include "core/stdafx.h"
#include "core/r5dev.h"
#include "core/init.h"
#include "core/logdef.h"
#include "core/logger.h"
#include "tier0/cpu.h"
#include "tier0/basetypes.h"
#include "tier0/crashhandler.h"
#include "tier0/commandline.h"
/*****************************************************************************/
#ifndef DEDICATED
#include "windows/id3dx.h"
#include "windows/input.h"
#endif // !DEDICATED
#include "windows/console.h"
#include "windows/system.h"
#include "mathlib/mathlib.h"
#include "launcher/launcher.h"
#include "protobuf/stubs/common.h"

#ifndef DEDICATED
#define SDK_DEFAULT_CFG "cfg/system/startup_default.cfg"
#else
#define SDK_DEFAULT_CFG "cfg/system/startup_dedi_default.cfg"
#endif

bool g_bSdkInitialized = false;

bool g_bSdkInitCallInitiated = false;
bool g_bSdkShutdownCallInitiated = false;

bool g_bSdkShutdownInitiatedFromConsoleHandler = false;

static bool s_bConsoleInitialized = false;
static HMODULE s_hModuleHandle = NULL;

//#############################################################################
// UTILITY
//#############################################################################

void Crash_Callback()
{
    // Shutdown SpdLog to flush all buffers.
    SpdLog_Shutdown();

    // TODO[ AMOS ]: This is where we want to call backtrace from.
}

void Show_Emblem()
{
    // Logged as 'SYSTEM_ERROR' for its red color.
    for (size_t i = 0; i < SDK_ARRAYSIZE(R5R_EMBLEM); i++)
    {
        Msg(eDLL_T::SYSTEM_ERROR, "%s\n", R5R_EMBLEM[i]);
    }

    // Log the SDK's 'build_id' under the emblem.
    Msg(eDLL_T::SYSTEM_ERROR,
        "+------------------------------------------------[%s%010d%s]-+\n",
        g_svYellowF, g_SDKDll.GetNTHeaders()->FileHeader.TimeDateStamp, g_svRedF);
    Msg(eDLL_T::SYSTEM_ERROR, "\n");
}

//#############################################################################
// INITIALIZATION
//#############################################################################

void Tier0_Init()
{
#if !defined (DEDICATED)
    g_RadVideoToolsDll.InitFromName("bink2w64.dll");
    g_RadAudioDecoderDll.InitFromName("binkawin64.dll");
    g_RadAudioSystemDll.InitFromName("mileswin64.dll");
#endif // !DEDICATED
    g_CoreMsgVCallback = &EngineLoggerSink; // Setup logger callback sink.

    g_pCmdLine->CreateCmdLine(GetCommandLineA());
    g_CrashHandler.SetCrashCallback(&Crash_Callback);

    // This prevents the game from recreating it,
    // see 'CCommandLine::StaticCreateCmdLine' for
    // more information.
    g_bCommandLineCreated = true;
}

void SDK_Init()
{
    assert(!g_bSdkInitialized);

    CheckSystemCPU(); // Check CPU as early as possible; error out if CPU isn't supported.

    if (g_bSdkInitCallInitiated)
    {
        spdlog::error("Recursive initialization!\n");
        return;
    }

    // Set after checking cpu and initializing MathLib since we check CPU
    // features there. Else we crash on the recursive initialization error as
    // SpdLog uses SSE features.
    g_bSdkInitCallInitiated = true;

    MathLib_Init(); // Initialize Mathlib.

    PEB64* pEnv = CModule::GetProcessEnvironmentBlock();

    g_GameDll.InitFromBase(pEnv->ImageBaseAddress);
    g_SDKDll.InitFromBase((QWORD)s_hModuleHandle);

    Tier0_Init();

    if (!CommandLine()->CheckParm("-launcher"))
    {
        CommandLine()->AppendParametersFromFile(SDK_DEFAULT_CFG);
    }

    const bool bAnsiColor = CommandLine()->CheckParm("-ansicolor") ? true : false;

#ifndef DEDICATED
    if (CommandLine()->CheckParm("-wconsole"))
#else
    if (!CommandLine()->CheckParm("-noconsole"))
#endif  // !DEDICATED
    {
        s_bConsoleInitialized = Console_Init(bAnsiColor);
    }

    SpdLog_Init(bAnsiColor);
    Show_Emblem();

    Winsock_Startup(); // Initialize Winsock.
    DirtySDK_Startup();

    Systems_Init();

    WinSys_Init();
#ifndef DEDICATED
    Input_Init();
#endif // !DEDICATED

    GOOGLE_PROTOBUF_VERIFY_VERSION;
    curl_global_init(CURL_GLOBAL_ALL);
    lzham_enable_fail_exceptions(true);

    g_bSdkInitialized = true;
}

//#############################################################################
// SHUTDOWN
//#############################################################################

void SDK_Shutdown()
{
    assert(g_bSdkInitialized);

    // Also check CPU in shutdown, since this function is exported, if they
    // call this with an unsupported CPU we should let them know rather than
    // crashing the process.
    CheckSystemCPU();

    if (g_bSdkShutdownCallInitiated)
    {
        spdlog::error("Recursive shutdown!\n");
        return;
    }

    g_bSdkShutdownCallInitiated = true;

    if (!g_bSdkInitialized)
    {
        spdlog::error("Not initialized!\n");
        return;
    }

    Msg(eDLL_T::NONE, "GameSDK shutdown initiated\n");

    curl_global_cleanup();

#ifndef DEDICATED
    Input_Shutdown();
#endif // !DEDICATED

    WinSys_Shutdown();
    Systems_Shutdown();

    DirtySDK_Shutdown();
    Winsock_Shutdown();

    SpdLog_Shutdown();

    // If the shutdown was initiated from the console window itself, don't
    // shutdown the console as it would otherwise deadlock in FreeConsole!
    if (s_bConsoleInitialized && !g_bSdkShutdownInitiatedFromConsoleHandler)
        Console_Shutdown();

    g_bSdkInitialized = false;
}

//#############################################################################
// ENTRYPOINT
//#############################################################################

BOOL APIENTRY DllMain(HMODULE hModule, DWORD dwReason, LPVOID lpReserved)
{
    NOTE_UNUSED(lpReserved);

    switch (dwReason)
    {
        case DLL_PROCESS_ATTACH:
        {
            s_hModuleHandle = hModule;
            break;
        }
        case DLL_PROCESS_DETACH:
        {
            s_hModuleHandle = NULL;
            break;
        }
    }

    return TRUE;
}