fix: handle ctrl+c properly

Signed-off-by: Miguel A. Cabrera Minagorri <[email protected]>

Author: miguelaeh

pipeless/Cargo.lock

@@ -46,6 +46,7 @@ num_cpus = "1.16.0"
 gstreamer-rtsp = "0.21.0"
 inquire = "0.6.2"
 tabled = "0.15.0"
+ctrlc = "3.4.2"
 
 [dependencies.uuid]
 version = "1.4.1"

pipeless/Cargo.toml

@@ -4,10 +4,16 @@ use tokio::sync::RwLock;
 use gstreamer as gst;
 use glib;
 use tokio;
+use ctrlc;
 
 use crate as pipeless;
 
 pub fn start_pipeless_node(stages_dir: &str) {
+    ctrlc::set_handler(|| {
+        println!("Exiting...");
+        std::process::exit(0);
+    }).expect("Error setting Ctrl+C handler");
+
     pipeless::setup_logger();
     pyo3::prepare_freethreaded_python(); // Setup Pyo3
 
@@ -34,4 +40,4 @@ pub fn start_pipeless_node(stages_dir: &str) {
     });
 
     glib_main_loop.run();
-}
+}