/*
 * EMS-ESP - https://github.com/emsesp/EMS-ESP
 * Copyright 2020-2025  emsesp.org
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "emsesp.h"

#ifndef EMSESP_STANDALONE
#include <esp_ota_ops.h>
#endif

namespace emsesp {

WebStatusService::WebStatusService(AsyncWebServer * server, SecurityManager * securityManager)
    : _securityManager(securityManager) {
    // GET
    securityManager->addEndpoint(server, EMSESP_SYSTEM_STATUS_SERVICE_PATH, AuthenticationPredicates::IS_AUTHENTICATED, [this](AsyncWebServerRequest * request) {
        systemStatus(request);
    });

    // POST - generic action handler, handles both GET and POST
    securityManager->addEndpoint(
        server,
        EMSESP_ACTION_SERVICE_PATH,
        AuthenticationPredicates::IS_AUTHENTICATED,
        [this](AsyncWebServerRequest * request, JsonVariant json) { action(request, json); },
        HTTP_ANY);
}

// /rest/systemStatus
// This contains both system & hardware Status to avoid having multiple costly endpoints
// This is also used for polling during the SystemMonitor to see if EMS-ESP is alive
void WebStatusService::systemStatus(AsyncWebServerRequest * request) {
    EMSESP::system_.refreshHeapMem(); // refresh free heap and max alloc heap

    auto *     response = new AsyncJsonResponse(false);
    JsonObject root     = response->getRoot();

    //
    // System Status
    //
    root["emsesp_version"] = EMSESP_APP_VERSION;
    root["bus_status"]     = EMSESP::bus_status(); // 0, 1 or 2
    root["bus_uptime"]     = EMSbus::bus_uptime();
    root["num_devices"]    = EMSESP::count_devices();
    root["num_sensors"]    = EMSESP::temperaturesensor_.count_entities();
    root["num_analogs"]    = EMSESP::analogsensor_.count_entities();
    root["free_heap"]      = EMSESP::system_.getHeapMem();
    root["uptime"]         = uuid::get_uptime_sec();
    root["mqtt_status"]    = EMSESP::mqtt_.connected();

#ifndef EMSESP_STANDALONE
    uint8_t ntp_status = 0; // 0=disabled, 1=enabled, 2=connected
    if (esp_sntp_enabled()) {
        ntp_status = (EMSESP::system_.ntp_connected()) ? 2 : 1;
    }
    root["ntp_status"] = ntp_status;
    if (ntp_status == 2) {
        // send back actual time if NTP enabled and active
        time_t now = time(nullptr);
        if (now > 1500000000L) {
            char t[25];
            strftime(t, sizeof(t), "%FT%T", localtime(&now));
            root["ntp_time"] = t; // optional string
        }
    }
#endif

    root["ap_status"] = EMSESP::esp32React.apStatus();

    if (EMSESP::system_.ethernet_connected()) {
        root["network_status"] = 10; // custom code #10 - ETHERNET_STATUS_CONNECTED
        root["wifi_rssi"]      = 0;
    } else {
        root["network_status"] = static_cast<uint8_t>(WiFi.status());
#ifndef EMSESP_STANDALONE
        root["wifi_rssi"] = WiFi.RSSI();
#endif
    }

#if defined(EMSESP_DEBUG)
#ifdef EMSESP_TEST
    root["build_flags"] = "DEBUG,TEST";
#else
    root["build_flags"] = "DEBUG";
#endif
#elif defined(EMSESP_TEST)
    root["build_flags"] = "TEST";
#endif

    //
    // Hardware Status
    //
    root["esp_platform"] = EMSESP_PLATFORM;
#ifndef EMSESP_STANDALONE
    root["cpu_type"]         = ESP.getChipModel();
    root["cpu_rev"]          = ESP.getChipRevision();
    root["cpu_cores"]        = ESP.getChipCores();
    root["cpu_freq_mhz"]     = ESP.getCpuFreqMHz();
    root["max_alloc_heap"]   = EMSESP::system_.getMaxAllocMem();
    root["arduino_version"]  = ARDUINO_VERSION;
    root["sdk_version"]      = ESP.getSdkVersion();
    root["partition"]        = (const char *)esp_ota_get_running_partition()->label; // active partition
    root["flash_chip_size"]  = ESP.getFlashChipSize() / 1024;
    root["flash_chip_speed"] = ESP.getFlashChipSpeed();
    root["app_used"]         = EMSESP::system_.appUsed();
    root["app_free"]         = EMSESP::system_.appFree();
    uint32_t FSused          = LittleFS.usedBytes() / 1024;
    root["fs_used"]          = FSused;
    root["fs_free"]          = EMSESP::system_.FStotal() - FSused;
    root["free_caps"]        = heap_caps_get_free_size(MALLOC_CAP_8BIT) / 1024; // includes heap and psram
    root["psram"]            = (EMSESP::system_.PSram() > 0);                   // boolean
    if (EMSESP::system_.PSram()) {
        root["psram_size"] = EMSESP::system_.PSram();
        root["free_psram"] = ESP.getFreePsram() / 1024;
    }
    root["model"] = EMSESP::system_.getBBQKeesGatewayDetails();
    root["board"] = EMSESP::system_.getBBQKeesGatewayDetails(FUSE_VALUE::BOARD);
#if CONFIG_IDF_TARGET_ESP32S3 || CONFIG_IDF_TARGET_ESP32C3 || CONFIG_IDF_TARGET_ESP32S2
    root["temperature"] = (int)Helpers::transformNumFloat(EMSESP::system_.temperature(), 0, EMSESP::system_.fahrenheit() ? 2 : 0); // only 2 decimal places
#endif

    // check for a factory partition first
    const esp_partition_t * partition = esp_partition_find_first(ESP_PARTITION_TYPE_APP, ESP_PARTITION_SUBTYPE_APP_FACTORY, nullptr);
    root["has_loader"]                = partition != NULL && partition != esp_ota_get_running_partition();
    partition                         = esp_ota_get_next_update_partition(nullptr);
    if (partition) {
        uint64_t buffer;
        esp_partition_read(partition, 0, &buffer, 8);
        root["has_partition"] = (buffer != 0xFFFFFFFFFFFFFFFF);
    } else {
        root["has_partition"] = false;
    }

    // get the partition info for each partition, including the running one
    // the partition data is gathered once in System::start() and stored in partition_info_
    // install_date is stored as a UTC epoch and formatted to local time here so it honors
    // the current TZ (which may not have been set yet when System::start() ran).
    JsonArray partitions = root["partitions"].to<JsonArray>();
    for (const auto & partition : EMSESP::system_.partition_info_) {
        // Skip partition if it has no version, or it's size is 0
        if (partition.second.version.empty() || partition.second.size == 0) {
            continue;
        }
        JsonObject part   = partitions.add<JsonObject>();
        part["partition"] = partition.first;
        part["version"]   = partition.second.version;
        part["size"]      = partition.second.size;
        if (partition.second.install_date > 0) {
            char   time_string[25];
            time_t d = partition.second.install_date;
            strftime(time_string, sizeof(time_string), "%FT%T", localtime(&d));
            part["install_date"] = time_string;
        } else {
            part["install_date"] = "";
        }
    }

    root["developer_mode"] = EMSESP::system_.developer_mode();

    // Also used in SystemMonitor.tsx
    root["status"] = EMSESP::system_.systemStatus(); // send the status. See System.h for status codes
    if (EMSESP::system_.systemStatus() == SYSTEM_STATUS::SYSTEM_STATUS_PENDING_RESTART) {
        // we're ready to do the actual restart ASAP
        EMSESP::system_.systemStatus(SYSTEM_STATUS::SYSTEM_STATUS_RESTART_REQUESTED);
    }

#endif

    response->setLength();
    request->send(response);
}

// generic action handler - as a POST
void WebStatusService::action(AsyncWebServerRequest * request, JsonVariant json) {
    auto *     response = new AsyncJsonResponse();
    JsonObject root     = response->getRoot();

    // param is optional - https://arduinojson.org/news/2024/09/18/arduinojson-7-2/
    std::string param;
    bool        has_param      = false;
    JsonVariant param_optional = json["param"];
    if (json["param"].is<const char *>()) {
        param     = param_optional.as<std::string>();
        has_param = true;
    } else {
        has_param = false;
    }

    // check if we're authenticated for admin tasks, some actions are only for admins
    Authentication authentication = _securityManager->authenticateRequest(request);
    bool           is_admin       = AuthenticationPredicates::IS_ADMIN(authentication);

    // call action command
    bool        ok     = false;
    std::string action = json["action"];

    if (action == "checkUpgrade") {
        ok = checkUpgrade(root, param); // param could be empty, if so only send back version
    } else if (action == "setPartition") {
        ok = EMSESP::system_.set_partition(param.c_str());
    } else if (action == "export") {
        if (has_param) {
            ok = exportData(root, param);
        }
    } else if (action == "getCustomSupport") {
        ok = getCustomSupport(root);
    } else if (action == "uploadURL" && is_admin) {
        ok = uploadURL(param.c_str());
    } else if (action == "systemStatus" && is_admin) {
        ok = setSystemStatus(param.c_str());
    } else if (action == "resetMQTT" && is_admin) {
        EMSESP::mqtt_.reset_mqtt();
        ok = true;
    } else if (action == "upgradeImportantMessages") {
        root["upgradeImportantMessageType"] = upgradeImportantMessages(param);
        ok                                  = true;
    }

#if defined(EMSESP_STANDALONE) && !defined(EMSESP_UNITY)
    Serial.printf("%sweb output: %s[%s]", COLOR_WHITE, COLOR_BRIGHT_CYAN, request->url().c_str());
    Serial.printf(" %s(%d)%s ", ok ? COLOR_BRIGHT_GREEN : COLOR_BRIGHT_RED, ok ? 200 : 400, COLOR_YELLOW);
    serializeJson(root, Serial);
    Serial.println(COLOR_RESET);
#endif

    // check for error
    if (!ok) {
        EMSESP::logger().err("Action '%s' failed", action.c_str());
        request->send(400); // bad request
        return;
    }

    // send response
    response->setLength();
    request->send(response);
}

// action = upgradeImportantMessages
// returns the type of upgrade important message to display in the UI
// 0 = no message (if just a minor version upgrade)
// 1 = going from <= 3.8 to 3.9 (has new partition layout)
// 2 = major version upgrade
// version can be like 3.8.2 or a filename like EMS-ESP-3_8_2-dev_13-ESP32-16MB+.bin
uint8_t WebStatusService::upgradeImportantMessages(std::string & version) {
    if (version.empty()) {
        return 0;
    }

    // it's a filename with a .bin or .md extension, try and extract the version from it
    // e.g. EMS-ESP-3_8_2-dev_13-ESP32-16MB+.bin -> major=3 minor=8 patch=2
    version::EMSESP_Version latest_version;
    if ((version.find(".bin") != std::string::npos) || (version.find(".md") != std::string::npos)) {
        std::string filename = version;
        auto        pos      = filename.find("EMS-ESP-");
        if (pos == std::string::npos) {
            EMSESP::logger().err("Invalid version string: %s", version.c_str());
            return 0;
        }

        pos += 8; // skip past "EMS-ESP-"
        auto underscore1 = filename.find('_', pos);
        auto underscore2 = filename.find('_', underscore1 + 1);
        auto dash        = filename.find('-', underscore2 + 1);
        if (underscore1 == std::string::npos || underscore2 == std::string::npos || dash == std::string::npos) {
            EMSESP::logger().err("Invalid version string: %s", version.c_str());
            return 0;
        }

        std::string major_version = filename.substr(pos, underscore1 - pos);
        std::string minor_version = filename.substr(underscore1 + 1, underscore2 - underscore1 - 1);
        std::string patch_version = filename.substr(underscore2 + 1, dash - underscore2 - 1);
        latest_version            = version::EMSESP_Version(major_version + "." + minor_version + "." + patch_version);
    } else {
        // if it's .json file exit
        if (version.find(".json") != std::string::npos) {
            return 0;
        } else {
            // treat it like a version string like "3.9.0"
            latest_version = version::EMSESP_Version(version);
        }
    }

    version::EMSESP_Version current_version(current_version_s); // get current version

    if ((current_version.major() <= 3 && current_version.minor() <= 8) && (latest_version.major() == 3 && latest_version.minor() == 9)) {
        return 1; // if moving from below 3.8.x to 3.9.x return 1
    }

    if (latest_version > current_version && current_version.major() < latest_version.major()) {
        return 2; // if it's a major version upgrade return 2
    }

    if (latest_version > current_version && current_version.minor() < latest_version.minor()) {
        return 0; // if it's just a minor version upgrade return 0
    }

    return 0; // if it's not a valid version upgrade return 0
}

// action = checkUpgrade
// versions holds the latest development version and stable version in one string, comma separated
bool WebStatusService::checkUpgrade(JsonObject root, std::string & version) {
    if (!version.empty()) {
        version::EMSESP_Version current_version(current_version_s);
        version::EMSESP_Version latest_dev_version(version.substr(0, version.find(',')));
        version::EMSESP_Version latest_stable_version(version.substr(version.find(',') + 1));

        bool dev_upgradeable    = latest_dev_version > current_version;
        bool stable_upgradeable = latest_stable_version > current_version;

#if defined(EMSESP_DEBUG)
        // look for dev in the name to determine if we're using a dev release
        bool using_dev_version = !current_version.prerelease().find("dev");
        EMSESP::logger()
            .debug("Checking version upgrade. This version=%d.%d.%d-%s (%s),latest dev=%d.%d.%d-%s (%s upgradeable),latest stable=%d.%d.%d-%s (%s upgradeable)",
                   current_version.major(),
                   current_version.minor(),
                   current_version.patch(),
                   current_version.prerelease().c_str(),
                   using_dev_version ? "Dev" : "Stable",
                   latest_dev_version.major(),
                   latest_dev_version.minor(),
                   latest_dev_version.patch(),
                   latest_dev_version.prerelease().c_str(),
                   dev_upgradeable ? "is" : "is not",
                   latest_stable_version.major(),
                   latest_stable_version.minor(),
                   latest_stable_version.patch(),
                   latest_stable_version.prerelease().c_str(),
                   stable_upgradeable ? "is" : "is not");
#endif

        root["dev_upgradeable"]    = dev_upgradeable;
        root["stable_upgradeable"] = stable_upgradeable;
    }

    root["emsesp_version"] = current_version_s; // always send back current version

    return true;
}

// action = allvalues
// output all the devices and their values, including custom entities, scheduler and sensors
void WebStatusService::allvalues(JsonObject output) {
    JsonObject device_output;
    auto       value = F_(values);

    // EMS-Device Entities
    for (const auto & emsdevice : EMSESP::emsdevices) {
        std::string title = emsdevice->device_type_2_device_name_translated() + std::string(" ") + emsdevice->to_string();
        device_output     = output[title].to<JsonObject>();
        emsdevice->get_value_info(device_output, value, DeviceValueTAG::TAG_NONE);
    }

    // Custom Entities
    device_output = output["Custom Entities"].to<JsonObject>();
    EMSESP::webCustomEntityService.get_value_info(device_output, value);

    // Scheduler
    device_output = output["Scheduler"].to<JsonObject>();
    EMSESP::webSchedulerService.get_value_info(device_output, value);

    // Sensors
    device_output = output["Analog Sensors"].to<JsonObject>();
    EMSESP::analogsensor_.get_value_info(device_output, value);
    device_output = output["Temperature Sensors"].to<JsonObject>();
    EMSESP::temperaturesensor_.get_value_info(device_output, value);
}

// action = export
// returns data for a specific feature/settings as a json object
bool WebStatusService::exportData(JsonObject root, std::string & type) {
    if (type == "settings") {
        root["type"] = type; // add settings as a group
        System::exportSettings(type, NETWORK_SETTINGS_FILE, root);
        System::exportSettings(type, AP_SETTINGS_FILE, root);
        System::exportSettings(type, MQTT_SETTINGS_FILE, root);
        System::exportSettings(type, NTP_SETTINGS_FILE, root);
        System::exportSettings(type, SECURITY_SETTINGS_FILE, root);
        System::exportSettings(type, EMSESP_SETTINGS_FILE, root);
    } else if (type == "schedule") {
        System::exportSettings(type, EMSESP_SCHEDULER_FILE, root);
    } else if (type == "customizations") {
        System::exportSettings(type, EMSESP_CUSTOMIZATION_FILE, root);
    } else if (type == "entities") {
        System::exportSettings(type, EMSESP_CUSTOMENTITY_FILE, root);
    } else if (type == "allvalues") {
        allvalues(root);
    } else if (type == "systembackup") {
        System::exportSystemBackup(root);
    } else {
        return false; // error
    }

    return true;
}

// action = getCustomSupport
// reads any upload customSupport.json file and sends to to Help page to be shown as Guest
bool WebStatusService::getCustomSupport(JsonObject root) {
    JsonDocument doc;

#if defined(EMSESP_STANDALONE)
    // dummy test data for "test api3"
    deserializeJson(doc, "{\"type\":\"customSupport\",\"Support\":{\"html\":[\"html code\",\"here\"], \"img_url\": \"https://emsesp.org/media/images/designer.png\"}");
#else
    // check if we have custom support file uploaded
    File file = LittleFS.open(EMSESP_CUSTOMSUPPORT_FILE, "r");
    if (!file) {
        // there is no custom file, return empty object
#if defined(EMSESP_DEBUG)
        EMSESP::logger().debug("No custom support file found");
#endif
        return true;
    }

    // read the contents of the file into a json doc. We can't do this direct to object since 7.2.1
    DeserializationError error = deserializeJson(doc, file);
    if (error) {
        EMSESP::logger().err("Failed to read custom support file");
        return false;
    }

    file.close();
#endif

#if defined(EMSESP_DEBUG)
    EMSESP::logger().debug("Sending custom support page");
#endif

    root.set(doc.as<JsonObject>()); // add to web response root object

    return true;
}

// action = uploadURL
// uploads a firmware file from a URL
bool WebStatusService::uploadURL(const char * url) {
    // this will keep a copy of the URL, but won't initiate the download yet
    EMSESP::system_.uploadFirmwareURL(url);
    return true;
}

// action = systemStatus
// sets the system status
bool WebStatusService::setSystemStatus(const char * status) {
    EMSESP::system_.systemStatus(Helpers::atoint(status));
    return true;
}

} // namespace emsesp