bcl-ddog 0.2.0

A fast CLI for querying Datadog logs.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

mod cli;
mod dashboards;
mod logs;
mod monitors;
mod output;
mod time_spec;

use std::process::ExitCode;

use clap::Parser;
use cli::{Cli, Command};
use tracing_subscriber::EnvFilter;

#[tokio::main]
async fn main() -> ExitCode {
    let cli = Cli::parse();
    init_tracing(cli.verbose);

    let exit = match run(cli).await {
        Ok(()) => 0,
        Err(err) => {
            eprintln!("error: {err:#}");
            err.downcast_ref::<dd_api::ApiError>()
                .map(|e| e.exit_code())
                .unwrap_or(1)
        }
    };
    ExitCode::from(exit as u8)
}

fn init_tracing(verbose: u8) {
    let level = match verbose {
        0 => "warn",
        1 => "info",
        2 => "debug",
        _ => "trace",
    };
    let filter = EnvFilter::try_from_default_env()
        .unwrap_or_else(|_| EnvFilter::new(format!("ddog={level},dd_api={level},dd_cli={level}")));
    let _ = tracing_subscriber::fmt()
        .with_env_filter(filter)
        .with_writer(std::io::stderr)
        .try_init();
}

async fn run(cli: Cli) -> anyhow::Result<()> {
    let cfg = dd_config::resolve(dd_config::Overrides {
        api_key: cli.api_key.clone(),
        app_key: cli.app_key.clone(),
        site: cli.site.clone(),
        profile: cli.profile.clone(),
        config_path: cli.config.clone(),
    })?;

    let client = dd_api::ClientBuilder::new(&cfg.api_key, &cfg.app_key)
        .site(&cfg.site)
        .build()?;

    let ctx = cli::Ctx {
        client,
        output: output::resolve_mode(cli.output.as_deref()),
    };

    match cli.command {
        Command::Logs(logs) => logs::dispatch(ctx, logs).await,
        Command::Dashboards(d) => dashboards::dispatch(ctx, d).await,
        Command::Monitors(m) => monitors::dispatch(ctx, m).await,
    }
}