#include "CTitleStateIntro_patches.h"
#include <api/SWA.h>
#include <locale/locale.h>
#include <ui/fader.h>
#include <ui/message_window.h>
#include <user/achievement_manager.h>
#include <user/paths.h>
#include <app.h>

bool g_quitMessageOpen = false;
static bool g_quitMessageFaderBegun = false;
static int g_quitMessageResult = -1;

static std::atomic<bool> g_corruptSaveMessageOpen = false;
static int g_corruptSaveMessageResult = -1;

static bool g_corruptAchievementsMessageOpen = false;
static int g_corruptAchievementsMessageResult = -1;

static bool ProcessQuitMessage()
{
    if (g_corruptSaveMessageOpen)
        return false;

    if (!g_quitMessageOpen)
        return false;

    if (g_quitMessageFaderBegun)
        return true;

    std::array<std::string, 2> options = { Localise("Common_Yes"), Localise("Common_No") };

    if (MessageWindow::Open(Localise("Title_Message_Quit"), &g_quitMessageResult, options, 1) == MSG_CLOSED)
    {
        switch (g_quitMessageResult)
        {
            case 0:
                Fader::FadeOut(1, []() { App::Exit(); });
                g_quitMessageFaderBegun = true;
                break;

            case 1:
                g_quitMessageOpen = false;
                g_quitMessageResult = -1;
                break;
        }
    }

    return true;
}

static bool ProcessCorruptSaveMessage()
{
    if (!g_corruptSaveMessageOpen)
        return false;

    if (MessageWindow::Open(Localise("Title_Message_SaveDataCorrupt"), &g_corruptSaveMessageResult) == MSG_CLOSED)
    {
        g_corruptSaveMessageOpen = false;
        g_corruptSaveMessageOpen.notify_one();
        g_corruptSaveMessageResult = -1;
    }

    return true;
}

static bool ProcessCorruptAchievementsMessage()
{
    if (!g_corruptAchievementsMessageOpen)
        return false;

    auto message = AchievementManager::Status == EAchStatus::IOError
        ? Localise("Title_Message_AchievementDataIOError")
        : Localise("Title_Message_AchievementDataCorrupt");

    if (MessageWindow::Open(message, &g_corruptAchievementsMessageResult) == MSG_CLOSED)
    {
        // Allow user to proceed if the achievement data couldn't be loaded.
        // Restarting may fix this error, so it isn't worth clearing the data for.
        if (AchievementManager::Status != EAchStatus::IOError)
            AchievementManager::Save(true);

        g_corruptAchievementsMessageOpen = false;
        g_corruptAchievementsMessageResult = -1;
    }

    return true;
}

void StorageDevicePromptMidAsmHook()
{
    AchievementManager::Load();

    if (AchievementManager::Status != EAchStatus::Success)
        g_corruptAchievementsMessageOpen = true;
}

// Save data validation hook.
PPC_FUNC_IMPL(__imp__sub_822C55B0);
PPC_FUNC(sub_822C55B0)
{
    App::s_isSaveDataCorrupt = true;
    g_corruptSaveMessageOpen = true;
    g_corruptSaveMessageOpen.wait(true);
    ctx.r3.u32 = 0;
}

// SWA::CTitleStateIntro::Update
PPC_FUNC_IMPL(__imp__sub_82587E50);
PPC_FUNC(sub_82587E50)
{
    auto isAutoSaveWarningShown = *(bool*)g_memory.Translate(0x83367BC1);

    if (isAutoSaveWarningShown)
    {
        __imp__sub_82587E50(ctx, base);
    }
    else if (!ProcessCorruptSaveMessage() && !ProcessCorruptAchievementsMessage())
    {
        auto pInputState = SWA::CInputState::GetInstance();

        if (pInputState && pInputState->GetPadState().IsTapped(SWA::eKeyState_B))
            g_quitMessageOpen = true;

        if (!ProcessQuitMessage())
            __imp__sub_82587E50(ctx, base);
    }
}