Skip to content

Commit ecdb411

Browse files
ImUrXehuss
authored andcommitted
fix mistakes
1 parent a4e2061 commit ecdb411

File tree

1 file changed

+10
-4
lines changed

1 file changed

+10
-4
lines changed

src/renderer/html_handlebars/hbs_renderer.rs

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -813,7 +813,9 @@ fn build_header_links(html: &str) -> String {
813813

814814
insert_link_into_header(
815815
level,
816-
caps.get(2).map(|x| x.as_str()).unwrap_or(&caps[4]),
816+
&caps[4],
817+
caps.get(2).map(|x| x.as_str().to_string()),
818+
caps.get(3).map(|x| x.as_str().to_string()),
817819
&mut id_counter,
818820
)
819821
})
@@ -825,15 +827,19 @@ fn build_header_links(html: &str) -> String {
825827
fn insert_link_into_header(
826828
level: usize,
827829
content: &str,
830+
id: Option<String>,
831+
classes: Option<String>,
828832
id_counter: &mut HashMap<String, usize>,
829833
) -> String {
830-
let id = utils::unique_id_from_content(content, id_counter);
834+
let id = id.unwrap_or_else(|| utils::unique_id_from_content(content, id_counter));
835+
let classes = classes.unwrap_or("".to_string());
831836

832837
format!(
833-
r##"<h{level} id="{id}"><a class="header" href="#{id}">{text}</a></h{level}>"##,
838+
r##"<h{level} id="{id}" classes="{classes}"><a class="header" href="#{id}">{text}</a></h{level}>"##,
834839
level = level,
835840
id = id,
836-
text = content
841+
text = content,
842+
classes = classes
837843
)
838844
}
839845

0 commit comments

Comments
 (0)