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

extern crate reqwest;
extern crate scraper;

use crate::gen::file_utils;
use crate::sample_cases::SampleCases;
use reqwest::Url;
use scraper::Html;
use std::io;

/// Contains information of a problem in a contest.
pub struct Problem<'a> {
    pub contest_id: &'a str,
    pub problem_id: &'a str,
    pub sample_cases: SampleCases,
}

impl<'a> Problem<'a> {
    /// Create a new instance.
    pub fn new(contest_id: &'a str, problem_id: &'a str) -> Option<Problem<'a>> {
        let sc = Problem::fetch_sample_cases(contest_id, problem_id);
        if let Some(sc) = sc {
            return Some(Problem {
                contest_id,
                problem_id,
                sample_cases: sc,
            });
        }
        None
    }

    /// Create `SampleCases` of a specific problem.
    fn fetch_sample_cases(contest_id: &'a str, problem_id: &'a str) -> Option<SampleCases> {
        let html = Problem::fetch_problem_html(contest_id, problem_id);
        if let Some(html) = html {
            let sc = SampleCases::new(&html);
            return Some(sc);
        }
        None
    }

    /// Fetch `Html` of a problem.
    fn fetch_problem_html(contest_id: &'a str, problem_id: &'a str) -> Option<Html> {
        let problem_name = format!("{}_{}",contest_id.replace("-", "_"), problem_id);
        let url = format!(
            "https://atcoder.jp/contests/{}/tasks/{}",
            contest_id, problem_name
        );
        let parsed_url = Url::parse(&url).unwrap();
        let mut res = reqwest::get(parsed_url).unwrap();

        if res.status().as_u16() == 200 {
            let html_text = res.text().unwrap();
            let document = Html::parse_document(&html_text);
            return Some(document);
        }
        None
    }

    /// Create files which contains information of sample cases.
    pub fn create_sample_cases_files(&self) -> Result<(), io::Error> {
        file_utils::create_directory("io_examples".to_string())?;
        file_utils::create_test_files(&self.sample_cases, self.problem_id)?;
        Ok(())
    }
}