diff options
Diffstat (limited to 'src/http_client.rs')
| -rw-r--r-- | src/http_client.rs | 33 |
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) +} |
