#include "updatedisplay.h"

// system includes
#include <array>
#include <string>

// esp-idf includes
#include <esp_log.h>

// 3rdparty lib includes
#include <tftinstance.h>
#include <screenmanager.h>
#ifdef FEATURE_OTA
#include <espasyncota.h>
#include <esp_ota_ops.h>
#endif

// local includes
#include "globals.h"
#include "ota.h"
#include "displays/menus/otamenu.h"
#include "newsettings.h"

#ifdef FEATURE_OTA
void UpdateDisplay::initScreen()
{
    Base::initScreen();

    espgui::tft.setTextFont(4);
    espgui::tft.setTextColor(TFT_YELLOW);

    espgui::tft.drawString("Update", 5, 5, 4);

    espgui::tft.fillRect(0, 34, espgui::tft.width(), 3, TFT_WHITE);

    espgui::tft.setTextColor(TFT_WHITE, TFT_BLACK);

    espgui::tft.drawString("Status:", 20, m_statusLabel.y());
    m_statusLabel.start();

    espgui::tft.drawString("Progress:", 20, m_progressLabel.y());
    m_progressLabel.start();

    espgui::tft.drawString("Total:", 20, m_totalLabel.y());
    m_totalLabel.start();

    m_messageLabel.start();

    m_progressBar.start();

    if (const esp_app_desc_t *app_desc = esp_ota_get_app_description())
    {
        espgui::tft.setTextColor(TFT_ORANGE, TFT_BLACK);
        espgui::tft.drawString(app_desc->version, 20, 250);
        espgui::tft.setTextColor(TFT_WHITE, TFT_BLACK);
    }

    m_newVersionLabel.start();
}

void UpdateDisplay::redraw()
{
    Base::redraw();

    if (asyncOta)
    {
        m_statusLabel.redraw(toString(asyncOta->status()));
        const auto progress = asyncOta->progress();
        m_progressLabel.redraw(std::to_string(progress));
        if (const auto totalSize = asyncOta->totalSize(); totalSize && *totalSize > 0)
        {
            m_totalLabel.redraw(std::to_string(*totalSize));
            m_progressBar.redraw(float(progress) / *totalSize * 100);
        }
        else
        {
            m_totalLabel.clear();
            m_progressBar.redraw(0);
        }
        m_messageLabel.redraw(asyncOta->message());

        if (const auto &appDesc = asyncOta->appDesc())
        {
            espgui::tft.setTextColor(TFT_GREEN, TFT_BLACK);
            m_newVersionLabel.redraw(appDesc->version);
            espgui::tft.setTextColor(TFT_WHITE, TFT_BLACK);
        }
        else
            m_newVersionLabel.clear();
    }
    else
    {
        m_statusLabel.clear();
        m_progressLabel.clear();
        m_totalLabel.clear();
        m_messageLabel.clear();

        m_progressBar.redraw(0);

        m_newVersionLabel.clear();
    }
}

void UpdateDisplay::buttonPressed(espgui::Button button)
{
    Base::buttonPressed(button);

    switch (button)
    {
    using espgui::Button;
    case Button::Left:
        espgui::switchScreen<OtaMenu>();
        break;
    case Button::Right:
        if (const auto result = triggerOta(configs.otaUrl.value); !result)
            ESP_LOGE("BOBBY", "triggerOta() failed with %.*s", result.error().size(), result.error().data());
        break;
    default:;
    }
}
#endif