use std::collections::HashMap; use std::fs::File; use std::io::Read; use roxmltree::{Document, Node}; pub struct Epub { container: zip::ZipArchive, pub chapters: Vec<(String, String)>, pub meta: String, } impl Epub { pub fn new(path: &str) -> std::io::Result { let file = File::open(path)?; let mut epub = Epub { container: zip::ZipArchive::new(file)?, chapters: Vec::new(), meta: String::new(), }; epub.get_rootfile(); Ok(epub) } pub fn get_chapters(&mut self) { self.chapters = std::mem::take(&mut self.chapters) .into_iter() .filter_map(|(path, title)| { let xml = self.get_text(&path); // https://github.com/RazrFalcon/roxmltree/issues/12 // UnknownEntityReference for HTML entities let doc = Document::parse(&xml).unwrap(); let body = doc.root_element().last_element_child().unwrap(); let mut chapter = String::new(); Epub::render(&mut chapter, body); if chapter.is_empty() { None } else { Some((chapter, title)) } }) .collect(); } fn render(buf: &mut String, n: Node) { if n.is_text() { let text = n.text().unwrap(); if !text.trim().is_empty() { buf.push_str(text); } return; } match n.tag_name().name() { "h1" | "h2" | "h3" | "h4" | "h5" | "h6" => { buf.push_str("\n\x1b[1m"); for c in n.children() { Self::render(buf, c); } buf.push_str("\x1b[0m\n"); } "blockquote" | "p" | "tr" => { buf.push('\n'); for c in n.children() { Self::render(buf, c); } buf.push('\n'); } "li" => { buf.push_str("\n- "); for c in n.children() { Self::render(buf, c); } buf.push('\n'); } "br" => buf.push('\n'), _ => { for c in n.children() { Self::render(buf, c); } } } } fn get_text(&mut self, name: &str) -> String { let mut text = String::new(); self.container .by_name(name) .unwrap() .read_to_string(&mut text) .unwrap(); text } fn get_rootfile(&mut self) { let xml = self.get_text("META-INF/container.xml"); let doc = Document::parse(&xml).unwrap(); let path = doc .descendants() .find(|n| n.has_tag_name("rootfile")) .unwrap() .attribute("full-path") .unwrap(); let xml = self.get_text(path); let doc = Document::parse(&xml).unwrap(); // zip expects unix path even on windows let rootdir = match path.rfind('/') { Some(n) => &path[..=n], None => "", }; let mut manifest = HashMap::new(); let mut nav = HashMap::new(); let mut children = doc.root_element().children().filter(Node::is_element); let meta_node = children.next().unwrap(); let manifest_node = children.next().unwrap(); let spine_node = children.next().unwrap(); meta_node .children() .filter(|n| n.is_element() && n.tag_name().name() != "meta") .for_each(|n| { let name = n.tag_name().name(); let text = n.text().unwrap(); self.meta.push_str(&format!("{}: {}\n", name, text)); }); manifest_node .children() .filter(Node::is_element) .for_each(|n| { manifest.insert(n.attribute("id").unwrap(), n.attribute("href").unwrap()); }); if doc.root_element().attribute("version") == Some("3.0") { let path = manifest_node .children() .find(|n| n.attribute("properties") == Some("nav")) .unwrap() .attribute("href") .unwrap(); let xml = self.get_text(&format!("{}{}", rootdir, path)); let doc = Document::parse(&xml).unwrap(); doc.descendants() .find(|n| n.has_tag_name("nav")) .unwrap() .descendants() .filter(|n| n.has_tag_name("a")) .for_each(|n| { let path = n.attribute("href").unwrap().to_string(); let text = n .descendants() .filter(Node::is_text) .map(|n| n.text().unwrap()) .collect(); nav.insert(path, text); }) } else { let toc = spine_node.attribute("toc").unwrap_or("ncx"); let path = manifest.get(toc).unwrap(); let xml = self.get_text(&format!("{}{}", rootdir, path)); let doc = Document::parse(&xml).unwrap(); doc.descendants() .find(|n| n.has_tag_name("navMap")) .unwrap() .descendants() .filter(|n| n.has_tag_name("navPoint")) .for_each(|n| { let path = n .descendants() .find(|n| n.has_tag_name("content")) .unwrap() .attribute("src") .unwrap() .to_string(); let text = n .descendants() .find(|n| n.has_tag_name("text")) .unwrap() .text() .unwrap() .to_string(); nav.insert(path, text); }) } self.chapters = spine_node .children() .filter(Node::is_element) .enumerate() .map(|(i, n)| { let id = n.attribute("idref").unwrap(); let path = manifest.remove(id).unwrap(); let label = nav.remove(path).unwrap_or_else(|| i.to_string()); let path = format!("{}{}", rootdir, path); (path, label) }) .collect(); } } #[test] fn test_dir() { let path = "/mnt/lit/read"; for entry in std::fs::read_dir(path).unwrap() { let path = entry.unwrap().path(); let s = path.to_str().unwrap(); println!("testing: {}", s); Epub::new(s).unwrap(); } }