atcoder-util 0.3.2

Utility command for AtCoder.
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

use crate::sample_cases::SampleCases;
use std::env;
use std::fs::{DirBuilder, File};
use std::io::{self, Write};

/// Create file of sample cases for each element of `SampleCases` struct.
pub fn create_test_files(sc: &SampleCases, problem_id: &str) -> Result<(), io::Error> {
    create_directory(format!("io_examples/{}", problem_id))?;
    create_directory(format!("io_examples/{}/{}_input", problem_id, problem_id))?;
    create_directory(format!("io_examples/{}/{}_output", problem_id, problem_id))?;

    for (i, input_example) in sc.input.iter().enumerate() {
        let file_name = format!(
            r"io_examples/{}/{}_input/input{}.txt",
            problem_id,
            problem_id,
            i + 1
        );
        create_test_file(input_example, file_name).expect("Failed to create file.");
    }

    for (i, output_example) in sc.output.iter().enumerate() {
        let file_name = format!(
            r"io_examples/{}/{}_output/output{}.txt",
            problem_id,
            problem_id,
            i + 1
        );
        create_test_file(output_example, file_name).expect("Failed to create file.");
    }
    Ok(())
}

/// Create directory named `dir_name` under the current directory if it doesn't exist.
pub fn create_directory(dir_name: String) -> io::Result<()> {
    let mut dir = env::current_dir()?;
    dir.push(dir_name);
    if dir.is_dir() {
        Ok(())
    } else {
        DirBuilder::new().create(dir)
    }
}

/// Write `file_content` on a file.
fn create_test_file(file_content: &String, file_name: String) -> Result<(), io::Error> {
    let file = File::create(file_name)?;
    let mut writer = io::BufWriter::new(file);
    write!(writer, "{}", file_content).unwrap();
    Ok(())
}