#[derive(Debug, Clone, clap::Args)]
pub struct CommonArgs {
#[arg(short = 'c', long = "config")]
pub config: Option<String>,
#[arg(
short = 'l',
long = "log-level",
env = "LOG_LEVEL",
default_value = "info"
)]
pub log_level: String,
#[arg(long = "log-format", env = "LOG_FORMAT", default_value = "auto")]
pub log_format: String,
#[arg(
long = "metrics-addr",
env = "METRICS_ADDR",
default_value = "0.0.0.0:9090"
)]
pub metrics_addr: String,
#[arg(short = 'v', long, conflicts_with = "quiet")]
pub verbose: bool,
#[arg(short = 'q', long, conflicts_with = "verbose")]
pub quiet: bool,
}
impl CommonArgs {
#[must_use]
pub fn effective_log_level(&self) -> &str {
if self.verbose {
"debug"
} else if self.quiet {
"error"
} else {
&self.log_level
}
}
#[cfg(feature = "logger")]
pub fn to_logger_options(&self) -> Result<crate::logger::LoggerOptions, super::CliError> {
use std::str::FromStr;
let level: tracing::Level =
self.effective_log_level()
.to_uppercase()
.parse()
.map_err(|_| {
super::CliError::InvalidArgument(format!(
"invalid log level: {}",
self.effective_log_level()
))
})?;
let format = crate::logger::LogFormat::from_str(&self.log_format)
.map_err(|e| super::CliError::InvalidArgument(format!("invalid log format: {e}")))?;
Ok(crate::logger::LoggerOptions {
level,
format,
..Default::default()
})
}
#[cfg(feature = "config")]
#[must_use]
pub fn to_config_options(&self, env_prefix: &str) -> crate::config::ConfigOptions {
let mut opts = crate::config::ConfigOptions {
env_prefix: env_prefix.to_string(),
..Default::default()
};
if let Some(ref path) = self.config {
opts.config_paths.push(path.into());
}
opts
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_effective_log_level_default() {
let args = CommonArgs {
config: None,
log_level: "info".to_string(),
log_format: "auto".to_string(),
metrics_addr: "0.0.0.0:9090".to_string(),
verbose: false,
quiet: false,
};
assert_eq!(args.effective_log_level(), "info");
}
#[test]
fn test_effective_log_level_verbose() {
let args = CommonArgs {
config: None,
log_level: "info".to_string(),
log_format: "auto".to_string(),
metrics_addr: "0.0.0.0:9090".to_string(),
verbose: true,
quiet: false,
};
assert_eq!(args.effective_log_level(), "debug");
}
#[test]
fn test_effective_log_level_quiet() {
let args = CommonArgs {
config: None,
log_level: "info".to_string(),
log_format: "auto".to_string(),
metrics_addr: "0.0.0.0:9090".to_string(),
verbose: false,
quiet: true,
};
assert_eq!(args.effective_log_level(), "error");
}
#[test]
fn test_effective_log_level_custom() {
let args = CommonArgs {
config: None,
log_level: "warn".to_string(),
log_format: "auto".to_string(),
metrics_addr: "0.0.0.0:9090".to_string(),
verbose: false,
quiet: false,
};
assert_eq!(args.effective_log_level(), "warn");
}
}