Revision control
Copy as Markdown
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
use std::fs;
use anyhow::{anyhow, Result};
use camino::Utf8PathBuf;
use clap::Parser;
pub mod component_config;
pub mod config;
mod main_dashboard;
pub mod metrics;
pub mod schema;
mod sql;
mod team_config;
pub mod util;
#[derive(Parser, Debug)]
#[clap(name = "generate-rust-dashboard")]
struct Cli {
/// Your team name
team_name: String,
/// Directory to write JSON files to
output_dir: Utf8PathBuf,
}
fn main() -> Result<()> {
let cli = Cli::parse();
let config = team_config::all_dashboards()
.into_iter()
.find(|d| {
d.team_name
.to_ascii_lowercase()
.contains(&cli.team_name.to_ascii_lowercase())
})
.ok_or_else(|| anyhow!("Dashboard not found: {}", cli.team_name))?;
let mut main_dashboard_builder = main_dashboard::start_dashboard(&config);
let mut extra_dashboards: Vec<schema::Dashboard> = vec![];
if config.component_errors {
metrics::rust_component_errors::add_to_dashboard(&mut main_dashboard_builder, &config)?;
extra_dashboards.push(metrics::rust_component_errors::extra_dashboard(&config)?);
}
if config.sync_metrics {
metrics::sync::add_to_main_dashboard(&mut main_dashboard_builder, &config)?;
extra_dashboards.push(metrics::sync::extra_dashboard(&config)?);
}
for metric in config.main_dashboard_metrics.iter() {
metric.add_to_dashboard(&mut main_dashboard_builder, &config)?;
}
for extra_dash_config in config.extra_dashboards.iter() {
let mut builder = schema::DashboardBuilder::new(
extra_dash_config.name,
util::slug(extra_dash_config.name),
);
for metric in extra_dash_config.metrics.iter() {
metric.add_to_dashboard(&mut builder, &config)?;
}
extra_dashboards.push(builder.dashboard);
}
if !cli.output_dir.exists() {
fs::create_dir_all(&cli.output_dir)?;
}
println!();
println!("Generating Dashboards:");
let dashboards = std::iter::once(main_dashboard_builder.dashboard).chain(extra_dashboards);
for dashboard in dashboards {
let mut value = serde_json::to_value(&dashboard)?;
value.sort_all_objects();
let content = serde_json::to_string_pretty(&value)?;
let path = cli.output_dir.join(format!("{}.json", dashboard.uid));
fs::write(&path, content)?;
println!("{path}");
}
println!();
println!("* Create/navigate to a folder for your team");
println!("* Click New -> Import and import each generated JSON file");
Ok(())
}