initial comms

common/src/logger.rs

@@ -0,0 +1,52 @@
+use anyhow::Result;
+use fern::colors::{Color, ColoredLevelConfig};
+use log::debug;
+use std::fs::create_dir_all;
+use std::str::FromStr;
+use std::{env, thread};
+
+pub fn setup_logger(package_name: &'static str) -> Result<()> {
+    let log_file = env::var("LOG_FILE").or_else(|_| {
+        create_dir_all("logs/")?;
+
+        anyhow::Ok(format!(
+            "logs/{}_{}.log",
+            package_name,
+            chrono::Local::now().format("%Y%m%d_%H%M%S")
+        ))
+    })?;
+    let log_level = match env::var("LOG_LEVEL") {
+        Ok(log_level) => log::LevelFilter::from_str(&log_level).unwrap_or(log::LevelFilter::Info),
+        Err(_) => log::LevelFilter::Info,
+    };
+
+    let colors = ColoredLevelConfig::new()
+        .error(Color::Red)
+        .warn(Color::Yellow)
+        .info(Color::White)
+        .debug(Color::White)
+        .trace(Color::BrightBlack);
+
+    fern::Dispatch::new()
+        .format(move |out, message, record| {
+            let binding = thread::current();
+            let thread_name = binding.name().unwrap_or("<unnamed>");
+
+            out.finish(format_args!(
+                "[{time} {level} {thread_name} {target}] {message}",
+                level = colors.color(record.level()),
+                time = chrono::Local::now().format("%Y-%m-%dT%H:%M:%S%.9f"),
+                target = record.target(),
+            ))
+        })
+        .chain(
+            fern::Dispatch::new()
+                .level(log_level)
+                .chain(std::io::stdout()),
+        )
+        .chain(fern::log_file(log_file.clone())?)
+        .apply()?;
+
+    debug!("Logging to {} at level {}", log_file, log_level);
+    Ok(())
+}