mktoc 5.0.0

Generate Table of Contents from Markdown files
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59

use clap::Parser;
use mktoc::Config;

#[derive(Parser, Debug)]
#[command(author, version, about, long_about = None)]
struct Args {
    #[arg(default_value = "README.md")]
    file: String,

    /// If set will output to stdout instead of replacing content in file
    #[arg(long, short)]
    stdout: bool,

    /// Minimum heading level
    #[arg(long, short = 'm', default_value_t = 1, env = "MKTOC_MIN_DEPTH")]
    min_depth: i32,

    /// Maximum heading level
    #[arg(long, short = 'M', default_value_t = 6, env = "MKTOC_MAX_DEPTH")]
    max_depth: i32,

    /// Wrap ToC in details html element
    #[arg(long, short = 'w', default_value_t = false, env = "MKTOC_WRAP_IN_DETAILS")]
    wrap_in_details: bool,
}

fn handle_write(args: Args, new_toc: String) {
    if !args.stdout {
        let res_write = std::fs::write(&args.file, new_toc.as_bytes());
        match res_write {
            Ok(_r) => {
                std::process::exit(0);
            }
            Err(e) => {
                eprintln!("Failed to write file. Error kind: {:?}", e.kind());
                std::process::exit(1);
            }
        }
    } else {
        println!("{}", new_toc);
    }
}

fn main() {
    let args = Args::parse();
    let config = Config{min_depth: args.min_depth, max_depth: args.max_depth, wrap_in_details: args.wrap_in_details, ..Default::default()};
    let res = mktoc::make_toc(&args.file, config);

    match res {
        Ok(new_toc) => {
            handle_write(args, new_toc);
            std::process::exit(0);
        }
        Err(e) => {
            eprintln!("Error: {}", e);
            std::process::exit(1);
        }
    };
}