bobatea 0.0.2

Terminal layout + styling you won't hate.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

//! Small badge / pill labels.
//!
//! ```rust
//! use bobatea::components::badge::{Badge, BadgeStyle};
//! let badge = Badge::new("v1.0.0").style(BadgeStyle::Success);
//! ```

use {
    crate::components::Component,
    ratatui::{
        prelude::{Buffer, Frame, Rect},
        style::Color,
        widgets::{Paragraph, Widget},
    },
    std::fmt::Display,
};

pub enum BadgeStyle {
    Default,
    Primary,
    Success,
    Warn,
    Danger,
    Info,
}

impl BadgeStyle {
    pub fn colors(&self) -> (Color, Color) {
        match self {
            BadgeStyle::Default => (Color::White, Color::DarkGray),
            BadgeStyle::Primary => (Color::White, Color::Blue),
            BadgeStyle::Success => (Color::White, Color::Green),
            BadgeStyle::Warn => (Color::Black, Color::Yellow),
            BadgeStyle::Danger => (Color::White, Color::Red),
            BadgeStyle::Info => (Color::White, Color::Cyan),
        }
    }
}

/// A small pill-shaped badge label.
pub struct Badge {
    text: String,
    style: BadgeStyle,
}

impl Badge {
    pub fn new(text: impl Display) -> Self { Self { text: text.to_string(), style: BadgeStyle::Default } }

    pub fn style(mut self, s: BadgeStyle) -> Self {
        self.style = s;
        self
    }

    pub fn render_to_buf(&self, area: Rect, buf: &mut Buffer, theme: &crate::theme::Theme) {
        let style = match self.style {
            BadgeStyle::Default => theme.badge_primary,
            BadgeStyle::Primary => theme.badge_primary,
            BadgeStyle::Success => theme.badge_success,
            BadgeStyle::Warn => theme.badge_warn,
            BadgeStyle::Danger => theme.badge_error,
            BadgeStyle::Info => theme.badge_info,
        };
        Paragraph::new(format!(" {} ", self.text)).style(style).render(area, buf);
    }
}

impl Component for Badge {
    fn height(&self) -> Option<usize> { Some(1) }

    fn render(&mut self, ctx: &mut Frame<'_>, theme: &crate::theme::Theme) {
        let area = ctx.area();
        self.render_to_buf(area, ctx.buffer_mut(), theme);
    }
}