#include "system_service.h"

namespace system_service {

static const char *TAG = "SystemService";

esp_err_t handleReset(PsychicRequest *request) {
    reset();
    return request->reply(200);
}

esp_err_t handleRestart(PsychicRequest *request) {
    restart();
    return request->reply(200);
}

esp_err_t handleSleep(PsychicRequest *request) {
    sleep();
    return request->reply(200);
}

void reset() {
    ESP_LOGI(TAG, "Resetting device");
    File root = ESP_FS.open(FS_CONFIG_DIRECTORY);
    File file;
    while (file = root.openNextFile()) {
        std::string path = file.path();
        file.close();
        ESP_FS.remove(path.c_str());
    }
    restart();
}

void restart() {
    xTaskCreate(
        [](void *pvParameters) {
            for (;;) {
                vTaskDelay(250 / portTICK_PERIOD_MS);
                MDNS.end();
                vTaskDelay(100 / portTICK_PERIOD_MS);
                WiFi.disconnect(true);
                vTaskDelay(500 / portTICK_PERIOD_MS);
                ESP.restart();
            }
        },
        "Restart task", 4096, nullptr, 10, nullptr);
}

void sleep() {
    xTaskCreate(
        [](void *pvParameters) {
            for (;;) {
                vTaskDelay(250 / portTICK_PERIOD_MS);
                MDNS.end();
                vTaskDelay(100 / portTICK_PERIOD_MS);
                WiFi.disconnect(true);
                vTaskDelay(500 / portTICK_PERIOD_MS);

                uint64_t bitmask = (uint64_t)1 << (WAKEUP_PIN_NUMBER);

#ifdef CONFIG_IDF_TARGET_ESP32C3
                esp_deep_sleep_enable_gpio_wakeup(bitmask, (esp_deepsleep_gpio_wake_up_mode_t)WAKEUP_SIGNAL);
#else
                esp_sleep_enable_ext1_wakeup(bitmask, (esp_sleep_ext1_wakeup_mode_t)WAKEUP_SIGNAL);
                esp_sleep_pd_config(ESP_PD_DOMAIN_RTC_PERIPH, ESP_PD_OPTION_OFF);
#endif
                esp_deep_sleep_start();
            }
        },
        "Sleep task", 4096, nullptr, 10, nullptr);
    ESP_LOGI(TAG, "Setting device to sleep");
}

void getStaticSystemInformation(socket_message_StaticSystemInformation &info) {
    size_t fs_total = 0, fs_used = 0;
    esp_littlefs_info("spiffs", &fs_total, &fs_used);

    info.esp_platform = (char *)ESP_PLATFORM_NAME;
    info.firmware_version = APP_VERSION;
    info.cpu_freq_mhz = ESP.getCpuFreqMHz();
    info.cpu_type = (char *)ESP.getChipModel();
    info.cpu_rev = ESP.getChipRevision();
    info.cpu_cores = ESP.getChipCores();
    info.sketch_size = ESP.getSketchSize();
    info.free_sketch_space = ESP.getFreeSketchSpace();
    info.sdk_version = (char *)ESP.getSdkVersion();
    info.arduino_version = ARDUINO_VERSION;
    info.flash_chip_size = ESP.getFlashChipSize();
    info.flash_chip_speed = ESP.getFlashChipSpeed();
    info.cpu_reset_reason = (char *)resetReason(esp_reset_reason());
}

void getAnalytics(socket_message_AnalyticsData &analytics) {
    size_t fs_total = 0, fs_used = 0;
    esp_littlefs_info("spiffs", &fs_total, &fs_used);

    analytics.max_alloc_heap = heap_caps_get_largest_free_block(MALLOC_CAP_8BIT);
    analytics.psram_size = heap_caps_get_total_size(MALLOC_CAP_SPIRAM);
    analytics.free_psram = heap_caps_get_free_size(MALLOC_CAP_SPIRAM);
    analytics.free_heap = esp_get_free_heap_size();
    analytics.total_heap = heap_caps_get_total_size(MALLOC_CAP_8BIT);
    analytics.min_free_heap = esp_get_minimum_free_heap_size();
    analytics.core_temp = temperatureRead();
    analytics.fs_total = fs_total;
    analytics.fs_used = fs_used;
    analytics.uptime = esp_timer_get_time() / 1000;
}

const char *resetReason(esp_reset_reason_t reason) {
    switch (reason) {
        case ESP_RST_UNKNOWN: return "Reset reason can not be determined";
        case ESP_RST_POWERON: return "Reset due to power-on event";
        case ESP_RST_EXT: return "Reset by external pin (not applicable for ESP32)";
        case ESP_RST_SW: return "Software reset via esp_restart";
        case ESP_RST_PANIC: return "Software reset due to exception/panic";
        case ESP_RST_INT_WDT: return "Reset (software or hardware) due to interrupt watchdog";
        case ESP_RST_TASK_WDT: return "Reset due to task watchdog";
        case ESP_RST_WDT: return "Reset due to other watchdogs";
        case ESP_RST_DEEPSLEEP: return "Reset after exiting deep sleep mode";
        case ESP_RST_BROWNOUT: return "Brownout reset (software or hardware)";
        case ESP_RST_SDIO: return "Reset over SDIO";
#ifdef ESP_RST_USB
        case ESP_RST_USB: return "Reset by USB peripheral";
#endif
#ifdef ESP_RST_JTAG
        case ESP_RST_JTAG: return "Reset by JTAG";
#endif
#ifdef ESP_RST_EFUSE
        case ESP_RST_EFUSE: return "Reset due to efuse error";
#endif
#ifdef ESP_RST_PWR_GLITCH
        case ESP_RST_PWR_GLITCH: return "Reset due to power glitch detected";
#endif
#ifdef ESP_RST_CPU_LOCKUP
        case ESP_RST_CPU_LOCKUP: return "Reset due to CPU lock up (double exception)";
#endif
        default:
            static char buffer[48];
            snprintf(buffer, sizeof(buffer), "Unknown reset reason (%d)", reason);
            return buffer;
    }
}

} // namespace system_service