diff --git a/src/main.rs b/src/main.rs index 6194d56..0d3c8c1 100644 --- a/src/main.rs +++ b/src/main.rs @@ -9,7 +9,7 @@ use tokio_tungstenite::{ connect_async_with_config, tungstenite::{ client::IntoClientRequest, - http::header::{AUTHORIZATION, HeaderValue, USER_AGENT}, + http::header::{HeaderValue, AUTHORIZATION, USER_AGENT}, Message, }, }; @@ -36,7 +36,11 @@ struct Args { async fn main() -> Result<(), Box> { let args = Args::parse(); - let log_level = if args.debug { Level::DEBUG } else { Level::INFO }; + let log_level = if args.debug { + Level::DEBUG + } else { + Level::INFO + }; tracing_subscriber::fmt() .with_target(false) .with_thread_ids(false) @@ -77,20 +81,41 @@ async fn main() -> Result<(), Box> { info!("[{}] Connecting to {}", letter, url); if debug_enabled { - debug!("[{}] Request: {} {}", letter, request.method(), request.uri()); + debug!( + "[{}] Request: {} {}", + letter, + request.method(), + request.uri() + ); for (name, value) in request.headers() { - debug!("[{}] {}: {}", letter, name, value.to_str().unwrap_or("")); + debug!( + "[{}] {}: {}", + letter, + name, + value.to_str().unwrap_or("") + ); } } - let (ws_stream, response) = match connect_async_with_config(request, None, false).await { + let (ws_stream, response) = match connect_async_with_config(request, None, false).await + { Ok(result) => result, Err(e) => { if debug_enabled { if let tokio_tungstenite::tungstenite::Error::Http(ref response) = e { - debug!("[{}] Error response: {} {}", letter, response.status().as_u16(), response.status().canonical_reason().unwrap_or("")); + debug!( + "[{}] Error response: {} {}", + letter, + response.status().as_u16(), + response.status().canonical_reason().unwrap_or("") + ); for (name, value) in response.headers() { - debug!("[{}] {}: {}", letter, name, value.to_str().unwrap_or("")); + debug!( + "[{}] {}: {}", + letter, + name, + value.to_str().unwrap_or("") + ); } if let Some(body) = response.body() { if let Ok(body_str) = std::str::from_utf8(body) { @@ -103,12 +128,26 @@ async fn main() -> Result<(), Box> { } }; - info!("[{}] Connected successfully (status: {})", letter, response.status()); + info!( + "[{}] Connected successfully (status: {})", + letter, + response.status() + ); if debug_enabled { - debug!("[{}] Response: {} {}", letter, response.status().as_u16(), response.status().canonical_reason().unwrap_or("")); + debug!( + "[{}] Response: {} {}", + letter, + response.status().as_u16(), + response.status().canonical_reason().unwrap_or("") + ); for (name, value) in response.headers() { - debug!("[{}] {}: {}", letter, name, value.to_str().unwrap_or("")); + debug!( + "[{}] {}: {}", + letter, + name, + value.to_str().unwrap_or("") + ); } } @@ -151,7 +190,8 @@ async fn main() -> Result<(), Box> { let truncated = if text.len() > 50 { "..." } else { "" }; info!("[{}:{}] Text: {}{}", letter, seq_num, preview, truncated); - let filename = session_dir.join(format!("{}{}.txt", letter, seq_num)); + let filename = + session_dir.join(format!("{}{}.txt", letter, seq_num)); if let Err(e) = fs::write(&filename, &text) { error!("[{}] Failed to write {:?}: {}", letter, filename, e); } @@ -159,7 +199,8 @@ async fn main() -> Result<(), Box> { Message::Binary(data) => { info!("[{}:{}] Binary: {} bytes", letter, seq_num, data.len()); - let filename = session_dir.join(format!("{}{}.bin", letter, seq_num)); + let filename = + session_dir.join(format!("{}{}.bin", letter, seq_num)); if let Err(e) = fs::write(&filename, &data) { error!("[{}] Failed to write {:?}: {}", letter, filename, e); } @@ -174,7 +215,10 @@ async fn main() -> Result<(), Box> { } Message::Close(frame) => { if let Some(cf) = frame { - info!("[{}] Connection closed: {} - {}", letter, cf.code, cf.reason); + info!( + "[{}] Connection closed: {} - {}", + letter, cf.code, cf.reason + ); } else { info!("[{}] Connection closed", letter); } @@ -209,7 +253,10 @@ async fn main() -> Result<(), Box> { } } - info!("All connections closed. Session saved to: {}", session_dir.display()); + info!( + "All connections closed. Session saved to: {}", + session_dir.display() + ); Ok(()) }