From 5830cb1e94e63f1aa5fd954f20ec36e38e118381 Mon Sep 17 00:00:00 2001 From: 魏曹先生 <1992414357@qq.com> Date: Sat, 18 Oct 2025 18:23:15 +0800 Subject: feat: add new binary files and utilities - Add jvref.rs binary - Add build_env_logger.rs utility module --- src/utils/build_env_logger.rs | 66 +++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 66 insertions(+) create mode 100644 src/utils/build_env_logger.rs (limited to 'src/utils') diff --git a/src/utils/build_env_logger.rs b/src/utils/build_env_logger.rs new file mode 100644 index 0000000..51af4c4 --- /dev/null +++ b/src/utils/build_env_logger.rs @@ -0,0 +1,66 @@ +use std::path::Path; + +use colored::Colorize; +use env_logger::{Builder, Target}; +use log::{Level, LevelFilter}; + +pub fn build_env_logger(log_path: impl AsRef) { + use std::io::{self, Write}; + + struct MultiWriter { + a: A, + b: B, + } + + impl MultiWriter { + fn new(a: A, b: B) -> Self { + Self { a, b } + } + } + + impl Write for MultiWriter { + fn write(&mut self, buf: &[u8]) -> io::Result { + let _ = self.a.write(buf); + self.b.write(buf) + } + + fn flush(&mut self) -> io::Result<()> { + let _ = self.a.flush(); + self.b.flush() + } + } + + let mut builder = Builder::new(); + + let log_format = |buf: &mut env_logger::fmt::Formatter, record: &log::Record| { + let now = chrono::Local::now(); + + let level_style = match record.level() { + Level::Error => record.args().to_string().red().bold(), + Level::Warn => record.args().to_string().yellow().bold(), + Level::Info => record.args().to_string().white(), + Level::Debug => record.args().to_string().white(), + Level::Trace => record.args().to_string().cyan(), + }; + + writeln!( + buf, + "{} {}", + now.format("%H:%M:%S") + .to_string() + .truecolor(105, 105, 105) + .bold(), + level_style + ) + }; + + let log_file = std::fs::File::create(log_path).expect("Failed to create log file"); + let combined_target = Target::Pipe(Box::new(MultiWriter::new(std::io::stdout(), log_file))); + + builder + .format(log_format) + .filter(None, LevelFilter::Info) + .filter_module("just_enough_vcs", LevelFilter::Trace) + .target(combined_target) + .init(); +} -- cgit