Add authentication using cookies

1 files changed, 33 insertions, 0 deletions

diff --git a/src/http_client.rs b/src/http_client.rs
new file mode 100644
index 0000000..6e06474
--- /dev/null
+++ b/src/http_client.rs
@@ -0,0 +1,33 @@
+use anyhow::{Context, Result};
+use reqwest::{Client, cookie::Jar};
+use std::{collections::HashMap, fs, sync::Arc};
+
+/// Reads the cookies.json file and builds an authenticated reqwest client.
+pub fn build_authenticated_client(cookies_path: &str) -> Result<Client> {
+    // Read the JSON file.
+    let cookies_content = fs::read_to_string(cookies_path)
+        .with_context(|| format!("Failed to read cookies file from: {}", cookies_path))?;
+
+    // Parse the JSON into a Rust HashMap.
+    let cookies_map: HashMap<String, String> = serde_json::from_str(&cookies_content)
+        .context("Failed to parse cookies file. Ensure it is a flat key-value JSON object.")?;
+
+    // Create a Cookie Jar.
+    let jar = Arc::new(Jar::default());
+    // `reqwest` needs a URL to associate the cookies with.
+    let url = "https://learning.oreilly.com".parse::<reqwest::Url>()?;
+    for (key, value) in cookies_map {
+        // `reqwest` expects cookies in the standard "key=value" string format.
+        let cookie_str = format!("{}={}", key, value);
+        jar.add_cookie_str(&cookie_str, &url);
+    }
+
+    // Build the client with the cookie provider and a standard User-Agent.
+    let client = Client::builder()
+        .cookie_provider(Arc::clone(&jar))
+        .user_agent("Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36")
+        .build()
+        .context("Failed to build the HTTP client")?;
+
+    Ok(client)
+}