#include "inspire_patches.h"
#include <api/SWA.h>
#include <ui/game_window.h>
#include <os/logger.h>
#include <app.h>
#include <sdl_events.h>

static SWA::Inspire::CScene* g_pScene;
static std::string g_sceneName;
static bool g_isFirstFrameChecked;
static uint32_t g_eventDispatchCount;

static std::array<std::string_view, 8> g_alwaysEvilSonic =
{
    "evrt_m2_02", // Same As Ever
    "evrt_s1_05", // Chun-nan Temple
    "evrt_s3_04", // Holoskan Temple
    "evrt_t0_02", // Shamaran Temple
    "evrt_m7_02", // The Final Temple
    "evrt_m7_04", // Congratulations
    "evrt_m8_02", // The Egg Dragoon
    "evrt_m8_03"  // Planet's End
};

static std::unordered_map<std::string_view, std::pair<float, float>> g_evilSonicTimings =
{
    { "evrt_m0_01_05", { 8189.97f, 10821 } }, // Opening
    { "evrt_m0_06", { 0, 5104.07f } },        // A New Journey
    { "evrt_m1_02", { 1162.46f, 3513 } },     // The First Night
    { "evrt_m6_03", { 2445, 5744 } },         // No Reason
    { "evrt_m8_04", { 0, 2314 } }             // Dark Gaia Appears
};

// SWA::Inspire::CScene
PPC_FUNC_IMPL(__imp__sub_82B98D80);
PPC_FUNC(sub_82B98D80)
{
    __imp__sub_82B98D80(ctx, base);

    g_pScene = (SWA::Inspire::CScene*)g_memory.Translate(ctx.r3.u32);
    g_isFirstFrameChecked = false;
    g_eventDispatchCount = 0;
}

// ~SWA::Inspire::CScene
PPC_FUNC_IMPL(__imp__sub_82B98D30);
PPC_FUNC(sub_82B98D30)
{
    __imp__sub_82B98D30(ctx, base);

    g_pScene = nullptr;
    g_sceneName.clear();

    SDL_User_EvilSonic(App::s_isWerehog);
}

PPC_FUNC_IMPL(__imp__sub_82B9BA98);
PPC_FUNC(sub_82B9BA98)
{
    auto sceneName = (Hedgehog::Base::CSharedString*)g_memory.Translate(ctx.r5.u32);

    g_sceneName = sceneName->c_str();

    __imp__sub_82B9BA98(ctx, base);
}

void InspirePatches::DrawDebug()
{
    if (!g_pScene)
    {
        ImGui::Text("There is no active scene.");
        return;
    }

    ImGui::Text("Name: %s", g_sceneName.c_str());
    ImGui::Text("Frame: %f", g_pScene->m_pData->Frame.get());
    ImGui::Text("Cut: %d", g_pScene->m_pData->Cut.get());

    static std::vector<float> g_loggedFrames{};

    ImGui::Separator();

    if (ImGui::Button("Log"))
        g_loggedFrames.push_back(g_pScene->m_pData->Frame);

    if (ImGui::Button("Clear"))
        g_loggedFrames.clear();

    if (g_loggedFrames.size())
    {
        ImGui::Separator();

        for (auto& frame : g_loggedFrames)
            ImGui::Text("%f", frame);
    }
}

void InspirePatches::Update()
{
    if (!g_pScene || !g_sceneName.size())
        return;

    if (!g_isFirstFrameChecked && std::find(g_alwaysEvilSonic.begin(), g_alwaysEvilSonic.end(), g_sceneName) != g_alwaysEvilSonic.end())
    {
        SDL_User_EvilSonic(true);
        g_isFirstFrameChecked = true;
        return;
    }

    auto findResult = g_evilSonicTimings.find(g_sceneName);

    if (findResult != g_evilSonicTimings.end())
    {
        auto& timings = findResult->second;
        auto& frame = g_pScene->m_pData->Frame;

        if (!g_isFirstFrameChecked && timings.first > 0)
        {
            SDL_User_EvilSonic(false);
            g_isFirstFrameChecked = true;
        }

        if (!g_eventDispatchCount && (frame > timings.first && frame < timings.second))
        {
            SDL_User_EvilSonic(true);
            g_eventDispatchCount++;
        }
        else if (g_eventDispatchCount == 1 && frame > timings.second)
        {
            SDL_User_EvilSonic(false);
            g_eventDispatchCount++;
        }
    }
}