flowstdlib 0.36.0

The standard library of functions for 'flow' programs
use std::io;
use std::path::Path;
use std::path::PathBuf;
use std::process::{Command, Stdio};

use glob::{glob_with, MatchOptions};
use simpath::{FileType, FoundType, Simpath};

use lib_path::check_flow_lib_path;

mod lib_path;

// Build script to compile the flowstdlib WASM files and generate manifest - using flowc
fn main() -> io::Result<()> {
    let flowc = get_flowc()?;

    let mut command = Command::new(flowc);
    // Options for flowc: -g for debug symbols, -z to dump graphs, -l for a library build
    let command_args = vec!["-v", "info", "-g", "-z", "-l", env!("CARGO_MANIFEST_DIR")];

    command
        .args(command_args)
        .stdin(Stdio::inherit())
        .stdout(Stdio::inherit())
        .stderr(Stdio::inherit())
        .spawn()
        .unwrap();

    check_flow_lib_path();

    generate_svgs(env!("CARGO_MANIFEST_DIR"))?;

    Ok(())
}

fn get_flowc() -> io::Result<PathBuf> {
    let root = Path::new(env!("CARGO_MANIFEST_DIR")).parent().unwrap();

    let dev = root.join("target/debug/flowc");
    if dev.exists() {
        return Ok(dev);
    }

    let dev = root.join("target/release/flowc");
    if dev.exists() {
        return Ok(dev);
    }

    if let Ok(FoundType::File(flowc)) = Simpath::new("PATH").find_type("flowc", FileType::File) {
        return Ok(flowc);
    }

    Err(io::Error::new(
        io::ErrorKind::Other,
        "`flowc` could not be found in `$PATH` or `target/`",
    ))
}

/*
   Generate SVG files from the .dot files created by flowc
*/
fn generate_svgs(root_dir: &str) -> io::Result<()> {
    if let Ok(FoundType::File(dot)) = Simpath::new("PATH").find_type("dot", FileType::File) {
        println!("Generating .dot.svg files from .dot files, using 'dot' command from $PATH");

        let mut dot_command = Command::new(dot);
        let options = MatchOptions {
            case_sensitive: false,
            ..Default::default()
        };
        let pattern = format!("{}/**/*.dot", root_dir);

        for path in glob_with(&pattern, &options).unwrap().flatten() {
            let dot_child = dot_command
                .args(vec!["-Tsvg", "-O", path.to_str().unwrap()])
                .stdin(Stdio::inherit())
                .stdout(Stdio::inherit())
                .stderr(Stdio::inherit())
                .spawn()?;

            let dot_output = dot_child.wait_with_output()?;
            match dot_output.status.code() {
                Some(0) => {}
                Some(_) => {
                    return Err(io::Error::new(
                        io::ErrorKind::Other,
                        "`dot` exited with non-zero status code",
                    ))
                }
                _ => {}
            }
        }
    } else {
        println!("Could not find 'dot' command in $PATH so SVG generation skipped");
    }

    Ok(())
}