From 1baa28bb126636cf39c8df97e826547e1eecff34 Mon Sep 17 00:00:00 2001
From: chanderlud <me@chanchan.dev>
Date: Tue, 26 Dec 2023 15:40:57 +0000
Subject: [PATCH] added speed to progress bar

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

diff --git a/src/main.rs b/src/main.rs
index b61be9d..083d109 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -19,7 +19,7 @@ use simple_logging::{log_to_file, log_to_stderr};
 use tokio::fs::File;
 use tokio::io::{AsyncReadExt, BufReader};
 use tokio::net::{TcpListener, TcpSocket, TcpStream, UdpSocket};
-use tokio::time::{interval, sleep};
+use tokio::time::{Instant, interval, sleep};
 use tokio::{io, select};
 
 use crate::cipher::CipherStream;
@@ -330,10 +330,11 @@ fn password_auth() -> io::Result<AuthMethod> {
 async fn print_progress(stats: TransferStats) {
     let bar = ProgressBar::new(100);
     let mut interval = interval(Duration::from_secs(1));
+    let now = Instant::now();
 
     bar.set_style(
         ProgressStyle::default_bar()
-            .template("{spinner:.green} [{elapsed_precise}] [{bar:40.cyan/blue}] {pos:>7}% ({eta})")
+            .template("{spinner:.green} [{elapsed_precise}] [{bar:40.cyan/blue}] {pos:>7}% {msg}MB/s ({eta})")
             .unwrap()
             .progress_chars("=>-"),
     );
@@ -342,6 +343,8 @@ async fn print_progress(stats: TransferStats) {
         interval.tick().await;
         let progress =
             stats.confirmed_data.load(Relaxed) as f64 / stats.total_data.load(Relaxed) as f64;
+        let speed = stats.confirmed_data.load(Relaxed) as f64 / now.elapsed().as_secs_f64() / 1_000_000_f64;
+        bar.set_message(format!("{:.2}", speed));
         bar.set_position((progress * 100_f64) as u64);
     }
 }