hematite-kokoros 0.1.2

Hematite-maintained fork of the Kokoros Rust text-to-speech engine
Documentation 
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
51
52

use std::time::Instant;

// ANSI color codes for request ID colorization
const COLORS: &[&str] = &[
    "\x1b[31m", "\x1b[32m", "\x1b[33m", "\x1b[34m", "\x1b[35m", "\x1b[36m", "\x1b[91m", "\x1b[92m",
    "\x1b[93m", "\x1b[94m", "\x1b[95m", "\x1b[96m", "\x1b[37m", "\x1b[90m",
];
const RESET: &str = "\x1b[0m";

/// Get consistent color for a request ID using hash-based assignment
pub fn get_request_id_color(request_id: &str) -> &'static str {
    let mut hash = 0u32;
    for byte in request_id.bytes() {
        hash = hash.wrapping_mul(31).wrapping_add(byte as u32);
    }
    let color_index = (hash as usize) % COLORS.len();
    COLORS[color_index]
}

/// Format a debug prefix with colored request ID and instance ID
pub fn format_debug_prefix(request_id: Option<&str>, instance_id: Option<&str>) -> String {
    match (request_id, instance_id) {
        (Some(req_id), Some(inst_id)) => {
            let color = get_request_id_color(req_id);
            format!("{}[{}]{}[{}]", color, req_id, RESET, inst_id)
        }
        (Some(req_id), None) => {
            let color = get_request_id_color(req_id);
            format!("{}[{}]{}", color, req_id, RESET)
        }
        (None, Some(inst_id)) => format!("[{}]", inst_id),
        (None, None) => String::new(),
    }
}

/// Get colored request ID with relative timing (enhanced version)
pub fn get_colored_request_id_with_relative(request_id: &str, start_time: Instant) -> String {
    let color = get_request_id_color(request_id);

    // Get relative time from request start
    let elapsed_ms = start_time.elapsed().as_millis();
    let relative_time = if elapsed_ms < 1 {
        "    0".to_string() // Show "0" right-aligned for initial request
    } else {
        format!("{:5}", elapsed_ms) // Right-aligned 5 digits
    };

    format!(
        "{}[{}]{} \x1b[90m{}\x1b[0m",
        color, request_id, RESET, relative_time
    )
}