actions
/
mdBook
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

inject main

This commit is contained in:
steveklabnik 2017-03-06 13:27:25 -05:00
parent 79f00eeea3
commit c5f9625feb
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/renderer/html_handlebars/hbs_renderer.rs
View File

@ -294,14 +294,23 @@ fn fix_code_blocks(html: String) -> String {
}
fn add_playpen_pre(html: String) -> String {
let regex = Regex::new(r##"((?s)<code[^>]?class="([^"]+)".*?>.*?</code>)"##).unwrap();
let regex = Regex::new(r##"((?s)<code[^>]?class="([^"]+)".*?>(.*?)</code>)"##).unwrap();
regex.replace_all(&html, |caps: &Captures| {
let text = &caps[1];
let classes = &caps[2];
let code = &caps[3];
if classes.contains("language-rust") && !classes.contains("ignore") {
// wrap the contents in an external pre block
format!("<pre class=\"playpen\">{}</pre>", text)
if text.contains("fn main") {
format!("<pre class=\"playpen\">{}</pre>", text)
} else {
// we need to inject our own main
format!("<pre class=\"playpen\"><code class=\"{}\">#fn main() {{
{}
#}}</code></pre>", classes, code)
}
} else {
// not language-rust, so no-op
format!("{}", text)