diff options
| -rw-r--r-- | Cargo.lock | 10 | ||||
| -rw-r--r-- | Cargo.toml | 1 | ||||
| -rw-r--r-- | src/epub.rs | 36 | ||||
| -rw-r--r-- | src/main.rs | 5 |
4 files changed, 48 insertions, 4 deletions
@@ -855,6 +855,7 @@ version = "0.1.0" dependencies = [ "anyhow", "clap", + "relative-path", "reqwest", "serde", "serde_json", @@ -1062,6 +1063,15 @@ dependencies = [ ] [[package]] +name = "relative-path" +version = "2.0.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "bca40a312222d8ba74837cb474edef44b37f561da5f773981007a10bbaa992b0" +dependencies = [ + "serde", +] + +[[package]] name = "reqwest" version = "0.13.2" source = "registry+https://github.com/rust-lang/crates.io-index" @@ -6,6 +6,7 @@ edition = "2024" [dependencies] anyhow = "1.0.102" clap = { version = "4.5.60", features = ["derive"] } +relative-path = "2.0.1" reqwest = { version = "0.13.2", features = ["cookies", "json"] } serde = { version = "1.0.228", features = ["derive"] } serde_json = "1.0.149" diff --git a/src/epub.rs b/src/epub.rs index 7eb47f3..4fc4915 100644 --- a/src/epub.rs +++ b/src/epub.rs @@ -1,7 +1,9 @@ -use crate::models::FileEntry; +use crate::models::{Chapter, FileEntry}; use anyhow::{Context, Result}; -use reqwest::Client; +use relative_path::RelativePath; +use reqwest::{Client, Url}; use std::{ + collections::HashMap, io::{Read, Write}, path::Path, }; @@ -66,6 +68,7 @@ pub fn create_epub_archive( epub_root: &Path, output_epub: &Path, file_entries: &[FileEntry], + chapters: &HashMap<String, Chapter>, ) -> Result<()> { let out_file = std::fs::File::create(output_epub)?; let mut zip = ZipWriter::new(out_file); @@ -83,6 +86,12 @@ pub fn create_epub_archive( .context("No OPF file with the correct MIME type was found.")?; write_container_xml_to_zip(&mut zip, &opf_entry.full_path)?; + // Prepare url path to local path mapping to clean xhtml files from external dependencies. + let url_to_local: HashMap<String, String> = file_entries + .iter() + .map(url_path_to_local) + .collect::<Result<HashMap<_, _>>>()?; + // Add the rest of the files according to file_entries. let options: FileOptions<()> = FileOptions::default().compression_method(CompressionMethod::Deflated); @@ -91,10 +100,31 @@ pub fn create_epub_archive( let mut src_file = std::fs::File::open(epub_root.join(&entry.full_path))?; let mut buffer = Vec::new(); src_file.read_to_end(&mut buffer)?; - zip.write_all(&buffer)?; + if chapters.contains_key(&entry.ourn) { + let mut html = String::from_utf8(buffer)?; + let chapter_dir = RelativePath::new(&entry.full_path) + .parent() + .unwrap_or(RelativePath::new("")); + for (url_path, local_path) in &url_to_local { + let rel_path = chapter_dir + .to_relative_path_buf() + .relative(RelativePath::new(local_path)); + html = html.replace(url_path, rel_path.as_str()); + } + zip.write_all(html.as_bytes())?; + } else { + zip.write_all(&buffer)?; + } } zip.finish()?; Ok(()) } + +/// Helper function. Maps FileEntry to (url path, full_path) pair. +fn url_path_to_local(entry: &FileEntry) -> Result<(String, String)> { + let url = Url::parse(&entry.url).with_context(|| format!("Could not parse: {}", entry.url))?; + let url_path = url.path().to_string(); + Ok((url_path, entry.full_path.clone())) +} diff --git a/src/main.rs b/src/main.rs index 3c5956e..80f81e4 100644 --- a/src/main.rs +++ b/src/main.rs @@ -2,6 +2,7 @@ mod epub; mod http_client; mod models; +use std::collections::HashMap; use std::path::Path; use crate::epub::{create_epub_archive, download_all_files}; @@ -107,6 +108,8 @@ async fn main() -> Result<()> { println!("Fetching book structure..."); let chapters: Vec<Chapter> = fetch_all_pages(&client, epub_data.chapters.clone()).await?; + let chapters: HashMap<String, Chapter> = + chapters.into_iter().map(|c| (c.ourn.clone(), c)).collect(); let file_entries: Vec<FileEntry> = fetch_all_pages(&client, epub_data.files.clone()).await?; let spine_items: Vec<SpineItem> = fetch_all_pages(&client, epub_data.spine.clone()).await?; let toc_vec: Vec<TocNode> = fetch_direct_array(&client, &epub_data.table_of_contents).await?; @@ -116,7 +119,7 @@ async fn main() -> Result<()> { download_all_files(&client, &file_entries, dest_root).await?; let epub_path = format!("Books/{0}/{0}.epub", args.bookid); let epub_path = Path::new(&epub_path); - create_epub_archive(dest_root, &epub_path, &file_entries)?; + create_epub_archive(dest_root, epub_path, &file_entries, &chapters)?; Ok(()) } |