Stop showing ETA after download is complete

rust-lang · May 15, 2024 · eca5013 · eca5013
1 parent fcff18c
commit eca5013
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/src/cli/download_tracker.rs b/src/cli/download_tracker.rs
@@ -186,13 +186,13 @@ impl DownloadTracker {
                             (remaining / speed) as u64
                         });
                         format!(
-                            "{} / {} ({:3.0} %) {} in {} ETA: {}",
+                            "{} / {} ({:3.0} %) {} in {}{}",
                             total_h,
                             content_len_h,
                             percent,
                             speed_h,
                             elapsed_h.display(),
-                            eta_h.display(),
+                            Eta(eta_h),
                         )
                     }
                     None => format!(
@@ -220,6 +220,17 @@ impl DownloadTracker {
     }
 }
 
+struct Eta(Duration);
+
+impl fmt::Display for Eta {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        match self.0 {
+            Duration::ZERO => Ok(()),
+            _ => write!(f, "{}", self.0.display()),
+        }
+    }
+}
+
 trait DurationDisplay {
     fn display(self) -> Display;
 }