OTA: Add firmware update callbacks

Signed-off-by: Vitaliy Andreevich <[email protected]>

Author: VitaliyAndreevich

src/SettingsESP.h

@@ -85,13 +85,20 @@ class SettingsESP : public sets::SettingsBase {
             String auth = server.arg(F("auth"));
             if (!authenticate(Text(auth).toInt32HEX())) return;
 
+            size_t maxSketchSpace = sets::OtaGetMaxSize();
             HTTPUpload& upload = server.upload();
             if (upload.status == UPLOAD_FILE_START) {
                 sets::beginOta();
+                if (_onStart)
+                  _onStart();
             } else if (upload.status == UPLOAD_FILE_WRITE) {
                 Update.write(upload.buf, upload.currentSize);
+                if (_onProgress)
+                  _onProgress(upload.totalSize, maxSketchSpace);
             } else if (upload.status == UPLOAD_FILE_END) {
-                Update.end(true);
+                bool result = Update.end(true);
+                if (_onDone)
+                  _onDone(result);
             } });
 
         server.onNotFound([this]() {

src/core/SettingsBase.h

@@ -26,11 +26,19 @@
 
 namespace sets {
 
+  typedef void (*onUpdateFWStart_t)();
+  typedef void (*onUpdateFWProgress_t)(size_t current, size_t final);
+  typedef void (*onUpdateFWDone_t)(bool success);
+
 class SettingsBase {
     static const uint16_t FOCUS_TOUT = 5000;
     static const uint16_t DB_WS_UPDATE_PRD = 300;
 
    protected:
+    sets::onUpdateFWStart_t _onStart = nullptr;
+    sets::onUpdateFWProgress_t _onProgress = nullptr;
+    sets::onUpdateFWDone_t _onDone = nullptr;
+
     typedef std::function<void(Builder& b)> BuildCallback;
     typedef std::function<void(Updater& upd)> UpdateCallback;
     typedef std::function<void(Text path)> FileCallback;
@@ -116,6 +124,18 @@ class SettingsBase {
     }
 
    public:
+     void onUpdateFWStart(sets::onUpdateFWStart_t fn) {
+       _onStart = fn;
+     };
+
+     void onUpdateFWProgress(sets::onUpdateFWProgress_t fn) {
+       _onProgress = fn;
+     };
+
+     void onUpdateFWDone(sets::onUpdateFWDone_t fn) {
+       _onDone = fn;
+     };
+
 #ifndef SETT_NO_DB
     SettingsBase(const String& title = "", GyverDB* db = nullptr) : _title(title), _db(db) {
 #else

src/core/ota.cpp

@@ -8,26 +8,38 @@
 #endif
 
 namespace sets {
+size_t OtaGetMaxSize(bool ota_flash)
+{
+  size_t ota_size = 0;
+  if (ota_flash) {
+#ifdef ESP8266
+    ota_size = (size_t)((ESP.getFreeSketchSpace() - 0x1000) & 0xFFFFF000);
+#else
+    ota_size = UPDATE_SIZE_UNKNOWN;
+#endif
+  } else {
+#ifdef ESP8266
+    close_all_fs();
+    ota_size = (size_t)&_FS_end - (size_t)&_FS_start;
+#else
+    ota_size = UPDATE_SIZE_UNKNOWN;
+#endif
+  }
+  return ota_size;
+}
 
 bool beginOta(bool ota_flash, bool async) {
-    size_t ota_size = 0;
+    size_t ota_size = OtaGetMaxSize(ota_flash);
     int ota_type = 0;
 
     if (ota_flash) {
         ota_type = U_FLASH;
-#ifdef ESP8266
-        ota_size = (size_t)((ESP.getFreeSketchSpace() - 0x1000) & 0xFFFFF000);
-#else
-        ota_size = UPDATE_SIZE_UNKNOWN;
-#endif
     } else {
 #ifdef ESP8266
         ota_type = U_FS;
         close_all_fs();
-        ota_size = (size_t)&_FS_end - (size_t)&_FS_start;
 #else
         ota_type = U_SPIFFS;
-        ota_size = UPDATE_SIZE_UNKNOWN;
 #endif
     }
 

src/core/ota.h

@@ -7,7 +7,7 @@
 #endif
 
 namespace sets {
-
+size_t OtaGetMaxSize(bool ota_flash = true);
 bool beginOta(bool ota_flash = true, bool async = false);
 
 }  // namespace sets