From 56dd74b65f099b07a88a14fc5ac4d91da992d790 Mon Sep 17 00:00:00 2001
From: Joscha <joscha@plugh.de>
Date: Sat, 5 Aug 2023 00:05:34 +0200
Subject: [PATCH] Improve verbosity levels

---
 src/main.rs | 46 +++++++++++++++++++++++++++++++---------------
 1 file changed, 31 insertions(+), 15 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 4fc373d..e5fab2d 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -13,8 +13,10 @@ use directories::ProjectDirs;
 use sqlx::SqlitePool;
 use state::AppState;
 use tokio::{select, signal::unix::SignalKind};
-use tracing::{debug, info};
-use tracing_subscriber::filter::LevelFilter;
+use tracing::{debug, info, Level};
+use tracing_subscriber::{
+    filter::Targets, prelude::__tracing_subscriber_SubscriberExt, util::SubscriberInitExt,
+};
 
 use crate::config::Config;
 
@@ -31,23 +33,37 @@ struct Args {
     /// Path to the config file.
     #[arg(long, short)]
     config: Option<PathBuf>,
-    /// Enable more verbose output
-    #[arg(long, short)]
-    verbose: bool,
+    /// Enable increasingly more verbose output
+    #[arg(long, short, action = clap::ArgAction::Count)]
+    verbose: u8,
 }
 
-fn set_up_logging(verbose: bool) {
-    if verbose {
-        tracing_subscriber::fmt()
-            .with_max_level(LevelFilter::TRACE)
-            .pretty()
-            .init();
-    } else {
-        tracing_subscriber::fmt()
-            .with_max_level(LevelFilter::INFO)
+fn set_up_logging(verbose: u8) {
+    let filter = Targets::new()
+        .with_default(Level::TRACE)
+        .with_target("sqlx", Level::INFO);
+    match verbose {
+        0 => tracing_subscriber::fmt()
+            .with_max_level(Level::INFO)
             .without_time()
             .with_target(false)
-            .init();
+            .init(),
+        1 => tracing_subscriber::fmt()
+            .with_max_level(Level::TRACE)
+            .with_target(false)
+            .finish()
+            .with(filter)
+            .init(),
+        2 => tracing_subscriber::fmt()
+            .with_max_level(Level::TRACE)
+            .pretty()
+            .finish()
+            .with(filter)
+            .init(),
+        _ => tracing_subscriber::fmt()
+            .with_max_level(Level::TRACE)
+            .pretty()
+            .init(),
     }
 }