teo_runtime/message/
mod.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
use std::time::Duration;
use chrono::{DateTime, Local};
use colored::{ColoredString, Colorize};
use array_tool::vec::Join;

fn timestamp() -> ColoredString {
    let local: DateTime<Local> = Local::now();
    let local_formatted = format!("[{}]", local.format("%Y-%m-%d %H:%M:%S")).bright_blue();
    local_formatted
}

pub fn info_message(content: impl AsRef<str>) {
    println!("{} {}", timestamp(), content.as_ref())
}

pub fn request_message(
    time_elapsed: Duration,
    method: &str,
    path: &str,
    handler_group_path: &Vec<String>,
    action: &str,
    code: u16,
) {
    let handler_str: String = handler_group_path.join(".") + ".";
    let code_string = format_code_into_string(code);
    let ms = time_elapsed.as_millis();
    let ms_str = format!("{ms}ms").normal().clear();
    println!("{} {} {} => {}{} {} {}", timestamp(), method.bright_blue().bold(), path.bright_yellow(), handler_str.magenta(), action.purple(), code_string, ms_str)
}

pub fn unhandled_request_message(
    time_elapsed: Duration,
    method: &str,
    path: &str,
    code: u16,
) {
    let code_string = format_code_into_string(code);
    let ms = time_elapsed.as_millis();
    let ms_str = format!("{ms}ms").normal().clear();
    println!("{} {} {} {} {}", timestamp(), method.bright_blue().bold(), path.bright_yellow(), code_string, ms_str)
}

fn format_code_into_string(code: u16) -> ColoredString {
    match code {
        0..=199 => code.to_string().purple().bold(),
        200..=299 => code.to_string().green().bold(),
        300..=399 => code.to_string().yellow().bold(),
        _ => code.to_string().red().bold(),
    }
}