#ifndef ESP32_HTTP_UPDATE_SERVER_H #define ESP32_HTTP_UPDATE_SERVER_H /* Based on the HTTP update exemple of ESP32 core */ #include #include #define ESP32_WEB_UPDATE_HTML "" #define ESP32_WEB_UPDATE_SUCCESS_RESPONSE "Update Success! Rebooting...\n" class ESP32HTTPUpdateServer { private: WebServer* _server; String _username; String _password; bool _serialDebugging; public: ESP32HTTPUpdateServer(bool serialDebugging = false) { _server = NULL; _username = ""; _password = ""; } void setup(WebServer* server, const char* path = "/", const char* username = "", const char* password = "") { _server = server; _username = username; _password = password; // Get of the index handling _server->on(path, HTTP_GET, [&]() { // Force authentication if a user and password are defined if (_username.length() > 0 && _password.length() > 0 && !_server->authenticate(_username.c_str(), _password.c_str())) return _server->requestAuthentication(); _server->sendHeader("Connection", "close"); _server->send(200, "text/html", ESP32_WEB_UPDATE_HTML); }); // Post of the file handling _server->on(path, HTTP_POST, [&]() { _server->client().setNoDelay(true); _server->send_P(200, "text/html", (Update.hasError()) ? "FAIL" : ESP32_WEB_UPDATE_SUCCESS_RESPONSE); delay(100); _server->client().stop(); ESP.restart(); }, [&]() { HTTPUpload& upload = _server->upload(); if (upload.status == UPLOAD_FILE_START) { // Check if we are authenticated if (!(_username.length() == 0 || _password.length() == 0 || _server->authenticate(_username.c_str(), _password.c_str()))) { if (_serialDebugging) Serial.printf("Unauthenticated Update\n"); return; } // Debugging message for upload start if (_serialDebugging) { Serial.setDebugOutput(true); Serial.printf("Update: %s\n", upload.filename.c_str()); } // Starting update bool error = Update.begin(UPDATE_SIZE_UNKNOWN); if (_serialDebugging && error) Update.printError(Serial); } else if (upload.status == UPLOAD_FILE_WRITE) { if (Update.write(upload.buf, upload.currentSize) != upload.currentSize && _serialDebugging) Update.printError(Serial); } else if (upload.status == UPLOAD_FILE_END) { if (Update.end(true) && _serialDebugging) Serial.printf("Update Success: %u\nRebooting...\n", upload.totalSize); else if(_serialDebugging) Update.printError(Serial); if(_serialDebugging) Serial.setDebugOutput(false); } else if(_serialDebugging) Serial.printf("Update Failed Unexpectedly (likely broken connection): status=%d\n", upload.status); }); _server->begin(); } }; #endif