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
75
76
77
78
79

//! This module contains the `content style` that can be applied to an `styled content`.

use std::fmt::Display;

use crate::style::{Attribute, Color, StyledContent};

/// The style that can be put on content.
#[derive(Debug, Clone, Default)]
pub struct ContentStyle {
    /// The foreground color.
    pub foreground_color: Option<Color>,
    /// The background color.
    pub background_color: Option<Color>,
    /// List of attributes.
    pub attributes: Vec<Attribute>,
}

impl ContentStyle {
    /// Creates a `StyledContent` by applying the style to the given `val`.
    pub fn apply<D: Display + Clone>(&self, val: D) -> StyledContent<D> {
        StyledContent::new(self.clone(), val)
    }

    /// Creates a new `ContentStyle`.
    pub fn new() -> ContentStyle {
        ContentStyle::default()
    }

    /// Sets the background color.
    pub fn background(mut self, color: Color) -> ContentStyle {
        self.background_color = Some(color);
        self
    }

    /// Sets the foreground color.
    pub fn foreground(mut self, color: Color) -> ContentStyle {
        self.foreground_color = Some(color);
        self
    }

    /// Adds the attribute.
    ///
    /// You can add more attributes by calling this method multiple times.
    pub fn attribute(mut self, attr: Attribute) -> ContentStyle {
        self.attributes.push(attr);
        self
    }
}

#[cfg(test)]
mod tests {
    use crate::style::{Attribute, Color, ContentStyle};

    #[test]
    fn test_set_fg_bg_add_attr() {
        let content_style = ContentStyle::new()
            .foreground(Color::Blue)
            .background(Color::Red)
            .attribute(Attribute::Reset);

        assert_eq!(content_style.foreground_color, Some(Color::Blue));
        assert_eq!(content_style.background_color, Some(Color::Red));
        assert_eq!(content_style.attributes[0], Attribute::Reset);
    }

    #[test]
    fn test_apply_content_style_to_text() {
        let content_style = ContentStyle::new()
            .foreground(Color::Blue)
            .background(Color::Red)
            .attribute(Attribute::Reset);

        let styled_content = content_style.apply("test");

        assert_eq!(styled_content.style().foreground_color, Some(Color::Blue));
        assert_eq!(styled_content.style().background_color, Some(Color::Red));
        assert_eq!(styled_content.style().attributes[0], Attribute::Reset);
    }
}