diff --git a/src/book/mdbook.rs b/src/book/mdbook.rs index 4425ff40..f5ee7838 100644 --- a/src/book/mdbook.rs +++ b/src/book/mdbook.rs @@ -132,7 +132,7 @@ impl MDBook { if !path.exists() { debug!("[*]: {:?} does not exist, trying to create file", path); - ::std::fs::create_dir_all(path.parent().unwrap()); + try!(::std::fs::create_dir_all(path.parent().unwrap())); let mut f = try!(File::create(path)); debug!("[*]: Writing to {:?}", path); @@ -153,15 +153,12 @@ impl MDBook { pub fn build(&mut self) -> Result<(), Box> { debug!("[fn]: build"); - self.init(); + try!(self.init()); // Clean output directory try!(utils::remove_dir_content(&self.config.get_dest())); - try!(self.renderer.render( - self.iter(), - &self.config, - )); + try!(self.renderer.render(&self)); Ok(()) } @@ -288,11 +285,18 @@ impl MDBook { self } + pub fn get_title(&self) -> &str { + &self.config.title + } + pub fn set_author(mut self, author: &str) -> Self { self.config.author = author.to_owned(); self } + pub fn get_author(&self) -> &str { + &self.config.author + } // Construct book fn parse_summary(&mut self) -> Result<(), Box> { diff --git a/src/renderer/html_handlebars/hbs_renderer.rs b/src/renderer/html_handlebars/hbs_renderer.rs index af9011ff..ffce05e1 100644 --- a/src/renderer/html_handlebars/hbs_renderer.rs +++ b/src/renderer/html_handlebars/hbs_renderer.rs @@ -4,7 +4,7 @@ extern crate pulldown_cmark; use renderer::html_handlebars::helpers; use renderer::Renderer; -use book::{BookItems, BookConfig}; +use book::MDBook; use {utils, theme}; use std::path::PathBuf; @@ -26,12 +26,12 @@ impl HtmlHandlebars { } impl Renderer for HtmlHandlebars { - fn render(&self, book: BookItems, config: &BookConfig) -> Result<(), Box> { + fn render(&self, book: &MDBook) -> Result<(), Box> { debug!("[fn]: render"); let mut handlebars = Handlebars::new(); // Load theme - let theme = theme::Theme::new(&config.get_src()); + let theme = theme::Theme::new(book.get_src()); // Register template debug!("[*]: Register handlebars template"); @@ -43,22 +43,22 @@ impl Renderer for HtmlHandlebars { handlebars.register_helper("previous", Box::new(helpers::navigation::previous)); handlebars.register_helper("next", Box::new(helpers::navigation::next)); - let mut data = try!(make_data(book.clone(), config)); + let mut data = try!(make_data(book)); // Check if dest directory exists debug!("[*]: Check if destination directory exists"); - match utils::create_path(config.get_dest()) { + match utils::create_path(book.get_dest()) { Err(_) => return Err(Box::new(io::Error::new(io::ErrorKind::Other, "Unexcpected error when constructing destination path"))), _ => {}, }; // Render a file for every entry in the book let mut index = true; - for (_, item) in book { + for (_, item) in book.iter() { if item.path != PathBuf::new() { - let path = config.get_src().join(&item.path); + let path = book.get_src().join(&item.path); debug!("[*]: Opening file: {:?}", path); let mut f = try!(File::open(&path)); @@ -90,10 +90,10 @@ impl Renderer for HtmlHandlebars { debug!("[*]: Render template"); let rendered = try!(handlebars.render("index", &data)); - debug!("[*]: Create file {:?}", &config.get_dest().join(&item.path).with_extension("html")); + debug!("[*]: Create file {:?}", &book.get_dest().join(&item.path).with_extension("html")); // Write to file - let mut file = try!(utils::create_file(&config.get_dest().join(&item.path).with_extension("html"))); - output!("[*] Creating {:?} ✓", &config.get_dest().join(&item.path).with_extension("html")); + let mut file = try!(utils::create_file(&book.get_dest().join(&item.path).with_extension("html"))); + output!("[*] Creating {:?} ✓", &book.get_dest().join(&item.path).with_extension("html")); try!(file.write_all(&rendered.into_bytes())); @@ -101,13 +101,13 @@ impl Renderer for HtmlHandlebars { if index { debug!("[*]: index.html"); try!(fs::copy( - config.get_dest().join(&item.path.with_extension("html")), - config.get_dest().join("index.html") + book.get_dest().join(&item.path.with_extension("html")), + book.get_dest().join("index.html") )); output!( "[*] Creating index.html from {:?} ✓", - config.get_dest().join(&item.path.with_extension("html")) + book.get_dest().join(&item.path.with_extension("html")) ); index = false; } @@ -118,33 +118,33 @@ impl Renderer for HtmlHandlebars { debug!("[*] Copy static files"); // JavaScript - let mut js_file = try!(File::create(config.get_dest().join("book.js"))); + let mut js_file = try!(File::create(book.get_dest().join("book.js"))); try!(js_file.write_all(&theme.js)); // Css - let mut css_file = try!(File::create(config.get_dest().join("book.css"))); + let mut css_file = try!(File::create(book.get_dest().join("book.css"))); try!(css_file.write_all(&theme.css)); // syntax highlighting - let mut highlight_css = try!(File::create(config.get_dest().join("highlight.css"))); + let mut highlight_css = try!(File::create(book.get_dest().join("highlight.css"))); try!(highlight_css.write_all(&theme.highlight_css)); - let mut highlight_js = try!(File::create(config.get_dest().join("highlight.js"))); + let mut highlight_js = try!(File::create(book.get_dest().join("highlight.js"))); try!(highlight_js.write_all(&theme.highlight_js)); Ok(()) } } -fn make_data(book: BookItems, config: &BookConfig) -> Result, Box> { +fn make_data(book: &MDBook) -> Result, Box> { debug!("[fn]: make_data"); let mut data = BTreeMap::new(); data.insert("language".to_string(), "en".to_json()); - data.insert("title".to_string(), config.title.to_json()); + data.insert("title".to_string(), book.get_title().to_json()); let mut chapters = vec![]; - for (section, item) in book { + for (section, item) in book.iter() { let mut chapter = BTreeMap::new(); chapter.insert("section".to_string(), section.to_json()); chapter.insert("name".to_string(), item.name.to_json()); diff --git a/src/renderer/renderer.rs b/src/renderer/renderer.rs index cd751791..615efbf7 100644 --- a/src/renderer/renderer.rs +++ b/src/renderer/renderer.rs @@ -1,8 +1,5 @@ -use book::BookItems; -use book::BookConfig; - use std::error::Error; pub trait Renderer { - fn render(&self, book: BookItems, config: &BookConfig) -> Result<(), Box>; + fn render(&self, book: &::book::MDBook) -> Result<(), Box>; }