feat: [#639] Tracker Checker. Setup logging

Author: josecelano

src/console/clients/checker/app.rs

@@ -17,6 +17,7 @@ use std::sync::Arc;
 
 use anyhow::{Context, Result};
 use clap::Parser;
+use log::{debug, LevelFilter};
 
 use super::config::Configuration;
 use super::console::Console;
@@ -39,6 +40,8 @@ struct Args {
 ///
 /// Will return an error if the configuration was not provided.
 pub async fn run() -> Result<Vec<CheckResult>> {
+    setup_logging(LevelFilter::Info);
+
     let args = Args::parse();
 
     let config = setup_config(args)?;
@@ -53,6 +56,27 @@ pub async fn run() -> Result<Vec<CheckResult>> {
     Ok(service.run_checks().await)
 }
 
+fn setup_logging(level: LevelFilter) {
+    if let Err(_err) = fern::Dispatch::new()
+        .format(|out, message, record| {
+            out.finish(format_args!(
+                "{} [{}][{}] {}",
+                chrono::Local::now().format("%+"),
+                record.target(),
+                record.level(),
+                message
+            ));
+        })
+        .level(level)
+        .chain(std::io::stdout())
+        .apply()
+    {
+        panic!("Failed to initialize logging.")
+    }
+
+    debug!("logging initialized.");
+}
+
 fn setup_config(args: Args) -> Result<Configuration> {
     match (args.config_path, args.config_content) {
         (Some(config_path), _) => load_config_from_file(&config_path),

src/console/clients/checker/service.rs

@@ -58,12 +58,16 @@ impl Service {
         self.console.println("UDP trackers ...");
 
         for udp_tracker in &self.config.udp_trackers {
+            debug!("UDP tracker: {:?}", udp_tracker);
+
             let colored_tracker_url = udp_tracker.to_string().yellow();
 
             let transaction_id = TransactionId(RANDOM_TRANSACTION_ID);
 
             let mut client = checker::Client::default();
 
+            debug!("Bind and connect");
+
             let Ok(bound_to) = client.bind_and_connect(ASSIGNED_BY_OS, udp_tracker).await else {
                 check_results.push(Err(CheckError::UdpError {
                     socket_addr: *udp_tracker,
@@ -73,6 +77,8 @@ impl Service {
                 break;
             };
 
+            debug!("Send connection request");
+
             let Ok(connection_id) = client.send_connection_request(transaction_id).await else {
                 check_results.push(Err(CheckError::UdpError {
                     socket_addr: *udp_tracker,
@@ -87,6 +93,8 @@ impl Service {
 
             let info_hash = InfoHash(hex!("9c38422213e30bff212b30c360d26f9a02136422")); // # DevSkim: ignore DS173237
 
+            debug!("Send announce request");
+
             if (client
                 .send_announce_request(connection_id, transaction_id, info_hash, Port(bound_to.port()))
                 .await)
@@ -104,6 +112,8 @@ impl Service {
                     .println(&format!("{} - Announce at {} is failing", "✗".red(), colored_tracker_url));
             }
 
+            debug!("Send scrape request");
+
             let info_hashes = vec![InfoHash(hex!("9c38422213e30bff212b30c360d26f9a02136422"))]; // # DevSkim: ignore DS173237
 
             if (client.send_scrape_request(connection_id, transaction_id, info_hashes).await).is_ok() {