rush_sync_server/ui/
color.rs1use crate::core::prelude::*;
2use log::Level;
3use once_cell::sync::Lazy;
4use std::collections::HashMap;
5
6#[derive(Debug, Clone, Copy, PartialEq)]
7pub struct AppColor(Color);
8
9static COLOR_MAP: Lazy<HashMap<&'static str, Color>> = Lazy::new(|| {
10 let mut map = HashMap::new();
11
12 map.insert("black", Color::Black);
14 map.insert("red", Color::Red);
15 map.insert("green", Color::Green);
16 map.insert("yellow", Color::Yellow);
17 map.insert("blue", Color::Blue);
18 map.insert("magenta", Color::Magenta);
19 map.insert("cyan", Color::Cyan);
20 map.insert("gray", Color::Gray);
21 map.insert("darkgray", Color::DarkGray);
22 map.insert("lightred", Color::LightRed);
23 map.insert("lightgreen", Color::LightGreen);
24 map.insert("lightyellow", Color::LightYellow);
25 map.insert("lightblue", Color::LightBlue);
26 map.insert("lightmagenta", Color::LightMagenta);
27 map.insert("lightcyan", Color::LightCyan);
28 map.insert("white", Color::White);
29
30 map.insert("error", Color::Red);
32 map.insert("warning", Color::Yellow);
33 map.insert("warn", Color::Yellow);
34 map.insert("info", Color::Green);
35 map.insert("debug", Color::Blue);
36 map.insert("trace", Color::White);
37 map.insert("lang", Color::Cyan);
38 map.insert("version", Color::LightBlue);
39 map.insert("startup", Color::Magenta);
40
41 map
42});
43
44impl AppColor {
45 pub fn new(color: Color) -> Self {
46 Self(color)
47 }
48
49 pub fn from_any<T: Into<String>>(source: T) -> Self {
55 let key = source.into().to_lowercase();
56
57 if let Some(&color) = COLOR_MAP.get(key.as_str()) {
59 return Self(color);
60 }
61
62 let mapped_category = crate::i18n::get_color_category_for_display(&key);
64 let fallback_color = COLOR_MAP
65 .get(mapped_category.as_str())
66 .unwrap_or(&Color::Gray);
67
68 Self(*fallback_color)
69 }
70
71 pub fn from_log_level(level: Level) -> Self {
72 Self::from_any(level.to_string())
73 }
74
75 pub fn from_string(color_str: &str) -> crate::core::error::Result<Self> {
76 let normalized = color_str.trim().to_lowercase();
77
78 let color = COLOR_MAP.get(normalized.as_str()).copied().ok_or_else(|| {
80 log::error!("❌ Color '{}' not found in COLOR_MAP", normalized);
81 AppError::Validation(format!("Invalid color: {}", color_str))
82 })?;
83
84 let app_color = Self(color);
85
86 Ok(app_color)
87 }
88
89 pub fn format_message(&self, level: &str, message: &str) -> String {
90 if level.is_empty() {
91 format!("\x1B[{}m{}\x1B[0m", self.to_ansi_code(), message)
92 } else {
93 format!(
94 "\x1B[{}m[{}] {}\x1B[0m",
95 self.to_ansi_code(),
96 level,
97 message
98 )
99 }
100 }
101
102 pub fn to_ansi_code(&self) -> u8 {
103 match self.0 {
104 Color::Black => 30,
105 Color::Red => 31,
106 Color::Green => 32,
107 Color::Yellow => 33,
108 Color::Blue => 34,
109 Color::Magenta => 35,
110 Color::Cyan => 36,
111 Color::Gray => 37,
112 Color::DarkGray => 90,
113 Color::LightRed => 91,
114 Color::LightGreen => 92,
115 Color::LightYellow => 93,
116 Color::LightBlue => 94,
117 Color::LightMagenta => 95,
118 Color::LightCyan => 96,
119 Color::White => 97,
120 _ => 37,
121 }
122 }
123
124 pub fn to_name(&self) -> &'static str {
125 COLOR_MAP
126 .iter()
127 .find(|(_, &v)| v == self.0)
128 .map(|(k, _)| *k)
129 .unwrap_or("gray")
130 }
131}
132
133impl From<AppColor> for Color {
134 fn from(app_color: AppColor) -> Self {
135 app_color.0
136 }
137}
138
139impl From<&AppColor> for Color {
140 fn from(app_color: &AppColor) -> Self {
141 app_color.0
142 }
143}
144
145impl Default for AppColor {
146 fn default() -> Self {
147 Self(Color::Gray)
148 }
149}