1use ratatui::style::{Color, Modifier, Style};
2
3#[derive(Debug, Clone, Copy, PartialEq)]
4pub enum TerminalBackground {
5 Dark,
6 Light,
7}
8
9pub fn detect_terminal_background() -> TerminalBackground {
10 if let Ok(val) = std::env::var("TERM_BACKGROUND") {
11 match val.to_lowercase().as_str() {
12 "light" => return TerminalBackground::Light,
13 "dark" => return TerminalBackground::Dark,
14 _ => {}
15 }
16 }
17
18 if let Ok(val) = std::env::var("COLORFGBG")
19 && let Some(bg_str) = val.rsplit(';').next()
20 && let Ok(bg) = bg_str.trim().parse::<u8>()
21 {
22 return if bg < 8 || (232..232 + 12).contains(&bg) {
23 TerminalBackground::Dark
24 } else if (8..=15).contains(&bg) {
25 TerminalBackground::Light
26 } else {
27 TerminalBackground::Dark
28 };
29 }
30
31 if let Ok(term_program) = std::env::var("TERM_PROGRAM")
32 && term_program.to_lowercase().contains("apple_terminal")
33 {
34 return TerminalBackground::Light;
35 }
36
37 TerminalBackground::Dark
38}
39
40#[derive(Debug, Clone, Copy)]
41pub struct SyntaxStyles {
42 pub keyword: Style,
43 pub string: Style,
44 pub comment: Style,
45 pub function: Style,
46 pub type_name: Style,
47 pub number: Style,
48 pub constant: Style,
49 pub attribute: Style,
50}
51
52pub struct Theme {
53 pub bg: Color,
54 pub fg: Color,
55 pub dim: Style,
56 pub accent: Color,
57 pub user_label: Style,
58 pub assistant_label: Style,
59 pub border: Style,
60 pub input_prompt: Style,
61 pub status_bar: Style,
62 pub code_bg: Color,
63 pub inline_code: Style,
64 pub error: Style,
65 pub tool_name: Style,
66 pub tool_output: Style,
67 pub heading: Style,
68 pub bold: Style,
69 pub italic: Style,
70 pub blockquote: Style,
71 pub link: Style,
72 pub list_bullet: Style,
73 pub scrollbar_track: Style,
74 pub scrollbar_thumb: Style,
75 pub tool_success: Style,
76 pub highlight: Style,
77 pub muted_fg: Color,
78 pub tool_file_read: Style,
79 pub tool_file_write: Style,
80 pub tool_directory: Style,
81 pub tool_search: Style,
82 pub tool_command: Style,
83 pub tool_mcp: Style,
84 pub tool_skill: Style,
85 pub tool_badge_bg: Color,
86 pub tool_path: Style,
87 pub thinking: Style,
88 pub mode_normal_fg: Color,
89 pub mode_normal_bg: Color,
90 pub mode_insert_fg: Color,
91 pub mode_insert_bg: Color,
92 pub cost: Style,
93 pub user_text: Style,
94 pub tool_action: Style,
95 pub separator: Style,
96 pub tool_exit_ok: Style,
97 pub tool_exit_err: Style,
98 pub syntax: Option<SyntaxStyles>,
99 pub syntect_theme: Option<&'static str>,
100 pub diff_add: Style,
101 pub diff_remove: Style,
102 pub diff_hunk: Style,
103 pub input_bg: Color,
104 pub input_fg: Color,
105 pub input_dim_fg: Color,
106 pub progress_bar_filled: Style,
107 pub progress_bar_empty: Style,
108 pub streaming_dot: Style,
109 pub user_text_bg: Color,
110 pub subagent_border: Style,
111 pub subagent_header: Style,
112 pub subagent_done: Style,
113 pub subagent_working: Style,
114}
115
116impl Theme {
117 pub fn from_config(name: &str) -> Self {
118 match name {
119 "light" => Self::light(),
120 "terminal" => Self::terminal(),
121 "auto" => match detect_terminal_background() {
122 TerminalBackground::Light => Self::light(),
123 TerminalBackground::Dark => Self::dark(),
124 },
125 _ => Self::dark(),
126 }
127 }
128
129 pub fn dark() -> Self {
130 let muted = Color::Rgb(88, 91, 112);
131 let surface = Color::Rgb(42, 44, 60);
132 let accent = Color::Rgb(110, 150, 215);
133 let green = Color::Rgb(140, 190, 135);
134 let peach = Color::Rgb(210, 155, 115);
135 let red = Color::Rgb(200, 120, 145);
136 let mauve = Color::Rgb(170, 140, 210);
137 let yellow = Color::Rgb(210, 190, 150);
138 let teal = Color::Rgb(120, 185, 175);
139 let sapphire = Color::Rgb(95, 165, 200);
140 let base = Color::Rgb(30, 30, 46);
141
142 Self {
143 bg: Color::Reset,
144 fg: Color::White,
145 dim: Style::default().fg(muted),
146 accent,
147 muted_fg: muted,
148 user_label: Style::default().fg(mauve).add_modifier(Modifier::BOLD),
149 assistant_label: Style::default().fg(accent).add_modifier(Modifier::BOLD),
150 border: Style::default().fg(surface),
151 input_prompt: Style::default().fg(accent),
152 status_bar: Style::default().fg(muted),
153 code_bg: surface,
154 inline_code: Style::default().fg(peach),
155 error: Style::default().fg(red),
156 tool_name: Style::default().fg(yellow).add_modifier(Modifier::BOLD),
157 tool_output: Style::default().fg(muted),
158 tool_success: Style::default().fg(green),
159 heading: Style::default().fg(accent).add_modifier(Modifier::BOLD),
160 bold: Style::default().add_modifier(Modifier::BOLD),
161 italic: Style::default().add_modifier(Modifier::ITALIC),
162 blockquote: Style::default().fg(muted),
163 link: Style::default()
164 .fg(accent)
165 .add_modifier(Modifier::UNDERLINED),
166 list_bullet: Style::default().fg(muted),
167 scrollbar_track: Style::default().fg(surface),
168 scrollbar_thumb: Style::default().fg(muted),
169 highlight: Style::default().fg(base).bg(accent),
170 tool_file_read: Style::default().fg(sapphire),
171 tool_file_write: Style::default().fg(peach),
172 tool_directory: Style::default().fg(accent),
173 tool_search: Style::default().fg(mauve),
174 tool_command: Style::default().fg(green),
175 tool_mcp: Style::default().fg(teal),
176 tool_skill: Style::default().fg(mauve),
177 tool_badge_bg: surface,
178 tool_path: Style::default()
179 .fg(Color::White)
180 .add_modifier(Modifier::UNDERLINED),
181 thinking: Style::default().fg(muted),
182 mode_normal_fg: base,
183 mode_normal_bg: muted,
184 mode_insert_fg: base,
185 mode_insert_bg: accent,
186 cost: Style::default().fg(Color::Rgb(165, 135, 80)),
187 user_text: Style::default().fg(Color::Rgb(205, 214, 244)),
188 tool_action: Style::default().fg(muted),
189 separator: Style::default().fg(Color::Rgb(52, 54, 72)),
190 tool_exit_ok: Style::default().fg(green),
191 tool_exit_err: Style::default().fg(red),
192 syntax: None,
193 syntect_theme: Some("base16-ocean.dark"),
194 diff_add: Style::default().fg(green),
195 diff_remove: Style::default().fg(red),
196 diff_hunk: Style::default().fg(accent),
197 input_bg: Color::Rgb(36, 38, 55),
198 input_fg: Color::White,
199 input_dim_fg: muted,
200 progress_bar_filled: Style::default().fg(accent).add_modifier(Modifier::BOLD),
201 progress_bar_empty: Style::default().fg(surface),
202 streaming_dot: Style::default().fg(accent),
203 user_text_bg: Color::Rgb(38, 40, 58),
204 subagent_border: Style::default().fg(surface),
205 subagent_header: Style::default().fg(accent).add_modifier(Modifier::BOLD),
206 subagent_done: Style::default().fg(green),
207 subagent_working: Style::default().fg(accent),
208 }
209 }
210
211 pub fn light() -> Self {
212 let muted = Color::Rgb(140, 143, 161);
213 let surface = Color::Rgb(204, 208, 218);
214 let accent = Color::Rgb(35, 90, 210);
215 let green = Color::Rgb(55, 135, 40);
216 let peach = Color::Rgb(210, 90, 20);
217 let red = Color::Rgb(175, 30, 60);
218 let mauve = Color::Rgb(110, 55, 190);
219 let yellow = Color::Rgb(185, 120, 30);
220 let teal = Color::Rgb(25, 125, 130);
221 let sapphire = Color::Rgb(30, 130, 155);
222 let text = Color::Rgb(76, 79, 105);
223
224 Self {
225 bg: Color::Reset,
226 fg: text,
227 dim: Style::default().fg(muted),
228 accent,
229 muted_fg: muted,
230 user_label: Style::default().fg(mauve).add_modifier(Modifier::BOLD),
231 assistant_label: Style::default().fg(accent).add_modifier(Modifier::BOLD),
232 border: Style::default().fg(surface),
233 input_prompt: Style::default().fg(accent),
234 status_bar: Style::default().fg(muted),
235 code_bg: surface,
236 inline_code: Style::default().fg(peach),
237 error: Style::default().fg(red),
238 tool_name: Style::default().fg(yellow).add_modifier(Modifier::BOLD),
239 tool_output: Style::default().fg(muted),
240 tool_success: Style::default().fg(green),
241 heading: Style::default().fg(accent).add_modifier(Modifier::BOLD),
242 bold: Style::default().add_modifier(Modifier::BOLD),
243 italic: Style::default().add_modifier(Modifier::ITALIC),
244 blockquote: Style::default().fg(muted),
245 link: Style::default()
246 .fg(accent)
247 .add_modifier(Modifier::UNDERLINED),
248 list_bullet: Style::default().fg(muted),
249 scrollbar_track: Style::default().fg(surface),
250 scrollbar_thumb: Style::default().fg(muted),
251 highlight: Style::default().fg(Color::White).bg(accent),
252 tool_file_read: Style::default().fg(sapphire),
253 tool_file_write: Style::default().fg(peach),
254 tool_directory: Style::default().fg(accent),
255 tool_search: Style::default().fg(mauve),
256 tool_command: Style::default().fg(green),
257 tool_mcp: Style::default().fg(teal),
258 tool_skill: Style::default().fg(mauve),
259 tool_badge_bg: surface,
260 tool_path: Style::default().fg(text).add_modifier(Modifier::UNDERLINED),
261 thinking: Style::default().fg(muted),
262 mode_normal_fg: Color::White,
263 mode_normal_bg: muted,
264 mode_insert_fg: Color::White,
265 mode_insert_bg: accent,
266 cost: Style::default().fg(Color::Rgb(150, 110, 35)),
267 user_text: Style::default().fg(text),
268 tool_action: Style::default().fg(muted),
269 separator: Style::default().fg(surface),
270 tool_exit_ok: Style::default().fg(green),
271 tool_exit_err: Style::default().fg(red),
272 syntax: None,
273 syntect_theme: Some("base16-ocean.light"),
274 diff_add: Style::default().fg(green),
275 diff_remove: Style::default().fg(red),
276 diff_hunk: Style::default().fg(accent),
277 input_bg: Color::Rgb(210, 214, 225),
278 input_fg: text,
279 input_dim_fg: muted,
280 progress_bar_filled: Style::default().fg(accent).add_modifier(Modifier::BOLD),
281 progress_bar_empty: Style::default().fg(surface),
282 streaming_dot: Style::default().fg(accent),
283 user_text_bg: Color::Rgb(218, 222, 232),
284 subagent_border: Style::default().fg(surface),
285 subagent_header: Style::default().fg(accent).add_modifier(Modifier::BOLD),
286 subagent_done: Style::default().fg(green),
287 subagent_working: Style::default().fg(accent),
288 }
289 }
290
291 pub fn terminal() -> Self {
292 let dim = Style::default().add_modifier(Modifier::DIM);
293 let bold = Style::default().add_modifier(Modifier::BOLD);
294 let muted = Color::Indexed(8);
295
296 Self {
297 bg: Color::Reset,
298 fg: Color::Reset,
299 dim,
300 accent: Color::Reset,
301 muted_fg: muted,
302 user_label: bold,
303 assistant_label: Style::default(),
304 border: dim,
305 input_prompt: bold,
306 status_bar: dim,
307 code_bg: Color::Indexed(0),
308 inline_code: Style::default().fg(muted),
309 error: Style::default().add_modifier(Modifier::BOLD | Modifier::REVERSED),
310 tool_name: bold,
311 tool_output: dim,
312 tool_success: bold,
313 heading: Style::default().add_modifier(Modifier::BOLD | Modifier::UNDERLINED),
314 bold,
315 italic: Style::default().add_modifier(Modifier::ITALIC),
316 blockquote: dim,
317 link: Style::default().add_modifier(Modifier::UNDERLINED),
318 list_bullet: dim,
319 scrollbar_track: dim,
320 scrollbar_thumb: Style::default(),
321 highlight: Style::default().add_modifier(Modifier::REVERSED),
322 tool_file_read: Style::default().fg(muted),
323 tool_file_write: Style::default().fg(muted),
324 tool_directory: bold,
325 tool_search: Style::default().fg(muted),
326 tool_command: Style::default().fg(muted),
327 tool_mcp: Style::default().fg(muted),
328 tool_skill: Style::default().fg(muted),
329 tool_badge_bg: muted,
330 tool_path: Style::default().add_modifier(Modifier::UNDERLINED),
331 thinking: dim,
332 mode_normal_fg: Color::Reset,
333 mode_normal_bg: muted,
334 mode_insert_fg: Color::Indexed(0),
335 mode_insert_bg: Color::Reset,
336 cost: dim,
337 user_text: bold,
338 tool_action: dim,
339 separator: dim,
340 tool_exit_ok: bold,
341 tool_exit_err: Style::default().add_modifier(Modifier::BOLD),
342 syntax: Some(SyntaxStyles {
343 keyword: bold,
344 string: Style::default().fg(muted),
345 comment: dim.add_modifier(Modifier::ITALIC),
346 function: bold,
347 type_name: Style::default().fg(muted),
348 number: Style::default().fg(muted),
349 constant: bold,
350 attribute: Style::default().fg(muted),
351 }),
352 syntect_theme: None,
353 diff_add: bold,
354 diff_remove: dim,
355 diff_hunk: Style::default().fg(muted),
356 input_bg: muted,
357 input_fg: Color::Reset,
358 input_dim_fg: muted,
359 progress_bar_filled: bold,
360 progress_bar_empty: dim,
361 streaming_dot: dim,
362 user_text_bg: Color::Indexed(0),
363 subagent_border: dim,
364 subagent_header: bold,
365 subagent_done: bold,
366 subagent_working: dim,
367 }
368 }
369}
370
371impl Default for Theme {
372 fn default() -> Self {
373 Self::terminal()
374 }
375}