Init

aoc-helper/src/client.rs

@@ -0,0 +1,70 @@
+use log::debug;
+use std::{env, fs};
+
+use reqwest::{
+    blocking::{self, Response},
+    header,
+};
+
+use crate::{
+    cli::Part,
+    cookies::session::{CONFIG_DIR, SESSION_FILE},
+};
+
+const BASE_URL: &str = "https://adventofcode.com";
+
+pub struct AocClient {
+    client: blocking::Client,
+    base_url: String,
+}
+
+impl AocClient {
+    pub fn new() -> Result<AocClient, reqwest::Error> {
+        let home_dir = env::home_dir().expect("cannot read HOME var env");
+        let session = fs::read_to_string(home_dir.join(CONFIG_DIR).join(SESSION_FILE))
+            .expect("missing session file. Did you correctly set up the session using get-session or set-session commands ?");
+
+        let mut headers = header::HeaderMap::new();
+
+        let mut session_value =
+            header::HeaderValue::from_str(format!("session={session}").as_str()).unwrap();
+        session_value.set_sensitive(true);
+
+        headers.insert(header::COOKIE, session_value);
+
+        Ok(AocClient {
+            client: blocking::Client::builder()
+                .default_headers(headers)
+                .build()?,
+            base_url: String::from(BASE_URL),
+        })
+    }
+
+    pub fn get_input_file(&self, year: usize, day: usize) -> Result<Response, reqwest::Error> {
+        let url = format!("{}/{year}/day/{day}/input", self.base_url);
+
+        debug!("preparing: GET {url}");
+        self.client.get(url).send()
+    }
+
+    pub fn post_answer(
+        &self,
+        year: usize,
+        day: usize,
+        part: &Part,
+        answer: &str,
+    ) -> Result<Response, reqwest::Error> {
+        let url = format!("{}/{year}/day/{day}/answer", self.base_url);
+
+        let part_str = match part {
+            Part::One => "1",
+            Part::Two => "2",
+        };
+
+        debug!("preparing: POST {url} - level: {part_str}, answer: {answer}");
+        self.client
+            .post(url)
+            .form(&[("level", part_str), ("answer", answer)])
+            .send()
+    }
+}