use log::{log_enabled, Level};
use zzz::{ProgressBarIterExt, ProgressBarIter};


// Helper type to hold the iterator, with a progress bar if desired
pub(crate) enum OptionalProgressBarIter<I: Iterator> {
    Enabled(ProgressBarIter<I>),
    Disabled(I),
}

impl<I: Iterator> Iterator for OptionalProgressBarIter<I> {
    type Item = I::Item;

    fn next(&mut self) -> Option<Self::Item> {
        // Simply iterate from the underlying iterator
        match self {
            OptionalProgressBarIter::Enabled(iter) => iter.next(),
            OptionalProgressBarIter::Disabled(iter) => iter.next(),
        }
    }
}

pub(crate) trait ProgressBarIterLogExt: Iterator {
    fn progress_with_level<'a>(self, level: Level) -> OptionalProgressBarIter<Self>
    where
        Self: Sized + 'a,
    {
        if log_enabled!(level) {
            OptionalProgressBarIter::Enabled(self.progress())
        } else {
            OptionalProgressBarIter::Disabled(self)
        }
    }
}
impl<I> ProgressBarIterLogExt for I where I: Iterator {}

some_iterator.progress_with_level(Level::Info)

use std::thread;
use std::time::Duration;

use rayon::prelude::*;
use zzz::ProgressBar;

fn main() {
    let elems = vec![0; 10000];
    let pb = ProgressBar::with_target(elems.len());
    elems.par_iter().for_each(|_| {
        pb.add_sync(1);
        thread::sleep(Duration::from_millis(5));
    });
}