use {
crate::{
compound_style::CompoundStyle,
errors::Result,
},
crossterm::{
QueueableCommand,
style::{Color, PrintStyledContent, StyledContent},
},
std::fmt::{self, Display},
};
#[derive(Clone)]
pub struct StyledChar {
compound_style: CompoundStyle,
nude_char: char,
styled_char: StyledContent<char>, }
impl StyledChar {
pub fn new(compound_style: CompoundStyle, nude_char: char) -> StyledChar {
Self {
nude_char,
styled_char: compound_style.apply_to(nude_char),
compound_style,
}
}
pub fn from_fg_char(fg: Color, nude_char: char) -> StyledChar {
Self::new(CompoundStyle::with_fg(fg), nude_char)
}
pub fn set_char(&mut self, nude_char: char) {
self.nude_char = nude_char;
self.styled_char = self.compound_style.apply_to(self.nude_char);
}
pub fn get_char(&self) -> char {
self.nude_char
}
pub fn set_fg(&mut self, color: Color) {
self.compound_style.set_fg(color);
self.styled_char = self.compound_style.apply_to(self.nude_char);
}
pub fn get_fg(&self) -> Option<Color> {
self.compound_style.get_fg()
}
pub fn set_bg(&mut self, color: Color) {
self.compound_style.set_bg(color);
self.styled_char = self.compound_style.apply_to(self.nude_char);
}
pub fn get_bg(&self) -> Option<Color> {
self.compound_style.get_bg()
}
pub fn set_compound_style(&mut self, compound_style: CompoundStyle) {
self.compound_style = compound_style;
self.styled_char = self.compound_style.apply_to(self.nude_char);
}
pub fn repeated(&self, count: usize) -> StyledContent<String> {
let mut s = String::new();
for _ in 0..count {
s.push(self.nude_char);
}
self.compound_style.apply_to(s)
}
pub fn queue_repeat<W>(&self, w: &mut W, count: usize) -> Result<()>
where
W: std::io::Write,
{
let mut s = String::new();
for _ in 0..count {
s.push(self.nude_char);
}
self.compound_style.queue(w, s)
}
pub fn queue<W>(&self, w: &mut W) -> Result<()>
where
W: std::io::Write,
{
w.queue(PrintStyledContent(self.styled_char.clone()))?;
Ok(())
}
}
impl Display for StyledChar {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
self.styled_char.fmt(f)
}
}