diff options
| -rw-r--r-- | src/main.rs | 79 |
1 files changed, 62 insertions, 17 deletions
diff --git a/src/main.rs b/src/main.rs index 02e6b52..e8f4639 100644 --- a/src/main.rs +++ b/src/main.rs @@ -33,7 +33,7 @@ struct Bk { chapter_idx: usize, pos: usize, rows: usize, - toc: Vec<String>, + toc: Vec<(String, String)>, pad: u16, } @@ -54,8 +54,7 @@ impl Epub { .unwrap(); text } - fn get_toc(&mut self) -> Vec<String> { - // container.xml -> <rootfile> -> opf -> <spine> -> <manifest> + fn get_toc(&mut self) -> Vec<(String, String)> { let xml = self.get_text("META-INF/container.xml"); let doc = Document::parse(&xml).unwrap(); let path = doc @@ -65,7 +64,8 @@ impl Epub { .attribute("full-path") .unwrap(); - let mut manifest = HashMap::new(); + // in theory the toc is enough, in reality it isn't + let mut id_path = HashMap::new(); let xml = self.get_text(path); let doc = Document::parse(&xml).unwrap(); doc.root_element() @@ -75,25 +75,68 @@ impl Epub { .children() .filter(|n| n.is_element()) .for_each(|n| { - manifest.insert( + id_path.insert( n.attribute("id").unwrap(), n.attribute("href").unwrap(), ); }); - - let path = std::path::Path::new(&path).parent().unwrap(); - doc.root_element() + let dirname = std::path::Path::new(&path).parent().unwrap(); + let paths: Vec<&str> = doc.root_element() .children() .find(|n| n.has_tag_name("spine")) .unwrap() .children() .filter(|n| n.is_element()) - .map(|n| { - let name = - manifest.get(n.attribute("idref").unwrap()).unwrap(); - path.join(name).to_str().unwrap().to_string() - }) - .collect() + .map(|n| id_path.remove(n.attribute("idref").unwrap()).unwrap()) + .collect(); + + // epub2: item id="ncx" (spine toc=id) + // epub3: item properties="nav" + let titles: Vec<String> = { + if let Some(path) = id_path.get("ncx") { + let xml = self.get_text(dirname.join(path).to_str().unwrap()); + let doc = Document::parse(&xml).unwrap(); + + doc.descendants() + .find(|n| n.has_tag_name("navMap")) + .unwrap() + .descendants() + .filter(|n| n.has_tag_name("navPoint")) + .map(|n| n.descendants() + .find(|n| n.has_tag_name("text")) + .unwrap() + .text() + .unwrap() + .to_string() + ) + .collect() + } else if let Some(path) = id_path.get("toc.xhtml") { + let xml = self.get_text(dirname.join(path).to_str().unwrap()); + let doc = Document::parse(&xml).unwrap(); + + doc.descendants() + .find(|n| n.has_tag_name("nav")) + .unwrap() + .descendants() + .filter(|n| n.has_tag_name("a")) + .map(|n| { + n.descendants() + .filter(|n| n.is_text()) + .map(|n| n.text().unwrap()) + .collect() + }) + .collect() + } else { + panic!("can't read epub"); + } + }; + + paths.into_iter().enumerate().map(|(i, path)| { + let title = titles.get(i).unwrap_or(&path.to_string()).to_string(); + let path = dirname.join(path).to_str().unwrap().to_string(); + (title, path) + }) + .collect() } } @@ -152,7 +195,7 @@ impl Bk { } fn get_chapter(&mut self, idx: usize) { let mut chapter = Vec::new(); - let xml = self.epub.get_text(&self.toc[idx]); + let xml = self.epub.get_text(&self.toc[idx].1); let doc = Document::parse(&xml).unwrap(); for n in doc.descendants() { @@ -290,10 +333,12 @@ impl Bk { queue!( stdout, terminal::Clear(terminal::ClearType::All), - cursor::MoveTo(0, 0), - Print("TODO: nav") ) .unwrap(); + let end = std::cmp::min(self.rows, self.toc.len()); + for i in 0..end { + queue!(stdout, cursor::MoveTo(0, i as u16), Print(&self.toc[i].0)).unwrap(); + } stdout.flush().unwrap(); } fn render_help(&self) { |