diff --git a/helix-view/src/theme.rs b/helix-view/src/theme.rs index f1219ec5dc170..d82a77f871366 100644 --- a/helix-view/src/theme.rs +++ b/helix-view/src/theme.rs @@ -350,8 +350,23 @@ impl ThemePalette { Self { palette: default } } - pub fn hex_string_to_rgb(s: &str) -> Result { - if s.starts_with('#') && s.len() >= 7 { + pub fn string_to_rgb(s: &str) -> Result { + if s.starts_with('#') { + Self::hex_string_to_rgb(s) + } else { + Self::ansi_string_to_rgb(s) + } + } + + fn ansi_string_to_rgb(s: &str) -> Result { + if let Ok(index) = u8::from_str_radix(s, 10) { + return Ok(Color::Indexed(index)) + } + Err(format!("Theme: malformed ANSI: {}", s)) + } + + fn hex_string_to_rgb(s: &str) -> Result { + if s.len() >= 7 { if let (Ok(red), Ok(green), Ok(blue)) = ( u8::from_str_radix(&s[1..3], 16), u8::from_str_radix(&s[3..5], 16), @@ -377,7 +392,7 @@ impl ThemePalette { .get(value) .copied() .ok_or("") - .or_else(|_| Self::hex_string_to_rgb(value)) + .or_else(|_| Self::string_to_rgb(value)) } pub fn parse_modifier(value: &Value) -> Result { @@ -453,7 +468,7 @@ impl TryFrom for ThemePalette { let mut palette = HashMap::with_capacity(map.len()); for (name, value) in map { let value = Self::parse_value_as_str(&value)?; - let color = Self::hex_string_to_rgb(value)?; + let color = Self::string_to_rgb(value)?; palette.insert(name, color); }