feat(backend): allow to set log level via config file (#182)

Co-authored-by: ravenclaw900 <50060110+ravenclaw900@users.noreply.github.com>

config.toml

@@ -1,11 +1,18 @@
+# Log level
+# - Options: "off", "error", "warn", "info", "debug", "trace"
+# - Default: "info"
+#log_level = "info"
+
 # TCP network port
 # - Default: 5252
 #port = 5252
+
 # TLS for HTTPS
 # - Default: false
 #tls = false
 #cert = "/path/to/cert"
 #key = "/path/to/key"
+
 # Password protection
 # - Default: false
 #pass = false
@@ -16,6 +23,7 @@
 # Token expiry time in seconds
 # - Default: 3600
 #expiry = 3600
+
 # Other nodes viewable on frontend page
 # - Default: []
 # Please insert the IP addresses/domains and ports of these nodes

src/backend/src/config.rs

@@ -1,6 +1,9 @@
+use std::str::FromStr;
 use toml::Value;
 
 pub struct Config {
+    pub log_level: log::LevelFilter,
+
     pub port: u16,
 
     pub tls: bool,
@@ -32,6 +35,14 @@ pub fn config() -> Config {
     .parse::<Value>()
     .expect("Invalid config file");
 
+    let log_level = log::LevelFilter::from_str(
+        cfg.get("log_level")
+            .unwrap_or(&Value::String("info".to_string()))
+            .as_str()
+            .unwrap(),
+    )
+    .unwrap();
+
     #[allow(clippy::cast_sign_loss)]
     #[allow(clippy::cast_possible_truncation)]
     let port: u16 = cfg
@@ -106,6 +117,7 @@ pub fn config() -> Config {
     }
 
     Config {
+        log_level,
         port,
         tls,
         cert,

src/backend/src/main.rs

@@ -23,7 +23,7 @@ fn main() {
             const DIR: include_dir::Dir = include_dir::include_dir!("dist");
 
             SimpleLogger::new()
-                .with_level(log::LevelFilter::Info)
+                .with_level(CONFIG.log_level)
                 .env()
                 .init()
                 .unwrap();