dora/binaries/cli/src/main.rs

use clap::Parser;
use eyre::Context;
use std::{io::Write, path::PathBuf};
use tempfile::NamedTempFile;

mod build;
mod check;
mod graph;

#[derive(Debug, clap::Parser)]
#[clap(version)]
struct Args {
    #[clap(subcommand)]
    command: Command,
}

#[derive(Debug, clap::Subcommand)]
enum Command {
    Check {
        dataflow: PathBuf,
        runtime_path: PathBuf,
    },
    Graph {
        dataflow: PathBuf,
        #[clap(long, action)]
        mermaid: bool,
        #[clap(long, action)]
        open: bool,
    },
    Build {
        dataflow: PathBuf,
    },
    Templates,
    Dashboard,
    Start,
    Stop,
    Logs,
    Metrics,
    Stats,
    List,
    Get,
    Upgrade,
}

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

    match args.command {
        Command::Check {
            dataflow,
            runtime_path,
        } => check::check(&dataflow, &runtime_path)?,
        Command::Graph {
            dataflow,
            mermaid,
            open,
        } => {
            if mermaid {
                let visualized = graph::visualize_as_mermaid(&dataflow)?;
                println!("{visualized}");
                println!(
                    "Paste the above output on https://mermaid.live/ or in a \
                    ```mermaid code block on GitHub to display it."
                );
            } else {
                let html = graph::visualize_as_html(&dataflow)?;
                let mut file = NamedTempFile::new().context("failed to create temp file")?;
                file.as_file_mut().write_all(html.as_bytes())?;

                let path = file.path().to_owned();
                file.keep()?;

                println!(
                    "View graph by opening the following in your browser:\n  file://{}",
                    path.display()
                );

                if open {
                    webbrowser::open(path.as_os_str().to_str().unwrap())?;
                }
            }
        }
        Command::Build { dataflow } => {
            build::build(&dataflow)?;
        }
        Command::Templates => todo!(),
        Command::Dashboard => todo!(),
        Command::Start => todo!(),
        Command::Stop => todo!(),
        Command::Logs => todo!(),
        Command::Metrics => todo!(),
        Command::Stats => todo!(),
        Command::List => todo!(),
        Command::Get => todo!(),
        Command::Upgrade => todo!(),
    }

    Ok(())
}