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
80
81
82
83
84
85
86
87
88
use std::fmt::Display;
use crate::style::{Attribute, Attributes, Color, StyledContent};
#[derive(Debug, Copy, Clone, Default, PartialEq, Eq)]
pub struct ContentStyle {
pub foreground_color: Option<Color>,
pub background_color: Option<Color>,
pub attributes: Attributes,
}
impl ContentStyle {
#[inline]
pub fn apply<D: Display>(self, val: D) -> StyledContent<D> {
StyledContent::new(self, val)
}
#[inline]
pub fn new() -> ContentStyle {
ContentStyle::default()
}
#[inline]
pub fn background(self, color: Color) -> ContentStyle {
Self {
background_color: Some(color),
..self
}
}
#[inline]
pub fn foreground(self, color: Color) -> ContentStyle {
Self {
foreground_color: Some(color),
..self
}
}
#[inline]
pub fn attribute(mut self, attr: Attribute) -> ContentStyle {
self.attributes.set(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::Bold);
assert_eq!(content_style.foreground_color, Some(Color::Blue));
assert_eq!(content_style.background_color, Some(Color::Red));
assert!(content_style.attributes.has(Attribute::Bold));
}
#[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!(styled_content.style().attributes.has(Attribute::Reset));
}
}