2018-01-07 07:31:14 +11:00
|
|
|
use std;
|
|
|
|
use std::fmt;
|
|
|
|
use std::io::Write;
|
|
|
|
|
|
|
|
use terminal_style::{Color, ObjectStyle};
|
2018-01-04 00:43:54 +11:00
|
|
|
|
|
|
|
/// Struct that contains both the style and the content wits is styled.
|
2018-01-07 07:31:14 +11:00
|
|
|
pub struct StyledObject<D> {
|
2018-01-04 00:43:54 +11:00
|
|
|
pub object_style: ObjectStyle,
|
2018-01-07 07:31:14 +11:00
|
|
|
pub content: D,
|
2018-01-04 00:43:54 +11:00
|
|
|
}
|
|
|
|
|
2018-01-07 07:31:14 +11:00
|
|
|
impl<D> StyledObject<D> {
|
|
|
|
/// Paints the foreground color with the passed `Color`
|
|
|
|
///
|
|
|
|
/// #Example
|
|
|
|
///
|
2018-01-04 00:43:54 +11:00
|
|
|
/// ```rust
|
2018-01-07 07:31:14 +11:00
|
|
|
/// // create an styled object with the foreground color red.
|
2018-01-04 00:43:54 +11:00
|
|
|
/// let styledobject = paint("I am colored red").with(Color::Red);
|
|
|
|
/// // create an styled object with the foreground color blue.
|
|
|
|
/// let styledobject1 = paint("I am colored blue").with(Color::Blue);
|
2018-01-07 07:31:14 +11:00
|
|
|
///
|
2018-01-10 06:36:48 +11:00
|
|
|
/// // print the styledobject to see the result
|
2018-01-04 00:43:54 +11:00
|
|
|
/// println!("{}", styledobject);
|
|
|
|
/// println!("{}", styledobject1);
|
2018-01-10 06:36:48 +11:00
|
|
|
/// // print an styled object directly.
|
|
|
|
/// println!("{}", paint("I am colored green").with(Color::Green));
|
|
|
|
///
|
2018-01-04 00:43:54 +11:00
|
|
|
/// ```
|
2018-01-07 07:31:14 +11:00
|
|
|
pub fn with(mut self, foreground_color: Color) -> StyledObject<D> {
|
2018-01-04 00:43:54 +11:00
|
|
|
self.object_style = self.object_style.fg(foreground_color);
|
|
|
|
self
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Paints the background color with the passed `Color`
|
2018-01-07 07:31:14 +11:00
|
|
|
///
|
|
|
|
/// #Example
|
|
|
|
///
|
2018-01-04 00:43:54 +11:00
|
|
|
/// ```rust
|
2018-01-10 06:36:48 +11:00
|
|
|
///
|
2018-01-07 07:31:14 +11:00
|
|
|
/// // create an styled object with the background color red.
|
2018-01-04 00:43:54 +11:00
|
|
|
/// let styledobject = paint("I am colored red").on(Color::Red);
|
|
|
|
/// // create an styled object with the background color blue.
|
|
|
|
/// let styledobject1 = paint("I am colored blue").on(Color::Blue);
|
2018-01-07 07:31:14 +11:00
|
|
|
///
|
2018-01-10 06:36:48 +11:00
|
|
|
/// // print the styledobjects
|
2018-01-04 00:43:54 +11:00
|
|
|
/// println!("{}", styledobject);
|
|
|
|
/// println!("{}", styledobject1);
|
2018-01-10 06:36:48 +11:00
|
|
|
/// // print an styled object directly.
|
2018-01-04 00:43:54 +11:00
|
|
|
/// println!("{}", paint("I am colored green").on(Color::Green))
|
2018-01-10 06:36:48 +11:00
|
|
|
///
|
2018-01-04 00:43:54 +11:00
|
|
|
/// ```
|
2018-01-07 07:31:14 +11:00
|
|
|
pub fn on(mut self, background_color: Color) -> StyledObject<D> {
|
2018-01-04 00:43:54 +11:00
|
|
|
self.object_style = self.object_style.bg(background_color);
|
|
|
|
self
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-01-07 07:31:14 +11:00
|
|
|
/// This is used to make StyledObject able to be displayed.
|
2018-01-10 06:36:48 +11:00
|
|
|
/// This macro will set the styles stored in Styled Object
|
2018-01-07 07:31:14 +11:00
|
|
|
macro_rules! impl_fmt
|
|
|
|
{
|
|
|
|
($name:ident) => {
|
|
|
|
impl<D: fmt::$name> fmt::$name for StyledObject<D> {
|
|
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result
|
|
|
|
{
|
|
|
|
let mut colored_terminal = super::super::colored_terminal();
|
|
|
|
let mut reset = true;
|
|
|
|
|
|
|
|
if let Some(bg) = self.object_style.bg_color
|
|
|
|
{
|
|
|
|
colored_terminal.set_bg(bg);
|
|
|
|
reset = true;
|
|
|
|
}
|
|
|
|
if let Some(fg) = self.object_style.fg_color
|
|
|
|
{
|
|
|
|
colored_terminal.set_fg(fg);
|
|
|
|
reset = true;
|
|
|
|
}
|
|
|
|
fmt::$name::fmt(&self.content, f)?;
|
|
|
|
std::io::stdout().flush().expect("Flush stdout failed");
|
|
|
|
|
|
|
|
if reset
|
|
|
|
{
|
|
|
|
colored_terminal.reset();
|
|
|
|
}
|
|
|
|
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2018-01-04 00:43:54 +11:00
|
|
|
|
2018-01-07 07:31:14 +11:00
|
|
|
impl_fmt!(Debug);
|
|
|
|
impl_fmt!(Display);
|