56 lines
1.5 KiB
Rust
56 lines
1.5 KiB
Rust
use super::Segment;
|
|
use ansi_term::{Color, Style};
|
|
use clap::ArgMatches;
|
|
|
|
/// Creates a segment for the prompt character
|
|
///
|
|
/// The char segment prints an arrow character in a color dependant on the exit-
|
|
/// code of the last executed command:
|
|
/// - If the exit-code was "0", the arrow will be formatted with `COLOR_SUCCESS`
|
|
/// (green by default)
|
|
/// - If the exit-code was anything else, the arrow will be formatted with
|
|
/// `COLOR_FAILURE` (red by default)
|
|
pub fn segment(args: &ArgMatches) -> Segment {
|
|
const PROMPT_CHAR: &str = "➜ ";
|
|
const COLOR_SUCCESS: Color = Color::Green;
|
|
const COLOR_FAILURE: Color = Color::Red;
|
|
|
|
let color = if args.value_of("status_code").unwrap() == "0" {
|
|
COLOR_SUCCESS
|
|
} else {
|
|
COLOR_FAILURE
|
|
};
|
|
|
|
Segment {
|
|
value: String::from(PROMPT_CHAR),
|
|
style: Style::from(color),
|
|
..Default::default()
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use clap::{App, Arg};
|
|
|
|
#[test]
|
|
fn char_section_success_status() {
|
|
let args = App::new("starship")
|
|
.arg(Arg::with_name("status_code"))
|
|
.get_matches_from(vec!["starship", "0"]);
|
|
|
|
let segment = segment(&args);
|
|
assert_eq!(segment.style, Style::from(Color::Green));
|
|
}
|
|
|
|
#[test]
|
|
fn char_section_failure_status() {
|
|
let args = App::new("starship")
|
|
.arg(Arg::with_name("status_code"))
|
|
.get_matches_from(vec!["starship", "1"]);
|
|
|
|
let segment = segment(&args);
|
|
assert_eq!(segment.style, Style::from(Color::Red));
|
|
}
|
|
}
|