acr-cli 0.6.0

A CLI tool for AtCoder competitive programming in Rust
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

use crate::atcoder;
use crate::browser;
use crate::workspace;
use crate::workspace::CurrentContext;

fn resolve_contest_id(
    current: CurrentContext,
    contest_id: Option<String>,
) -> anyhow::Result<String> {
    match current {
        CurrentContext::ProblemDir(ctx) => match contest_id {
            Some(_) => anyhow::bail!("Cannot specify a contest ID from a problem directory."),
            None => Ok(ctx.contest_id),
        },
        CurrentContext::ContestDir(ctx) => match contest_id {
            Some(_) => anyhow::bail!("Cannot specify a contest ID from a contest directory."),
            None => Ok(ctx.contest_id),
        },
        CurrentContext::Outside => match contest_id {
            Some(id) => {
                workspace::find_contest_dir_by_id(&id)?;
                Ok(id)
            }
            None => anyhow::bail!("Specify a contest ID or run from a contest directory."),
        },
    }
}

fn build_url(contest_id: &str) -> String {
    format!(
        "{}/contests/{}/submissions/me",
        atcoder::BASE_URL,
        contest_id
    )
}

pub fn execute(contest_id: Option<String>) -> anyhow::Result<()> {
    let current = workspace::detect_current_context();
    let contest_id = resolve_contest_id(current, contest_id)?;
    let url = build_url(&contest_id);
    browser::open(&url);
    println!("{}", url);
    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::workspace::{ContestContext, ProblemContext};
    use std::path::PathBuf;

    fn make_problem_context(contest_id: &str) -> ProblemContext {
        ProblemContext {
            contest_id: contest_id.to_string(),
            problem_alphabet: "a".to_string(),
            task_screen_name: format!("{}_a", contest_id),
            problem_dir: PathBuf::from(format!("/tmp/{}/a", contest_id)),
            problem_url: format!(
                "https://atcoder.jp/contests/{}/tasks/{}_a",
                contest_id, contest_id
            ),
        }
    }

    fn make_contest_context(contest_id: &str) -> ContestContext {
        ContestContext {
            contest_id: contest_id.to_string(),
            contest_dir: PathBuf::from(format!("/tmp/{}", contest_id)),
        }
    }

    #[test]
    fn test_resolve_contest_id_problem_dir() {
        let ctx = CurrentContext::ProblemDir(make_problem_context("abc001"));
        let id = resolve_contest_id(ctx, None).unwrap();
        assert_eq!(id, "abc001");
    }

    #[test]
    fn test_resolve_contest_id_problem_dir_with_id_error() {
        let ctx = CurrentContext::ProblemDir(make_problem_context("abc001"));
        assert!(resolve_contest_id(ctx, Some("abc002".to_string())).is_err());
    }

    #[test]
    fn test_resolve_contest_id_contest_dir() {
        let ctx = CurrentContext::ContestDir(make_contest_context("abc001"));
        let id = resolve_contest_id(ctx, None).unwrap();
        assert_eq!(id, "abc001");
    }

    #[test]
    fn test_resolve_contest_id_contest_dir_with_id_error() {
        let ctx = CurrentContext::ContestDir(make_contest_context("abc001"));
        assert!(resolve_contest_id(ctx, Some("abc002".to_string())).is_err());
    }

    #[test]
    fn test_resolve_contest_id_outside_no_id_error() {
        let ctx = CurrentContext::Outside;
        assert!(resolve_contest_id(ctx, None).is_err());
    }

    #[test]
    fn test_build_url() {
        assert_eq!(
            build_url("abc001"),
            "https://atcoder.jp/contests/abc001/submissions/me"
        );
    }
}