1 files changed, 9 insertions, 2 deletions
diff --git a/src/main.rs b/src/main.rs
index 4f28d52..ed73ec7 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,12 +1,15 @@
+mod epub;
 mod http_client;
 mod models;
 
 use std::collections::HashMap;
+use std::path::Path;
 
+use crate::epub::download_all_files;
+use crate::http_client::build_authenticated_client;
+use crate::models::{Chapter, EpubResponse, FileEntry, Paginated, SpineItem, TocNode};
 use anyhow::{Context, Result, ensure};
 use clap::Parser;
-use http_client::build_authenticated_client;
-use models::{Chapter, EpubResponse, FileEntry, Paginated, SpineItem, TocNode};
 use reqwest::Client;
 
 /// Download and generate an EPUB from Safari Books Online.
@@ -115,6 +118,10 @@ async fn main() -> Result<()> {
         .find(|f| f.filename_ext == ".opf" && f.media_type == "application/oebps-package+xml")
         .context("No OPF file with the correct MIME type was found.")?;
 
+    let dest_root = format!("Books/{}/epub_root", args.bookid);
+    let dest_root = Path::new(&dest_root);
+    download_all_files(&client, &file_entries, dest_root).await?;
+
     // Sanity check: Every entry in spine exists in chapters.
     let chapters: HashMap<String, Chapter> =
         chapters.into_iter().map(|c| (c.ourn.clone(), c)).collect();