#include "logvisor/logvisor.hpp"
#include "boo/boo.hpp"
#include "specter/specter.hpp"
#include "hecl/CVarManager.hpp"
#include "Runtime/CBasics.hpp"
#include "ViewManager.hpp"
#include "hecl/hecl.hpp"
#include "hecl/CVarCommons.hpp"
#include "hecl/Console.hpp"

static logvisor::Module AthenaLog("Athena");
static void AthenaExc(athena::error::Level level, const char* file,
                      const char*, int line, const char* fmt, ...)
{
    va_list ap;
    va_start(ap, fmt);
    AthenaLog.report(logvisor::Level(level), fmt, ap);
    va_end(ap);
}

namespace urde
{
static logvisor::Module Log{"URDE"};

static hecl::SystemString CPUFeatureString(const zeus::CPUInfo& cpuInf)
{
    hecl::SystemString features;
    auto AddFeature = [&features](const hecl::SystemChar* str)
    {
        if (!features.empty())
            features += _S(", ");
        features += str;
    };
    if (cpuInf.AESNI)
        AddFeature(_S("AES-NI"));
    if (cpuInf.SSE1)
        AddFeature(_S("SSE"));
    if (cpuInf.SSE2)
        AddFeature(_S("SSE2"));
    if (cpuInf.SSE3)
        AddFeature(_S("SSE3"));
    if (cpuInf.SSSE3)
        AddFeature(_S("SSSE3"));
    if (cpuInf.SSE4a)
        AddFeature(_S("SSE4a"));
    if (cpuInf.SSE41)
        AddFeature(_S("SSE4.1"));
    if (cpuInf.SSE42)
        AddFeature(_S("SSE4.2"));
    if (cpuInf.AVX)
        AddFeature(_S("AVX"));
    if (cpuInf.AVX2)
        AddFeature(_S("AVX2"));
    return features;
}

struct Application : boo::IApplicationCallback
{
    hecl::Runtime::FileStoreManager m_fileMgr;
    hecl::CVarManager m_cvarManager;
    hecl::CVarCommons m_cvarCommons;
    hecl::Console m_console;
    std::unique_ptr<ViewManager> m_viewManager;

    bool m_running = true;

    Application() :
        m_fileMgr(_S("urde")),
        m_cvarManager(m_fileMgr),
        m_cvarCommons(m_cvarManager),
        m_console(&m_cvarManager)
    {
        //hecl::Console::RegisterLogger(&m_console);
        m_viewManager = std::make_unique<ViewManager>(m_fileMgr, m_cvarManager);
        m_console.registerCommand("quit", "Quits application instantly", "", std::bind(&Application::quit, this, std::placeholders::_1, std::placeholders::_2));
    }

    virtual ~Application() = default;

    int appMain(boo::IApplication* app)
    {
        initialize(app);
        m_viewManager->init(app);
        while (m_running)
        {
            if (!m_viewManager->proc())
                break;
        }
        m_viewManager->stop();
        m_viewManager->projectManager().saveProject();
        m_cvarManager.serialize();
        m_viewManager.reset();
        return 0;
    }
    void appQuitting(boo::IApplication*)
    {
        m_running = false;
    }
    void appFilesOpen(boo::IApplication*, const std::vector<boo::SystemString>& paths)
    {
        for (const auto& path : paths)
        {
            hecl::ProjectRootPath projPath = hecl::SearchForProject(path);
            if (projPath)
            {
                m_viewManager->deferOpenProject(path);
                break;
            }
        }
    }

    void initialize(boo::IApplication* app)
    {
        zeus::detectCPU();
        for (const boo::SystemString& arg : app->getArgs())
        {
            if (arg.find(_S("--verbosity=")) == 0 || arg.find(_S("-v=")) == 0)
            {
                hecl::SystemUTF8Conv utf8Arg(arg.substr(arg.find_last_of('=') + 1));
                hecl::VerbosityLevel = atoi(utf8Arg.c_str());
                hecl::LogModule.report(logvisor::Info, "Set verbosity level to %i", hecl::VerbosityLevel);
            }
        }

        const zeus::CPUInfo& cpuInf = zeus::cpuFeatures();
        Log.report(logvisor::Info, "CPU Name: %s", cpuInf.cpuBrand);
        Log.report(logvisor::Info, "CPU Vendor: %s", cpuInf.cpuVendor);
        Log.report(logvisor::Info, _S("CPU Features: %s"), CPUFeatureString(cpuInf).c_str());
    }

    std::string getGraphicsApi() const
    {
        return m_cvarCommons.getGraphicsApi();
    }

    uint32_t getSamples() const
    {
        return m_cvarCommons.getSamples();
    }

    uint32_t getAnisotropy() const
    {
        return m_cvarCommons.getAnisotropy();
    }

    void quit(hecl::Console* con = nullptr, const std::vector<std::string>& arg = std::vector<std::string>())
    {
        m_running = false;
    }
};

}

static hecl::SystemChar CwdBuf[1024];
hecl::SystemString ExeDir;

static void SetupBasics(bool logging)
{
    auto result = zeus::validateCPU();
    if (!result.first)
    {
#if _WIN32 && !WINDOWS_STORE
        char* msg;
        asprintf(&msg, "ERROR: This build of URDE requires the following CPU features:\n%s\n",
                 urde::CPUFeatureString(result.second).c_str());
        MessageBoxA(nullptr, msg, "CPU error", MB_OK | MB_ICONERROR);
        free(msg);
#else
        fprintf(stderr, "ERROR: This build of URDE requires the following CPU features:\n%s\n",
                urde::CPUFeatureString(result.second).c_str());
#endif
        exit(1);
    }

    logvisor::RegisterStandardExceptions();
    if (logging)
        logvisor::RegisterConsoleLogger();
    atSetExceptionHandler(AthenaExc);
}

static bool IsClientLoggingEnabled(int argc, const boo::SystemChar** argv)
{
    for (int i = 1; i < argc; ++i)
        if (!hecl::StrNCmp(argv[i], _S("-l"), 2))
            return true;
    return false;
}

#if !WINDOWS_STORE
#if _WIN32
int wmain(int argc, const boo::SystemChar** argv)
#else
int main(int argc, const boo::SystemChar** argv)
#endif
{
    if (argc > 1 && !hecl::StrCmp(argv[1], _S("--dlpackage")))
    {
        printf("%s\n", URDE_DLPACKAGE);
        return 100;
    }

    SetupBasics(IsClientLoggingEnabled(argc, argv));

    if (hecl::SystemChar* cwd = hecl::Getcwd(CwdBuf, 1024))
    {
        if (hecl::PathRelative(argv[0]))
            ExeDir = hecl::SystemString(cwd) + _S('/');
        hecl::SystemString Argv0(argv[0]);
        hecl::SystemString::size_type lastIdx = Argv0.find_last_of(_S("/\\"));
        if (lastIdx != hecl::SystemString::npos)
            ExeDir.insert(ExeDir.end(), Argv0.begin(), Argv0.begin() + lastIdx);
    }

    urde::Application appCb;
    int ret = boo::ApplicationRun(boo::IApplication::EPlatformType::Auto,
        appCb, _S("urde"), _S("URDE"), argc, argv, appCb.getGraphicsApi(),
        appCb.getSamples(), appCb.getAnisotropy(), false);
    //printf("IM DYING!!\n");
    return ret;
}
#endif

#if WINDOWS_STORE
#include "boo/UWPViewProvider.hpp"
using namespace Windows::ApplicationModel::Core;

[Platform::MTAThread]
int WINAPIV main(Platform::Array<Platform::String^>^ params)
{
    SetupBasics(false);
    urde::Application appCb;
    auto viewProvider = ref new boo::ViewProvider(appCb, _S("urde"), _S("URDE"), _S("urde"), params, false);
    CoreApplication::Run(viewProvider);
    return 0;
}

#elif _WIN32
int APIENTRY wWinMain(HINSTANCE hInstance, HINSTANCE, LPWSTR lpCmdLine, int)
{
    int argc = 0;
    const boo::SystemChar** argv = (const wchar_t**)(CommandLineToArgvW(lpCmdLine, &argc));
    static boo::SystemChar selfPath[1024];
    GetModuleFileNameW(nullptr, selfPath, 1024);
    static const boo::SystemChar* booArgv[32] = {};
    booArgv[0] = selfPath;
    for (int i=0 ; i<argc ; ++i)
        booArgv[i+1] = argv[i];

    if (IsClientLoggingEnabled(argc+1, booArgv))
        logvisor::CreateWin32Console();
    return wmain(argc+1, booArgv);
}
#endif