schematools-cli 1.0.2

use clap::{Parser, Subcommand};

use schematools::Client;

use schematools::storage::SchemaStorage;
use schematools::{discovery::Discovery, schema::Schema};

use crate::error::Error;

#[cfg(feature = "codegen")]
use super::codegen;
use super::process;
use super::registry;
use super::validate;
use super::GetSchemaCommand;

use std::fmt::Display;
use std::time::Instant;

#[derive(Clone, Debug, Parser)]
pub struct OutputOpts {
    #[clap(flatten)]
    output: crate::commands::Output,
}

#[derive(Clone, Debug, Subcommand)]
pub enum ChainCommandOption {
    #[cfg(feature = "codegen")]
    Codegen(codegen::Opts),
    Process(process::Opts),
    Validate(validate::Opts),
    Registry(registry::Opts),
    Output(OutputOpts),
}

impl Display for OutputOpts {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match &self.output.to_file {
            Some(path) => write!(f, "to {} in {} format", path, self.output.output),
            None => write!(f, "to stdout in {} format", self.output.output),
        }
    }
}

impl Display for ChainCommandOption {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match &self {
            #[cfg(feature = "codegen")]
            Self::Codegen(p) => write!(f, "codegen: {p}"),
            Self::Process(p) => write!(f, "process: {p}"),
            Self::Validate(p) => write!(f, "validate: {p}"),
            Self::Registry(p) => write!(f, "registry: {p}"),
            Self::Output(p) => write!(f, "output: {p}"),
        }
    }
}

fn parse_command(cmd: &str) -> Result<ChainCommandOption, Error> {
    let parts = schematools::tools::ArgumentsExtractor::new(cmd).collect::<Vec<String>>();

    match parts.first().unwrap().as_ref() {
        "registry" => Ok(ChainCommandOption::Registry(
            registry::Opts::try_parse_from(parts)
                .map_err(|e| Error::ChainWrongParameters("registry".to_string(), e))?,
        )),
        #[cfg(feature = "codegen")]
        "codegen" => Ok(ChainCommandOption::Codegen(
            codegen::Opts::try_parse_from(parts)
                .map_err(|e| Error::ChainWrongParameters("codegen".to_string(), e))?,
        )),
        "process" => Ok(ChainCommandOption::Process(
            process::Opts::try_parse_from(parts)
                .map_err(|e| Error::ChainWrongParameters("process".to_string(), e))?,
        )),
        "validate" => Ok(ChainCommandOption::Validate(
            validate::Opts::try_parse_from(parts)
                .map_err(|e| Error::ChainWrongParameters("validate".to_string(), e))?,
        )),
        "output" => Ok(ChainCommandOption::Output(
            OutputOpts::try_parse_from(parts)
                .map_err(|e| Error::ChainWrongParameters("output".to_string(), e))?,
        )),
        s => Err(Error::ChainUnknownCommand(s.to_string())),
    }
}

#[derive(Debug, Parser)]
pub struct Opts {
    #[clap(short = 'c', value_parser = parse_command, number_of_values = 1)]
    commands: Vec<ChainCommandOption>,

    #[clap(flatten)]
    verbose: crate::commands::Verbosity,
}

pub fn execute(opts: Opts, client: &Client) -> Result<(), Error> {
    opts.verbose.start()?;

    let mut schemas: Vec<(Schema, Vec<ChainCommandOption>)> = vec![];
    let mut discovery = Discovery::default();

    let timing_load = Instant::now();
    for command in opts.commands {
        let schema = match &command {
            #[cfg(feature = "codegen")]
            ChainCommandOption::Codegen(c) => c.get_schema(client),
            ChainCommandOption::Process(c) => c.get_schema(client),
            ChainCommandOption::Validate(c) => c.get_schema(client),
            ChainCommandOption::Registry(c) => {
                c.run(&mut discovery)?;

                Err(Error::SchemaNotApplicable)
            }
            ChainCommandOption::Output(_) => Err(Error::SchemaNotApplicable),
        };

        match schema {
            Ok(s) => {
                schemas.push((s, vec![]));
                Ok(())
            }
            Err(e) => match e {
                Error::Schematools(schematools::error::Error::SchemaAsReference) => Ok(()),
                Error::SchemaNotApplicable => Ok(()),
                e => Err(e),
            },
        }?;

        if let Some((_, commands)) = schemas.last_mut() {
            commands.push(command);
        }
    }

    log::info!(
        "\x1b[1;4mloading registry and root schemas: {:.2?}\x1b[0m",
        timing_load.elapsed()
    );

    // create resolver
    let timing_resolve = Instant::now();

    let mut storage =
        SchemaStorage::new_multi(&schemas.iter().map(|(s, _)| s).collect::<Vec<_>>(), client);

    log::info!(
        "\x1b[1;4mresolving schema dependencies took: {:.2?}\x1b[0m",
        timing_resolve.elapsed()
    );

    let processing_time = Instant::now();
    for (ref mut current, ref mut actions) in schemas {
        for cmd in actions {
            log::info!("\x1b[1;70mCHAINING: {} {}\x1b[0m", cmd, current.get_url());
            let timing_step = Instant::now();

            match cmd {
                #[cfg(feature = "codegen")]
                ChainCommandOption::Codegen(c) => c.run(current, &discovery, &storage),
                ChainCommandOption::Process(c) => c.run(current, &storage).inspect(|_| {
                    storage
                        .schemas
                        .insert(current.get_url().clone(), current.clone());
                }),
                ChainCommandOption::Validate(v) => v.run(current),
                ChainCommandOption::Output(o) => {
                    o.output.show(current.get_body());
                    Ok(())
                }
                _ => Ok(()),
            }?;

            log::info!(
                "\x1b[1;4m{} took: {:.2?}\x1b[0m",
                cmd,
                timing_step.elapsed()
            );
        }
    }

    log::info!(
        "\x1b[1;4mprocessing and rendering took: {:.2?}, total execution: {:.2?}\x1b[0m",
        processing_time.elapsed(),
        timing_load.elapsed()
    );

    Ok(())
}