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

use std::io::Write;
use std::path::Path;

use anyhow::Result;

use super::util::prompt_yes_no;
use super::Ui;
use crate::entities::Scope;
use crate::{command, service, RP};

pub fn get_samples(ui: &mut Ui<impl RP>, scope: &Scope, params: command::Samples) -> Result<()> {
    let test_cases = service::fetch_samples(&ui.res, scope, &params.task_id)?;
    let existing_files =
        service::test_cases_exist(&ui.res, test_cases.len() as u64, params.dir_name.as_deref());
    if !existing_files.is_empty() {
        let mut overwrite_message = format!(
            "The following files are already present in the {}:\n",
            format_dir_name(params.dir_name.as_deref())
        );
        for file in existing_files {
            overwrite_message.push_str(format!("  {}\n", file).as_str());
        }
        overwrite_message.push_str("Do you want to overwrite them? (yes/No) ");
        if !prompt_yes_no(ui, &overwrite_message)? {
            return Ok(());
        }
    }
    service::create_dir_all(&ui.res, params.dir_name.as_deref())?;
    let amount = test_cases.len();
    service::save_test_cases(&ui.res, test_cases, params.dir_name.as_deref())?;
    Ok(writeln!(
        ui.term,
        "{} sample test cases successfully saved to the {}",
        amount,
        format_dir_name(params.dir_name.as_deref())
    )?)
}

fn format_dir_name(dir_name: Option<&Path>) -> String {
    if let Some(dir_name) = dir_name {
        format!("directory {}", dir_name.display())
    } else {
        "current directory".to_owned()
    }
}