148 lines
4.1 KiB
Rust
148 lines
4.1 KiB
Rust
use std::path::Path;
|
|
use std::fs::{self, File, metadata};
|
|
use std::io::{self, Write};
|
|
use std::error::Error;
|
|
|
|
use {BookConfig, BookItem};
|
|
use book::BookItems;
|
|
use parse;
|
|
use renderer::Renderer;
|
|
use renderer::HtmlHandlebars;
|
|
|
|
pub struct MDBook {
|
|
config: BookConfig,
|
|
pub content: Vec<BookItem>,
|
|
renderer: Box<Renderer>,
|
|
}
|
|
|
|
impl MDBook {
|
|
|
|
pub fn new(path: &Path) -> MDBook {
|
|
|
|
// Hacky way to check if the path exists... Until PathExt moves to stable
|
|
match metadata(path) {
|
|
Err(_) => panic!("Directory does not exist"),
|
|
Ok(f) => {
|
|
if !f.is_dir() {
|
|
panic!("Is not a directory");
|
|
}
|
|
}
|
|
}
|
|
|
|
MDBook {
|
|
content: vec![],
|
|
config: BookConfig::new()
|
|
.set_src(&path.join("src"))
|
|
.set_dest(&path.join("book"))
|
|
.to_owned(),
|
|
renderer: Box::new(HtmlHandlebars::new()),
|
|
}
|
|
}
|
|
|
|
pub fn iter(&self) -> BookItems {
|
|
BookItems {
|
|
items: &self.content[..],
|
|
current_index: 0,
|
|
stack: Vec::new(),
|
|
}
|
|
}
|
|
|
|
pub fn init(&self) -> Result<(), Box<Error>> {
|
|
|
|
let dest = self.config.dest();
|
|
let src = self.config.src();
|
|
|
|
// Hacky way to check if the directory exists... Until PathExt moves to stable
|
|
match metadata(&dest) {
|
|
Err(_) => {
|
|
// There is a very high chance that the error is due to the fact that
|
|
// the directory / file does not exist
|
|
fs::create_dir(&dest).unwrap();
|
|
},
|
|
Ok(_) => { /* If there is no error, the directory / file does exist */ }
|
|
}
|
|
|
|
// Hacky way to check if the directory exists... Until PathExt moves to stable
|
|
match metadata(&src) {
|
|
Err(_) => {
|
|
// There is a very high chance that the error is due to the fact that
|
|
// the directory / file does not exist
|
|
fs::create_dir(&src).unwrap();
|
|
},
|
|
Ok(_) => { /* If there is no error, the directory / file does exist */ }
|
|
}
|
|
|
|
// Hacky way to check if the directory exists... Until PathExt moves to stable
|
|
let summary = match metadata(&src.join("SUMMARY.md")) {
|
|
Err(_) => {
|
|
// There is a very high chance that the error is due to the fact that
|
|
// the directory / file does not exist
|
|
Ok(File::create(&src.join("SUMMARY.md")).unwrap())
|
|
},
|
|
Ok(_) => {
|
|
/* If there is no error, the directory / file does exist */
|
|
Err("SUMMARY.md does already exist")
|
|
}
|
|
};
|
|
|
|
if let Ok(mut f) = summary {
|
|
try!(writeln!(f, "# Summary"));
|
|
try!(writeln!(f, ""));
|
|
try!(writeln!(f, "- [Chapter 1](./chapter_1.md)"));
|
|
|
|
let mut chapter_1 = File::create(&src.join("chapter_1.md")).unwrap();
|
|
try!(writeln!(chapter_1, "# Chapter 1"));
|
|
}
|
|
|
|
return Ok(());
|
|
}
|
|
|
|
pub fn build(&mut self) -> Result<(), Box<Error>> {
|
|
|
|
try!(self.parse_summary());
|
|
|
|
try!(self.renderer.render(
|
|
self.iter(),
|
|
&self.config,
|
|
));
|
|
|
|
Ok(())
|
|
}
|
|
|
|
|
|
// Builder functions
|
|
pub fn set_dest(mut self, dest: &Path) -> Self {
|
|
self.config.set_dest(dest);
|
|
self
|
|
}
|
|
|
|
pub fn set_src(mut self, src: &Path) -> Self {
|
|
self.config.set_src(src);
|
|
self
|
|
}
|
|
|
|
pub fn set_title(mut self, title: &str) -> Self {
|
|
self.config.set_title(title);
|
|
self
|
|
}
|
|
|
|
pub fn set_author(mut self, author: &str) -> Self {
|
|
self.config.set_author(author);
|
|
self
|
|
}
|
|
|
|
|
|
// Construct book
|
|
fn parse_summary(&mut self) -> Result<(), Box<Error>> {
|
|
|
|
// When append becomes stable, use self.content.append() ...
|
|
let book_items = try!(parse::construct_bookitems(&self.config.src().join("SUMMARY.md")));
|
|
|
|
for item in book_items {
|
|
self.content.push(item)
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
}
|