1 files changed, 33 insertions, 0 deletions
diff --git a/src/utils/log.rs b/src/utils/log.rs
new file mode 100644
index 0000000..5fc6160
--- /dev/null
+++ b/src/utils/log.rs
@@ -0,0 +1,33 @@
+use env_logger::Builder;
+use log::Level;
+use std::io::Write;
+
+pub fn init_logger(level_filter: Option<log::LevelFilter>) {
+    let mut builder = match level_filter {
+        Some(f) => {
+            let mut b = Builder::new();
+            b.filter_level(f);
+            b
+        }
+        None => return,
+    };
+
+    builder
+        .format(|buf, record| {
+            let level = record.level();
+            let args = record.args();
+
+            let (prefix, color_code) = match level {
+                Level::Error => ("error: ", "\x1b[1;31m"),
+                Level::Warn => ("warn: ", "\x1b[1;33m"),
+                Level::Info => ("", "\x1b[37m"),
+                Level::Debug => ("debug: ", "\x1b[90m"),
+                Level::Trace => ("trace: ", "\x1b[36m"),
+            };
+
+            let colored_prefix = format!("{}{}\x1b[0m", color_code, prefix);
+
+            writeln!(buf, "{}{}", colored_prefix, args)
+        })
+        .init();
+}