diff --git a/src/config/mod.rs b/src/config/mod.rs index a06e163..c8d2d96 100644 --- a/src/config/mod.rs +++ b/src/config/mod.rs @@ -14,6 +14,7 @@ pub struct Config { pub extension: String, pub headings: Headings, pub links_classes: String, + pub file_tree_title_classes: String, } pub struct Headings { @@ -141,6 +142,11 @@ pub fn parse(yaml_str: &String) -> Result { _ => String::from(""), }; + let file_tree_title_classes = match config_yaml.get(ystr!("file_tree_title_classes")) { + Some(Yaml::String(input)) => input.clone(), + _ => String::from(""), + }; + Ok(Config { input: input_folder.clone(), @@ -149,6 +155,7 @@ pub fn parse(yaml_str: &String) -> Result { extension: file_extension, headings: headings_yaml, links_classes: link_classes, + file_tree_title_classes, }) } diff --git a/src/generator/heading.rs b/src/generator/heading.rs index 022bc2e..de54c8f 100644 --- a/src/generator/heading.rs +++ b/src/generator/heading.rs @@ -28,7 +28,7 @@ impl Printable for Heading { format!( "{}", - self.depth, extra_classes, html_fragment_text, html_fragment_text, text, self.depth + self.depth, html_fragment_text, extra_classes, html_fragment_text, text, self.depth ) } else { format!("{}", self.depth, text, self.depth) diff --git a/src/pages/mod.rs b/src/pages/mod.rs index 38ddf2f..fa420c8 100644 --- a/src/pages/mod.rs +++ b/src/pages/mod.rs @@ -122,9 +122,10 @@ pub fn generate_pages_html(file_tree: &Node, current_path: &Path, config: &Confi } else { format!( "
  • -
    {}
    +
    {}
      {}
  • ", + &config.file_tree_title_classes, folder.name, sub_nodes_html.join("") )