use anyhow::Result;
use clap::{CommandFactory, Parser};
use clap_complete::generate;
use std::io;
use std::process;
use svccat::cli::{Cli, Commands, AuditFormat, DiffFormat, ExportFormat, GraphFormat, HookKind, ImportSource, OutputFormat, ReportFormat};
use svccat::{
audit, config, diff, discovery, drift, fix, hooks, import, init, lint, manifest, output, ping,
report, serve, since, stats, watch,
};
fn main() {
match run() {
Ok(code) => process::exit(code),
Err(e) => {
eprintln!("error: {e:#}");
process::exit(2);
}
}
}
fn run() -> Result<i32> {
let cli = Cli::parse();
let root = cli.root.unwrap_or_else(|| std::path::PathBuf::from("."));
let cfg = config::SvccatConfig::load(&root)?;
match cli.command {
Commands::Check {
manifest: manifest_path,
format,
fail_on_drift,
ping: do_ping,
ignore: cli_ignore,
team,
since,
fail_on_new_drift,
depth,
baseline,
} => {
let format = if format == OutputFormat::Terminal
&& std::env::var("GITHUB_ACTIONS").as_deref() == Ok("true")
{
OutputFormat::GithubAnnotation
} else {
format
};
let path = manifest_path.unwrap_or_else(|| manifest::find_default(&root));
let full_m = manifest::Manifest::load(&path)?;
let mut m = full_m.clone();
if let Some(ref t) = team {
m.services.retain(|s| {
s.team
.as_deref()
.map(|v| v.eq_ignore_ascii_case(t))
.unwrap_or(false)
});
}
let mut ignore: Vec<String> = cfg.ignore.clone();
ignore.extend(cli_ignore);
let discovered_all = discovery::discover_services_with_opts(&root, &full_m, &ignore, depth);
let in_scope_names: std::collections::HashSet<&str> =
m.services.iter().map(|s| s.name.as_str()).collect();
let other_declared_names: std::collections::HashSet<&str> = full_m
.services
.iter()
.filter(|s| !in_scope_names.contains(s.name.as_str()))
.map(|s| s.name.as_str())
.collect();
let discovered: Vec<_> = discovered_all
.into_iter()
.filter(|d| !other_declared_names.contains(d.name.as_str()))
.collect();
let mut report = drift::analyze(&m, &discovered, &root);
report.manifest = path.display().to_string();
if let Some(ref baseline_path) = baseline {
use std::collections::HashSet;
#[derive(serde::Deserialize)]
struct BaselineFile {
drift: Vec<drift::DriftItem>,
}
let text = std::fs::read_to_string(baseline_path)
.map_err(|e| anyhow::anyhow!("cannot read baseline {}: {e}", baseline_path.display()))?;
let snap: BaselineFile = serde_json::from_str(&text)
.map_err(|e| anyhow::anyhow!("cannot parse baseline JSON: {e}"))?;
let baseline_keys: HashSet<String> = snap.drift.iter()
.map(|d| format!("{:?}|{}|{}", d.kind, d.service, d.detail.as_deref().unwrap_or("")))
.collect();
report.drifts.retain(|d| {
!baseline_keys.contains(&format!(
"{:?}|{}|{}",
d.kind,
d.service,
d.detail.as_deref().unwrap_or("")
))
});
}
let ping_results = if do_ping {
ping::ping_services(&m)
} else {
vec![]
};
if let Some(ref git_ref) = since {
let old_m = since::load_at_ref(&root, &path, git_ref)?;
let mut old_report = drift::analyze(&old_m, &discovered, &root);
old_report.manifest = path.display().to_string();
let new_count = match format {
OutputFormat::Markdown => {
let md = output::markdown::render_since_diff_markdown(
&old_report,
&report,
git_ref,
);
print!("{}", md);
use std::collections::HashSet;
let old_keys: HashSet<String> = old_report
.drifts
.iter()
.map(|d| {
format!(
"{:?}|{}|{}",
d.kind,
d.service,
d.detail.as_deref().unwrap_or("")
)
})
.collect();
report
.drifts
.iter()
.filter(|d| {
let k = format!(
"{:?}|{}|{}",
d.kind,
d.service,
d.detail.as_deref().unwrap_or("")
);
!old_keys.contains(&k)
})
.count()
}
OutputFormat::GithubAnnotation => {
output::github_annotation::render_since_annotations(&old_report, &report)
}
OutputFormat::Junit => {
output::junit::render_since(&old_report, &report, git_ref)
}
_ => {
let (new_count, _) =
output::terminal::render_since_diff(&old_report, &report, git_ref);
new_count
}
};
if fail_on_new_drift && new_count > 0 {
return Ok(1);
}
} else {
match format {
OutputFormat::Terminal => {
output::terminal::render_check(&report, &ping_results)
}
OutputFormat::Compact => {
output::terminal::render_compact(&m, &report);
}
OutputFormat::Json => output::json::render_check(&report, &ping_results)?,
OutputFormat::Sarif => output::sarif::render_check(&report, &ping_results)?,
OutputFormat::Markdown => {
let md = output::markdown::render_check_markdown(&report, &ping_results);
print!("{}", md);
}
OutputFormat::Junit => output::junit::render_check(&report, &ping_results)?,
OutputFormat::GithubAnnotation => {
output::github_annotation::render_check(&report);
}
OutputFormat::Csv => output::csv::render_check(&report),
OutputFormat::Slack => output::slack::render_check(&report)?,
OutputFormat::Teams => output::teams::render_check(&report)?,
}
}
let should_fail = fail_on_drift || cfg.fail_on_drift;
if should_fail && !report.drifts.is_empty() {
Ok(1)
} else {
Ok(0)
}
}
Commands::Graph {
manifest: manifest_path,
format,
team,
} => {
let path = manifest_path.unwrap_or_else(|| manifest::find_default(&root));
let m = manifest::Manifest::load(&path)?;
match format {
GraphFormat::Mermaid => output::mermaid::render_graph_filtered(&m, team.as_deref()),
GraphFormat::Markdown => output::mermaid::render_markdown_table(&m),
GraphFormat::Dot => output::mermaid::render_dot(&m, team.as_deref()),
GraphFormat::Plantuml => output::mermaid::render_plantuml(&m, team.as_deref()),
}
Ok(0)
}
Commands::Export {
manifest: manifest_path,
format,
ignore: cli_ignore,
depth,
since: since_ref,
} => {
let path = manifest_path.unwrap_or_else(|| manifest::find_default(&root));
let mut m = manifest::Manifest::load(&path)?;
let mut ignore: Vec<String> = cfg.ignore.clone();
ignore.extend(cli_ignore);
let discovered = discovery::discover_services_with_opts(&root, &m, &ignore, depth);
let mut report = drift::analyze(&m, &discovered, &root);
report.manifest = path.display().to_string();
if let Some(ref git_ref) = since_ref {
if let Ok(old_m) = since::load_at_ref(&root, &path, git_ref) {
let old_map: std::collections::HashMap<String, &manifest::ServiceEntry> =
old_m.services.iter().map(|s| (s.name.clone(), s)).collect();
m.services.retain(|svc| {
if let Some(old_svc) = old_map.get(&svc.name) {
svc != *old_svc
} else {
true }
});
let discovered2 = discovery::discover_services_with_opts(&root, &m, &ignore, depth);
report = drift::analyze(&m, &discovered2, &root);
report.manifest = path.display().to_string();
}
}
match format {
ExportFormat::Json => output::json::render_export(&m, &report)?,
ExportFormat::Markdown => output::mermaid::render_export_markdown(&m, &report),
ExportFormat::Csv => output::csv::render_export(&m),
}
Ok(0)
}
Commands::Init { output, force } => {
let output_path = output.unwrap_or_else(|| root.join("services.yaml"));
init::run(&root, output_path, force)?;
Ok(0)
}
Commands::Diff { before, after, format } => {
let report = diff::diff_snapshots(&before, &after)?;
match format {
DiffFormat::Terminal => diff::render_diff(&report),
DiffFormat::Markdown => diff::render_diff_markdown(&report),
}
Ok(0)
}
Commands::Watch {
manifest: manifest_path,
fail_on_drift,
team,
ignore: cli_ignore,
depth,
since: watch_since,
notify,
} => {
let path = manifest_path.unwrap_or_else(|| manifest::find_default(&root));
let mut ignore: Vec<String> = cfg.ignore.clone();
ignore.extend(cli_ignore);
let initial_errors = watch::run(&path, &root, &ignore, team.as_deref(), depth, watch_since.as_deref(), notify)?;
let should_fail = fail_on_drift || cfg.fail_on_drift;
if should_fail && initial_errors > 0 {
Ok(1)
} else {
Ok(0)
}
}
Commands::Report {
manifest: manifest_path,
format,
output: output_path,
ignore: cli_ignore,
history,
badge,
} => {
let path = manifest_path.unwrap_or_else(|| manifest::find_default(&root));
let m = manifest::Manifest::load(&path)?;
let mut ignore: Vec<String> = cfg.ignore.clone();
ignore.extend(cli_ignore);
let discovered = discovery::discover_services_with_ignore(&root, &m, &ignore);
let mut drift_report = drift::analyze(&m, &discovered, &root);
drift_report.manifest = path.display().to_string();
if badge {
println!("{}", report::render_badge(&drift_report));
return Ok(0);
}
let content = if let Some(n) = history {
report::render_history_markdown(&root, &path, &discovered, n)?
} else {
match format {
ReportFormat::Markdown => report::render_markdown(&m, &drift_report),
ReportFormat::Html => report::render_html(&m, &drift_report),
}
};
if let Some(out_path) = output_path {
std::fs::write(&out_path, &content)?;
eprintln!("wrote report to {}", out_path.display());
} else {
print!("{}", content);
}
Ok(0)
}
Commands::Lint {
manifest: manifest_path,
} => {
let path = manifest_path.unwrap_or_else(|| manifest::find_default(&root));
let m = manifest::Manifest::load(&path)?;
let result = lint::run(&m);
lint::render(&result);
if result.error_count() > 0 {
Ok(1)
} else {
Ok(0)
}
}
Commands::Import {
from,
output: output_path,
force,
} => {
let out = output_path.unwrap_or_else(|| root.join("services.yaml"));
match from {
ImportSource::Backstage => import::run_backstage(&root, out, force)?,
ImportSource::DockerCompose => import::run_docker_compose(&root, out, force)?,
ImportSource::Openapi => import::run_openapi(&root, out, force)?,
}
Ok(0)
}
Commands::Fix {
manifest: manifest_path,
prune,
dry_run,
ignore: cli_ignore,
depth,
} => {
let path = manifest_path.unwrap_or_else(|| manifest::find_default(&root));
let mut ignore: Vec<String> = cfg.ignore.clone();
ignore.extend(cli_ignore);
fix::run(&path, &root, &ignore, depth, prune, dry_run)?;
Ok(0)
}
Commands::InstallHooks { hook, fail_on_drift } => {
let hook_name = match hook {
HookKind::PreCommit => "pre-commit",
HookKind::PrePush => "pre-push",
};
hooks::install(&root, hook_name, fail_on_drift)?;
Ok(0)
}
Commands::Audit {
manifest: manifest_path,
format,
ping: do_ping,
ignore: cli_ignore,
depth,
} => {
let path = manifest_path.unwrap_or_else(|| manifest::find_default(&root));
let mut ignore: Vec<String> = cfg.ignore.clone();
ignore.extend(cli_ignore);
let (result, lint_result, drift_report, ping_results) =
audit::run(&path, &root, &ignore, depth, do_ping)?;
match format {
AuditFormat::Terminal => audit::render_terminal(&result, &lint_result, &drift_report, &ping_results),
AuditFormat::Json => audit::render_json(&result)?,
}
if result.passed {
Ok(0)
} else {
Ok(1)
}
}
Commands::Stats {
manifest: manifest_path,
} => {
let path = manifest_path.unwrap_or_else(|| manifest::find_default(&root));
let m = manifest::Manifest::load(&path)?;
stats::run(&m);
Ok(0)
}
Commands::Serve {
manifest: _manifest_path,
port,
refresh,
} => {
serve::serve(&root, port, refresh)?;
Ok(0)
}
Commands::Completions { shell } => {
let mut cmd = Cli::command();
generate(shell, &mut cmd, "svccat", &mut io::stdout());
Ok(0)
}
}
}