Add debug! macro that can be activated using the "debug" feature. #19

This commit is contained in:
Mathieu David 2015-08-03 18:06:01 +02:00
parent fd58a4d441
commit 22280bbb16
8 changed files with 96 additions and 33 deletions

View File

@ -8,3 +8,6 @@ clap = "*"
handlebars = "*" handlebars = "*"
rustc-serialize = "*" rustc-serialize = "*"
pulldown-cmark = "*" pulldown-cmark = "*"
[features]
debug = []

View File

@ -4,8 +4,6 @@ use std::fs::File;
use std::io::Read; use std::io::Read;
use std::path::{Path, PathBuf}; use std::path::{Path, PathBuf};
use utils;
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
pub struct BookConfig { pub struct BookConfig {
pub title: String, pub title: String,
@ -31,12 +29,18 @@ impl BookConfig {
pub fn read_config(&mut self) -> &mut Self { pub fn read_config(&mut self) -> &mut Self {
debug!("[fn]: read_config");
// If the file does not exist, return early // If the file does not exist, return early
let mut config_file = match File::open(self.src.join("book.json")) { let mut config_file = match File::open(self.src.join("book.json")) {
Ok(f) => f, Ok(f) => f,
Err(_) => return self, Err(_) => {
debug!("[*]: Failed to open {:?}", self.src.join("book.json"));
return self
},
}; };
debug!("[*]: Reading config");
let mut data = String::new(); let mut data = String::new();
config_file.read_to_string(&mut data).unwrap(); config_file.read_to_string(&mut data).unwrap();
@ -45,6 +49,7 @@ impl BookConfig {
// Extract data // Extract data
debug!("[*]: Extracting data from config");
// Title & author // Title & author
if let Some(a) = config.find_path(&["title"]) { self.title = a.to_string().replace("\"", "") } if let Some(a) = config.find_path(&["title"]) { self.title = a.to_string().replace("\"", "") }
if let Some(a) = config.find_path(&["author"]) { self.author = a.to_string().replace("\"", "") } if let Some(a) = config.find_path(&["author"]) { self.author = a.to_string().replace("\"", "") }

View File

@ -49,6 +49,8 @@ impl MDBook {
pub fn init(&self) -> Result<(), Box<Error>> { pub fn init(&self) -> Result<(), Box<Error>> {
debug!("[fn]: init");
let dest = self.config.dest(); let dest = self.config.dest();
let src = self.config.src(); let src = self.config.src();
@ -57,6 +59,7 @@ impl MDBook {
Err(_) => { Err(_) => {
// There is a very high chance that the error is due to the fact that // There is a very high chance that the error is due to the fact that
// the directory / file does not exist // the directory / file does not exist
debug!("[*]: {:?} does not exist, trying to create directory", dest);
fs::create_dir(&dest).unwrap(); fs::create_dir(&dest).unwrap();
}, },
Ok(_) => { /* If there is no error, the directory / file does exist */ } Ok(_) => { /* If there is no error, the directory / file does exist */ }
@ -67,6 +70,7 @@ impl MDBook {
Err(_) => { Err(_) => {
// There is a very high chance that the error is due to the fact that // There is a very high chance that the error is due to the fact that
// the directory / file does not exist // the directory / file does not exist
debug!("[*]: {:?} does not exist, trying to create directory", src);
fs::create_dir(&src).unwrap(); fs::create_dir(&src).unwrap();
}, },
Ok(_) => { /* If there is no error, the directory / file does exist */ } Ok(_) => { /* If there is no error, the directory / file does exist */ }
@ -77,6 +81,7 @@ impl MDBook {
Err(_) => { Err(_) => {
// There is a very high chance that the error is due to the fact that // There is a very high chance that the error is due to the fact that
// the directory / file does not exist // the directory / file does not exist
debug!("[*]: {:?} does not exist, trying to create SUMMARY.md", src.join("SUMMARY.md"));
Ok(File::create(&src.join("SUMMARY.md")).unwrap()) Ok(File::create(&src.join("SUMMARY.md")).unwrap())
}, },
Ok(_) => { Ok(_) => {
@ -86,6 +91,8 @@ impl MDBook {
}; };
if let Ok(mut f) = summary { if let Ok(mut f) = summary {
debug!("[*]: Writing to SUMMARY.md");
try!(writeln!(f, "# Summary")); try!(writeln!(f, "# Summary"));
try!(writeln!(f, "")); try!(writeln!(f, ""));
try!(writeln!(f, "- [Chapter 1](./chapter_1.md)")); try!(writeln!(f, "- [Chapter 1](./chapter_1.md)"));
@ -98,6 +105,7 @@ impl MDBook {
} }
pub fn build(&mut self) -> Result<(), Box<Error>> { pub fn build(&mut self) -> Result<(), Box<Error>> {
debug!("[fn]: build");
try!(self.parse_summary()); try!(self.parse_summary());

View File

@ -1,3 +1,5 @@
#[macro_use]
pub mod macros;
mod book; mod book;
mod parse; mod parse;
pub mod renderer; pub mod renderer;

11
src/macros.rs Normal file
View File

@ -0,0 +1,11 @@
#[cfg(feature = "debug")]
macro_rules! debug {
($fmt:expr) => (println!($fmt));
($fmt:expr, $($arg:tt)*) => (println!($fmt, $($arg)*));
}
#[cfg(not(feature = "debug"))]
macro_rules! debug {
($fmt:expr) => ();
($fmt:expr, $($arg:tt)*) => ();
}

View File

@ -14,18 +14,18 @@ pub enum LineType {
*/ */
pub fn construct_bookitems(path: &PathBuf) -> Result<Vec<BookItem>> { pub fn construct_bookitems(path: &PathBuf) -> Result<Vec<BookItem>> {
debug!("[fn]: construct_bookitems");
let mut summary = String::new(); let mut summary = String::new();
try!(try!(File::open(path)).read_to_string(&mut summary)); try!(try!(File::open(path)).read_to_string(&mut summary));
debug!("[*]: Parse SUMMARY.md");
let top_items = try!(parse_level(&mut summary.split('\n').collect(), 0)); let top_items = try!(parse_level(&mut summary.split('\n').collect(), 0));
Ok(top_items) Ok(top_items)
} }
fn parse_level(summary: &mut Vec<&str>, current_level: i32) -> Result<Vec<BookItem>> { fn parse_level(summary: &mut Vec<&str>, current_level: i32) -> Result<Vec<BookItem>> {
debug!("[fn]: parse_level");
let mut items: Vec<BookItem> = vec![]; let mut items: Vec<BookItem> = vec![];
loop { loop {
@ -50,6 +50,7 @@ fn parse_level(summary: &mut Vec<&str>, current_level: i32) -> Result<Vec<BookIt
} }
fn level(line: &str, spaces_in_tab: i32) -> Result<i32> { fn level(line: &str, spaces_in_tab: i32) -> Result<i32> {
debug!("[fn]: level");
let mut spaces = 0; let mut spaces = 0;
let mut level = 0; let mut level = 0;
@ -67,9 +68,12 @@ fn level(line: &str, spaces_in_tab: i32) -> Result<i32> {
// If there are spaces left, there is an indentation error // If there are spaces left, there is an indentation error
if spaces > 0 { if spaces > 0 {
debug!("[SUMMARY.md]:");
debug!("\t[line]: {}", line);
debug!("[*]: There is an indentation error on this line. Indentation should be {} spaces", spaces_in_tab);
return Err(Error::new( return Err(Error::new(
ErrorKind::Other, ErrorKind::Other,
format!("There is an indentation error on line:\n\n{}", line) format!("Indentation error on line:\n\n{}", line)
) )
) )
} }
@ -79,6 +83,7 @@ fn level(line: &str, spaces_in_tab: i32) -> Result<i32> {
fn parse_line(l: &str) -> Option<BookItem> { fn parse_line(l: &str) -> Option<BookItem> {
debug!("[fn]: parse_line");
let mut name; let mut name;
let mut path; let mut path;
// Remove leading and trailing spaces or tabs // Remove leading and trailing spaces or tabs
@ -88,18 +93,25 @@ fn parse_line(l: &str) -> Option<BookItem> {
match c { match c {
// List item // List item
'-' | '*' => { '-' | '*' => {
debug!("[*]: Line is list element");
let mut start_delimitor; let mut start_delimitor;
let mut end_delimitor; let mut end_delimitor;
// In the future, support for list item that is not a link // In the future, support for list item that is not a link
// Not sure if I should error on line I can't parse or just ignore them... // Not sure if I should error on line I can't parse or just ignore them...
if let Some(i) = line.find('[') { start_delimitor = i; } if let Some(i) = line.find('[') { start_delimitor = i; }
else { return None } else {
debug!("[*]: '[' not found, this line is not a link. Ignoring...");
return None
}
if let Some(i) = line[start_delimitor..].find("](") { if let Some(i) = line[start_delimitor..].find("](") {
end_delimitor = start_delimitor +i; end_delimitor = start_delimitor +i;
} }
else { return None } else {
debug!("[*]: '](' not found, this line is not a link. Ignoring...");
return None
}
name = line[start_delimitor + 1 .. end_delimitor].to_string(); name = line[start_delimitor + 1 .. end_delimitor].to_string();
@ -107,7 +119,10 @@ fn parse_line(l: &str) -> Option<BookItem> {
if let Some(i) = line[start_delimitor..].find(')') { if let Some(i) = line[start_delimitor..].find(')') {
end_delimitor = start_delimitor + i; end_delimitor = start_delimitor + i;
} }
else { return None } else {
debug!("[*]: ')' not found, this line is not a link. Ignoring...");
return None
}
path = PathBuf::from(line[start_delimitor + 1 .. end_delimitor].to_string()); path = PathBuf::from(line[start_delimitor + 1 .. end_delimitor].to_string());

View File

@ -24,16 +24,18 @@ pub struct HtmlHandlebars;
impl Renderer for HtmlHandlebars { impl Renderer for HtmlHandlebars {
fn render(&self, book: BookItems, config: &BookConfig) -> Result<(), Box<Error>> { fn render(&self, book: BookItems, config: &BookConfig) -> Result<(), Box<Error>> {
debug!("[fn]: render");
let mut handlebars = Handlebars::new(); let mut handlebars = Handlebars::new();
// Load template // Load template
let t = theme::get_index_hbs(); let t = theme::get_index_hbs();
// Register template // Register template
debug!("[*]: Register handlebars template");
try!(handlebars.register_template_string("index", t.to_owned())); try!(handlebars.register_template_string("index", t.to_owned()));
// Register helpers // Register helpers
debug!("[*]: Register handlebars helpers");
handlebars.register_helper("toc", Box::new(RenderToc)); handlebars.register_helper("toc", Box::new(RenderToc));
handlebars.register_helper("previous", Box::new(hbs_navigation_helper::previous)); handlebars.register_helper("previous", Box::new(hbs_navigation_helper::previous));
handlebars.register_helper("next", Box::new(hbs_navigation_helper::next)); handlebars.register_helper("next", Box::new(hbs_navigation_helper::next));
@ -41,6 +43,7 @@ impl Renderer for HtmlHandlebars {
let mut data = try!(make_data(book.clone(), config)); let mut data = try!(make_data(book.clone(), config));
// Check if dest directory exists // Check if dest directory exists
debug!("[*]: Check if destination directory exists");
match utils::path::create_path(config.dest()) { match utils::path::create_path(config.dest()) {
Err(_) => return Err(Box::new(io::Error::new(io::ErrorKind::Other, "Unexcpected error when constructing destination path"))), Err(_) => return Err(Box::new(io::Error::new(io::ErrorKind::Other, "Unexcpected error when constructing destination path"))),
_ => {}, _ => {},
@ -54,9 +57,11 @@ impl Renderer for HtmlHandlebars {
let path = config.src().join(&item.path); let path = config.src().join(&item.path);
debug!("[*]: Opening file: {:?}", path);
let mut f = try!(File::open(&path)); let mut f = try!(File::open(&path));
let mut content: String = String::new(); let mut content: String = String::new();
debug!("[*]: Reading file");
try!(f.read_to_string(&mut content)); try!(f.read_to_string(&mut content));
// Render markdown using the pulldown-cmark crate // Render markdown using the pulldown-cmark crate
@ -75,25 +80,26 @@ impl Renderer for HtmlHandlebars {
data.insert("path_to_root".to_string(), utils::path::path_to_root(&item.path).to_json()); data.insert("path_to_root".to_string(), utils::path::path_to_root(&item.path).to_json());
// Rendere the handlebars template with the data // Rendere the handlebars template with the data
debug!("[*]: Render template");
let rendered = try!(handlebars.render("index", &data)); let rendered = try!(handlebars.render("index", &data));
println!("Write file..."); debug!("[*] Write to file");
// Write to file // Write to file
let mut file = try!(create_file(config.dest(), &item.path)); let mut file = try!(create_file(config.dest(), &item.path));
try!(file.write_all(&rendered.into_bytes())); try!(file.write_all(&rendered.into_bytes()));
// Create an index.html from the first element in SUMMARY.md // Create an index.html from the first element in SUMMARY.md
if index { if index {
print!( debug!("[*] index.html");
"Creating index.html from {:?}",
config.dest().join(&item.path.with_extension("html"))
);
try!(fs::copy( try!(fs::copy(
config.dest().join(&item.path.with_extension("html")), config.dest().join(&item.path.with_extension("html")),
config.dest().join("index.html") config.dest().join("index.html")
)); ));
println!("");
println!(
"[*] Creating index.html from {:?} ✓",
config.dest().join(&item.path.with_extension("html"))
);
index = false; index = false;
} }
} }
@ -101,6 +107,7 @@ impl Renderer for HtmlHandlebars {
// Copy static files (js, css, images, ...) // Copy static files (js, css, images, ...)
debug!("[*] Copy static files");
// JavaScript // JavaScript
let mut js_file = try!(File::create(config.dest().join("book.js"))); let mut js_file = try!(File::create(config.dest().join("book.js")));
try!(js_file.write_all(theme::get_js())); try!(js_file.write_all(theme::get_js()));
@ -118,18 +125,19 @@ impl HtmlHandlebars {
HtmlHandlebars HtmlHandlebars
} }
fn _load_template(&self, path: &Path) -> Result<String, Box<Error>> { /*fn _load_template(&self, path: &Path) -> Result<String, Box<Error>> {
let mut file = try!(File::open(path)); let mut file = try!(File::open(path));
let mut s = String::new(); let mut s = String::new();
try!(file.read_to_string(&mut s)); try!(file.read_to_string(&mut s));
Ok(s) Ok(s)
} }*/
} }
fn create_file(working_directory: &Path, path: &Path) -> Result<File, Box<Error>> { fn create_file(working_directory: &Path, path: &Path) -> Result<File, Box<Error>> {
println!("[fn]: create_file"); debug!("[fn]: create_file");
debug!("[*]: extract filename");
// Extract filename // Extract filename
let mut file_name; let mut file_name;
if let Some(name) = path.file_stem() { if let Some(name) = path.file_stem() {
@ -160,31 +168,33 @@ fn create_file(working_directory: &Path, path: &Path) -> Result<File, Box<Error>
match metadata(&constructed_path) { match metadata(&constructed_path) {
// Any way to combine the Err and first Ok branch ?? // Any way to combine the Err and first Ok branch ??
Err(_) => { Err(_) => {
debug!("[*]: Create {:?}", constructed_path);
try!(fs::create_dir(&constructed_path)) try!(fs::create_dir(&constructed_path))
}, },
Ok(f) => { Ok(f) => {
if !f.is_dir() { if !f.is_dir() {
debug!("[*]: Create {:?}", constructed_path);
try!(fs::create_dir(&constructed_path)) try!(fs::create_dir(&constructed_path))
} else { } else {
println!("[*]: {:?} --> exists", constructed_path); debug!("[*]: Directory exists: {:?}", constructed_path);
continue continue
} }
}, },
} }
} }
debug!("[*]: Create {:?}", constructed_path.join(&file_name));
print!("Create file: {:?}", constructed_path.join(&file_name));
let file = try!(File::create( let file = try!(File::create(
constructed_path.join(&file_name) constructed_path.join(&file_name)
)); ));
println!(""); println!("[*] Create file: {:?}", constructed_path.join(&file_name));
Ok(file) Ok(file)
} }
fn make_data(book: BookItems, config: &BookConfig) -> Result<BTreeMap<String,Json>, Box<Error>> { fn make_data(book: BookItems, config: &BookConfig) -> Result<BTreeMap<String,Json>, Box<Error>> {
debug!("[fn]: make_data");
let mut data = BTreeMap::new(); let mut data = BTreeMap::new();
data.insert("language".to_string(), "en".to_json()); data.insert("language".to_string(), "en".to_json());
@ -203,6 +213,7 @@ fn make_data(book: BookItems, config: &BookConfig) -> Result<BTreeMap<String,Jso
data.insert("chapters".to_string(), chapters.to_json()); data.insert("chapters".to_string(), chapters.to_json());
debug!("[*]: JSON constructed");
Ok(data) Ok(data)
} }

View File

@ -3,13 +3,16 @@ use std::error::Error;
use std::fs::{self, metadata}; use std::fs::{self, metadata};
pub fn path_to_root(path: &Path) -> String { pub fn path_to_root(path: &Path) -> String {
debug!("[fn]: path_to_root");
// Remove filename and add "../" for every directory // Remove filename and add "../" for every directory
path.to_path_buf().parent().expect("") path.to_path_buf().parent().expect("")
.components().fold(String::new(), |mut s, c| { .components().fold(String::new(), |mut s, c| {
match c { match c {
Component::Normal(_) => s.push_str("../"), Component::Normal(_) => s.push_str("../"),
_ => {} _ => {
debug!("[*]: Other path component... {:?}", c);
}
} }
s s
}) })
@ -17,8 +20,7 @@ pub fn path_to_root(path: &Path) -> String {
pub fn create_path(path: &Path) -> Result<(), Box<Error>> { pub fn create_path(path: &Path) -> Result<(), Box<Error>> {
debug!("[fn]: create_path");
println!("[fn]: create_path");
// Create directories if they do not exist // Create directories if they do not exist
let mut constructed_path = PathBuf::new(); let mut constructed_path = PathBuf::new();
@ -28,32 +30,38 @@ pub fn create_path(path: &Path) -> Result<(), Box<Error>> {
let mut dir; let mut dir;
match component { match component {
Component::Normal(_) => { dir = PathBuf::from(component.as_os_str()); }, Component::Normal(_) => { dir = PathBuf::from(component.as_os_str()); },
Component::RootDir => { constructed_path.push("/"); continue }, Component::RootDir => {
debug!("[*]: Root directory");
// This doesn't look very compatible with Windows...
constructed_path.push("/");
continue
},
_ => continue, _ => continue,
} }
constructed_path.push(&dir); constructed_path.push(&dir);
debug!("[*]: {:?}", constructed_path);
// Check if path exists // Check if path exists
match metadata(&constructed_path) { match metadata(&constructed_path) {
// Any way to combine the Err and first Ok branch ?? // Any way to combine the Err and first Ok branch ??
Err(_) => { Err(_) => {
try!(fs::create_dir(&constructed_path)); try!(fs::create_dir(&constructed_path));
println!("[*]: Directory created {:?}", constructed_path); debug!("[*]: Directory created {:?}", constructed_path);
}, },
Ok(f) => { Ok(f) => {
if !f.is_dir() { if !f.is_dir() {
try!(fs::create_dir(&constructed_path)); try!(fs::create_dir(&constructed_path));
println!("[*]: Directory created {:?}", constructed_path); debug!("[*]: Directory created {:?}", constructed_path);
} else { } else {
println!("[*]: Directory exists {:?}", constructed_path); debug!("[*]: Directory exists {:?}", constructed_path);
continue continue
} }
}, },
} }
} }
println!("[*]: Constructed path: {:?}", constructed_path); debug!("[*]: Constructed path: {:?}", constructed_path);
Ok(()) Ok(())
} }