From a65395d94beb614bc17301601109042077e55e83 Mon Sep 17 00:00:00 2001 From: Blaž Hrastnik Date: Thu, 25 Mar 2021 16:42:14 +0900 Subject: Load theme from toml file. --- helix-view/src/editor.rs | 5 +- helix-view/src/theme.rs | 131 +++++++++++++++++++++++++++-------------------- 2 files changed, 80 insertions(+), 56 deletions(-) (limited to 'helix-view/src') diff --git a/helix-view/src/editor.rs b/helix-view/src/editor.rs index 08dd4d00..374eddfd 100644 --- a/helix-view/src/editor.rs +++ b/helix-view/src/editor.rs @@ -24,7 +24,10 @@ pub enum Action { impl Editor { pub fn new(executor: &'static smol::Executor<'static>, mut area: tui::layout::Rect) -> Self { - let theme = Theme::default(); + // TODO: load from config dir + let toml = include_str!("../../theme.toml"); + let theme: Theme = toml::from_str(&toml).expect("failed to parse theme.toml"); + let language_servers = helix_lsp::Registry::new(); // HAXX: offset the render area height by 1 to account for prompt/commandline diff --git a/helix-view/src/theme.rs b/helix-view/src/theme.rs index e51fab3e..f7731929 100644 --- a/helix-view/src/theme.rs +++ b/helix-view/src/theme.rs @@ -1,6 +1,8 @@ -use helix_core::hashmap; use std::collections::HashMap; +use serde::{Deserialize, Deserializer}; +use toml::Value; + #[cfg(feature = "term")] pub use tui::style::{Color, Style}; @@ -83,69 +85,88 @@ pub use tui::style::{Color, Style}; // } /// Color theme for syntax highlighting. +#[derive(Debug)] pub struct Theme { scopes: Vec, - mapping: HashMap<&'static str, Style>, + styles: HashMap, +} + +impl<'de> Deserialize<'de> for Theme { + fn deserialize(deserializer: D) -> Result + where + D: Deserializer<'de>, + { + let mut styles = HashMap::new(); + + if let Ok(colors) = HashMap::::deserialize(deserializer) { + // scopes.reserve(colors.len()); + styles.reserve(colors.len()); + for (name, style_value) in colors { + let mut style = Style::default(); + parse_style(&mut style, style_value); + // scopes.push(name); + styles.insert(name, style); + } + } + + let scopes = styles.keys().map(ToString::to_string).collect(); + Ok(Theme { scopes, styles }) + } +} + +fn parse_style(style: &mut Style, value: Value) { + if let Value::Table(entries) = value { + for (name, value) in entries { + match name.as_str() { + "fg" => { + if let Some(color) = parse_color(value) { + *style = style.fg(color); + } + } + "bg" => { + if let Some(color) = parse_color(value) { + *style = style.bg(color); + } + } + _ => (), + } + } + } else if let Some(color) = parse_color(value) { + *style = style.fg(color); + } +} + +fn hex_string_to_rgb(s: &str) -> Option<(u8, u8, u8)> { + if s.starts_with("#") && 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), + u8::from_str_radix(&s[5..7], 16), + ) { + Some((red, green, blue)) + } else { + None + } + } else { + None + } } -impl Default for Theme { - fn default() -> Self { - let mapping = hashmap! { - "attribute" => Style::default().fg(Color::Rgb(219, 191, 239)), // lilac - "keyword" => Style::default().fg(Color::Rgb(236, 205, 186)), // almond - "punctuation" => Style::default().fg(Color::Rgb(164, 160, 232)), // lavender - "punctuation.delimiter" => Style::default().fg(Color::Rgb(164, 160, 232)), // lavender - "operator" => Style::default().fg(Color::Rgb(219, 191, 239)), // lilac - "property" => Style::default().fg(Color::Rgb(164, 160, 232)), // lavender - "variable.parameter" => Style::default().fg(Color::Rgb(164, 160, 232)), // lavender - // TODO distinguish type from type.builtin? - "type" => Style::default().fg(Color::Rgb(255, 255, 255)), // white - "type.builtin" => Style::default().fg(Color::Rgb(255, 255, 255)), // white - "constructor" => Style::default().fg(Color::Rgb(219, 191, 239)), // lilac - "function" => Style::default().fg(Color::Rgb(255, 255, 255)), // white - "function.macro" => Style::default().fg(Color::Rgb(219, 191, 239)), // lilac - "comment" => Style::default().fg(Color::Rgb(105, 124, 129)), // sirocco - "variable.builtin" => Style::default().fg(Color::Rgb(159, 242, 143)), // mint - "constant" => Style::default().fg(Color::Rgb(255, 255, 255)), // white - "constant.builtin" => Style::default().fg(Color::Rgb(255, 255, 255)), // white - "string" => Style::default().fg(Color::Rgb(204, 204, 204)), // silver - "escape" => Style::default().fg(Color::Rgb(239, 186, 93)), // honey - // used for lifetimes - "label" => Style::default().fg(Color::Rgb(239, 186, 93)), // honey - - // TODO: diferentiate number builtin - // TODO: diferentiate doc comment - // TODO: variable as lilac - // TODO: mod/use statements as white - // TODO: mod stuff as chamoise - // TODO: add "(scoped_identifier) @path" for std::mem:: - // - // concat (ERROR) @syntax-error and "MISSING ;" selectors for errors - - "module" => Style::default().fg(Color::Rgb(255, 0, 0)), // white - "variable" => Style::default().fg(Color::Rgb(255, 0, 0)), // white - "function.builtin" => Style::default().fg(Color::Rgb(255, 0, 0)), // white - - "ui.background" => Style::default().bg(Color::Rgb(59, 34, 76)), // midnight - "ui.linenr" => Style::default().fg(Color::Rgb(90, 89, 119)), // comet - "ui.statusline" => Style::default().bg(Color::Rgb(40, 23, 51)), // revolver - "ui.popup" => Style::default().bg(Color::Rgb(40, 23, 51)), // revolver - - "warning" => Style::default().fg(Color::Rgb(255, 205, 28)), - "error" => Style::default().fg(Color::Rgb(244, 120, 104)), - "info" => Style::default().fg(Color::Rgb(111, 68, 240)), - "hint" => Style::default().fg(Color::Rgb(204, 204, 204)), - }; - - let scopes = mapping.keys().map(ToString::to_string).collect(); - - Self { scopes, mapping } +fn parse_color(value: Value) -> Option { + if let Value::String(s) = value { + if let Some((red, green, blue)) = hex_string_to_rgb(&s) { + Some(Color::Rgb(red, green, blue)) + } else { + None + } + } else { + None } } impl Theme { pub fn get(&self, scope: &str) -> Style { - self.mapping + self.styles .get(scope) .copied() .unwrap_or_else(|| Style::default().fg(Color::Rgb(0, 0, 255))) -- cgit v1.2.3-70-g09d2