From 81afa96eae91f1110936476aea56ec2b7df731ee Mon Sep 17 00:00:00 2001
From: Nathan LeRoy <NLeRoy917@gmail.com>
Date: Thu, 18 Jan 2024 19:53:52 -0500
Subject: [PATCH] progress bar

---
 genimtools/src/tools/mod.rs | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/genimtools/src/tools/mod.rs b/genimtools/src/tools/mod.rs
index 47bf4884..e9278504 100644
--- a/genimtools/src/tools/mod.rs
+++ b/genimtools/src/tools/mod.rs
@@ -1,6 +1,7 @@
 use std::fs::File;
 use std::io::{BufReader, Write, BufRead};
 
+use indicatif::{ProgressBar, ProgressStyle};
 use walkdir::WalkDir;
 
 pub mod cli;
@@ -29,7 +30,19 @@ pub fn data_dir_stat(path: &str, out: &str) {
     // create the writer
     let mut writer = File::create(out).unwrap();
 
+    // get num files in dir
+    println!("Counting files in directory...");
+    let num_files = WalkDir::new(path).into_iter().filter_map(|e| e.ok()).count();
+
+    // create the progress bar
+    let pb = ProgressBar::new(num_files as u64);
+    pb.set_style(ProgressStyle::with_template("[{elapsed_precise}] {bar:40.cyan/blue} {pos:>7}/{len:7} {msg}")
+        .unwrap()
+        .progress_chars("##-")
+    );
+
     // iterate over the files in the directory
+    println!("Iterating over files in directory...");
     for entry in WalkDir::new(path).into_iter().filter_map(|e| e.ok()) {
         if entry.file_type().is_file() {
 
@@ -44,5 +57,7 @@ pub fn data_dir_stat(path: &str, out: &str) {
             // write the line
             writer.write_all(line.as_bytes()).unwrap();
         }
+
+        pb.inc(1);
     }
 }
\ No newline at end of file