fraiseql_cli/

runner.rs

//! CLI command dispatch and helper utilities.

use std::{env, process, str::FromStr};

use clap::{CommandFactory, Parser};
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};

use crate::cli::{
    Cli, Commands, FederationCommands, GenerateClientCommands, IntrospectCommands, MigrateCommands,
    PerfCommands, PerfExploreCommands, SchemaCommands, ValidateCommands,
};

/// Run the FraiseQL CLI. Called from both the `fraiseql-cli` and `fraiseql` binary entry points.
#[allow(clippy::cognitive_complexity)] // Reason: CLI dispatch with many subcommand branches
pub async fn run() {
    use crate::{commands, output};

    if let Some(code) = handle_introspection_flags() {
        process::exit(code);
    }

    let cli = Cli::parse();

    init_logging(cli.verbose, cli.debug);

    let json_output = cli.json;
    let debug_output = cli.debug;

    let result = match cli.command {
        Commands::Compile {
            input,
            types,
            schema_dir,
            type_files,
            query_files,
            mutation_files,
            output,
            check,
            skip_hash,
            database,
            emit_ddl,
            check_migrations,
        } => {
            commands::compile::run(
                &input,
                types.as_deref(),
                schema_dir.as_deref(),
                type_files,
                query_files,
                mutation_files,
                &output,
                check,
                database.as_deref(),
                emit_ddl.as_deref(),
                check_migrations,
                skip_hash,
            )
            .await
        },

        Commands::Extract {
            input,
            language,
            recursive,
            output,
        } => commands::extract::run(&input, language.as_deref(), recursive, &output),

        Commands::Explain { query } => match commands::explain::run(&query) {
            Ok(result) => {
                println!("{}", output::OutputFormatter::new(cli.json, cli.quiet).format(&result));
                Ok(())
            },
            Err(e) => Err(e),
        },

        Commands::Cost { query } => match commands::cost::run(&query) {
            Ok(result) => {
                println!("{}", output::OutputFormatter::new(cli.json, cli.quiet).format(&result));
                Ok(())
            },
            Err(e) => Err(e),
        },

        Commands::Analyze { schema } => match commands::analyze::run(&schema) {
            Ok(result) => {
                println!("{}", output::OutputFormatter::new(cli.json, cli.quiet).format(&result));
                Ok(())
            },
            Err(e) => Err(e),
        },

        Commands::DependencyGraph { schema, format } => {
            match commands::dependency_graph::GraphFormat::from_str(&format) {
                Ok(fmt) => match commands::dependency_graph::run(&schema, fmt) {
                    Ok(result) => {
                        println!(
                            "{}",
                            output::OutputFormatter::new(cli.json, cli.quiet).format(&result)
                        );
                        Ok(())
                    },
                    Err(e) => Err(e),
                },
                Err(e) => Err(anyhow::anyhow!(e)),
            }
        },

        Commands::Lint {
            schema,
            federation,
            cost,
            cache,
            auth,
            compilation,
            fail_on_critical,
            fail_on_warning,
        } => {
            let opts = commands::lint::LintOptions {
                fail_on_critical,
                fail_on_warning,
                filter: commands::lint::LintCategoryFilter {
                    federation,
                    cost,
                    cache,
                    auth,
                    compilation,
                },
            };
            match commands::lint::run(&schema, opts) {
                Ok(result) => {
                    println!(
                        "{}",
                        output::OutputFormatter::new(cli.json, cli.quiet).format(&result)
                    );
                    // H21: `--fail-on-critical`/`--fail-on-warning` must affect the
                    // exit code, not just the printed status.
                    enforce_exit_code(&result);
                    Ok(())
                },
                Err(e) => Err(e),
            }
        },

        Commands::Federation { command } => match command {
            FederationCommands::Graph { schema, format } => {
                match commands::federation::graph::GraphFormat::from_str(&format) {
                    Ok(fmt) => match commands::federation::graph::run(&schema, fmt) {
                        Ok(result) => {
                            println!(
                                "{}",
                                output::OutputFormatter::new(cli.json, cli.quiet).format(&result)
                            );
                            Ok(())
                        },
                        Err(e) => Err(e),
                    },
                    Err(e) => Err(anyhow::anyhow!(e)),
                }
            },
            FederationCommands::Check {
                schema,
                against,
                json,
            } => {
                match commands::federation::check::run(
                    &schema,
                    against.as_deref(),
                    json || cli.json,
                ) {
                    Ok(result) => {
                        // `check::run` already prints the JSON result when either the
                        // subcommand `--json` or the global `--json` is set (it receives
                        // `json || cli.json`). Only format/print here when neither is set,
                        // otherwise the result is emitted twice on global `--json`.
                        if !(json || cli.json) {
                            println!(
                                "{}",
                                output::OutputFormatter::new(cli.json, cli.quiet).format(&result)
                            );
                        }
                        // H22: a composition failure must fail the process, not
                        // print the error and exit 0.
                        enforce_exit_code(&result);
                        Ok(())
                    },
                    Err(e) => Err(e),
                }
            },
        },

        Commands::GenerateViews {
            schema,
            entity,
            view,
            refresh_strategy,
            output,
            include_composition_views,
            include_monitoring,
            validate,
            gen_verbose,
        } => match commands::generate_views::RefreshStrategy::parse(&refresh_strategy) {
            Ok(refresh_strat) => {
                let config = commands::generate_views::GenerateViewsConfig {
                    schema_path: schema,
                    entity,
                    view,
                    refresh_strategy: refresh_strat,
                    output,
                    include_composition_views,
                    include_monitoring,
                    validate_only: validate,
                    verbose: cli.verbose || gen_verbose,
                };

                let formatter = output::OutputFormatter::new(cli.json, cli.quiet);
                commands::generate_views::run(config, &formatter)
            },
            Err(e) => Err(anyhow::anyhow!(e)),
        },

        Commands::GenerateCaptureTriggers {
            schema,
            output,
            no_function,
        } => {
            let config = commands::generate_capture_triggers::GenerateCaptureTriggersConfig {
                schema_path: schema,
                output,
                include_function: !no_function,
            };
            let formatter = output::OutputFormatter::new(cli.json, cli.quiet);
            commands::generate_capture_triggers::run(config, &formatter)
        },

        Commands::Validate {
            command,
            input,
            check_cycles,
            check_unused,
            strict,
            types,
            against_db,
        } => match command {
            Some(ValidateCommands::Facts { schema, database }) => {
                let formatter = output::OutputFormatter::new(cli.json, cli.quiet);
                commands::validate_facts::run(std::path::Path::new(&schema), &database, &formatter)
                    .await
            },
            None => match (against_db, input) {
                (Some(db_url), Some(input)) => {
                    commands::validate::run_against_db(&input, &db_url, cli.json).await
                },
                (Some(_), None) => Err(anyhow::anyhow!(
                    "INPUT (schema.compiled.json) is required with --against-db"
                )),
                (None, Some(input)) => {
                    let opts = commands::validate::ValidateOptions {
                        check_cycles,
                        check_unused,
                        strict,
                        filter_types: types,
                    };
                    match commands::validate::run_with_options(&input, opts) {
                        Ok(result) => {
                            println!(
                                "{}",
                                output::OutputFormatter::new(cli.json, cli.quiet).format(&result)
                            );
                            if result.status == "validation-failed" {
                                Err(anyhow::anyhow!("Validation failed"))
                            } else {
                                Ok(())
                            }
                        },
                        Err(e) => Err(e),
                    }
                },
                (None, None) => Err(anyhow::anyhow!("INPUT required when no subcommand provided")),
            },
        },

        Commands::Introspect { command } => match command {
            IntrospectCommands::Facts { database, format } => {
                match commands::introspect_facts::OutputFormat::parse(&format) {
                    Ok(fmt) => {
                        let formatter = output::OutputFormatter::new(cli.json, cli.quiet);
                        commands::introspect_facts::run(&database, fmt, &formatter).await
                    },
                    Err(e) => Err(anyhow::anyhow!(e)),
                }
            },
        },

        Commands::Generate {
            input,
            language,
            output,
        } => match commands::init::Language::from_str(&language) {
            Ok(lang) => commands::generate::run(&input, lang, output.as_deref()),
            Err(e) => Err(anyhow::anyhow!(e)),
        },

        Commands::GenerateClient { language } => match language {
            GenerateClientCommands::Typescript { schema, out, force } => {
                commands::generate_client::run(
                    commands::generate_client::ClientLanguage::TypeScript,
                    schema.as_deref(),
                    &out,
                    force,
                )
            },
        },

        Commands::Init {
            project_name,
            language,
            database,
            size,
            no_git,
        } => {
            match (
                commands::init::Language::from_str(&language),
                commands::init::Database::from_str(&database),
                commands::init::ProjectSize::from_str(&size),
            ) {
                (Ok(lang), Ok(db), Ok(sz)) => {
                    let config = commands::init::InitConfig {
                        project_name,
                        language: lang,
                        database: db,
                        size: sz,
                        no_git,
                    };
                    commands::init::run(&config)
                },
                (Err(e), _, _) | (_, Err(e), _) | (_, _, Err(e)) => Err(anyhow::anyhow!(e)),
            }
        },

        Commands::Migrate { command } => {
            let formatter = output::OutputFormatter::new(cli.json, cli.quiet);
            match command {
                MigrateCommands::Up { database, dir } => {
                    let db_url = commands::migrate::resolve_database_url(database.as_deref());
                    match db_url {
                        Ok(url) => {
                            let mig_dir = commands::migrate::resolve_migration_dir(dir.as_deref());
                            let action = commands::migrate::MigrateAction::Up {
                                database_url: url,
                                dir:          mig_dir,
                            };
                            commands::migrate::run(&action, &formatter)
                        },
                        Err(e) => Err(e),
                    }
                },
                MigrateCommands::Down {
                    database,
                    dir,
                    steps,
                } => {
                    let db_url = commands::migrate::resolve_database_url(database.as_deref());
                    match db_url {
                        Ok(url) => {
                            let mig_dir = commands::migrate::resolve_migration_dir(dir.as_deref());
                            let action = commands::migrate::MigrateAction::Down {
                                database_url: url,
                                dir: mig_dir,
                                steps,
                            };
                            commands::migrate::run(&action, &formatter)
                        },
                        Err(e) => Err(e),
                    }
                },
                MigrateCommands::Status { database, dir } => {
                    let db_url = commands::migrate::resolve_database_url(database.as_deref());
                    match db_url {
                        Ok(url) => {
                            let mig_dir = commands::migrate::resolve_migration_dir(dir.as_deref());
                            let action = commands::migrate::MigrateAction::Status {
                                database_url: url,
                                dir:          mig_dir,
                            };
                            commands::migrate::run(&action, &formatter)
                        },
                        Err(e) => Err(e),
                    }
                },
                MigrateCommands::Create { name, dir } => {
                    let mig_dir = commands::migrate::resolve_migration_dir(dir.as_deref());
                    let action = commands::migrate::MigrateAction::Create { name, dir: mig_dir };
                    commands::migrate::run(&action, &formatter)
                },
                MigrateCommands::Generate { name, dir } => {
                    let mig_dir = commands::migrate::resolve_migration_dir(dir.as_deref());
                    let action = commands::migrate::MigrateAction::Generate { name, dir: mig_dir };
                    commands::migrate::run(&action, &formatter)
                },
                MigrateCommands::Validate { dir } => {
                    let mig_dir = commands::migrate::resolve_migration_dir(dir.as_deref());
                    let action = commands::migrate::MigrateAction::Validate { dir: mig_dir };
                    commands::migrate::run(&action, &formatter)
                },
                MigrateCommands::Preflight { dir } => {
                    let mig_dir = commands::migrate::resolve_migration_dir(dir.as_deref());
                    let action = commands::migrate::MigrateAction::Preflight { dir: mig_dir };
                    commands::migrate::run(&action, &formatter)
                },
            }
        },

        Commands::Sbom { format, output } => match commands::sbom::SbomFormat::from_str(&format) {
            Ok(fmt) => commands::sbom::run(fmt, output.as_deref()),
            Err(e) => Err(anyhow::anyhow!(e)),
        },

        #[cfg(feature = "run-server")]
        Commands::Run {
            input,
            database,
            port,
            bind,
            watch,
            introspection,
        } => {
            // Box::pin: `run` dispatches across up to 4 adapter init futures,
            // pushing the combined future past clippy's `large_futures` 16-KiB
            // stack threshold. Heap-allocating once per CLI invocation is fine.
            Box::pin(commands::run::run(
                input.as_deref(),
                database,
                port,
                bind,
                watch,
                introspection,
            ))
            .await
        },

        Commands::Watch {
            input,
            output,
            reload_url,
            admin_token,
            database,
        } => {
            commands::watch::run(
                &input,
                &output,
                reload_url.as_deref(),
                admin_token.as_deref(),
                database.as_deref(),
            )
            .await
        },

        Commands::ValidateDocuments { manifest } => {
            let formatter = output::OutputFormatter::new(cli.json, cli.quiet);
            match commands::validate_documents::run(&manifest, &formatter) {
                Ok(true) => Ok(()),
                Ok(false) => {
                    process::exit(2);
                },
                Err(e) => Err(e),
            }
        },

        Commands::Setup { database, dry_run } => {
            let formatter = output::OutputFormatter::new(cli.json, cli.quiet);
            commands::setup::run(database.as_deref(), dry_run, &formatter).await
        },

        Commands::Schema { command } => match command {
            SchemaCommands::Metadata { server, token } => {
                commands::schema::metadata::run(&server, token.as_deref()).await
            },
        },

        Commands::Doctor {
            config,
            schema,
            db_url,
            against_db,
            schemas,
            json: json_flag,
        } => {
            let all_passed = commands::doctor::run_with_db_checks(
                &config,
                &schema,
                db_url.as_deref(),
                against_db.as_deref(),
                &schemas,
                json_flag,
            )
            .await;
            if all_passed {
                Ok(())
            } else {
                process::exit(1);
            }
        },

        Commands::Perf { command } => match command {
            PerfCommands::RegressionScan {
                database,
                recent_days,
                baseline_days,
                min_samples,
                threshold_pct,
                min_delta_ms,
                object_type,
                fail_on_regression,
            } => {
                let args = commands::perf::RegressionScanArgs {
                    database,
                    recent_days,
                    baseline_days,
                    min_samples,
                    threshold_pct,
                    min_delta_ms,
                    object_type,
                    fail_on_regression,
                    json: cli.json,
                    quiet: cli.quiet,
                };
                match commands::perf::run_regression_scan(args).await {
                    Ok(true) => Ok(()),
                    Ok(false) => process::exit(1),
                    Err(e) => Err(e),
                }
            },
            PerfCommands::Explore { command } => match command {
                PerfExploreCommands::Slowest {
                    database,
                    days,
                    object_type,
                    limit,
                } => {
                    commands::perf::run_explore_slowest(
                        database,
                        days,
                        object_type,
                        limit,
                        cli.json,
                        cli.quiet,
                    )
                    .await
                },
                PerfExploreCommands::NullRate {
                    database,
                    days,
                    object_type,
                } => {
                    commands::perf::run_explore_null_rate(
                        database,
                        days,
                        object_type,
                        cli.json,
                        cli.quiet,
                    )
                    .await
                },
                PerfExploreCommands::Summary {
                    database,
                    days,
                    object_type,
                } => {
                    commands::perf::run_explore_summary(
                        database,
                        days,
                        object_type,
                        cli.json,
                        cli.quiet,
                    )
                    .await
                },
            },
        },
    };

    if let Err(e) = result {
        let debug_info = if debug_output {
            Some(format!("{e:?}"))
        } else {
            None
        };
        eprintln!("{}", format_cli_error(&e.to_string(), debug_info.as_deref(), json_output, 1));
        process::exit(1);
    }
}

/// Enforce the CLI exit-code contract for a command that returns a [`CommandResult`].
///
/// Commands that print a [`CommandResult`] must also reflect its status in the
/// process exit code. The contract (mirroring [`crate::output::get_exit_codes`]):
///
/// - `"success"` → return normally (the process exits 0)
/// - `"validation-failed"` → exit **2** (the command ran, but the checked artifact failed its gate
///   — e.g. a lint quality gate or a federation composition error)
/// - any other status (e.g. `"error"`) → exit **1** (operational failure)
///
/// Operational errors surfaced as `Err(_)` are handled separately at the end of
/// [`run`] and also exit 1.
fn enforce_exit_code(result: &crate::output::CommandResult) {
    match result.status.as_str() {
        "success" => {},
        "validation-failed" => process::exit(2),
        _ => process::exit(1),
    }
}

/// Format a CLI error as either plain text or a JSON object for machine-readable output.
///
/// When `json` is `true`, produces a JSON object:
/// ```json
/// { "error": { "message": "...", "code": 1 } }
/// ```
/// When `json` is `false`, produces a human-readable string:
/// ```text
/// Error: <message>
/// ```
/// If `debug_info` is provided and `json` is `false`, a debug section is appended.
pub(crate) fn format_cli_error(
    message: &str,
    debug_info: Option<&str>,
    json: bool,
    code: i32,
) -> String {
    if json {
        serde_json::json!({
            "error": {
                "message": message,
                "code": code
            }
        })
        .to_string()
    } else {
        let mut out = format!("Error: {message}");
        if let Some(debug) = debug_info {
            out.push_str("\n\nDebug info:\n");
            out.push_str(debug);
        }
        out
    }
}

fn init_logging(verbose: bool, debug: bool) {
    let filter = if debug {
        "fraiseql=debug,fraiseql_core=debug"
    } else if verbose {
        "fraiseql=info,fraiseql_core=info"
    } else {
        "fraiseql=warn,fraiseql_core=warn"
    };

    tracing_subscriber::registry()
        .with(
            tracing_subscriber::EnvFilter::try_from_default_env().unwrap_or_else(|_| filter.into()),
        )
        // Diagnostics go to stderr, never stdout — so a command whose stdout is
        // the payload (e.g. `generate-capture-triggers … | psql`) is never
        // corrupted by a stray WARN/INFO line.
        .with(tracing_subscriber::fmt::layer().with_writer(std::io::stderr))
        .init();
}

/// Serialize a value to a JSON `Value`, printing to stderr and exiting with code 2 on failure.
fn serialize_or_exit<T: serde::Serialize>(value: &T, context: &str) -> serde_json::Value {
    serde_json::to_value(value).unwrap_or_else(|e| {
        eprintln!("fraiseql: failed to serialize {context}: {e}");
        std::process::exit(2);
    })
}

/// Serialize a value to pretty-printed JSON, printing to stderr and exiting with code 2 on failure.
fn pretty_or_exit<T: serde::Serialize>(value: &T, context: &str) -> String {
    serde_json::to_string_pretty(value).unwrap_or_else(|e| {
        eprintln!("fraiseql: failed to format {context}: {e}");
        std::process::exit(2);
    })
}

fn handle_introspection_flags() -> Option<i32> {
    let args: Vec<String> = env::args().collect();

    if args.iter().any(|a| a == "--help-json") {
        let cmd = Cli::command();
        let version = env!("CARGO_PKG_VERSION");
        let help = crate::introspection::extract_cli_help(&cmd, version);
        let result =
            crate::output::CommandResult::success("help", serialize_or_exit(&help, "help output"));
        println!("{}", pretty_or_exit(&result, "command result"));
        return Some(0);
    }

    if args.iter().any(|a| a == "--list-commands") {
        let cmd = Cli::command();
        let commands = crate::introspection::list_commands(&cmd);
        let result = crate::output::CommandResult::success(
            "list-commands",
            serialize_or_exit(&commands, "command list"),
        );
        println!("{}", pretty_or_exit(&result, "command result"));
        return Some(0);
    }

    let idx = args.iter().position(|a| a == "--show-output-schema")?;
    let available = crate::output_schemas::list_schema_commands().join(", ");

    let Some(cmd_name) = args.get(idx + 1) else {
        let result = crate::output::CommandResult::error(
            "show-output-schema",
            &format!("Missing command name. Available: {available}"),
            "MISSING_ARGUMENT",
        );
        println!("{}", pretty_or_exit(&result, "command result"));
        return Some(1);
    };

    if let Some(schema) = crate::output_schemas::get_output_schema(cmd_name) {
        let result = crate::output::CommandResult::success(
            "show-output-schema",
            serialize_or_exit(&schema, "output schema"),
        );
        println!("{}", pretty_or_exit(&result, "command result"));
        return Some(0);
    }

    let result = crate::output::CommandResult::error(
        "show-output-schema",
        &format!("Unknown command: {cmd_name}. Available: {available}"),
        "UNKNOWN_COMMAND",
    );
    println!("{}", pretty_or_exit(&result, "command result"));
    Some(1)
}