dagger-rs/src/cli_generate.rs

use clap::{Arg, ArgMatches};

use crate::{code_generation::CodeGeneration, config::Config, engine::Engine, session::Session};

#[allow(dead_code)]
pub struct GenerateCommand;

#[allow(dead_code)]
impl GenerateCommand {
    pub fn new_cmd() -> clap::Command {
        clap::Command::new("generate").arg(Arg::new("output").long("output"))
    }

    pub fn exec(arg_matches: &ArgMatches) -> eyre::Result<()> {
        let cfg = Config::default();
        let (conn, _proc) = Engine::new().start(&cfg)?;
        let session = Session::new();
        let req = session.start(&cfg, &conn)?;
        let schema = session.schema(req)?;
        let code = CodeGeneration::generate(&schema)?;

        if let Some(output) = arg_matches.get_one::<String>("output") {
            // TODO: Write to file
        } else {
            println!("{}", code);
        }

        Ok(())
    }
}