65 lines
1.8 KiB
Rust
65 lines
1.8 KiB
Rust
use dotenvy::dotenv;
|
|
use reqwest::blocking::Client;
|
|
use reqwest::header;
|
|
use std::{env, fs::OpenOptions, io::Write, process};
|
|
|
|
pub fn parse_args() -> Result<u8, pico_args::Error> {
|
|
let mut args = pico_args::Arguments::from_env();
|
|
args.free_from_str()
|
|
}
|
|
|
|
fn main() {
|
|
let day: u8 = match parse_args() {
|
|
Ok(day) => day,
|
|
Err(_) => {
|
|
eprintln!("Need to specify a day (as integer). example: `cargo download 7`");
|
|
process::exit(1);
|
|
}
|
|
};
|
|
dotenv().ok();
|
|
|
|
let day_padded = format!("{day:02}");
|
|
let token = env::var("TOKEN").expect("$TOKEN is not set");
|
|
let year = env::var("YEAR")
|
|
.expect("$YEAR is not set")
|
|
.parse::<u32>()
|
|
.expect("$YEAR must be a number");
|
|
|
|
let mut headers = header::HeaderMap::new();
|
|
let mut session_header = header::HeaderValue::from_str(format!("session={token}").as_str())
|
|
.expect("Error building cookie header");
|
|
session_header.set_sensitive(true);
|
|
headers.insert(header::COOKIE, session_header);
|
|
|
|
let client = Client::builder().default_headers(headers).build().unwrap();
|
|
let res = client
|
|
.get(format!("https://adventofcode.com/{year}/day/{day}/input"))
|
|
.send()
|
|
.unwrap()
|
|
.text()
|
|
.unwrap();
|
|
|
|
let input_path = format!("src/inputs/{day_padded}.txt");
|
|
let mut file = match OpenOptions::new()
|
|
.write(true)
|
|
.create(true)
|
|
.open(&input_path)
|
|
{
|
|
Ok(file) => file,
|
|
Err(e) => {
|
|
eprintln!("Failed to create module file: {e}");
|
|
process::exit(1);
|
|
}
|
|
};
|
|
|
|
match file.write_all(res.as_bytes()) {
|
|
Ok(_) => {
|
|
println!("Downloaded input file \"{}\"", &input_path);
|
|
}
|
|
Err(e) => {
|
|
eprintln!("Failed to write module contents: {e}");
|
|
process::exit(1);
|
|
}
|
|
}
|
|
}
|