Display download and installation completion using Reporter

crates/uv-installer/src/installer.rs

@@ -74,7 +74,13 @@ impl<'a> Installer<'a> {
 
                 Ok::<(), Error>(())
             })
-        })
+        })?;
+
+        if let Some(reporter) = self.reporter.as_ref() {
+            reporter.on_install_complete();
+        }
+
+        Ok(())
     }
 }
 

crates/uv/src/commands/pip_install.rs

@@ -739,8 +739,6 @@ async fn install(
     let wheels = if remote.is_empty() {
         vec![]
     } else {
-        let start = std::time::Instant::now();
-
         let downloader = Downloader::new(cache, tags, hasher, client, build_dispatch)
             .with_reporter(DownloadReporter::from(printer).with_length(remote.len() as u64));
 
@@ -749,18 +747,6 @@ async fn install(
             .await
             .context("Failed to download distributions")?;
 
-        let s = if wheels.len() == 1 { "" } else { "s" };
-        writeln!(
-            printer.stderr(),
-            "{}",
-            format!(
-                "Downloaded {} in {}",
-                format!("{} package{}", wheels.len(), s).bold(),
-                elapsed(start.elapsed())
-            )
-            .dimmed()
-        )?;
-
         wheels
     };
 
@@ -794,23 +780,10 @@ async fn install(
     // Install the resolved distributions.
     let wheels = wheels.into_iter().chain(cached).collect::<Vec<_>>();
     if !wheels.is_empty() {
-        let start = std::time::Instant::now();
         uv_installer::Installer::new(venv)
             .with_link_mode(link_mode)
             .with_reporter(InstallReporter::from(printer).with_length(wheels.len() as u64))
             .install(&wheels)?;
-
-        let s = if wheels.len() == 1 { "" } else { "s" };
-        writeln!(
-            printer.stderr(),
-            "{}",
-            format!(
-                "Installed {} in {}",
-                format!("{} package{}", wheels.len(), s).bold(),
-                elapsed(start.elapsed())
-            )
-            .dimmed()
-        )?;
     }
 
     if compile {

crates/uv/src/commands/reporters.rs

@@ -1,5 +1,6 @@
+use std::fmt::Write;
 use std::sync::{Arc, Mutex};
-use std::time::Duration;
+use std::time::{Duration, Instant};
 
 use indicatif::{MultiProgress, ProgressBar, ProgressStyle};
 use owo_colors::OwoColorize;
@@ -11,6 +12,7 @@ use distribution_types::{
 };
 use uv_normalize::PackageName;
 
+use crate::commands::elapsed;
 use crate::printer::Printer;
 
 #[derive(Debug)]
@@ -19,6 +21,7 @@ pub(crate) struct DownloadReporter {
     multi_progress: MultiProgress,
     progress: ProgressBar,
     bars: Arc<Mutex<Vec<ProgressBar>>>,
+    start_time: Instant,
 }
 
 impl From<Printer> for DownloadReporter {
@@ -36,6 +39,8 @@ impl From<Printer> for DownloadReporter {
             multi_progress,
             progress,
             bars: Arc::new(Mutex::new(Vec::new())),
+            // TODO(zanieb): Consider setting the start time on the start of the first download insteasd
+            start_time: Instant::now(),
         }
     }
 }
@@ -78,6 +83,20 @@ impl uv_installer::DownloadReporter for DownloadReporter {
 
     fn on_complete(&self) {
         self.progress.finish_and_clear();
+        if let Some(length) = self.progress.length() {
+            let s = if length == 1 { "" } else { "s" };
+            writeln!(
+                self.printer.stderr(),
+                "{}",
+                format!(
+                    "Downloaded {} in {}",
+                    format!("{} package{}", length, s).bold(),
+                    elapsed(self.start_time.elapsed())
+                )
+                .dimmed()
+            )
+            .unwrap();
+        };
     }
 
     fn on_build_start(&self, source: &BuildableSource) -> usize {
@@ -130,7 +149,9 @@ impl uv_installer::DownloadReporter for DownloadReporter {
 
 #[derive(Debug)]
 pub(crate) struct InstallReporter {
+    printer: Printer,
     progress: ProgressBar,
+    start_time: Instant,
 }
 
 impl From<Printer> for InstallReporter {
@@ -140,7 +161,11 @@ impl From<Printer> for InstallReporter {
             ProgressStyle::with_template("{bar:20} [{pos}/{len}] {wide_msg:.dim}").unwrap(),
         );
         progress.set_message("Installing wheels...");
-        Self { progress }
+        Self {
+            printer,
+            progress,
+            start_time: Instant::now(),
+        }
     }
 }
 
@@ -160,6 +185,20 @@ impl uv_installer::InstallReporter for InstallReporter {
 
     fn on_install_complete(&self) {
         self.progress.finish_and_clear();
+        if let Some(length) = self.progress.length() {
+            let s = if length == 1 { "" } else { "s" };
+            writeln!(
+                self.printer.stderr(),
+                "{}",
+                format!(
+                    "Installed {} in {}",
+                    format!("{} package{}", length, s).bold(),
+                    elapsed(self.start_time.elapsed())
+                )
+                .dimmed()
+            )
+            .unwrap();
+        };
     }
 }