Merge pull request #7 from mazdak/codex/add-color-to-profiles-for-distinction

Add profile color support

Author: mazdak

src/config.rs

@@ -1,12 +1,58 @@
 use serde::{Deserialize, Serialize};
 use std::{fs, path::{Path, PathBuf}};
+use ratatui::style::Color;
 
 #[derive(Deserialize, Serialize, Debug, Clone, PartialEq)]
 pub struct ConnectionProfile {
     pub name: String,
     pub url: String,
     pub db: Option<u8>,
     pub dev: Option<bool>,
+    pub color: Option<String>,
+}
+
+impl ConnectionProfile {
+    pub fn resolved_color(&self) -> Color {
+        self.color
+            .as_deref()
+            .map(parse_color)
+            .unwrap_or(Color::White)
+    }
+}
+
+fn parse_color(spec: &str) -> Color {
+    match spec.trim().to_lowercase().as_str() {
+        "black" => Color::Black,
+        "red" => Color::Red,
+        "green" => Color::Green,
+        "yellow" => Color::Yellow,
+        "blue" => Color::Blue,
+        "magenta" => Color::Magenta,
+        "cyan" => Color::Cyan,
+        "gray" | "grey" => Color::Gray,
+        "darkgray" | "darkgrey" => Color::DarkGray,
+        "lightred" => Color::LightRed,
+        "lightgreen" => Color::LightGreen,
+        "lightyellow" => Color::LightYellow,
+        "lightblue" => Color::LightBlue,
+        "lightmagenta" => Color::LightMagenta,
+        "lightcyan" => Color::LightCyan,
+        "white" => Color::White,
+        other => {
+            if let Some(hex) = other.strip_prefix('#') {
+                if hex.len() == 6 {
+                    if let (Ok(r), Ok(g), Ok(b)) = (
+                        u8::from_str_radix(&hex[0..2], 16),
+                        u8::from_str_radix(&hex[2..4], 16),
+                        u8::from_str_radix(&hex[4..6], 16),
+                    ) {
+                        return Color::Rgb(r, g, b);
+                    }
+                }
+            }
+            Color::White
+        }
+    }
 }
 
 #[derive(Deserialize, Serialize, Debug, Default, PartialEq)]
@@ -24,6 +70,7 @@ impl Config {
                     url: "redis://127.0.0.1:6379".to_string(),
                     db: Some(0),
                     dev: Some(true),
+                    color: Some("green".to_string()),
                 }
             ]
         }
@@ -112,6 +159,7 @@ mod tests {
         assert_eq!(cfg, loaded);
         assert_eq!(cfg.profiles.len(), 1);
         assert_eq!(cfg.profiles[0].name, "Default");
+        assert_eq!(cfg.profiles[0].color.as_deref(), Some("green"));
     }
 
     #[test]
@@ -128,6 +176,7 @@ mod tests {
                 url: "redis://localhost:6379".to_string(),
                 db: Some(1),
                 dev: Some(false),
+                color: Some("red".to_string()),
             }],
         };
         fs::write(&cfg_file, toml::to_string(&custom_cfg).unwrap()).unwrap();

src/ui.rs

@@ -118,7 +118,14 @@ fn draw_profiles_or_db_list(f: &mut Frame, app: &App, area: Rect) {
     f.render_stateful_widget(db_list_widget, db_list_area, &mut db_list_state);
 
     // Render Status
+    let profile_color = app
+        .profiles
+        .get(app.current_profile_index)
+        .map(|p| p.resolved_color())
+        .unwrap_or(Color::White);
+
     let connection_status_paragraph = Paragraph::new(app.connection_status.as_str())
+        .style(Style::default().fg(profile_color))
         .wrap(Wrap { trim: true })
         .alignment(Alignment::Center); // Center status text
     f.render_widget(connection_status_paragraph, status_area);
@@ -326,18 +333,27 @@ fn draw_profile_selector_modal(f: &mut Frame, app: &App) {
     let area = centered_rect(60, 50, f.area());
     f.render_widget(Clear, area); // Clear the background
 
-    let profiles: Vec<ListItem> = app.profiles.iter().enumerate().map(|(idx, profile)| {
-        let style = if idx == app.selected_profile_list_index {
-            Style::default().fg(Color::Black).bg(Color::White) // Highlight selected
-        } else {
-            Style::default()
-        };
-        ListItem::new(format!("{} ({})", profile.name, profile.url)).style(style)
-    }).collect();
+    let profiles: Vec<ListItem> = app
+        .profiles
+        .iter()
+        .enumerate()
+        .map(|(idx, profile)| {
+            let item_color = profile.resolved_color();
+            let style = if idx == app.selected_profile_list_index {
+                Style::default()
+                    .fg(Color::Black)
+                    .bg(item_color)
+                    .add_modifier(Modifier::BOLD)
+            } else {
+                Style::default().fg(item_color)
+            };
+            ListItem::new(format!("{} ({})", profile.name, profile.url)).style(style)
+        })
+        .collect();
 
     let list_widget = List::new(profiles)
         .block(Block::default().borders(Borders::ALL).title("Select Connection Profile (p/Esc to close)"))
-        .highlight_style(Style::default().add_modifier(Modifier::BOLD).fg(Color::Black).bg(Color::White)) // Ensure highlight is visible
+        .highlight_style(Style::default().add_modifier(Modifier::BOLD))
         .highlight_symbol(">> ");
 
     let mut list_state = ListState::default();