Add --verbose flag

Author: rikhuijzer

Cargo.toml

@@ -6,11 +6,13 @@ authors = ["Rik Huijzer"]
 license = "MIT"
 
 [dependencies]
+anyhow = "1"
 clap = { version = "4.5.29", features = ["derive"] }
+futures-util = "0.3.31"
 tokio = { version = "1", features = ["macros", "rt-multi-thread"] }
+tracing-subscriber = "0.3"
+tracing = "0.1"
 transformrs = "0.6"
-anyhow = "1"
-futures-util = "0.3.31"
 
 [dev-dependencies]
 assert_cmd = "2"

src/main.rs

@@ -4,6 +4,7 @@ mod tts;
 use chat::ChatArgs;
 use clap::Parser;
 use std::io::Read;
+use tracing::subscriber::SetGlobalDefaultError;
 use transformrs::Key;
 use tts::TextToSpeechArgs;
 
@@ -30,6 +31,12 @@ enum Commands {
 struct Arguments {
     #[command(subcommand)]
     command: Commands,
+    /// Verbose output.
+    ///
+    /// The output of the logs is printed to stderr because the output is
+    /// printed to stdout.
+    #[arg(long)]
+    verbose: bool,
 }
 
 pub enum Task {
@@ -46,9 +53,25 @@ fn find_single_key(keys: transformrs::Keys) -> Key {
     keys[0].clone()
 }
 
+/// Initialize logging with the given level.
+fn init_subscriber(level: tracing::Level) -> Result<(), SetGlobalDefaultError> {
+    let subscriber = tracing_subscriber::FmtSubscriber::builder()
+        .with_max_level(level)
+        .with_writer(std::io::stderr)
+        .without_time()
+        .with_target(false)
+        .finish();
+    tracing::subscriber::set_global_default(subscriber)
+}
+
 #[tokio::main]
 async fn main() {
     let args = Arguments::parse();
+    if args.verbose {
+        init_subscriber(tracing::Level::DEBUG).unwrap();
+    } else {
+        init_subscriber(tracing::Level::INFO).unwrap();
+    }
 
     let mut input = String::new();
     std::io::stdin().read_to_string(&mut input).unwrap();