use std::convert::From;
use std::str::FromStr;
#[derive(Debug,PartialEq, Eq)]
pub enum Color {
Black,
Red,
Green,
Yellow,
Blue,
Magenta,
Cyan,
White,
}
impl Color {
pub fn to_fg_str(&self) -> &str {
match *self {
Color::Black => "30",
Color::Red => "31",
Color::Green => "32",
Color::Yellow => "33",
Color::Blue => "34",
Color::Magenta => "35",
Color::Cyan => "36",
Color::White => "37",
}
}
pub fn to_bg_str(&self) -> &str {
match *self {
Color::Black => "40",
Color::Red => "41",
Color::Green => "42",
Color::Yellow => "43",
Color::Blue => "44",
Color::Magenta => "45",
Color::Cyan => "46",
Color::White => "47",
}
}
}
impl<'a> From<&'a str> for Color {
fn from(src: &str) -> Self {
src.parse().unwrap_or(Color::White)
}
}
impl From<String> for Color {
fn from(src: String) -> Self {
src.parse().unwrap_or(Color::White)
}
}
impl FromStr for Color {
type Err = ();
fn from_str(src: &str) -> Result<Self, Self::Err> {
let src = src.to_lowercase();
match src.as_ref() {
"black" => Ok(Color::Black),
"red" => Ok(Color::Red),
"green" => Ok(Color::Green),
"yellow" => Ok(Color::Yellow),
"blue" => Ok(Color::Blue),
"magenta" => Ok(Color::Magenta),
"cyan" => Ok(Color::Cyan),
"white" => Ok(Color::White),
_ => Err(()),
}
}
}
#[cfg(test)]
mod tests {
pub use super::*;
mod from_str {
pub use super::*;
macro_rules! make_test {
( $( $name:ident: $src:expr => $dst:expr),* ) => {
$(
#[test]
fn $name() {
let color : Color = $src.into();
assert_eq!($dst, color)
}
)*
}
}
make_test!(
black: "black" => Color::Black,
red: "red" => Color::Red,
green: "green" => Color::Green,
yellow: "yellow" => Color::Yellow,
blue: "blue" => Color::Blue,
magenta: "magenta" => Color::Magenta,
cyan: "cyan" => Color::Cyan,
white: "white" => Color::White,
invalid: "invalid" => Color::White,
capitalized: "BLUE" => Color::Blue,
mixed_case: "bLuE" => Color::Blue
);
}
mod from_string {
pub use super::*;
macro_rules! make_test {
( $( $name:ident: $src:expr => $dst:expr),* ) => {
$(
#[test]
fn $name() {
let src = String::from($src);
let color : Color = src.into();
assert_eq!($dst, color)
}
)*
}
}
make_test!(
black: "black" => Color::Black,
red: "red" => Color::Red,
green: "green" => Color::Green,
yellow: "yellow" => Color::Yellow,
blue: "blue" => Color::Blue,
magenta: "magenta" => Color::Magenta,
cyan: "cyan" => Color::Cyan,
white: "white" => Color::White,
invalid: "invalid" => Color::White,
capitalized: "BLUE" => Color::Blue,
mixed_case: "bLuE" => Color::Blue
);
}
mod fromstr {
pub use super::*;
#[test]
fn parse() {
let color: Result<Color, _> = "blue".parse();
assert_eq!(Ok(Color::Blue), color)
}
#[test]
fn error() {
let color: Result<Color, ()> = "bloublou".parse();
assert_eq!(Err(()), color)
}
}
}