feat: only debug mode can use RUST_LOG env control the log level

wendajiang · wendajiang · commit 0c59b3a7362d · 2022-12-14T16:24:36.000+08:00
diff --git a/src/cli.rs b/src/cli.rs
@@ -8,6 +8,7 @@ use crate::{
     flag::{Debug, Flag},
 };
 use clap::{crate_name, crate_version};
+use log::LevelFilter;
 
 /// This should be called before calling any cli method or printing any output.
 pub fn reset_signal_pipe_handler() {
@@ -41,10 +42,11 @@ pub async fn main() -> Result<(), Error> {
         .arg_required_else_help(true)
         .get_matches();
 
-    if m.contains_id("debug") {
+    if m.get_flag("debug") {
         Debug::handler()?;
     } else {
-        env_logger::Builder::from_env(env_logger::Env::new().default_filter_or("info"))
+        env_logger::Builder::new()
+            .filter_level(LevelFilter::Info)
             .format_timestamp(None)
             .init();
     }
diff --git a/src/flag.rs b/src/flag.rs
@@ -7,7 +7,7 @@
 //!     -V, --version    Prints version information
 //! ```
 use crate::err::Error;
-use clap::Arg;
+use clap::{Arg, ArgAction};
 use env_logger::Env;
 
 /// Abstract flag trait
@@ -25,10 +25,11 @@ impl Flag for Debug {
             .short('d')
             .long("debug")
             .help("debug mode")
+            .action(ArgAction::SetTrue)
     }
 
     fn handler() -> Result<(), Error> {
-        env_logger::Builder::from_env(Env::default().default_filter_or("leetcode")).init();
+        env_logger::Builder::from_env(Env::default().default_filter_or("debug")).init();
 
         Ok(())
     }
