feat: add logging capabilities

Signed-off-by: addrian-77 <lunguadrian30@gmail.com>

Author: addrian-77

tockloader-cli/Cargo.toml

@@ -26,3 +26,5 @@ glob = "0.3.1"
 inquire = "0.7.5"
 tockloader-lib = { path = "../tockloader-lib/" }
 anyhow = "1.0.89"
+env_logger = "0.11.8"
+log = "0.4.27"

tockloader-cli/src/cli.rs

@@ -14,9 +14,13 @@ pub fn make_cli() -> Command {
         .version(crate_version!())
         .subcommand_required(true)
         .subcommands(get_subcommands())
-        .args([
-            arg!(--debug "Print additional debugging information").action(clap::ArgAction::SetTrue)
-        ])
+        .arg(
+            arg!(--"log-level" <LEVEL>)
+                .required(false)
+                .value_parser(["error", "warn", "info", "debug", "trace"])
+                .default_value("info")
+                .global(true),
+        )
     // Note: arg_require_else_help will trigger the help command if no argument/subcommand is given.
     // This means that the --debug flag will not trigger the help menu, even if alone it does nothing.
 }

tockloader-cli/src/main.rs

@@ -134,6 +134,28 @@ async fn main() -> Result<()> {
     let mut cmd = cli::make_cli();
     let matches = cmd.get_matches_mut();
 
+    let user_level = matches
+        .get_one::<String>("log-level")
+        .map(String::as_str)
+        .unwrap();
+
+    let mut builder = env_logger::Builder::new();
+
+    let cli_level = match user_level {
+        "error" => log::LevelFilter::Error,
+        "warn" => log::LevelFilter::Warn,
+        "info" => log::LevelFilter::Info,
+        "debug" => log::LevelFilter::Debug,
+        "trace" => log::LevelFilter::Trace,
+        level => panic!("Unknown log level: {}", level),
+    };
+
+    builder.filter_level(log::LevelFilter::Off);
+    builder.filter_module("tockloader-lib", cli_level);
+    builder.filter_module("tockloader", cli_level);
+
+    builder.init();
+
     match matches.subcommand() {
         Some(("listen", sub_matches)) => {
             cli::validate(&mut cmd, sub_matches);

tockloader-lib/Cargo.toml

@@ -20,3 +20,4 @@ toml = "0.8.19"
 serde = { version = "1.0.210", features = ["derive"] }
 thiserror = "1.0.63"
 async-trait = "0.1.88"
+log = "0.4.27"

tockloader-lib/src/connection.rs

@@ -7,7 +7,7 @@ use tokio::io::AsyncWriteExt;
 use tokio_serial::{FlowControl, Parity, SerialPort, SerialStream, StopBits};
 
 use crate::errors::TockloaderError;
-
+use log::info;
 pub struct ProbeTargetInfo {
     pub chip: String,
     pub core: usize,
@@ -78,13 +78,15 @@ impl Connection for ProbeRSConnection {
         let probe = self.debug_probe.open()?;
 
         self.session = Some(probe.attach(&self.target_info.chip, Permissions::default())?);
+        info!("ProbeRS connection opened succesfully.");
 
         Ok(())
     }
 
     async fn close(&mut self) -> Result<(), TockloaderError> {
         // Session implements Drop, so we don't need to explicitly close it.
         self.session = None;
+        info!("ProbeRS connection closed.");
         Ok(())
     }
 
@@ -130,6 +132,7 @@ impl Connection for SerialConnection {
         stream.write_data_terminal_ready(self.target_info.data_terminal_ready)?;
 
         self.stream = Some(stream);
+        info!("Serial connection opened succesfully.");
         Ok(())
     }
 
@@ -140,6 +143,7 @@ impl Connection for SerialConnection {
                 .await
                 .map_err(|e| TockloaderError::Serial(e.into()))?;
         }
+        info!("Serial connection closed.");
         Ok(())
     }