From 25cbf4d2f27be4bcd728ed4c14c0f93cb7cd08cf Mon Sep 17 00:00:00 2001
From: Trevor Merritt <trevor.merritt@gmail.com>
Date: Mon, 21 Jul 2025 08:12:23 -0400
Subject: [PATCH] Adds colored logs for when a client connects or disconnects

---
 cli/src/main.rs | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/cli/src/main.rs b/cli/src/main.rs
index 7de6c02..b3e9f12 100644
--- a/cli/src/main.rs
+++ b/cli/src/main.rs
@@ -53,7 +53,11 @@ fn main() -> std::io::Result<()> {
     Ok(())
 }
 
+use colored::*; // Add this at the top of your file
+
 fn handle_client(mut client: TcpStream, target: &str, client_id: usize) -> std::io::Result<()> {
+    println!("{}", format!("Client {} connected", client_id).green());
+
     let mut server = TcpStream::connect(target)?;
     let timestamp = SystemTime::now()
         .duration_since(SystemTime::UNIX_EPOCH)
@@ -80,6 +84,7 @@ fn handle_client(mut client: TcpStream, target: &str, client_id: usize) -> std::
     client_to_server.join().unwrap()?;
     server_to_client.join().unwrap()?;
 
+    println!("{}", format!("Client {} disconnected", client_id).red());
     Ok(())
 }