feat: restart esp while ws disconnect on update

Author: filipton

src/main.rs

@@ -92,6 +92,9 @@ async fn main(spawner: Spawner) {
         }
     }
 
+    #[cfg(feature = "e2e")]
+    log::info!("This firmware is E2E! (HIL TESTING)");
+
     let nvs =
         esp_hal_wifimanager::Nvs::new_from_part_table().expect("Wrong partition configuration!");
 

src/ws.rs

@@ -1,6 +1,6 @@
 use crate::{
     consts::WS_RETRY_MS,
-    state::{GlobalState, Scene},
+    state::{ota_state, GlobalState, Scene},
     structs::{ApiError, FromPacket, TimerPacket, TimerPacketInner},
 };
 use alloc::{rc::Rc, string::ToString};
@@ -198,6 +198,14 @@ async fn ws_loop(
             .await;
 
             if let Err(e) = res {
+                if ota_state() {
+                    global_state.state.lock().await.custom_message =
+                        Some(("Connection lost".to_string(), "during update".to_string()));
+
+                    Timer::after_millis(5000).await;
+                    esp_hal::reset::software_reset();
+                }
+
                 log::error!("ws_rw_error: {e:?}");
                 Timer::after_millis(WS_RETRY_MS).await;
                 break;