jer/mdbook-toc
1
Fork 0
Code Actions Activity

Start normalizing the levels based on the first header's level

Browse source
This commit is contained in:
Jan-Erik Rediger 2020-06-24 11:33:46 +02:00
parent 5706d2e6bf
commit 2d5b40214c
1 changed files with 44 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
src/lib.rs
View file

@ -39,7 +39,13 @@ fn build_toc<'a>(toc: &[(u32, String)]) -> String {
// Otherwise the markdown render will escape nested levels.
//
// This is a rough approximation only.
let mut last_lower = 0;
let mut toc_iter = toc.iter().peekable();
// Start from the level of the first header.
let mut last_lower = match toc_iter.peek() {
Some((lvl, _)) => *lvl,
None => 0
};
let toc = toc.iter().map(|(lvl, name)| {
let lvl = *lvl;
let lvl = if last_lower + 1 == lvl {
@ -347,4 +353,41 @@ mod test {
assert_eq!(expected, add_toc(content).unwrap());
}
#[test]
fn multi_header_linear_regression_3() {
let content = r#"# Heading
<!-- toc -->
## Level 1.1
### Level 1.1.1
### Level 1.1.2
## Level 1.2
### Level 1.2.1
text"#;
let expected = r#"# Heading
* [Level 1.1](#level-11)
* [Level 1.1.1](#level-111)
* [Level 1.1.2](#level-112)
* [Level 1.2](#level-12)
* [Level 1.2.1](#level-121)
## Level 1.1
### Level 1.1.1
### Level 1.1.2
## Level 1.2
### Level 1.2.1
text"#;
assert_eq!(expected, add_toc(content).unwrap());
}
}