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 {
50 let key = source.into().to_lowercase();
51 Self(*COLOR_MAP.get(key.as_str()).unwrap_or(&Color::Gray))
52 }
53
54 pub fn from_log_level(level: Level) -> Self {
55 Self::from_any(level.to_string())
56 }
57
58 pub fn from_string(color_str: &str) -> crate::core::error::Result<Self> {
59 COLOR_MAP
60 .get(&color_str.to_lowercase().as_str())
61 .map(|&c| Self(c))
62 .ok_or_else(|| AppError::Validation(format!("Ungültige Farbe: {}", color_str)))
63 }
64
65 pub fn format_message(&self, level: &str, message: &str) -> String {
66 if level.is_empty() {
67 format!("\x1B[{}m{}\x1B[0m", self.to_ansi_code(), message)
68 } else {
69 format!(
70 "\x1B[{}m[{}] {}\x1B[0m",
71 self.to_ansi_code(),
72 level,
73 message
74 )
75 }
76 }
77
78 pub fn to_ansi_code(&self) -> u8 {
79 match self.0 {
80 Color::Black => 30,
81 Color::Red => 31,
82 Color::Green => 32,
83 Color::Yellow => 33,
84 Color::Blue => 34,
85 Color::Magenta => 35,
86 Color::Cyan => 36,
87 Color::Gray => 37,
88 Color::DarkGray => 90,
89 Color::LightRed => 91,
90 Color::LightGreen => 92,
91 Color::LightYellow => 93,
92 Color::LightBlue => 94,
93 Color::LightMagenta => 95,
94 Color::LightCyan => 96,
95 Color::White => 97,
96 _ => 37,
97 }
98 }
99
100 pub fn to_name(&self) -> &'static str {
101 COLOR_MAP
102 .iter()
103 .find(|(_, &v)| v == self.0)
104 .map(|(k, _)| *k)
105 .unwrap_or("gray")
106 }
107}
108
109impl From<AppColor> for Color {
110 fn from(app_color: AppColor) -> Self {
111 app_color.0
112 }
113}
114
115impl From<&AppColor> for Color {
116 fn from(app_color: &AppColor) -> Self {
117 app_color.0
118 }
119}
120
121impl Default for AppColor {
122 fn default() -> Self {
123 Self(Color::Gray)
124 }
125}