51 lines
1022 B
Rust
51 lines
1022 B
Rust
use nvim;
|
|
|
|
#[derive(PartialEq)]
|
|
pub enum NvimMode {
|
|
Normal,
|
|
Insert,
|
|
Other,
|
|
}
|
|
|
|
pub struct Mode {
|
|
mode: NvimMode,
|
|
idx: usize,
|
|
info: Option<Vec<nvim::ModeInfo>>,
|
|
}
|
|
|
|
impl Mode {
|
|
pub fn new() -> Self {
|
|
Mode {
|
|
mode: NvimMode::Normal,
|
|
idx: 0,
|
|
info: None,
|
|
}
|
|
}
|
|
|
|
pub fn is(&self, mode: &NvimMode) -> bool {
|
|
self.mode == *mode
|
|
}
|
|
|
|
pub fn mode_info(&self) -> Option<&nvim::ModeInfo> {
|
|
self.info.as_ref().and_then(|i| i.get(self.idx))
|
|
}
|
|
|
|
pub fn update(&mut self, mode: &str, idx: usize) {
|
|
match mode {
|
|
"normal" => self.mode = NvimMode::Normal,
|
|
"insert" => self.mode = NvimMode::Insert,
|
|
_ => self.mode = NvimMode::Other,
|
|
}
|
|
|
|
self.idx = idx;
|
|
}
|
|
|
|
pub fn set_info(&mut self, cursor_style_enabled: bool, info: Vec<nvim::ModeInfo>) {
|
|
self.info = if cursor_style_enabled {
|
|
Some(info)
|
|
} else {
|
|
None
|
|
};
|
|
}
|
|
}
|