#include "Mutex.h"
#include "Timer.h"
#include "Tactility.h"

#include "hal/Power.h"
#include "hal/SdCard.h"
#include "lvgl/Statusbar.h"
#include "service/ServiceContext.h"
#include "service/wifi/Wifi.h"
#include "service/ServiceRegistry.h"
#include "TactilityHeadless.h"
#include "lvgl/LvglSync.h"

namespace tt::service::statusbar {

#define TAG "statusbar_service"

// SD card status
#define STATUSBAR_ICON_SDCARD "sdcard.png"
#define STATUSBAR_ICON_SDCARD_ALERT "sdcard_alert.png"

// Wifi status
#define STATUSBAR_ICON_WIFI_OFF_WHITE "wifi_off_white.png"
#define STATUSBAR_ICON_WIFI_SCAN_WHITE "wifi_scan_white.png"
#define STATUSBAR_ICON_WIFI_SIGNAL_WEAK_WHITE "wifi_signal_weak_white.png"
#define STATUSBAR_ICON_WIFI_SIGNAL_MEDIUM_WHITE "wifi_signal_medium_white.png"
#define STATUSBAR_ICON_WIFI_SIGNAL_STRONG_WHITE "wifi_signal_strong_white.png"

// Power status
#define STATUSBAR_ICON_POWER_0 "power_0.png"
#define STATUSBAR_ICON_POWER_10 "power_10.png"
#define STATUSBAR_ICON_POWER_20 "power_20.png"
#define STATUSBAR_ICON_POWER_30 "power_30.png"
#define STATUSBAR_ICON_POWER_40 "power_40.png"
#define STATUSBAR_ICON_POWER_50 "power_50.png"
#define STATUSBAR_ICON_POWER_60 "power_60.png"
#define STATUSBAR_ICON_POWER_70 "power_70.png"
#define STATUSBAR_ICON_POWER_80 "power_80.png"
#define STATUSBAR_ICON_POWER_90 "power_90.png"
#define STATUSBAR_ICON_POWER_100 "power_100.png"

extern const ServiceManifest manifest;

struct ServiceData {
    Mutex mutex;
    std::unique_ptr<Timer> updateTimer;
    int8_t wifi_icon_id = lvgl::statusbar_icon_add();
    const char* wifi_last_icon = nullptr;
    int8_t sdcard_icon_id = lvgl::statusbar_icon_add();
    const char* sdcard_last_icon = nullptr;
    int8_t power_icon_id = lvgl::statusbar_icon_add();
    const char* power_last_icon = nullptr;

    std::unique_ptr<service::Paths> paths;

    ~ServiceData() {
        lvgl::statusbar_icon_remove(wifi_icon_id);
        lvgl::statusbar_icon_remove(sdcard_icon_id);
        lvgl::statusbar_icon_remove(power_icon_id);
    }

    void lock() const {
        tt_check(mutex.acquire(TtWaitForever) == TtStatusOk);
    }

    void unlock() const {
        tt_check(mutex.release() == TtStatusOk);
    }
};

// region wifi

const char* getWifiStatusIconForRssi(int rssi) {
    if (rssi >= -60) {
        return STATUSBAR_ICON_WIFI_SIGNAL_STRONG_WHITE;
    } else if (rssi >= -70) {
        return STATUSBAR_ICON_WIFI_SIGNAL_MEDIUM_WHITE;
    } else {
        return STATUSBAR_ICON_WIFI_SIGNAL_WEAK_WHITE;
    }
}

static const char* getWifiStatusIcon(wifi::WifiRadioState state, bool secure) {
    int rssi;
    switch (state) {
        case wifi::WIFI_RADIO_ON:
        case wifi::WIFI_RADIO_ON_PENDING:
        case wifi::WIFI_RADIO_CONNECTION_PENDING:
            return STATUSBAR_ICON_WIFI_SCAN_WHITE;
        case wifi::WIFI_RADIO_OFF_PENDING:
        case wifi::WIFI_RADIO_OFF:
            return STATUSBAR_ICON_WIFI_OFF_WHITE;
        case wifi::WIFI_RADIO_CONNECTION_ACTIVE:
            rssi = wifi::getRssi();
            return getWifiStatusIconForRssi(rssi);
        default:
            tt_crash("not implemented");
    }
}

static void updateWifiIcon(const service::Paths* paths, const std::shared_ptr<ServiceData>& data) {
    wifi::WifiRadioState radio_state = wifi::getRadioState();
    bool is_secure = wifi::isConnectionSecure();
    const char* desired_icon = getWifiStatusIcon(radio_state, is_secure);
    if (data->wifi_last_icon != desired_icon) {
        if (desired_icon != nullptr) {
            auto icon_path = paths->getSystemPathLvgl(desired_icon);
            lvgl::statusbar_icon_set_image(data->wifi_icon_id, icon_path);
            lvgl::statusbar_icon_set_visibility(data->wifi_icon_id, true);
        } else {
            lvgl::statusbar_icon_set_visibility(data->wifi_icon_id, false);
        }
        data->wifi_last_icon = desired_icon;
    }
}

// endregion wifi

// region sdcard

static const char* getSdCardStatusIcon(hal::SdCard::State state) {
    switch (state) {
        case hal::SdCard::StateMounted:
            return STATUSBAR_ICON_SDCARD;
        case hal::SdCard::StateError:
        case hal::SdCard::StateUnmounted:
        case hal::SdCard::StateUnknown:
            return STATUSBAR_ICON_SDCARD_ALERT;
        default:
            tt_crash("Unhandled SdCard state");
    }
}

static void updateSdCardIcon(const service::Paths* paths, const std::shared_ptr<ServiceData>& data) {
    auto sdcard = tt::hal::getConfiguration()->sdcard;
    if (sdcard != nullptr) {
        auto state = sdcard->getState();
        if (state != hal::SdCard::StateUnknown) {
            auto* desired_icon = getSdCardStatusIcon(state);
            if (data->sdcard_last_icon != desired_icon) {
                auto icon_path = paths->getSystemPathLvgl(desired_icon);
                lvgl::statusbar_icon_set_image(data->sdcard_icon_id, icon_path);
                lvgl::statusbar_icon_set_visibility(data->sdcard_icon_id, true);
                data->sdcard_last_icon = desired_icon;
            }
        }
        // TODO: Consider tracking how long the SD card has been in unknown status and then show error
    }
}

// endregion sdcard

// region power

static _Nullable const char* getPowerStatusIcon() {
    auto get_power = getConfiguration()->hardware->power;
    if (get_power == nullptr) {
        return nullptr;
    }

    auto power = get_power();

    hal::Power::MetricData charge_level;
    if (!power->getMetric(hal::Power::MetricType::CHARGE_LEVEL, charge_level)) {
        return nullptr;
    }

    uint8_t charge = charge_level.valueAsUint8;

    if (charge >= 95) {
        return STATUSBAR_ICON_POWER_100;
    } else if (charge >= 85) {
        return STATUSBAR_ICON_POWER_90;
    } else if (charge >= 75) {
        return STATUSBAR_ICON_POWER_80;
    } else if (charge >= 65) {
        return STATUSBAR_ICON_POWER_70;
    } else if (charge >= 55) {
        return STATUSBAR_ICON_POWER_60;
    } else if (charge >= 45) {
        return STATUSBAR_ICON_POWER_50;
    } else if (charge >= 35) {
        return STATUSBAR_ICON_POWER_40;
    } else if (charge >= 25) {
        return STATUSBAR_ICON_POWER_30;
    } else if (charge >= 15) {
        return STATUSBAR_ICON_POWER_20;
    } else if (charge >= 5) {
        return STATUSBAR_ICON_POWER_10;
    } else  {
        return STATUSBAR_ICON_POWER_0;
    }
}

static void updatePowerStatusIcon(const service::Paths* paths, const std::shared_ptr<ServiceData>& data) {
    const char* desired_icon = getPowerStatusIcon();
    if (data->power_last_icon != desired_icon) {
        if (desired_icon != nullptr) {
            auto icon_path = paths->getSystemPathLvgl(desired_icon);
            lvgl::statusbar_icon_set_image(data->power_icon_id, icon_path);
            lvgl::statusbar_icon_set_visibility(data->power_icon_id, true);
        } else {
            lvgl::statusbar_icon_set_visibility(data->power_icon_id, false);
        }
        data->power_last_icon = desired_icon;
    }
}

// endregion power

// region service

static void service_data_free(ServiceData* data) {
   free(data);
}

static void onUpdate(std::shared_ptr<void> parameter) {
    auto data = std::static_pointer_cast<ServiceData>(parameter);
    // TODO: Make thread-safe for LVGL
    auto* paths = data->paths.get();
    updateWifiIcon(paths, data);
    updateSdCardIcon(paths, data);
    updatePowerStatusIcon(paths, data);
}

static void onStart(ServiceContext& service) {
    lvgl::lock(TtWaitForever);
    auto data = std::make_shared<ServiceData>();
    lvgl::unlock();

    data->paths = std::move(service.getPaths());

    service.setData(data);

    // TODO: Make thread-safe for LVGL
    lvgl::statusbar_icon_set_visibility(data->wifi_icon_id, true);
    updateWifiIcon(data->paths.get(), data);
    updateSdCardIcon(data->paths.get(), data); // also updates visibility
    updatePowerStatusIcon(data->paths.get(), data);

    data->updateTimer = std::make_unique<Timer>(Timer::TypePeriodic, onUpdate, data);
    // We want to try and scan more often in case of startup or scan lock failure
    data->updateTimer->start(1000);
}

static void onStop(ServiceContext& service) {
    auto data = std::static_pointer_cast<ServiceData>(service.getData());

    // Stop thread
    data->updateTimer->stop();
    data->updateTimer = nullptr;
}

extern const ServiceManifest manifest = {
    .id = "Statusbar",
    .onStart = onStart,
    .onStop = onStop
};

// endregion service

} // namespace