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
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
87
88
89
90
91

//! **atcoder-util** make it easy to test for your code in sample cases in competitive programming.
//!
//! ## Gen
//!
//! First, you have to fetch input/output example from AtCoder's website.
//! If you provide only contest name, sample cases of all problem in the contest will be downloaded.
//!
//! ```
//! $ atcoder-util gen abc145
//! ```
//!
//! Or if you specify a contest name and a problem, sample cases of the problem will be downloaded.
//! ```
//! $ atcoder-util gen abc145 c
//! ```
//!
//! ## Test
//!
//! After you fetched sample cases, you can run check whether its answer is correct in all at once.
//!
//! ```
//! $ atcoder-util test c
//! ```
//!

#[macro_use]
extern crate clap;
extern crate atcoder_util;

use atcoder_util::gen;
use atcoder_util::tester;
use clap::{App, Arg, SubCommand};
use std::io;

fn main() -> Result<(), io::Error> {
    let matches = App::new(crate_name!())
        .version(crate_version!())
        .author(crate_authors!())
        .about(crate_description!())
        .subcommand(
            SubCommand::with_name("gen")
                .about("Generate input/output format example fetched from AtCoder's website.")
                .arg(
                    Arg::with_name("contest name")
                        .help("Specify which contest to fetch.")
                        .takes_value(true)
                        .required(true),
                )
                .arg(
                    Arg::with_name("problem id")
                        .help("Specify which problem to fetch when a contest name is set.")
                        .takes_value(true),
                ),
        )
        .subcommand(
            SubCommand::with_name("test")
                .about("Check whether your code will output right answer for each sample cases.")
                .arg(
                    Arg::with_name("problem id")
                        .help("Specify which problem to test.")
                        .takes_value(true)
                        .required(true),
                ),
        )
        .get_matches();

    // Fetch input/output examples and write each of them into text files.
    if let Some(ref matches) = matches.subcommand_matches("gen") {
        let contest_id = matches.value_of("contest name");
        let problem_id = matches.value_of("problem id");

        // Problem is specified (such as "a", "b", "c"...).
        match (contest_id, problem_id) {
            (Some(contest_id), Some(problem_id)) => {
                gen::execute_fetching_problem(contest_id, &problem_id)?;
            }
            (Some(contest_id), None) => {
                gen::execute_fetching_problems_in_contest(contest_id)?;
            }
            (_, _) => {}
        }
    }

    if let Some(ref matches) = matches.subcommand_matches("test") {
        let problem_id = matches.value_of("problem id");
        if let Some(problem_id) = problem_id {
            tester::run_test::run_test_all(problem_id);
        }
    }
    Ok(())
}