acr-cli 0.4.4

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

pub mod auth;
pub mod contest;
pub mod scraper;

use std::sync::Arc;

use anyhow::Context;
use reqwest::cookie::Jar;

pub const BASE_URL: &str = "https://atcoder.jp";

pub struct AtCoderClient {
    client: reqwest::Client,
}

#[derive(Debug, Clone)]
pub struct ContestInfo {
    pub problems: Vec<Problem>,
}

#[derive(Debug, Clone)]
pub struct Problem {
    pub alphabet: String,
    pub task_screen_name: String,
    pub url: String,
}

#[derive(Debug, Clone)]
pub struct TestCase {
    pub index: usize,
    pub input: String,
    pub expected: String,
}

impl AtCoderClient {
    /// Create a client with an existing REVEL_SESSION cookie
    pub fn with_session(revel_session: &str) -> anyhow::Result<Self> {
        let jar = Jar::default();
        let cookie = format!("REVEL_SESSION={}", revel_session);
        let url = BASE_URL.parse().context("Failed to parse base URL")?;
        jar.add_cookie_str(&cookie, &url);

        let client = reqwest::Client::builder()
            .cookie_provider(Arc::new(jar))
            .build()
            .context("Failed to create HTTP client")?;
        Ok(Self { client })
    }
}