simple_logger update and configurable DEBUG flag

carlossless · Jan 12, 2024 · a48503f · a48503f
1 parent bb43762
commit a48503f
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 5 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -23,7 +23,7 @@ features = ["linux-static-libusb"]
 
 [dependencies.log]
 version = "0.4"
-features = ["max_level_debug", "release_max_level_info"]
+features = ["max_level_debug"]
 
 [dependencies.simple_logger]
 version = "4.3"

diff --git a/src/main.rs b/src/main.rs
@@ -1,5 +1,5 @@
 use std::{
-    fs,
+    env, fs,
     io::{self, Read},
     process::ExitCode,
 };
@@ -66,8 +66,26 @@ fn cli() -> Command {
         );
 }
 
+fn get_log_level() -> log::LevelFilter {
+    return if let Ok(debug) = env::var("DEBUG") {
+        if debug == "1" {
+            log::LevelFilter::Debug
+        } else {
+            log::LevelFilter::Info
+        }
+    } else {
+        #[cfg(debug_assertions)]
+        return log::LevelFilter::Debug;
+        #[cfg(not(debug_assertions))]
+        log::LevelFilter::Info
+    };
+}
+
 fn err_main() -> Result<(), CLIError> {
-    SimpleLogger::new().init().unwrap();
+    SimpleLogger::new()
+        .with_level(get_log_level())
+        .init()
+        .unwrap();
 
     let matches = cli().get_matches();