From 2e8f328278fb203648114edea17305b4c21ebb05 Mon Sep 17 00:00:00 2001
From: bwmp <oli@bwmp.dev>
Date: Sat, 25 Oct 2025 18:06:18 -0700
Subject: [PATCH] fix

---
 .../CommandManager/commands/device_commands.cpp        |  2 +-
 main/openiris_main.cpp                                 |  6 +++++-
 tools/setup_openiris.py                                | 10 ++++++++++
 3 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/components/CommandManager/CommandManager/commands/device_commands.cpp b/components/CommandManager/CommandManager/commands/device_commands.cpp
index 42a098b..37f64f1 100644
--- a/components/CommandManager/CommandManager/commands/device_commands.cpp
+++ b/components/CommandManager/CommandManager/commands/device_commands.cpp
@@ -113,7 +113,7 @@ CommandResult startStreamingCommand()
     esp_timer_handle_t activateStreamingTimer;
     esp_timer_create(&args, &activateStreamingTimer);
     esp_timer_start_once(activateStreamingTimer, pdMS_TO_TICKS(150));
-
+    streamServer.startStreamServer();
     return CommandResult::getSuccessResult("Streaming starting");
 }
 
diff --git a/main/openiris_main.cpp b/main/openiris_main.cpp
index c57a3ea..75ba6e2 100644
--- a/main/openiris_main.cpp
+++ b/main/openiris_main.cpp
@@ -215,7 +215,11 @@ void startWiFiMode()
     wifiManager->Begin();
     mdnsManager.start();
     restAPI->begin();
-
+    StreamingMode mode = deviceConfig->getDeviceMode();
+    if (mode == StreamingMode::WIFI)
+    {
+        streamServer.startStreamServer();
+    }
     xTaskCreate(
         HandleRestAPIPollTask,
         "HandleRestAPIPollTask",
diff --git a/tools/setup_openiris.py b/tools/setup_openiris.py
index 7d1b1a2..ac1cd0d 100644
--- a/tools/setup_openiris.py
+++ b/tools/setup_openiris.py
@@ -482,6 +482,15 @@ def get_settings_summary(device: OpenIrisDevice, *args, **kwargs):
         configured = wifi.get("networks_configured", 0)
         print(f"📶 WiFi: {status}  |  IP: {ip}  |  Networks configured: {configured}")
 
+def restart_device_command(device: OpenIrisDevice, *args, **kwargs):
+    print("🔄 Restarting device...")
+    response = device.send_command("restart_device")
+    if has_command_failed(response):
+        print(f"❌ Failed to restart device: {response['error']}")
+        return
+
+    print("✅ Device restart command sent successfully")
+    print("💡 Please wait a few seconds for the device to reboot")
 
 def scan_networks(wifi_scanner: WiFiScanner, *args, **kwargs):
     use_custom_timeout = (
@@ -668,6 +677,7 @@ def handle_menu(menu_context: dict | None = None) -> str:
     menu.add_action("🔄 Switch device mode (WiFi/UVC/Auto)", switch_device_mode_command)
     menu.add_action("💡 Update PWM Duty Cycle", set_led_duty_cycle)
     menu.add_action("🧩 Get settings summary", get_settings_summary)
+    menu.add_action("🔪 Restart device", restart_device_command)
     menu.show()