From a345b210d034b642419044dfc39318a4acc61022 Mon Sep 17 00:00:00 2001
From: sgoudham <sgoudham@gmail.com>
Date: Sun, 30 Jan 2022 23:41:24 +0000
Subject: [PATCH] Update progress bar

---
 src/lib.rs | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index f246ca7..60a9338 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -17,12 +17,14 @@ mod constants;
 mod seeder;
 
 macro_rules! progress_bar {
-    () => {{
-        let progress_bar = ProgressBar::new_spinner();
+    ($bytes:expr) => {{
+        let progress_bar = ProgressBar::new($bytes);
         progress_bar.set_style(
-            ProgressStyle::default_spinner().template("{spinner:.green} [{elapsed_precise}] {msg}"),
+            ProgressStyle::default_spinner()
+                .template("{spinner:.magenta} [{elapsed_precise:.bold}] {msg:.green.bold}"),
         );
-        progress_bar.enable_steady_tick(100);
+        progress_bar.set_message("UwU'ifying In Progress...");
+        progress_bar.enable_steady_tick(30);
 
         progress_bar
     }};
@@ -113,7 +115,7 @@ impl<'a> UwUify<'a> {
                     ));
                 }
 
-                let uwu_progress_bar = progress_bar!();
+                let uwu_progress_bar = progress_bar!(self.text.len() as u64);
                 self.uwuify_sentence(self.text, &mut BufWriter::new(File::create(&self.output)?))?;
                 uwu_progress_bar.finish_with_message("UwU'ifying Complete!");
             } else {
@@ -128,9 +130,10 @@ impl<'a> UwUify<'a> {
                 ));
             }
 
-            let uwu_progress_bar = progress_bar!();
+            let infile = File::open(&self.input)?;
+            let uwu_progress_bar = progress_bar!(infile.metadata()?.len());
             self.uwuify_sentence(
-                unsafe { from_utf8_unchecked(Mmap::map(&File::open(&self.input)?)?.as_ref()) },
+                unsafe { from_utf8_unchecked(Mmap::map(&infile)?.as_ref()) },
                 &mut BufWriter::new(File::create(&self.output)?),
             )?;
             uwu_progress_bar.finish_with_message("UwU'ifying Complete!");