hugr_cli/
mermaid.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
46
//! Render mermaid diagrams.
use std::io::Write;

use clap::Parser;
use clio::Output;
use hugr::HugrView;

/// Dump the standard extensions.
#[derive(Parser, Debug)]
#[clap(version = "1.0", long_about = None)]
#[clap(about = "Render mermaid diagrams..")]
#[group(id = "hugr")]
#[non_exhaustive]
pub struct MermaidArgs {
    /// Common arguments
    #[command(flatten)]
    pub hugr_args: crate::HugrArgs,
    /// Validate package.
    #[arg(
        long,
        help = "Validate before rendering, includes extension inference."
    )]
    pub validate: bool,
    /// Output file '-' for stdout
    #[clap(long, short, value_parser, default_value = "-")]
    output: Output,
}

impl MermaidArgs {
    /// Write the mermaid diagram to the output.
    pub fn run_print(&mut self) -> Result<(), crate::CliError> {
        let hugrs = if self.validate {
            self.hugr_args.validate()?
        } else {
            let extensions = self.hugr_args.extensions()?;
            self.hugr_args
                .get_package_or_hugr(&extensions)?
                .into_hugrs()
        };

        for hugr in hugrs {
            write!(self.output, "{}", hugr.mermaid_string())?;
        }
        Ok(())
    }
}