rtop_dev 1.1.0-beta.1

//! # Components representing a `ProgressBar`
//!
//! ## Example
//! ```rust
//! use rtop_dev::components::bar::*;
//!
//!
//! let bar = Horizontal::new(3, 24, Some(Color::Blue)); // Create a blue Horizontal ProgressBar with 3 cells of height and 24 cells of width
//! bar.display(50.0);
//! ```

use std::fmt::Write;

/// Vertical `ProgressBar`, designed to be ready to use in Rtop
pub struct Vertical {
    rows: i32,
    cols: i32,
    color: Color,
}
impl Vertical {
    /// # Return a formatted String ready to be display in Rtop
    /// ## Arguments
    /// * `pourcent` - Represent the progress of the bar<br>
    /// **⚠️ The `pourcent` must be between 0.0 and 100.0**
    pub fn display(&self, pourcent: f32) -> String {
        let color = match self.color {
            Color::Red => "[[EFFECT_COLOR_RED_BLACK]]",
            Color::Green => "[[EFFECT_COLOR_GREEN_BLACK]]",
            Color::Yellow => "[[EFFECT_COLOR_YELLOW_BLACK]]",
            Color::Blue => "[[EFFECT_COLOR_BLUE_BLACK]]",
            Color::Magenta => "[[EFFECT_COLOR_MAGENTA_BLACK]]",
            Color::Cyan => "[[EFFECT_COLOR_CYAN_BLACK]]",
            Color::White => "[[EFFECT_COLOR_WHITE_BLACK]]",
        };

        let mut out = String::new();
        let mut bar_parts = std::collections::HashMap::new();
        bar_parts.insert(0, " ");
        bar_parts.insert(1, "▁");
        bar_parts.insert(2, "▂");
        bar_parts.insert(3, "▃");
        bar_parts.insert(4, "▄");
        bar_parts.insert(5, "▅");
        bar_parts.insert(6, "▆");
        bar_parts.insert(7, "▇");
        bar_parts.insert(8, "█");

        if pourcent == 100. {
            out += &format!("{}{}{}\n", color, "█".repeat(self.cols as usize), color)
                .repeat(self.rows as usize);
        } else {
            let block_filled = (self.rows as f32 * 8. * (pourcent / 100.)) as i32;
            let white_lines =
                (self.rows - 1 - (self.rows as f32 * (pourcent / 100.)) as i32) as usize;

            out += &format!("{}{}{}\n", color, " ".repeat(self.cols as usize), color)
                .repeat(white_lines);
            writeln!(
                &mut out,
                "{}{}{}",
                color,
                bar_parts[&(block_filled % 8)].repeat(self.cols as usize),
                color
            )
            .unwrap();
            out += &format!("{}{}{}\n", color, "█".repeat(self.cols as usize), color)
                .repeat(self.rows as usize - white_lines - 1);
        }

        out
    }

    /// # Create a new `Vertical`
    /// ## Arguments
    /// * `cols` - Represent the width of the bar in cells
    /// * `rows` - Represent the height of the bar in cells
    /// * `color` - *`Optional`* - If supplied, set the color of the progress of the bar, otherwise, it will be green
    pub fn new(rows: i32, cols: i32, color: Option<Color>) -> Self {
        Self {
            rows,
            cols,
            color: color.unwrap_or(Color::Green),
        }
    }
}

/// Horizontal `ProgressBar`, designed to be ready to use in Rtop to represent progress and other stuff
pub struct Horizontal {
    rows: i32,
    cols: i32,
    color: Color,
}
impl Horizontal {
    /// # Return a formatted String ready to be display in Rtop
    /// ## Arguments
    /// * `pourcent` - Represent the progress of the bar<br>
    /// **⚠️ The `pourcent` must be between 0.0 and 100.0**
    pub fn display(&self, pourcent: f32) -> String {
        let color = match self.color {
            Color::Red => "[[EFFECT_COLOR_RED_BLACK]]",
            Color::Green => "[[EFFECT_COLOR_GREEN_BLACK]]",
            Color::Yellow => "[[EFFECT_COLOR_YELLOW_BLACK]]",
            Color::Blue => "[[EFFECT_COLOR_BLUE_BLACK]]",
            Color::Magenta => "[[EFFECT_COLOR_MAGENTA_BLACK]]",
            Color::Cyan => "[[EFFECT_COLOR_CYAN_BLACK]]",
            Color::White => "[[EFFECT_COLOR_WHITE_BLACK]]",
        };

        let mut out = String::new();
        let mut bar_parts = std::collections::HashMap::new();
        bar_parts.insert(0, " ");
        bar_parts.insert(1, "▏");
        bar_parts.insert(2, "▎");
        bar_parts.insert(3, "▍");
        bar_parts.insert(4, "▌");
        bar_parts.insert(5, "▋");
        bar_parts.insert(6, "▊");
        bar_parts.insert(7, "▉");
        bar_parts.insert(8, "█");

        let block_filled = (self.cols as f32 * 8. * (pourcent / 100.)) as i32;
        let white_lines = (self.cols - 1 - (self.cols as f32 * (pourcent / 100.)) as i32) as usize;

        if pourcent == 100. {
            out += &format!(
                "{}{}{}\n",
                color,
                bar_parts[&8].repeat(self.cols as usize),
                color
            )
            .repeat(self.rows as usize);
        } else {
            out += &format!(
                "{}{}{}{}{}\n",
                color,
                bar_parts[&8].repeat((block_filled / 8) as usize),
                bar_parts[&(block_filled % 8)],
                " ".repeat(white_lines),
                color
            )
            .repeat(self.rows as usize);
        }

        out
    }

    /// # Create a new `Horizontal`
    /// ## Arguments
    /// * `cols` - Represent the width of the bar in cells
    /// * `rows` - Represent the height of the bar in cells
    /// * `color` - *`Optional`* - If supplied, set the color of the progress of the bar, otherwise, it will be green
    pub fn new(rows: i32, cols: i32, color: Option<Color>) -> Self {
        Self {
            rows,
            cols,
            color: color.unwrap_or(Color::Green),
        }
    }
}

/// Represent a Color of progress for the Bar
#[non_exhaustive]
pub enum Color {
    Red,
    Green,
    Yellow,
    Blue,
    Magenta,
    Cyan,
    White,
}