You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
47 lines
1.1 KiB
47 lines
1.1 KiB
4 years ago
|
extern crate reqwest;
|
||
|
|
extern crate tempfile;
|
||
|
|
|
||
|
|
use std::io::{copy, Write};
|
||
|
|
use std::fs::File;
|
||
|
|
use tempfile::Builder;
|
||
|
|
use std::collections::HashMap;
|
||
|
|
use reqwest::header::HeaderValue;
|
||
|
|
use std::fs;
|
||
|
|
use std::path::Path;
|
||
|
|
use std::env;
|
||
|
|
|
||
|
|
fn get_body(problem: i32) -> String {
|
||
|
|
|
||
|
|
fs::create_dir_all("./cache").expect("");
|
||
|
|
|
||
|
|
let filepath = format!("./cache/problem-{}", problem);
|
||
|
|
let url = format!("https://adventofcode.com/2020/day/{}/input", problem);
|
||
|
|
|
||
|
|
if Path::new(&filepath).exists() {
|
||
|
|
fs::read_to_string(filepath).unwrap()
|
||
|
|
} else {
|
||
|
|
|
||
|
|
let session_token = match env::var("SESSION") {
|
||
|
|
Ok(token) => token,
|
||
|
|
Err(e) => panic!("No session token"),
|
||
|
|
};
|
||
|
|
|
||
|
|
let session_token = format!("session={};", session_token);
|
||
|
|
|
||
|
|
let client = reqwest::Client::new();
|
||
|
|
let body = client
|
||
|
|
.get(&url)
|
||
|
|
.header("Cookie", session_token)
|
||
|
|
.send().expect("").text().expect("");
|
||
|
|
|
||
|
|
let mut f = File::create(format!("./cache/problem-{}", problem)).expect("");
|
||
|
|
f.write_all(body.as_bytes());
|
||
|
|
|
||
|
|
body
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn get_problem(problem: i32) -> String {
|
||
|
|
get_body(problem)
|
||
|
|
}
|