use crate::{Config, Result}; use clap::{Args, Parser}; use std::fs::File; use std::sync::Arc; use tracing_subscriber::{filter::LevelFilter, Layer, Registry}; /// Simple program to manage projects and ssh hosts using tmux #[derive(Parser, Debug)] #[command(author, version, about)] pub struct Cli { #[command(flatten)] pub verbose: Verbosity, /// Connect to ssh host #[arg(short, long)] pub ssh: Option, #[command(flatten)] pub config: Config, } impl Cli { pub fn as_layer(&self) -> Result + Send + Sync>>> { let mut layers = Vec::new(); let fmt_layer = tracing_subscriber::fmt::layer() .pretty() .with_filter(self.verbose.into_filter()) .boxed(); layers.push(fmt_layer); if self.config.enable_logging { let file = File::create(&self.config.log_file)?; let log_layer = tracing_subscriber::fmt::layer() .with_writer(Arc::new(file)) .boxed(); layers.push(log_layer); }; Ok(layers) } } #[derive(Debug, Default, Args)] pub struct Verbosity { /// Print additional information per occurrence #[arg(short, long, action = clap::ArgAction::Count, conflicts_with = "quiet")] pub verbose: u8, /// Suppress all output #[arg(short, long, global = true, conflicts_with = "verbose")] pub quiet: bool, } impl Verbosity { pub fn into_filter(&self) -> LevelFilter { self.into() } } impl From<&Verbosity> for LevelFilter { fn from(value: &Verbosity) -> Self { match value.verbose + 1 - u8::from(value.quiet) { 0 => LevelFilter::OFF, 1 => LevelFilter::ERROR, 2 => LevelFilter::WARN, 3 => LevelFilter::INFO, 4 => LevelFilter::DEBUG, _ => LevelFilter::TRACE, } } } #[cfg(test)] mod tests { #[test] fn test_start() { assert_eq!(1, 1); } }