From 9e7c0fd45e169929156bdb317b10d7bb3db65f8b Mon Sep 17 00:00:00 2001
From: 魏曹先生 <1992414357@qq.com>
Date: Sat, 7 Mar 2026 19:37:52 +0800
Subject: Add callback support to chunk_stream_with and implement stream
 writing

---
 src/utils/log.rs | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 src/utils/log.rs

(limited to 'src/utils/log.rs')

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();
+}
-- 
cgit