Add e-book metadata fetching

2 files changed, 73 insertions, 10 deletions

diff --git a/src/main.rs b/src/main.rs
index 33c8d82..165d6be 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,8 +1,11 @@
 mod http_client;
+mod models;
 
-use anyhow::Result;
+use anyhow::{Context, Result};
 use clap::Parser;
 use http_client::build_authenticated_client;
+use models::{EpubResponse, SearchResponse};
+use reqwest::Client;
 
 /// Download and generate an EPUB from Safari Books Online.
 #[derive(Parser, Debug)]
@@ -19,6 +22,34 @@ struct Args {
     preserve_log: bool,
 }
 
+/// Fetches book metadata from the search endpoint.
+async fn fetch_metadata(client: &Client, bookid: &str) -> Result<SearchResponse> {
+    let url = format!("https://learning.oreilly.com/api/v2/search/?query={bookid}&limit=1");
+    let response = client
+        .get(&url)
+        .send()
+        .await?
+        .error_for_status()?
+        .json::<SearchResponse>()
+        .await
+        .context("Failed to deserialize Search API response.")?;
+    Ok(response)
+}
+
+/// Fetches EPUB structural data (like the chapters URL)
+async fn fetch_epub_data(client: &Client, bookid: &str) -> Result<EpubResponse> {
+    let url = format!("https://learning.oreilly.com/api/v2/epubs/urn:orm:book:{bookid}/");
+    let response = client
+        .get(&url)
+        .send()
+        .await?
+        .error_for_status()?
+        .json::<EpubResponse>()
+        .await
+        .context("Failed to deserialize EPUB API response")?;
+    Ok(response)
+}
+
 #[tokio::main]
 async fn main() -> Result<()> {
     // Parse the command line arguments
@@ -31,17 +62,24 @@ async fn main() -> Result<()> {
     println!("Loading cookies and initialising the HTTP client...");
     let client = build_authenticated_client(&args.cookies)?;
 
-    // Quick test request to verify authentication.
-    let profile_url = "https://learning.oreilly.com/profile/";
-    let response = client.get(profile_url).send().await?;
-    if response.status().is_success() {
-        println!("Successfully authenticated!");
+    println!("Fetching book metadata...");
+    // Fetch from the search API.
+    let search_data = fetch_metadata(&client, &args.bookid).await?;
+    if let Some(book) = search_data.results.first() {
+        println!("\n--- Book Found ---");
+        println!("Title: {}", book.title);
+        println!("Authors: {}", book.authors.join(", "));
+        println!("Publisher: {}", book.publishers.join(", "));
+        println!("Cover URL: {}", book.cover_url);
     } else {
-        println!(
-            "Authentication might have failed. Status code: {}",
-            response.status()
-        );
+        anyhow::bail!("Could not find book metadata for ID: {}", args.bookid);
     }
+    // Fetch from the EPUB API.
+    let epub_data = fetch_epub_data(&client, &args.bookid).await?;
+    println!("Publication date: {}", epub_data.publication_date);
+    println!("Chapters URL: {}", epub_data.chapters);
+    println!("Resources URL: {}", epub_data.files);
+    println!("------------------\n");
 
     Ok(())
 }
diff --git a/src/models.rs b/src/models.rs
new file mode 100644
index 0000000..77afeb2
--- /dev/null
+++ b/src/models.rs
@@ -0,0 +1,25 @@
+use serde::Deserialize;
+
+// --- Models for the Search API ---
+
+#[derive(Debug, Deserialize)]
+pub struct SearchResponse {
+    pub results: Vec<SearchResult>,
+}
+
+#[derive(Debug, Deserialize)]
+pub struct SearchResult {
+    pub title: String,
+    pub authors: Vec<String>,
+    pub publishers: Vec<String>,
+    pub cover_url: String,
+}
+
+// --- Models for the EPUB API ---
+
+#[derive(Debug, Deserialize)]
+pub struct EpubResponse {
+    pub publication_date: String,
+    pub chapters: String, // This is a URL to the chapters list
+    pub files: String,    // This is a URL to the resource files
+}