Merge #44

44: Implement progress bar r=obbardc a=Razaloc

Display a progress bar for the copy process

Closes: #33 

Signed-off-by: Rafael Garcia Ruiz <rafael.garcia@collabora.com>

Co-authored-by: Rafael Garcia Ruiz <rafael.garcia@collabora.com>

Author: bors[bot]

bmap-rs/Cargo.toml

@@ -12,3 +12,4 @@ anyhow = "1.0.66"
 nix = "0.25.0"
 flate2 = "1.0.24"
 clap = { version = "4.0.18", features = ["derive"] }
+indicatif = "0.17.1"

bmap-rs/src/main.rs

@@ -2,8 +2,10 @@ use anyhow::{anyhow, bail, Context, Result};
 use bmap::{Bmap, Discarder, SeekForward};
 use clap::Parser;
 use flate2::read::GzDecoder;
+use indicatif::{ProgressBar, ProgressState, ProgressStyle};
 use nix::unistd::ftruncate;
 use std::ffi::OsStr;
+use std::fmt::Write;
 use std::fs::File;
 use std::io::Read;
 use std::os::unix::io::AsRawFd;
@@ -100,7 +102,7 @@ fn copy(c: Copy) -> Result<()> {
     b.read_to_string(&mut xml)?;
 
     let bmap = Bmap::from_xml(&xml)?;
-    let mut output = std::fs::OpenOptions::new()
+    let output = std::fs::OpenOptions::new()
         .write(true)
         .create(true)
         .open(c.dest)?;
@@ -111,7 +113,14 @@ fn copy(c: Copy) -> Result<()> {
     }
 
     let mut input = setup_input(&c.image)?;
-    bmap::copy(&mut input, &mut output, &bmap)?;
+    let pb = ProgressBar::new(bmap.total_mapped_size());
+    pb.set_style(ProgressStyle::with_template("{spinner:.green} [{elapsed_precise}] [{wide_bar:.cyan/blue}] {bytes}/{total_bytes} ({eta})")
+        .unwrap()
+        .with_key("eta", |state: &ProgressState, w: &mut dyn Write| write!(w, "{:.1}s", state.eta().as_secs_f64()).unwrap())
+        .progress_chars("#>-"));
+    bmap::copy(&mut input, &mut pb.wrap_write(&output), &bmap)?;
+    pb.finish_and_clear();
+
     println!("Done: Syncing...");
     output.sync_all().expect("Sync failure");
 

bmap/src/bmap.rs

@@ -92,6 +92,9 @@ impl Bmap {
     pub fn block_map(&self) -> impl ExactSizeIterator + Iterator<Item = &BlockRange> {
         self.blockmap.iter()
     }
+    pub fn total_mapped_size(&self) -> u64 {
+        self.block_size * self.mapped_blocks
+    }
 }
 
 #[derive(Clone, Debug, Error)]