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::*;
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) {
match message_type {
LogMessageType::Bot => println!("{} {}", colored_brackets(&"BOT".bold().blue()), message),
LogMessageType::Chat => println!("{} {}", colored_brackets(&"CHAT".bold().blue()), message),
LogMessageType::Bot => {
println!(
"{} {} {}",
current_time(),
colored_brackets(&"BOT".bold().blue()),
message
)
}
LogMessageType::Chat => {
println!(
"{} {} {}",
current_time(),
colored_brackets(&"CHAT".bold().blue()),
message
)
}
LogMessageType::Error => println!(
"{} {}",
"{} {} {}",
current_time(),
colored_brackets(&"ERROR".bold().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 {
format!("{}{}{}", "[".bold().yellow(), text, "]".bold().yellow())
}