Add a login check

Try to access the profile page to see if the cookies work or not.

2 files changed, 38 insertions, 2 deletions

diff --git a/src/main.rs b/src/main.rs
index 6981b8e..f79f191 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -3,14 +3,18 @@ mod config;
 mod cookies;
 mod display;
 mod http_client;
+mod orly;
 
 use clap::Parser;
 use cli::Args;
 use cookies::CookieStore;
 use display::Display;
 use http_client::HttpClient;
+use orly::check_login;
+use reqwest::Client;
 
-fn main() {
+#[tokio::main]
+async fn main() {
     let args = Args::parse();
     let mut ui = Display::new(&args.bookid);
 
@@ -40,12 +44,22 @@ fn main() {
     ));
 
     // Build the HTTP client with our cookies (no network calls yet).
-    let _client = match HttpClient::from_store(&store) {
+    let client = match HttpClient::from_store(&store) {
         Ok(c) => c,
         Err(e) => ui.error_and_exit(&format!("Failed to build HTTP client: {e}")),
     };
     ui.info("HTTP client initialized with cookies (no requests performed).");
 
+    // Check whether the cookies work (are we logged in?).
+    match check_login(&client).await {
+        Ok(true) => ui.info("Login confirmed..."),
+        Ok(false) => ui.error_and_exit(
+            "Logged out. Cookies could be stale or invalid.\n\
+            Try refreshing your cookies.json and trying again.",
+        ),
+        Err(e) => ui.error_and_exit(&format!("Login check failed: {e}")),
+    };
+
     let output_dir = config::books_root().join(format!("(pending) ({})", args.bookid));
 
     ui.set_output_dir(output_dir);
diff --git a/src/orly.rs b/src/orly.rs
new file mode 100644
index 0000000..cd8b645
--- /dev/null
+++ b/src/orly.rs
@@ -0,0 +1,22 @@
+use crate::http_client::HttpClient;
+use anyhow::{bail, Result};
+
+pub const PROFILE_URL: &str = "https://learning.oreilly.com/profile/";
+
+/// Check whether cookies keep us logged in by fetching the profile page.
+/// Returns:
+/// - Ok(true)  => HTTP 200 (assume logged in)
+/// - Ok(false) => Redirect or 401/403 (assume not logged in)
+/// - Err(..)   => Network/other error
+pub async fn check_login(client: &HttpClient) -> Result<bool> {
+    let res = client.client().get(PROFILE_URL).send().await?;
+    let status = res.status();
+
+    if status.is_redirection() {
+        Ok(false)
+    } else if status == 200 {
+        Ok(true)
+    } else {
+        bail!("Profile request returned unexpected status {}", status)
+    }
+}