EMS-ESP32/lib/framework/OTASettingsService.cpp

#include "OTASettingsService.h"

#include "../../src/emsesp_stub.hpp"

OTASettingsService::OTASettingsService(AsyncWebServer * server, FS * fs, SecurityManager * securityManager)
    : _httpEndpoint(OTASettings::read, OTASettings::update, this, server, OTA_SETTINGS_SERVICE_PATH, securityManager)
    , _fsPersistence(OTASettings::read, OTASettings::update, this, fs, OTA_SETTINGS_FILE)
    , _arduinoOTA(nullptr) {
    WiFi.onEvent([this](WiFiEvent_t event, WiFiEventInfo_t info) { WiFiEvent(event); });
    addUpdateHandler([this] { configureArduinoOTA(); }, false);
}

void OTASettingsService::begin() {
    _fsPersistence.readFromFS();
    configureArduinoOTA();
}

void OTASettingsService::loop() {
    if (_state.enabled && _arduinoOTA) {
        _arduinoOTA->handle();
    }
}

void OTASettingsService::configureArduinoOTA() {
    if (_arduinoOTA) {
        _arduinoOTA->end();
        delete _arduinoOTA;
        _arduinoOTA = nullptr;
    }

    if (_state.enabled) {
        _arduinoOTA = new ArduinoOTAClass;
        _arduinoOTA->setPort(static_cast<uint16_t>(_state.port));
        _arduinoOTA->setPassword(_state.password.c_str());

        _arduinoOTA->onStart([] { emsesp::EMSESP::system_.upload_status(true); });
        _arduinoOTA->onEnd([] { emsesp::EMSESP::system_.upload_status(false); });

        _arduinoOTA->onProgress([](unsigned int progress, unsigned int total) {
            // Serial.printf("Progress: %u%%\r\n", (progress / (total / 100)));
        });
        _arduinoOTA->onError([](ota_error_t error) {
            /*
#if defined(EMSESP_DEBUG)
            Serial.printf("Error[%u]: ", error);
            if (error == OTA_AUTH_ERROR)
                Serial.println("Auth Failed");
            else if (error == OTA_BEGIN_ERROR)
                Serial.println("Begin Failed");
            else if (error == OTA_CONNECT_ERROR)
                Serial.println("Connect Failed");
            else if (error == OTA_RECEIVE_ERROR)
                Serial.println("Receive Failed");
            else if (error == OTA_END_ERROR)
                Serial.println("End Failed");
#endif
*/
        });

        _arduinoOTA->setMdnsEnabled(false); // disable as handled in NetworkSettingsService.cpp. https://github.com/emsesp/EMS-ESP32/issues/161
        _arduinoOTA->begin();
    }
}

void OTASettingsService::WiFiEvent(WiFiEvent_t event) {
    switch (event) {
    case ARDUINO_EVENT_WIFI_STA_GOT_IP:
    case ARDUINO_EVENT_ETH_GOT_IP:
        configureArduinoOTA();
        break;
    default:
        break;
    }
}

void OTASettings::read(OTASettings & settings, JsonObject root) {
    root["enabled"]  = settings.enabled;
    root["port"]     = settings.port;
    root["password"] = settings.password;
}

StateUpdateResult OTASettings::update(JsonObject root, OTASettings & settings) {
    settings.enabled  = root["enabled"] | FACTORY_OTA_ENABLED;
    settings.port     = root["port"] | FACTORY_OTA_PORT;
    settings.password = root["password"] | FACTORY_OTA_PASSWORD;
    return StateUpdateResult::CHANGED;
}