Add a device search function and improve the orderly shutdown

- Add the 'searchDevices' function to initialize and search for devices
- Implement the 'stopAllDevices' function to stop all devices and delete device lists
- Update the 'consoleHandler' to use 'searchDevices'
- Integrate 'stopAllDevices' into 'signalHandler' for a more comprehensive shutdown process
- Add CLI flag '--search' to allow users to search for devices via the command line
- Ensure an orderly shutdown on SIGINT by stopping all devices

These changes improve the application by providing a dedicated device search function and ensuring a thorough cleanup on shutdown.

src/handler.cpp

@@ -34,6 +34,24 @@ void initDevices() {
   // std::cout << "Device initialization complete.\n";
 }
 
+void searchDevices() {
+    initDevices();
+    if (devices.empty()) {
+        std::cout << "No devices found.\n";
+    }
+    stopAllDevices();
+}
+
+void stopAllDevices() {
+    if (sampler.has_value()) {
+        for (auto &device : sampler->sampleDevices) {
+            device.first->send(Omniscope::Stop{});
+        }
+    }
+    devices.clear();
+    deviceManager.clearDevices();
+}
+
 void consoleHandler(bool &flagPaused, std::set<std::string>& selected_serials) {
     std::string input;
     while (running) {
@@ -53,9 +71,8 @@ void consoleHandler(bool &flagPaused, std::set<std::string>& selected_serials) {
                     input.erase(0, pos + 1);
                 }
                 selected_serials.insert(input);
-            }
-            else {
-                std::cout << "No devices fpound.\n";
+            } else {
+                std::cout << "No devices found.\n";
             }
         }
         else if (input == "Start") {
@@ -97,5 +114,6 @@ void signalHandler(int signal) {
     if (signal == SIGINT) {
         std::cout << "\nSIGINT received, shutting down gracefully...\n";
         running = false;
+        stopAllDevices();
     }
 }

src/handler.hpp

@@ -16,7 +16,10 @@ inline std::map<Omniscope::Id, std::vector<std::pair<double, double>>> captureDa
 inline std::atomic<bool> running{true};
 
 void initDevices();
+void searchDevices(std::set<std::string>& selected_serials);
 void consoleHandler(bool &flagPaused, std::set<std::string>& selected_serials);
 void signalHandler(int signal);
+void stopAllDevices();
+void searchDevices();
 
 #endif

src/main.cpp

@@ -9,13 +9,20 @@ int main(int argc, char** argv) {
     bool flagPaused{true};
     std::set<std::string> selected_serials;
     std::string wsURI;
+    bool search = false;
 
     CLI::App app{"OmniView"};
 
     app.add_option("-w, --wsuri", wsURI, "WebSocket URI")->type_name("");
+    app.add_flag("--search", search, "Search for devices");
 
     CLI11_PARSE(app, argc, argv);
 
+    if (search) {
+        searchDevices();
+        return 0;
+    }
+
     WebSocketHandler wsHandler(wsURI);
 
     // Signal handler for SIGINT