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

#![recursion_limit = "512"]

extern crate proc_macro;

mod codegen;
#[cfg(feature = "graphviz")]
mod diagramgen;
mod parser;

use syn::parse_macro_input;

// dot -Tsvg statemachine.gv -o statemachine.svg

#[proc_macro]
pub fn statemachine(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    // Parse the syntax into structures
    let input = parse_macro_input!(input as parser::StateMachine);

    // Validate syntax
    match parser::ParsedStateMachine::new(input) {
        // Generate code and hand the output tokens back to the compiler
        Ok(sm) => {
            #[cfg(feature = "graphviz")]
            {
                use std::io::Write;

                // Generate dot syntax for the statemachine.
                let diagram = diagramgen::generate_diagram(&sm);

                // Start the 'dot' process.
                let mut process = std::process::Command::new("dot")
                    .args(&["-Tsvg", "-o", "statemachine.svg"])
                    .stdin(std::process::Stdio::piped())
                    .spawn()
                    .expect("Failed to execute 'dot'. Are you sure graphviz is installed?");

                // Write the dot syntax string to the 'dot' process stdin.
                process
                    .stdin
                    .as_mut()
                    .map(|s| s.write_all(diagram.as_bytes()));

                // Check the graphviz return status to see if it was successful.
                match process.wait() {
                    Ok(status) => {
                        if !status.success() {
                            panic!("'dot' failed to run. Are you sure graphviz is installed?");
                        }
                    }
                    Err(_) => panic!("'dot' failed to run. Are you sure graphviz is installed?"),
                }
            }

            codegen::generate_code(&sm).into()
        }
        Err(error) => error.to_compile_error().into(),
    }
}