cses-cli 0.1.3

CSES CLI is a lightweight tool for using CSES from the command line.
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

use anyhow::{Context, Result};
use console::{style, Style, StyledObject};

use super::Ui;
use crate::entities::TaskStatus;
use crate::RP;

pub fn result_with_color(line: &str) -> StyledObject<&str> {
    let color = match line {
        "ACCEPTED" => Style::new().green(),
        "UNKNOWN" => Style::new().white(),
        _ => Style::new().red(),
    };
    color.apply_to(line)
}

pub fn format_code_time(time: Option<u64>) -> String {
    match time {
        Some(time) => format!("{:.2} s", time as f64 / 1000.0),
        None => "--".to_owned(),
    }
}

pub fn format_code_size(size: Option<u64>) -> Option<String> {
    size.map(|size| format!("{} ch.", size))
}

pub fn format_test_groups(groups: &Option<Vec<u64>>) -> Option<String> {
    groups.as_ref().map(|groups| {
        let mut text: String = groups
            .iter()
            .map(|&group| group.to_string() + ",")
            .collect();
        text.pop(); // Removes the trailing comma
        text
    })
}

pub fn styled_task_status_or_score(
    status: Option<TaskStatus>,
    score: Option<u64>,
) -> StyledObject<String> {
    if let Some(points) = score {
        return match points {
            points if points >= 100 => style(points.to_string()).green(),
            points if points > 0 => style(points.to_string()).yellow(),
            _ => style(points.to_string()).red(),
        }
        .bold();
    }

    match status {
        Some(TaskStatus::Pass) => style("PASS".to_string()).green(),
        Some(TaskStatus::Fail) => style("FAIL".to_string()).red(),
        Some(TaskStatus::None) | None => style("--".to_string()).dim(),
    }
    .bold()
}

pub fn prompt_yes_no(ui: &mut Ui<impl RP>, message: &str) -> Result<bool> {
    ui.term.write_str(message)?;
    let answer = ui.prompt_line().context("Failed reading confirmation")?;
    Ok(answer == "yes")
}