69 lines
1.5 KiB
Rust
69 lines
1.5 KiB
Rust
|
use std::collections::HashMap;
|
||
|
|
||
|
use neovim_lib::{Value, Neovim, NeovimApi};
|
||
|
|
||
|
use nvim::ErrorReport;
|
||
|
use color::Color;
|
||
|
use value::ValueMapExt;
|
||
|
|
||
|
pub struct Theme {
|
||
|
pub pmenu: Option<Pmenu>,
|
||
|
}
|
||
|
|
||
|
impl Theme {
|
||
|
pub fn new() -> Self {
|
||
|
Theme { pmenu: None }
|
||
|
}
|
||
|
|
||
|
pub fn update(&mut self, nvim: &mut Neovim) {
|
||
|
self.pmenu = Some(Pmenu::new(nvim));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
pub struct Pmenu {
|
||
|
pub bg: Option<Color>,
|
||
|
pub fg: Option<Color>,
|
||
|
pub bg_sel: Option<Color>,
|
||
|
pub fg_sel: Option<Color>,
|
||
|
}
|
||
|
|
||
|
impl Pmenu {
|
||
|
pub fn new(nvim: &mut Neovim) -> Self {
|
||
|
let (bg, fg) = get_hl_colors(nvim, "Pmenu");
|
||
|
let (bg_sel, fg_sel) = get_hl_colors(nvim, "PmenuSel");
|
||
|
|
||
|
Pmenu {
|
||
|
bg,
|
||
|
fg,
|
||
|
bg_sel,
|
||
|
fg_sel,
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
fn get_hl_color(map: &HashMap<&str, &Value>, color_name: &str) -> Option<Color> {
|
||
|
if let Some(col) = map.get(color_name) {
|
||
|
if let Some(col) = col.as_u64() {
|
||
|
Some(Color::from_indexed_color(col))
|
||
|
} else {
|
||
|
None
|
||
|
}
|
||
|
} else {
|
||
|
None
|
||
|
}
|
||
|
}
|
||
|
|
||
|
fn get_hl_colors(nvim: &mut Neovim, hl: &str) -> (Option<Color>, Option<Color>) {
|
||
|
nvim.get_hl_by_name(hl, true)
|
||
|
.ok_and_report(nvim)
|
||
|
.and_then(|m| if let Some(m) = m.to_attrs_map_report() {
|
||
|
Some((
|
||
|
get_hl_color(&m, "background"),
|
||
|
get_hl_color(&m, "foreground"),
|
||
|
))
|
||
|
} else {
|
||
|
None
|
||
|
})
|
||
|
.unwrap_or((None, None))
|
||
|
}
|