41 lines
1.4 KiB
Rust
41 lines
1.4 KiB
Rust
|
use std::fs::File;
|
||
|
|
|
||
|
|
use epub_builder::{EpubBuilder, EpubContent, ZipLibrary};
|
||
|
|
|
||
|
|
use crate::extractor::{self, Extractor};
|
||
|
|
|
||
|
|
pub fn generate_epub(extractor: Extractor) {
|
||
|
|
let file_name = format!(
|
||
|
|
"{}.epub",
|
||
|
|
extractor
|
||
|
|
.metadata()
|
||
|
|
.title()
|
||
|
|
.replace("/", " ")
|
||
|
|
.replace("\\", " ")
|
||
|
|
);
|
||
|
|
let mut out_file = File::create(&file_name).unwrap();
|
||
|
|
let mut html_buf = Vec::new();
|
||
|
|
extractor::serialize_to_xhtml(extractor.article().unwrap(), &mut html_buf)
|
||
|
|
.expect("Unable to serialize to xhtml");
|
||
|
|
let html_buf = std::str::from_utf8(&html_buf).unwrap();
|
||
|
|
let mut epub = EpubBuilder::new(ZipLibrary::new().unwrap()).unwrap();
|
||
|
|
if let Some(author) = extractor.metadata().byline() {
|
||
|
|
epub.metadata("author", author.replace("&", "&"))
|
||
|
|
.unwrap();
|
||
|
|
}
|
||
|
|
epub.metadata("title", extractor.metadata().title().replace("&", "&"))
|
||
|
|
.unwrap();
|
||
|
|
epub.add_content(EpubContent::new("index.xhtml", html_buf.as_bytes()))
|
||
|
|
.unwrap();
|
||
|
|
for img in extractor.img_urls {
|
||
|
|
let mut file_path = std::env::temp_dir();
|
||
|
|
file_path.push(&img.0);
|
||
|
|
|
||
|
|
let img_buf = File::open(&file_path).expect("Can't read file");
|
||
|
|
epub.add_resource(file_path.file_name().unwrap(), img_buf, img.1.unwrap())
|
||
|
|
.unwrap();
|
||
|
|
}
|
||
|
|
epub.generate(&mut out_file).unwrap();
|
||
|
|
println!("Created {:?}", file_name);
|
||
|
|
}
|