Add timestamps

This commit is contained in:
ErrorNoInternet 2023-01-18 19:12:35 +08:00
parent 145c9554e6
commit 45747625e7
Signed by untrusted user who does not match committer: ErrorNoInternet
GPG Key ID: 2486BFB7B1E6A4A3

View File

@ -1,3 +1,4 @@
use chrono::Local;
use colored::*; use colored::*;
pub enum LogMessageType { pub enum LogMessageType {
@ -15,16 +16,44 @@ pub fn log_error<T, E: std::fmt::Display>(result: Result<T, E>) {
pub fn log_message(message_type: LogMessageType, message: &String) { pub fn log_message(message_type: LogMessageType, message: &String) {
match message_type { match message_type {
LogMessageType::Bot => println!("{} {}", colored_brackets(&"BOT".bold().blue()), message), LogMessageType::Bot => {
LogMessageType::Chat => println!("{} {}", colored_brackets(&"CHAT".bold().blue()), message), println!(
"{} {} {}",
current_time(),
colored_brackets(&"BOT".bold().blue()),
message
)
}
LogMessageType::Chat => {
println!(
"{} {} {}",
current_time(),
colored_brackets(&"CHAT".bold().blue()),
message
)
}
LogMessageType::Error => println!( LogMessageType::Error => println!(
"{} {}", "{} {} {}",
current_time(),
colored_brackets(&"ERROR".bold().red()), colored_brackets(&"ERROR".bold().red()),
message.red() message.red()
), ),
} }
} }
fn current_time() -> String {
format!(
"{}{}{}",
"[".bold().white(),
Local::now()
.format("%Y/%m/%d %H:%M:%S")
.to_string()
.bold()
.white(),
"]".bold().white()
)
}
fn colored_brackets(text: &ColoredString) -> String { fn colored_brackets(text: &ColoredString) -> String {
format!("{}{}{}", "[".bold().yellow(), text, "]".bold().yellow()) format!("{}{}{}", "[".bold().yellow(), text, "]".bold().yellow())
} }