cli+github: show download progress

The spinner implementation has been improved to show
the remaining and total amount of bytes to download

Signed-off-by: Orhun Parmaksız <[email protected]>

src/cli/handlers/download.rs

@@ -1,4 +1,6 @@
+use std::cmp;
 use std::fs::File;
+use std::io::{Read, Write};
 use std::path::{Path, PathBuf};
 
 use crate::cli::get_env;
@@ -107,11 +109,22 @@ impl DownloadHandler {
     ) -> Result<(), HandlerError> {
         let spinner = Spinner::download(&selected_asset.name, output_path);
         spinner.start();
-        let mut stream =
+        let (mut stream, content_length) =
             github::download_asset(client, selected_asset).map_err(Self::download_error)?;
+        spinner.set_max_progress(content_length);
         let mut destination = Self::create_file(output_path)?;
-        std::io::copy(&mut stream, &mut destination)
-            .map_err(|x| Self::copy_err(&selected_asset.name, output_path, x))?;
+        let mut downloaded = 0;
+        let mut buffer = [0; 1024];
+        while let Ok(bytes_read) = stream.read(&mut buffer) {
+            if bytes_read == 0 {
+                break;
+            }
+            destination
+                .write(&buffer[..bytes_read])
+                .map_err(|x| Self::write_err(&selected_asset.name, output_path, x))?;
+            downloaded = cmp::min(downloaded + bytes_read as u64, content_length);
+            spinner.update_progress(downloaded);
+        }
         spinner.stop();
         Ok(())
     }
@@ -151,9 +164,9 @@ impl DownloadHandler {
         })
     }
 
-    pub fn copy_err(asset_name: &str, output_path: &Path, error: std::io::Error) -> HandlerError {
+    pub fn write_err(asset_name: &str, output_path: &Path, error: std::io::Error) -> HandlerError {
         HandlerError::new(format!(
-            "Error copying {} to {}: {}",
+            "Error saving {} to {}: {}",
             asset_name,
             output_path.display(),
             error

src/cli/spinner.rs

@@ -13,12 +13,12 @@ pub struct Spinner {
 }
 
 impl Spinner {
-    pub fn new(message: String, end_message: String) -> Self {
+    pub fn new(message: String, end_message: String, template: &str) -> Self {
         let pb = ProgressBar::new_spinner();
         pb.set_style(
             ProgressStyle::default_spinner()
                 .tick_strings(TICKS)
-                .template("{spinner:.blue} {msg}"),
+                .template(template),
         );
         pb.set_message(message);
         Self { pb, end_message }
@@ -38,24 +38,34 @@ impl Spinner {
         println!("{}", message);
     }
 
+    pub fn set_max_progress(&self, progress: u64) {
+        self.pb.set_length(progress);
+    }
+
+    pub fn update_progress(&self, progress: u64) {
+        self.pb.set_position(progress);
+    }
+
     pub fn download(download_asset: &str, output_path: &Path) -> Spinner {
         Spinner::new(
             format!("Downloading {}", Color::new(download_asset).bold()),
             format!(
                 "Saved to: {}",
                 Color::new(&format!("{}", output_path.display())).bold()
             ),
+            "{msg}\n{percent}% [{wide_bar}] {bytes}/{total_bytes} ({eta})",
         )
     }
 
     pub fn install() -> Spinner {
         Spinner::new(
             "Installing".into(),
             format!("{}", Color::new("Installation completed!").green()),
+            "{spinner:.blue} {msg}",
         )
     }
 
     pub fn no_messages() -> Spinner {
-        Spinner::new(String::new(), String::new())
+        Spinner::new(String::new(), String::new(), "{spinner:.blue} {msg}")
     }
 }

src/github/error.rs

@@ -4,6 +4,7 @@ use std::fmt::Formatter;
 pub enum GithubError {
     Http(Box<ureq::Error>),
     JsonDeserialization(std::io::Error),
+    InvalidContentLength,
     RepositoryOrReleaseNotFound,
     RateLimitExceeded,
     Unauthorized,
@@ -28,6 +29,9 @@ impl std::fmt::Display for GithubError {
             GithubError::JsonDeserialization(e) => {
                 f.write_str(&format!("Error deserializing response: {}", e))
             }
+            GithubError::InvalidContentLength => {
+                f.write_str("Content-Length header is missing or invalid")
+            }
             GithubError::RepositoryOrReleaseNotFound => {
                 f.write_str("Repository or release not found")
             }

src/github/mod.rs

@@ -49,11 +49,15 @@ fn deserialize(response: ureq::Response) -> Result<Release, GithubError> {
 pub fn download_asset(
     client: &GithubClient,
     asset: &Asset,
-) -> Result<impl Read + Send, GithubError> {
-    client
+) -> Result<(impl Read + Send, u64), GithubError> {
+    let response = client
         .get(&asset.download_url)
         .set("Accept", "application/octet-stream")
         .call()
-        .map_err(GithubError::from)
-        .map(|response| response.into_reader())
+        .map_err(GithubError::from)?;
+    let content_length = response
+        .header("Content-Length")
+        .and_then(|v| v.parse().ok())
+        .ok_or(GithubError::InvalidContentLength)?;
+    Ok((response.into_reader(), content_length))
 }