#include "JKSV.hpp"
#include "Colors.hpp"
#include "FsLib.hpp"
#include "Input.hpp"
#include "Logger.hpp"
#include "SDL.hpp"
#include "Strings.hpp"
#include <switch.h>

#include "UI/RenderFunctions.hpp"

#define ABORT_ON_FAILURE(x)                                                                                                                    \
    if (!x)                                                                                                                                    \
    {                                                                                                                                          \
        return;                                                                                                                                \
    }

namespace
{
    constexpr uint8_t BUILD_MON = 12;
    constexpr uint8_t BUILD_DAY = 5;
    constexpr uint16_t BUILD_YEAR = 2024;
} // namespace

template <typename... Args>
static bool InitializeService(Result (*Function)(Args...), const char *ServiceName, Args... Arguments)
{
    Result Error = (*Function)(Arguments...);
    if (R_FAILED(Error))
    {
        Logger::Log("Error initializing %s: 0x%X.", Error);
        return false;
    }
    return true;
}

JKSV::JKSV(void)
{
    ABORT_ON_FAILURE(FsLib::Initialize());
    // Need to init RomFS here for now until I update FsLib to take care of this.
    ABORT_ON_FAILURE(InitializeService(romfsInit, "RomFS"));
    // Let FsLib take care of calls to SDMC instead of fs_dev
    ABORT_ON_FAILURE(FsLib::Dev::InitializeSDMC());
    // SDL
    ABORT_ON_FAILURE(SDL::Initialize("JKSV", 1280, 720));
    ABORT_ON_FAILURE(SDL::Text::Initialize());

    // Services.
    // Using administrator so JKSV can still run in Applet mode.
    ABORT_ON_FAILURE(InitializeService(accountInitialize, "Account", AccountServiceType_Administrator));
    ABORT_ON_FAILURE(InitializeService(nsInitialize, "NS"));
    ABORT_ON_FAILURE(InitializeService(pdmqryInitialize, "PDMQry"));
    ABORT_ON_FAILURE(InitializeService(plInitialize, "PL", PlServiceType_User));
    ABORT_ON_FAILURE(InitializeService(pmshellInitialize, "PMShell"));
    ABORT_ON_FAILURE(InitializeService(setInitialize, "Set"));
    ABORT_ON_FAILURE(InitializeService(setsysInitialize, "SetSys"));
    ABORT_ON_FAILURE(InitializeService(socketInitializeDefault, "Socket"));

    // This needs Set. JKSV also has no internal strings anymore. This is FATAL now.
    ABORT_ON_FAILURE(Strings::Initialize());

    // Install/setup our color changing characters.
    SDL::Text::AddColorCharacter(L'#', Colors::Blue);
    SDL::Text::AddColorCharacter(L'*', Colors::Red);
    SDL::Text::AddColorCharacter(L'<', Colors::Yellow);
    SDL::Text::AddColorCharacter(L'>', Colors::Green);

    // This is to check whether the author wanted credit for their work.
    m_ShowTranslationInfo = std::char_traits<char>::compare(Strings::GetByName(Strings::Names::TranslationInfo, 1), "NULL", 4) != 0;

    // This can't be in an initializer list because it needs SDL initialized.
    m_HeaderIcon = SDL::TextureManager::CreateLoadTexture("HeaderIcon", "romfs:/Textures/HeaderIcon.png");

    Input::Initialize();

    m_IsRunning = true;
}

JKSV::~JKSV()
{
    socketExit();
    setsysExit();
    setExit();
    pmshellExit();
    plExit();
    pdmqryExit();
    nsExit();
    accountExit();
    SDL::Text::Exit();
    SDL::Exit();
    FsLib::Exit();
}

bool JKSV::IsRunning(void) const
{
    return m_IsRunning;
}

void JKSV::Update(void)
{
    Input::Update();

    if (Input::ButtonPressed(HidNpadButton_Plus))
    {
        m_IsRunning = false;
    }
}

void JKSV::Render(void)
{
    SDL::FrameBegin(Colors::ClearColor);
    // Top and bottom divider lines.
    SDL::RenderLine(NULL, 30, 88, 1250, 88, Colors::White);
    SDL::RenderLine(NULL, 30, 648, 1250, 648, Colors::White);
    // Icon
    m_HeaderIcon->Render(NULL, 66, 27);
    // "JKSV"
    SDL::Text::Render(NULL, 130, 32, 34, SDL::Text::NO_TEXT_WRAP, Colors::White, "JKSV");
    // Translation info in bottom left.
    if (m_ShowTranslationInfo)
    {
        SDL::Text::Render(NULL,
                          8,
                          682,
                          12,
                          SDL::Text::NO_TEXT_WRAP,
                          Colors::White,
                          Strings::GetByName(Strings::Names::TranslationInfo, 0),
                          Strings::GetByName(Strings::Names::TranslationInfo, 1));
    }
    SDL::Text::Render(NULL, 8, 700, 12, SDL::Text::NO_TEXT_WRAP, Colors::White, "v %02d.%02d.%04d", BUILD_MON, BUILD_DAY, BUILD_YEAR);

    UI::RenderDialogBox(NULL, 320, 240, 320, 240);

    SDL::FrameEnd();
}

void JKSV::PushState(std::shared_ptr<AppState> NewState)
{
    m_StateVector.push_back(NewState);
}