138 lines
5.2 KiB
Rust
138 lines
5.2 KiB
Rust
use std::path::Path;
|
|
use std::collections::{VecDeque, BTreeMap};
|
|
|
|
use serde_json;
|
|
use handlebars::{Handlebars, HelperDef, RenderError, RenderContext, Helper};
|
|
use pulldown_cmark::{Parser, html, Event, Tag};
|
|
|
|
// Handlebars helper to construct TOC
|
|
#[derive(Clone, Copy)]
|
|
pub struct RenderToc;
|
|
|
|
impl HelperDef for RenderToc {
|
|
fn call(&self, _h: &Helper, _: &Handlebars, rc: &mut RenderContext) -> Result<(), RenderError> {
|
|
|
|
// get value from context data
|
|
// rc.get_path() is current json parent path, you should always use it like this
|
|
// param is the key of value you want to display
|
|
let chapters = rc.context().navigate(rc.get_path(), &VecDeque::new(), "chapters").to_owned();
|
|
let current = rc.context().navigate(rc.get_path(), &VecDeque::new(), "path").to_string().replace("\"", "");
|
|
try!(rc.writer.write_all("<ul class=\"chapter\">".as_bytes()));
|
|
|
|
// Decode json format
|
|
let decoded: Vec<BTreeMap<String, String>> = serde_json::from_str(&chapters.to_string()).unwrap();
|
|
|
|
let mut current_level = 1;
|
|
|
|
for item in decoded {
|
|
|
|
// Spacer
|
|
if item.get("spacer").is_some() {
|
|
try!(rc.writer.write_all("<li class=\"spacer\"></li>".as_bytes()));
|
|
continue;
|
|
}
|
|
|
|
let level = if let Some(s) = item.get("section") {
|
|
s.matches(".").count()
|
|
} else {
|
|
1
|
|
};
|
|
|
|
if level > current_level {
|
|
while level > current_level {
|
|
try!(rc.writer.write_all("<li>".as_bytes()));
|
|
try!(rc.writer.write_all("<ul class=\"section\">".as_bytes()));
|
|
current_level += 1;
|
|
}
|
|
try!(rc.writer.write_all("<li>".as_bytes()));
|
|
} else if level < current_level {
|
|
while level < current_level {
|
|
try!(rc.writer.write_all("</ul>".as_bytes()));
|
|
try!(rc.writer.write_all("</li>".as_bytes()));
|
|
current_level -= 1;
|
|
}
|
|
try!(rc.writer.write_all("<li>".as_bytes()));
|
|
} else {
|
|
try!(rc.writer.write_all("<li".as_bytes()));
|
|
if item.get("section").is_none() {
|
|
try!(rc.writer.write_all(" class=\"affix\"".as_bytes()));
|
|
}
|
|
try!(rc.writer.write_all(">".as_bytes()));
|
|
}
|
|
|
|
// Link
|
|
let path_exists = if let Some(path) = item.get("path") {
|
|
if !path.is_empty() {
|
|
try!(rc.writer.write_all("<a href=\"".as_bytes()));
|
|
|
|
// Add link
|
|
try!(rc.writer.write_all(Path::new(item.get("path")
|
|
.expect("Error: path should be Some(_)"))
|
|
.with_extension("html")
|
|
.to_str()
|
|
.unwrap()
|
|
// Hack for windows who tends to use `\` as separator instead of `/`
|
|
.replace("\\", "/")
|
|
.as_bytes()));
|
|
|
|
try!(rc.writer.write_all("\"".as_bytes()));
|
|
|
|
if path == ¤t {
|
|
try!(rc.writer.write_all(" class=\"active\"".as_bytes()));
|
|
}
|
|
|
|
try!(rc.writer.write_all(">".as_bytes()));
|
|
true
|
|
} else {
|
|
false
|
|
}
|
|
} else {
|
|
false
|
|
};
|
|
|
|
// Section does not necessarily exist
|
|
if let Some(section) = item.get("section") {
|
|
try!(rc.writer.write_all("<strong>".as_bytes()));
|
|
try!(rc.writer.write_all(section.as_bytes()));
|
|
try!(rc.writer.write_all("</strong> ".as_bytes()));
|
|
}
|
|
|
|
if let Some(name) = item.get("name") {
|
|
// Render only inline code blocks
|
|
|
|
// filter all events that are not inline code blocks
|
|
let parser = Parser::new(name).filter(|event| {
|
|
match *event {
|
|
Event::Start(Tag::Code) |
|
|
Event::End(Tag::Code) |
|
|
Event::InlineHtml(_) |
|
|
Event::Text(_) => true,
|
|
_ => false,
|
|
}
|
|
});
|
|
|
|
// render markdown to html
|
|
let mut markdown_parsed_name = String::with_capacity(name.len() * 3 / 2);
|
|
html::push_html(&mut markdown_parsed_name, parser);
|
|
|
|
// write to the handlebars template
|
|
try!(rc.writer.write_all(markdown_parsed_name.as_bytes()));
|
|
}
|
|
|
|
if path_exists {
|
|
try!(rc.writer.write_all("</a>".as_bytes()));
|
|
}
|
|
|
|
try!(rc.writer.write_all("</li>".as_bytes()));
|
|
|
|
}
|
|
while current_level > 1 {
|
|
try!(rc.writer.write_all("</ul>".as_bytes()));
|
|
try!(rc.writer.write_all("</li>".as_bytes()));
|
|
current_level -= 1;
|
|
}
|
|
|
|
try!(rc.writer.write_all("</ul>".as_bytes()));
|
|
Ok(())
|
|
}
|
|
}
|