cses-cli 0.1.4

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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

use std::path::{Path, PathBuf};

use anyhow::{Context, Result};

use crate::entities::{Scope, TestCase};
use crate::{CsesApi, Filesystem, Resources, Storage, RP};

pub fn fetch_samples(
    res: &Resources<impl RP>,
    scope: &Scope,
    task_id: &str,
) -> Result<Vec<TestCase>> {
    (|| -> Result<_> {
        let token = res.storage.get().get_token();
        let response = res.api.get_samples(token, scope, task_id)?;
        Ok(response.test_cases)
    })()
    .context("Failed querying sample test cases from the server")
}

pub fn save_test_cases(
    res: &Resources<impl RP>,
    test_cases: Vec<TestCase>,
    dir_name: Option<&Path>,
) -> Result<()> {
    (|| -> Result<_> {
        let path = make_path(dir_name);
        let mut case_num = 0;
        for case in test_cases.iter() {
            case_num += 1;
            res.filesystem.write_file(
                &res.filesystem.decode_base64(&case.input)?,
                &format_path(path, case_num, "in"),
            )?;
            res.filesystem.write_file(
                &res.filesystem.decode_base64(&case.output)?,
                &format_path(path, case_num, "out"),
            )?;
        }
        Ok(())
    })()
    .context("Failed saving test cases")
}

pub fn test_cases_exist(
    res: &Resources<impl RP>,
    case_count: u64,
    dir_name: Option<&Path>,
) -> Vec<String> {
    let path = make_path(dir_name);
    let mut files_found = Vec::<String>::new();
    for case_num in 1..=case_count {
        if res
            .filesystem
            .file_exists(&format_path(path, case_num, "in"))
        {
            files_found.push(format!("{}.in", case_num));
        }
        if res
            .filesystem
            .file_exists(&format_path(path, case_num, "out"))
        {
            files_found.push(format!("{}.out", case_num));
        }
    }

    files_found
}

pub fn create_dir_all(res: &Resources<impl RP>, dir_name: Option<&Path>) -> Result<()> {
    res.filesystem
        .create_dir_all(make_path(dir_name))
        .context("Failed creating directory for the test cases")
}

fn make_path(dir_name: Option<&Path>) -> &Path {
    match dir_name {
        Some(path) => path,
        None => Path::new("."),
    }
}
fn format_path(path: &Path, case_num: u64, end: &str) -> PathBuf {
    let mut path = path.to_owned();
    path.push(format!("{}.{}", case_num, end));
    path
}