Add ProgressIterator::progress_with_style()

Style a ProgressIterator without manually handling the length of the bar

src/iter.rs

@@ -33,6 +33,16 @@ where
 
     /// Wrap an iterator with a custom progress bar.
     fn progress_with(self, progress: ProgressBar) -> ProgressBarIter<Self>;
+
+    /// Wrap an iterator with a progress bar and style it.
+    fn progress_with_style(self, style: crate::ProgressStyle) -> ProgressBarIter<Self>
+    where
+        Self: ExactSizeIterator,
+    {
+        let len = u64::try_from(self.len()).unwrap();
+        let bar = ProgressBar::new(len).with_style(style);
+        self.progress_with(bar)
+    }
 }
 
 /// Wraps an iterator to display its progress.
@@ -166,6 +176,7 @@ impl<S, T: Iterator<Item = S>> ProgressIterator for T {
 mod test {
     use crate::iter::{ProgressBarIter, ProgressIterator};
     use crate::progress_bar::ProgressBar;
+    use crate::ProgressStyle;
 
     #[test]
     fn it_can_wrap_an_iterator() {
@@ -180,5 +191,9 @@ mod test {
             let pb = ProgressBar::new(v.len() as u64);
             v.iter().progress_with(pb)
         });
+        wrap({
+            let style = ProgressStyle::default_bar().template("{wide_bar:.red} {percent}/100%");
+            v.iter().progress_with_style(style)
+        });
     }
 }

src/rayon.rs

@@ -28,6 +28,16 @@ where
         let len = u64::try_from(self.len()).unwrap();
         self.progress_count(len)
     }
+
+    /// Wrap an iterator with a progress bar and style it.
+    fn progress_with_style(self, style: crate::ProgressStyle) -> ProgressBarIter<Self>
+    where
+        Self: IndexedParallelIterator,
+    {
+        let len = u64::try_from(self.len()).unwrap();
+        let bar = ProgressBar::new(len).with_style(style);
+        self.progress_with(bar)
+    }
 }
 
 impl<S: Send, T: ParallelIterator<Item = S>> ParallelProgressIterator for T {
@@ -199,6 +209,7 @@ impl<S: Send, T: ParallelIterator<Item = S>> ParallelIterator for ProgressBarIte
 
 #[cfg(test)]
 mod test {
+    use crate::ProgressStyle;
     use crate::{ParallelProgressIterator, ProgressBar, ProgressBarIter};
     use rayon::iter::{IntoParallelRefIterator, ParallelIterator};
 
@@ -214,5 +225,10 @@ mod test {
             let pb = ProgressBar::new(v.len() as u64);
             v.par_iter().progress_with(pb)
         });
+
+        wrap({
+            let style = ProgressStyle::default_bar().template("{wide_bar:.red} {percent}/100%");
+            v.par_iter().progress_with_style(style)
+        });
     }
 }