mirror of
https://github.com/extrawurst/gitui.git
synced 2024-11-23 20:52:54 +03:00
253 lines
9.5 KiB
Rust
253 lines
9.5 KiB
Rust
use crate::get_app_config_path;
|
|
use anyhow::Result;
|
|
use crossterm::event::{KeyCode, KeyEvent, KeyModifiers};
|
|
use ron::{
|
|
de::from_bytes,
|
|
ser::{to_string_pretty, PrettyConfig},
|
|
};
|
|
use serde::{Deserialize, Serialize};
|
|
use std::{
|
|
fs::File,
|
|
io::{Read, Write},
|
|
path::PathBuf,
|
|
rc::Rc,
|
|
};
|
|
|
|
pub type SharedKeyConfig = Rc<KeyConfig>;
|
|
|
|
#[derive(Serialize, Deserialize, Debug)]
|
|
pub struct KeyConfig {
|
|
pub tab_status: KeyEvent,
|
|
pub tab_log: KeyEvent,
|
|
pub tab_stashing: KeyEvent,
|
|
pub tab_stashes: KeyEvent,
|
|
pub tab_toggle: KeyEvent,
|
|
pub tab_toggle_reverse: KeyEvent,
|
|
pub tab_toggle_reverse_windows: KeyEvent,
|
|
pub focus_workdir: KeyEvent,
|
|
pub focus_stage: KeyEvent,
|
|
pub focus_right: KeyEvent,
|
|
pub focus_left: KeyEvent,
|
|
pub focus_above: KeyEvent,
|
|
pub focus_below: KeyEvent,
|
|
pub exit: KeyEvent,
|
|
pub exit_popup: KeyEvent,
|
|
pub open_commit: KeyEvent,
|
|
pub open_commit_editor: KeyEvent,
|
|
pub open_help: KeyEvent,
|
|
pub move_left: KeyEvent,
|
|
pub move_right: KeyEvent,
|
|
pub home: KeyEvent,
|
|
pub end: KeyEvent,
|
|
pub move_up: KeyEvent,
|
|
pub move_down: KeyEvent,
|
|
pub page_down: KeyEvent,
|
|
pub page_up: KeyEvent,
|
|
pub shift_up: KeyEvent,
|
|
pub shift_down: KeyEvent,
|
|
pub enter: KeyEvent,
|
|
pub edit_file: KeyEvent,
|
|
pub status_stage_all: KeyEvent,
|
|
pub status_reset_item: KeyEvent,
|
|
pub status_ignore_file: KeyEvent,
|
|
pub stashing_save: KeyEvent,
|
|
pub stashing_toggle_untracked: KeyEvent,
|
|
pub stashing_toggle_index: KeyEvent,
|
|
pub stash_open: KeyEvent,
|
|
pub stash_drop: KeyEvent,
|
|
pub cmd_bar_toggle: KeyEvent,
|
|
pub log_tag_commit: KeyEvent,
|
|
pub commit_amend: KeyEvent,
|
|
pub copy: KeyEvent,
|
|
pub create_branch: KeyEvent,
|
|
}
|
|
|
|
#[rustfmt::skip]
|
|
impl Default for KeyConfig {
|
|
fn default() -> Self {
|
|
Self {
|
|
tab_status: KeyEvent { code: KeyCode::Char('1'), modifiers: KeyModifiers::empty()},
|
|
tab_log: KeyEvent { code: KeyCode::Char('2'), modifiers: KeyModifiers::empty()},
|
|
tab_stashing: KeyEvent { code: KeyCode::Char('3'), modifiers: KeyModifiers::empty()},
|
|
tab_stashes: KeyEvent { code: KeyCode::Char('4'), modifiers: KeyModifiers::empty()},
|
|
tab_toggle: KeyEvent { code: KeyCode::Tab, modifiers: KeyModifiers::empty()},
|
|
tab_toggle_reverse: KeyEvent { code: KeyCode::BackTab, modifiers: KeyModifiers::empty()},
|
|
tab_toggle_reverse_windows: KeyEvent { code: KeyCode::BackTab, modifiers: KeyModifiers::SHIFT},
|
|
focus_workdir: KeyEvent { code: KeyCode::Char('w'), modifiers: KeyModifiers::empty()},
|
|
focus_stage: KeyEvent { code: KeyCode::Char('s'), modifiers: KeyModifiers::empty()},
|
|
focus_right: KeyEvent { code: KeyCode::Right, modifiers: KeyModifiers::empty()},
|
|
focus_left: KeyEvent { code: KeyCode::Left, modifiers: KeyModifiers::empty()},
|
|
focus_above: KeyEvent { code: KeyCode::Up, modifiers: KeyModifiers::empty()},
|
|
focus_below: KeyEvent { code: KeyCode::Down, modifiers: KeyModifiers::empty()},
|
|
exit: KeyEvent { code: KeyCode::Char('c'), modifiers: KeyModifiers::CONTROL},
|
|
exit_popup: KeyEvent { code: KeyCode::Esc, modifiers: KeyModifiers::empty()},
|
|
open_commit: KeyEvent { code: KeyCode::Char('c'), modifiers: KeyModifiers::empty()},
|
|
open_commit_editor: KeyEvent { code: KeyCode::Char('e'), modifiers:KeyModifiers::CONTROL},
|
|
open_help: KeyEvent { code: KeyCode::Char('h'), modifiers: KeyModifiers::empty()},
|
|
move_left: KeyEvent { code: KeyCode::Left, modifiers: KeyModifiers::empty()},
|
|
move_right: KeyEvent { code: KeyCode::Right, modifiers: KeyModifiers::empty()},
|
|
home: KeyEvent { code: KeyCode::Home, modifiers: KeyModifiers::empty()},
|
|
end: KeyEvent { code: KeyCode::End, modifiers: KeyModifiers::empty()},
|
|
move_up: KeyEvent { code: KeyCode::Up, modifiers: KeyModifiers::empty()},
|
|
move_down: KeyEvent { code: KeyCode::Down, modifiers: KeyModifiers::empty()},
|
|
page_down: KeyEvent { code: KeyCode::PageDown, modifiers: KeyModifiers::empty()},
|
|
page_up: KeyEvent { code: KeyCode::PageUp, modifiers: KeyModifiers::empty()},
|
|
shift_up: KeyEvent { code: KeyCode::Up, modifiers: KeyModifiers::SHIFT},
|
|
shift_down: KeyEvent { code: KeyCode::Down, modifiers: KeyModifiers::SHIFT},
|
|
enter: KeyEvent { code: KeyCode::Enter, modifiers: KeyModifiers::empty()},
|
|
edit_file: KeyEvent { code: KeyCode::Char('e'), modifiers: KeyModifiers::empty()},
|
|
status_stage_all: KeyEvent { code: KeyCode::Char('a'), modifiers: KeyModifiers::empty()},
|
|
status_reset_item: KeyEvent { code: KeyCode::Char('D'), modifiers: KeyModifiers::SHIFT},
|
|
status_ignore_file: KeyEvent { code: KeyCode::Char('i'), modifiers: KeyModifiers::empty()},
|
|
stashing_save: KeyEvent { code: KeyCode::Char('s'), modifiers: KeyModifiers::empty()},
|
|
stashing_toggle_untracked: KeyEvent { code: KeyCode::Char('u'), modifiers: KeyModifiers::empty()},
|
|
stashing_toggle_index: KeyEvent { code: KeyCode::Char('i'), modifiers: KeyModifiers::empty()},
|
|
stash_open: KeyEvent { code: KeyCode::Right, modifiers: KeyModifiers::empty()},
|
|
stash_drop: KeyEvent { code: KeyCode::Char('D'), modifiers: KeyModifiers::SHIFT},
|
|
cmd_bar_toggle: KeyEvent { code: KeyCode::Char('.'), modifiers: KeyModifiers::empty()},
|
|
log_tag_commit: KeyEvent { code: KeyCode::Char('t'), modifiers: KeyModifiers::empty()},
|
|
commit_amend: KeyEvent { code: KeyCode::Char('a'), modifiers: KeyModifiers::CONTROL},
|
|
copy: KeyEvent { code: KeyCode::Char('y'), modifiers: KeyModifiers::empty()},
|
|
create_branch: KeyEvent { code: KeyCode::Char('b'), modifiers: KeyModifiers::empty()},
|
|
}
|
|
}
|
|
}
|
|
impl KeyConfig {
|
|
fn save(&self) -> Result<()> {
|
|
let config_file = Self::get_config_file()?;
|
|
let mut file = File::create(config_file)?;
|
|
let data = to_string_pretty(self, PrettyConfig::default())?;
|
|
file.write_all(data.as_bytes())?;
|
|
Ok(())
|
|
}
|
|
|
|
fn get_config_file() -> Result<PathBuf> {
|
|
let app_home = get_app_config_path()?;
|
|
Ok(app_home.join("key_config.ron"))
|
|
}
|
|
|
|
fn read_file(config_file: PathBuf) -> Result<Self> {
|
|
let mut f = File::open(config_file)?;
|
|
let mut buffer = Vec::new();
|
|
f.read_to_end(&mut buffer)?;
|
|
Ok(from_bytes(&buffer)?)
|
|
}
|
|
|
|
fn init_internal() -> Result<Self> {
|
|
let file = Self::get_config_file()?;
|
|
if file.exists() {
|
|
Ok(Self::read_file(file)?)
|
|
} else {
|
|
let def = Self::default();
|
|
if def.save().is_err() {
|
|
log::warn!(
|
|
"failed to store default key config to disk."
|
|
)
|
|
}
|
|
Ok(def)
|
|
}
|
|
}
|
|
|
|
pub fn init() -> Self {
|
|
Self::init_internal().unwrap_or_default()
|
|
}
|
|
}
|
|
|
|
// The hint follows apple design
|
|
// http://xahlee.info/comp/unicode_computing_symbols.html
|
|
pub fn get_hint(ev: KeyEvent) -> String {
|
|
match ev.code {
|
|
KeyCode::Char(c) => {
|
|
format!("{}{}", get_modifier_hint(ev.modifiers), c)
|
|
}
|
|
KeyCode::Enter => {
|
|
format!("{}\u{23ce}", get_modifier_hint(ev.modifiers)) //⏎
|
|
}
|
|
KeyCode::Left => {
|
|
format!("{}\u{2190}", get_modifier_hint(ev.modifiers)) //←
|
|
}
|
|
KeyCode::Right => {
|
|
format!("{}\u{2192}", get_modifier_hint(ev.modifiers)) //→
|
|
}
|
|
KeyCode::Up => {
|
|
format!("{}\u{2191}", get_modifier_hint(ev.modifiers)) //↑
|
|
}
|
|
KeyCode::Down => {
|
|
format!("{}\u{2193}", get_modifier_hint(ev.modifiers)) //↓
|
|
}
|
|
KeyCode::Backspace => {
|
|
format!("{}\u{232b}", get_modifier_hint(ev.modifiers)) //⌫
|
|
}
|
|
KeyCode::Home => {
|
|
format!("{}\u{2912}", get_modifier_hint(ev.modifiers)) //⤒
|
|
}
|
|
KeyCode::End => {
|
|
format!("{}\u{2913}", get_modifier_hint(ev.modifiers)) //⤓
|
|
}
|
|
KeyCode::PageUp => {
|
|
format!("{}\u{21de}", get_modifier_hint(ev.modifiers)) //⇞
|
|
}
|
|
KeyCode::PageDown => {
|
|
format!("{}\u{21df}", get_modifier_hint(ev.modifiers)) //⇟
|
|
}
|
|
KeyCode::Tab => {
|
|
format!("{}\u{21e5}", get_modifier_hint(ev.modifiers)) //⇥
|
|
}
|
|
KeyCode::BackTab => {
|
|
format!("{}\u{21e4}", get_modifier_hint(ev.modifiers)) //⇤
|
|
}
|
|
KeyCode::Delete => {
|
|
format!("{}\u{2326}", get_modifier_hint(ev.modifiers)) //⌦
|
|
}
|
|
KeyCode::Insert => {
|
|
format!("{}\u{2380}", get_modifier_hint(ev.modifiers)) //⎀
|
|
}
|
|
KeyCode::Esc => {
|
|
format!("{}\u{238b}", get_modifier_hint(ev.modifiers)) //⎋
|
|
}
|
|
KeyCode::F(u) => {
|
|
format!("{}F{}", get_modifier_hint(ev.modifiers), u)
|
|
}
|
|
KeyCode::Null => get_modifier_hint(ev.modifiers),
|
|
}
|
|
}
|
|
|
|
fn get_modifier_hint(modifier: KeyModifiers) -> String {
|
|
match modifier {
|
|
KeyModifiers::CONTROL => "^".to_string(),
|
|
KeyModifiers::SHIFT => {
|
|
"\u{21e7}".to_string() //⇧
|
|
}
|
|
KeyModifiers::ALT => {
|
|
"\u{2325}".to_string() //⌥
|
|
}
|
|
_ => "".to_string(),
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::{get_hint, KeyConfig};
|
|
use crossterm::event::{KeyCode, KeyEvent, KeyModifiers};
|
|
|
|
#[test]
|
|
fn test_get_hint() {
|
|
let h = get_hint(KeyEvent {
|
|
code: KeyCode::Char('c'),
|
|
modifiers: KeyModifiers::CONTROL,
|
|
});
|
|
assert_eq!(h, "^c");
|
|
}
|
|
|
|
#[test]
|
|
fn test_load_vim_style_example() {
|
|
assert_eq!(
|
|
KeyConfig::read_file(
|
|
"assets/vim_style_key_config.ron".into()
|
|
)
|
|
.is_ok(),
|
|
true
|
|
);
|
|
}
|
|
}
|