#include "appstates/ProgressState.hpp"

#include "appstates/FadeState.hpp"
#include "graphics/colors.hpp"
#include "input.hpp"
#include "sdl.hpp"
#include "strings/strings.hpp"
#include "stringutil.hpp"
#include "ui/PopMessageManager.hpp"

#include <cmath>

namespace
{
    constexpr int COORD_BAR_X = 312;
    constexpr int COORD_BAR_Y = 470;

    constexpr int COORD_TEXT_Y = 475;

    constexpr int COORD_DISPLAY_CENTER = 640;
    constexpr double SIZE_BAR_WIDTH    = 656.0f;
}

ProgressState::ProgressState(sys::threadpool::JobFunction function, sys::Task::TaskData taskData)
{
    initialize_static_members();
    m_task = std::make_unique<sys::ProgressTask>(function, taskData);
}

void ProgressState::update()
{
    sys::ProgressTask *task = static_cast<sys::ProgressTask *>(m_task.get());
    const double current    = task->get_progress();

    BaseTask::update_loading_glyph();
    BaseTask::pop_on_plus();

    if (!m_task->is_running()) { ProgressState::deactivate_state(); }

    m_progressBarWidth        = std::round(SIZE_BAR_WIDTH * current);
    m_progress                = std::round(current * 100);
    m_percentageString        = stringutil::get_formatted_string("%u%%", m_progress);
    const int percentageWidth = sdl::text::get_width(BaseTask::FONT_SIZE, m_percentageString);
    m_percentageX             = COORD_DISPLAY_CENTER - (percentageWidth / 2);
}

void ProgressState::render()
{
    static constexpr int RIGHT_EDGE_X = (COORD_BAR_X + SIZE_BAR_WIDTH) - 16;

    const bool hasFocus      = BaseState::has_focus();
    const int barWidth       = static_cast<int>(SIZE_BAR_WIDTH);
    const std::string status = m_task->get_status();

    sdl::render_rect_fill(sdl::Texture::Null, 0, 0, 1280, 720, colors::DIM_BACKGROUND);
    sm_dialog->render(sdl::Texture::Null, hasFocus);
    sdl::text::render(sdl::Texture::Null, 312, 288, BaseTask::FONT_SIZE, 656, colors::WHITE, status);

    sdl::render_line(sdl::Texture::Null, 280, 454, 999, 454, colors::DIV_COLOR);
    sdl::render_rect_fill(sdl::Texture::Null, COORD_BAR_X, COORD_BAR_Y, barWidth, 32, colors::BLACK);
    sdl::render_rect_fill(sdl::Texture::Null, COORD_BAR_X, COORD_BAR_Y, m_progressBarWidth, 32, colors::GREEN);

    sm_barEdges->render_part(sdl::Texture::Null, COORD_BAR_X, COORD_BAR_Y, 0, 0, 16, 32);
    sm_barEdges->render_part(sdl::Texture::Null, RIGHT_EDGE_X, COORD_BAR_Y, 16, 0, 16, 32);

    sdl::text::render(sdl::Texture::Null,
                      m_percentageX,
                      COORD_TEXT_Y,
                      BaseTask::FONT_SIZE,
                      sdl::text::NO_WRAP,
                      colors::WHITE,
                      m_percentageString);

    BaseTask::render_loading_glyph();
}

void ProgressState::initialize_static_members()
{
    static constexpr std::string_view BAR_EDGE_NAME = "BarEdges";

    if (sm_dialog && sm_barEdges) { return; }

    sm_dialog   = ui::DialogBox::create(280, 262, 720, 256);
    sm_barEdges = sdl::TextureManager::load(BAR_EDGE_NAME, "romfs:/Textures/BarEdges.png");
}

void ProgressState::deactivate_state()
{
    FadeState::create_and_push(colors::DIM_BACKGROUND, colors::ALPHA_FADE_END, colors::ALPHA_FADE_BEGIN, nullptr);
    BaseState::deactivate();
}