ransi 0.1.0

ransi - Utility for ANSI escape codes
Documentation 
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
89
90
91
92
93
94
95
96
97
98
99
100
101

/// Function that returns an ANSI escape code for a given color name
pub fn get_ansi_from_name(short_name: &str) -> Option<String> {
    match short_name {
        "red" => Some("\033[31m".to_string()),
        "green" => Some("\033[32m".to_string()),
        "yellow" => Some("\033[33m".to_string()),
        "blue" => Some("\033[34m".to_string()),
        "magenta" => Some("\033[35m".to_string()),
        "cyan" => Some("\033[36m".to_string()),
        _ => None,

    }
}

/// Function that returns an ANSI escape code for a given RGB color
pub fn get_ansi_from_rgb(r: u8, g: u8, b: u8) -> String {
    // NOTE: no need to check for out of range values, as this is u8 (max 255)
    format!("\033[38;2;{};{};{}m", r, g, b)
}

/// Trait that supports or the rgb color or the color name
pub trait ColorIn {
    fn to_ansi(&self) -> Option<String>;
}

/// Implement the ColorIn trait for when a string is passed
impl ColorIn for &str {
    fn to_ansi(&self) -> Option<String> {
        get_ansi_from_name(self)
    }
}

/// Implement the ColorIn trait for when a tuple of u8 is passed
impl ColorIn for (u8, u8, u8) {
    fn to_ansi(&self) -> Option<String> {
        let (r, g, b) = *self;
        format!("\033[38;2;{};{};{}m", r, g, b).into()
    }
}

/// Implement the ColorIn trait for when a single u8 is passed (256 colors)
impl ColorIn for u8 {
	fn to_ansi(&self) -> Option<String> {
		Some(format!("\x1B[38;5;{}m", self))
	}
}

/// reset() - Manually reset the color
///
/// Reset manually clears all atributes
pub fn reset() {
    println!("\033[0m");
}

/// set_fg(&str color OR (u8, u8, u8) color) - Prints and returns the ANSI 
/// escape code for the color
/// given on color.
///
/// # Examples
/// ```
/// use ransi::colors::{set_fg, reset};
///
/// set_fg("red");
/// reset();
/// println!("This text is red");
/// ```
pub fn set_fg<C: ColorIn>(color: C) -> Option<String> {

    let ret = color.to_ansi();
    
    if None != ret {
        println!("{}", ret.clone().unwrap()); // Change the color
        return ret;
    }

    None
    
}

/// set_bg(&str color OR (u8, u8, u8) color) - Prints and returns the ANSI 
/// escape code for the color
/// given on color.
///
/// # Examples
/// ```
/// use ransi::colors::{set_bg, reset};
///
/// set_bg("red");
/// reset();
/// println!("This text is on red");
/// ```
pub fn set_bg<C: ColorIn>(color: C) -> Option<String> {
	let ret = color.to_ansi();

	if None != ret {
		println!("{}", ret.clone().unwrap());
		return ret;
	}

	None
}