src/main.rs


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

use std::{
    fs::{read, write},
    path::PathBuf,
    process::Command,
};

use clap::Parser;
use tempfile::tempdir;

use roadmap::from_yaml;

fn main() -> anyhow::Result<()> {
    let args = Args::parse();

    let data = read(&args.filename)?;
    let yaml = String::from_utf8(data)?;
    let roadmap = from_yaml(&yaml)?;
    let graph = roadmap.format_as_dot(args.label_width)?;

    let tmp = tempdir()?;
    let filename = tmp.path().join("dot");
    write(&filename, graph)?;

    let output = Command::new("dot").arg("-Tsvg").arg(filename).output()?;
    if output.status.success() {
        let dot = String::from_utf8(output.stdout)?;
        println!("{}", dot);
    } else {
        let stderr = String::from_utf8_lossy(&output.stderr);
        eprintln!("ERROR: dot failed:\n{stderr}");
        std::process::exit(output.status.code().unwrap());
    }

    Ok(())
}

#[derive(Parser)]
struct Args {
    /// Width of labels in the graph in characters.
    #[clap(short, long, default_value = "20")]
    label_width: usize,

    /// Input filename.
    filename: PathBuf,
}