datagen-rs-cli 0.1.0

use colored::Colorize;
use indicatif::{ProgressBar, ProgressState, ProgressStyle};
use num_format::{Locale, ToFormattedString};
use std::fmt::Write;
use std::rc::Rc;
use std::sync::Mutex;
use std::time::Duration;

pub trait CliProgressTrait {
    fn increase(&self, current: usize, total: usize);
    fn set_message(&self, msg: &'static str);
    fn finish(&self, ok: bool);
}

#[derive(Default)]
pub struct CliProgress {
    pb: Option<ProgressBar>,
}

impl CliProgress {
    pub fn increase(&mut self, current: usize, total: usize) {
        if self.pb.is_none() {
            println!(
                "Generating {} records",
                format!("~{}", total.to_formatted_string(&Locale::en)).bright_cyan()
            );

            let len = total.to_string().len() + 1;
            let pb = ProgressBar::new(total as _);
            pb.enable_steady_tick(Duration::from_millis(80));
            pb.set_style(
                ProgressStyle::with_template(
                    &format!("{{spinner:.green}} {{msg}} [{{elapsed_precise}}] |{{wide_bar:.cyan/blue}}| {{cur:>{len}}}/{{total:{len}}} ({{per_sec}})"),
                )
                    .unwrap()
                    .with_key("cur", |state: &ProgressState, w: &mut dyn Write| write!(w, "{}", state.pos().to_formatted_string(&Locale::en)).unwrap())
                    .with_key("total", |state: &ProgressState, w: &mut dyn Write| write!(w, "{}", state.len().unwrap_or_default().to_formatted_string(&Locale::en)).unwrap())
                    .with_key("per_sec", |state: &ProgressState, w: &mut dyn Write| write!(w, "{:.1}/s", state.per_sec()).unwrap())
                    .progress_chars("#>-")
                    .tick_strings(&[
                        "⠋",
                        "⠙",
                        "⠹",
                        "⠸",
                        "⠼",
                        "⠴",
                        "⠦",
                        "⠧",
                        "⠇",
                        "⠏",
                        "✔"
                    ]),
            );
            pb.set_message("Generating records");

            self.pb.replace(pb);
        }

        self.pb.as_ref().unwrap().set_position(current as _);
    }

    pub fn set_message(&self, msg: &'static str) {
        if let Some(pb) = self.pb.as_ref() {
            pb.set_message(msg);
        }
    }

    pub fn finish(&self, ok: bool) {
        if let Some(pb) = self.pb.as_ref() {
            if ok {
                pb.finish_with_message("Done");
                println!(
                    "Success - Generated {} records in {}",
                    pb.position().to_formatted_string(&Locale::en).bright_blue(),
                    format!("{:.1?}", pb.elapsed()).bright_blue()
                );
            } else {
                pb.finish_with_message("Error");
            }
        }
    }
}

pub type CliProgressRef = Rc<Mutex<CliProgress>>;

impl CliProgressTrait for CliProgressRef {
    fn increase(&self, current: usize, total: usize) {
        self.lock().unwrap().increase(current, total);
    }

    fn set_message(&self, msg: &'static str) {
        self.lock().unwrap().set_message(msg);
    }

    fn finish(&self, ok: bool) {
        self.lock().unwrap().finish(ok);
    }
}