Skip to main content

tess/
keys.rs

1//! Custom keybindings loaded from `~/.config/tess/keys.toml`.
2//!
3//! Schema:
4//! ```toml
5//! [bindings]
6//! "j" = "scroll-down"
7//! "f1" = "!git status"
8//! ```
9
10use std::collections::HashMap;
11
12use crossterm::event::{KeyCode, KeyEvent, KeyModifiers};
13use serde::Deserialize;
14
15use crate::input::Command;
16
17#[derive(Debug, Clone)]
18pub enum BindingTarget {
19    Command(Command),
20    Shell(String),
21}
22
23#[derive(Debug, Clone)]
24pub struct KeyMap {
25    map: HashMap<KeyEvent, BindingTarget>,
26}
27
28impl KeyMap {
29    pub fn empty() -> Self {
30        Self { map: HashMap::new() }
31    }
32
33    /// Load keys from the global config dir (`/etc/tess/` or
34    /// `$TESS_GLOBAL_CONFIG_DIR`) and from `~/.config/tess/keys.toml`,
35    /// merging per individual binding key with local winning.
36    ///
37    /// Global parse errors warn on stderr and the global layer is treated as
38    /// empty; local parse errors fail startup.
39    pub fn load_layered() -> Result<Self, String> {
40        let mut bindings: HashMap<String, String> = HashMap::new();
41
42        // Global layer.
43        if let Some(dir) = crate::config_path::global_config_dir() {
44            let path = dir.join("keys.toml");
45            if path.exists() {
46                match std::fs::read_to_string(&path) {
47                    Ok(text) => {
48                        match toml::from_str::<KeysConfig>(&text) {
49                            Ok(cfg) => {
50                                for (k, v) in cfg.bindings {
51                                    bindings.insert(k, v);
52                                }
53                            }
54                            Err(e) => eprintln!(
55                                "tess: warning: keys.toml: {}: {e}; ignoring global config",
56                                path.display()
57                            ),
58                        }
59                    }
60                    Err(e) => eprintln!(
61                        "tess: warning: keys.toml: {}: {e}; ignoring global config",
62                        path.display()
63                    ),
64                }
65            }
66        }
67
68        // Local layer.
69        if let Some(dir) = crate::config_path::user_config_dir() {
70            let path = dir.join("keys.toml");
71            if path.exists() {
72                let text = std::fs::read_to_string(&path)
73                    .map_err(|e| format!("keys.toml: reading {}: {e}", path.display()))?;
74                let cfg: KeysConfig = toml::from_str(&text)
75                    .map_err(|e| format!("keys.toml: parsing {}: {e}", path.display()))?;
76                for (k, v) in cfg.bindings {
77                    bindings.insert(k, v);
78                }
79            }
80        }
81
82        // Build the final KeyMap from the merged bindings table.
83        let mut map = HashMap::with_capacity(bindings.len());
84        for (key_spec, action) in bindings {
85            let key = parse_key_spec(&key_spec)
86                .map_err(|e| format!("keys.toml: '{key_spec}': {e}"))?;
87            reject_forbidden_key(&key, &key_spec)
88                .map_err(|e| format!("keys.toml: {e}"))?;
89            let target = parse_action(&action)
90                .map_err(|e| format!("keys.toml: '{key_spec}': {e}"))?;
91            map.insert(key, target);
92        }
93        Ok(Self { map })
94    }
95
96    pub fn load_from_str(toml_text: &str) -> Result<Self, String> {
97        let cfg: KeysConfig = toml::from_str(toml_text)
98            .map_err(|e| format!("parsing: {e}"))?;
99        let mut map = HashMap::with_capacity(cfg.bindings.len());
100        for (key_spec, action) in cfg.bindings {
101            let key = parse_key_spec(&key_spec)
102                .map_err(|e| format!("'{key_spec}': {e}"))?;
103            reject_forbidden_key(&key, &key_spec)?;
104            let target = parse_action(&action)
105                .map_err(|e| format!("'{key_spec}': {e}"))?;
106            map.insert(key, target);
107        }
108        Ok(Self { map })
109    }
110
111    pub fn lookup(&self, key: &KeyEvent) -> Option<&BindingTarget> {
112        self.map.get(key)
113    }
114
115    pub fn is_empty(&self) -> bool {
116        self.map.is_empty()
117    }
118
119    /// Group user bindings by the kebab-case command name they target.
120    /// Shell-target bindings are excluded. Used by the help overlay to
121    /// replace default key displays with the user's chosen keys.
122    pub fn user_keys_by_command_name(&self) -> std::collections::HashMap<String, Vec<String>> {
123        let mut out: std::collections::HashMap<String, Vec<String>> =
124            std::collections::HashMap::new();
125        for (key, target) in &self.map {
126            let BindingTarget::Command(cmd) = target else { continue };
127            let Some(name) = command_to_kebab(cmd) else { continue };
128            out.entry(name.to_string())
129                .or_default()
130                .push(format_key_event(*key));
131        }
132        out
133    }
134}
135
136#[derive(Debug, Deserialize, Default)]
137struct KeysConfig {
138    #[serde(default)]
139    bindings: HashMap<String, String>,
140}
141
142/// Parse a key spec string into a `KeyEvent`.
143fn parse_key_spec(spec: &str) -> Result<KeyEvent, String> {
144    let lower = spec.to_lowercase();
145    let mut parts: Vec<&str> = lower.split('-').collect();
146    if parts.is_empty() {
147        return Err("empty key spec".to_string());
148    }
149    let key_part = parts.pop().unwrap();
150    let mut modifiers = KeyModifiers::NONE;
151    for m in &parts {
152        if m.is_empty() {
153            // Handle "ctrl--" (ctrl + dash). The trailing "-" became "".
154            continue;
155        }
156        match *m {
157            "ctrl" => modifiers |= KeyModifiers::CONTROL,
158            "alt" => modifiers |= KeyModifiers::ALT,
159            "shift" => modifiers |= KeyModifiers::SHIFT,
160            other => return Err(format!("unknown modifier '{other}'")),
161        }
162    }
163    let code = match key_part {
164        "esc" => KeyCode::Esc,
165        "enter" => KeyCode::Enter,
166        "tab" => KeyCode::Tab,
167        "backspace" => KeyCode::Backspace,
168        "space" => KeyCode::Char(' '),
169        "up" => KeyCode::Up,
170        "down" => KeyCode::Down,
171        "left" => KeyCode::Left,
172        "right" => KeyCode::Right,
173        "pgup" => KeyCode::PageUp,
174        "pgdn" => KeyCode::PageDown,
175        "home" => KeyCode::Home,
176        "end" => KeyCode::End,
177        "" => {
178            // The trailing piece was empty, meaning the key is literal "-"
179            // and the modifiers consumed everything else (e.g. "ctrl--").
180            KeyCode::Char('-')
181        }
182        s if s.starts_with('f') && s.len() > 1 => {
183            let n: u8 = s[1..].parse()
184                .map_err(|_| format!("unknown key '{s}'"))?;
185            KeyCode::F(n)
186        }
187        s if s.chars().count() == 1 => {
188            // For a BARE letter with no modifiers, an uppercase letter
189            // promotes to shift-prefix semantics ("J" == "shift-j").
190            // When modifiers are already present (e.g. "ctrl-J"), the
191            // user's intent is the literal letter — don't add SHIFT.
192            let original_char = spec.chars().last().unwrap();
193            if original_char.is_ascii_uppercase() && modifiers == KeyModifiers::NONE {
194                modifiers |= KeyModifiers::SHIFT;
195                KeyCode::Char(original_char.to_ascii_lowercase())
196            } else {
197                // Either lowercase letter, or uppercase with an explicit
198                // modifier — lowercase the char either way for consistency.
199                KeyCode::Char(original_char.to_ascii_lowercase())
200            }
201        }
202        other => return Err(format!("unknown key '{other}'")),
203    };
204    Ok(KeyEvent::new(code, modifiers))
205}
206
207fn reject_forbidden_key(key: &KeyEvent, original_spec: &str) -> Result<(), String> {
208    let forbidden = match (&key.code, key.modifiers) {
209        (KeyCode::Char('m'), KeyModifiers::NONE) => true,
210        (KeyCode::Char('\''), KeyModifiers::NONE) => true,
211        (KeyCode::Char('-'), KeyModifiers::NONE) => true,
212        (KeyCode::Char('x'), KeyModifiers::CONTROL) => true,
213        (KeyCode::Char(c), KeyModifiers::NONE) if c.is_ascii_digit() => true,
214        _ => false,
215    };
216    if forbidden {
217        return Err(format!(
218            "'{original_spec}' is part of a multi-key sequence and cannot be rebound"
219        ));
220    }
221    Ok(())
222}
223
224fn parse_action(action: &str) -> Result<BindingTarget, String> {
225    if let Some(shell_cmd) = action.strip_prefix('!') {
226        if shell_cmd.is_empty() {
227            return Err("shell binding requires a command after '!'".to_string());
228        }
229        return Ok(BindingTarget::Shell(shell_cmd.to_string()));
230    }
231    let cmd = command_from_kebab(action)
232        .ok_or_else(|| format!("unknown command '{action}'"))?;
233    Ok(BindingTarget::Command(cmd))
234}
235
236/// Map a kebab-case command name to the corresponding `Command` enum variant.
237fn command_from_kebab(name: &str) -> Option<Command> {
238    match name {
239        "scroll-down" => Some(Command::ScrollLines(1)),
240        "scroll-up" => Some(Command::ScrollLines(-1)),
241        "scroll-logical-down" => Some(Command::ScrollLogicalLines(1)),
242        "scroll-logical-up" => Some(Command::ScrollLogicalLines(-1)),
243        "page-down" => Some(Command::PageDown),
244        "page-up" => Some(Command::PageUp),
245        "half-page-down" => Some(Command::HalfPageDown),
246        "half-page-up" => Some(Command::HalfPageUp),
247        "quit" => Some(Command::Quit),
248        "refresh" => Some(Command::Refresh),
249        "reload" => Some(Command::Reload),
250        "toggle-line-numbers" => Some(Command::ToggleLineNumbers),
251        "toggle-chop" => Some(Command::ToggleChop),
252        "toggle-follow" => Some(Command::ToggleFollow),
253        "toggle-prettify" => Some(Command::TogglePrettify),
254        "search-forward" => Some(Command::SearchForward),
255        "search-backward" => Some(Command::SearchBackward),
256        "next-match" => Some(Command::NextMatch),
257        "previous-match" => Some(Command::PreviousMatch),
258        "option-prefix" => Some(Command::OptionPrefix),
259        "goto-line" => Some(Command::GotoLine),
260        "goto-record" => Some(Command::GotoRecord),
261        "goto-percent" => Some(Command::GotoPercent),
262        "mark-set" => Some(Command::MarkSet),
263        "mark-jump" => Some(Command::MarkJump),
264        "ctrl-x-prefix" => Some(Command::CtrlXPrefix),
265        "jump-previous" => Some(Command::JumpPrevious),
266        "shell-escape" => Some(Command::ShellEscape),
267        "cancel" => Some(Command::Cancel),
268        _ => None,
269    }
270}
271
272/// Inverse of `command_from_kebab` — covers the same command set.
273fn command_to_kebab(cmd: &Command) -> Option<&'static str> {
274    match cmd {
275        Command::ScrollLines(1) => Some("scroll-down"),
276        Command::ScrollLines(-1) => Some("scroll-up"),
277        Command::ScrollLogicalLines(1) => Some("scroll-logical-down"),
278        Command::ScrollLogicalLines(-1) => Some("scroll-logical-up"),
279        Command::PageDown => Some("page-down"),
280        Command::PageUp => Some("page-up"),
281        Command::HalfPageDown => Some("half-page-down"),
282        Command::HalfPageUp => Some("half-page-up"),
283        Command::Quit => Some("quit"),
284        Command::Refresh => Some("refresh"),
285        Command::Reload => Some("reload"),
286        Command::ToggleLineNumbers => Some("toggle-line-numbers"),
287        Command::ToggleChop => Some("toggle-chop"),
288        Command::ToggleFollow => Some("toggle-follow"),
289        Command::TogglePrettify => Some("toggle-prettify"),
290        Command::SearchForward => Some("search-forward"),
291        Command::SearchBackward => Some("search-backward"),
292        Command::NextMatch => Some("next-match"),
293        Command::PreviousMatch => Some("previous-match"),
294        Command::OptionPrefix => Some("option-prefix"),
295        Command::GotoLine => Some("goto-line"),
296        Command::GotoRecord => Some("goto-record"),
297        Command::GotoPercent => Some("goto-percent"),
298        Command::MarkSet => Some("mark-set"),
299        Command::MarkJump => Some("mark-jump"),
300        Command::CtrlXPrefix => Some("ctrl-x-prefix"),
301        Command::JumpPrevious => Some("jump-previous"),
302        Command::ShellEscape => Some("shell-escape"),
303        Command::Cancel => Some("cancel"),
304        _ => None,
305    }
306}
307
308fn format_key_event(ke: KeyEvent) -> String {
309    let ctrl  = ke.modifiers.contains(KeyModifiers::CONTROL);
310    let alt   = ke.modifiers.contains(KeyModifiers::ALT);
311    let shift = ke.modifiers.contains(KeyModifiers::SHIFT);
312
313    // Convention: SHIFT-alone on an ASCII letter displays as the uppercase
314    // letter with no "Shift-" prefix (matches KEY_REGISTRY's `"J"` form).
315    if shift && !ctrl && !alt {
316        if let KeyCode::Char(c) = ke.code {
317            if c.is_ascii_alphabetic() {
318                return c.to_ascii_uppercase().to_string();
319            }
320        }
321    }
322
323    let mut parts: Vec<&'static str> = Vec::new();
324    if ctrl  { parts.push("Ctrl"); }
325    if alt   { parts.push("Alt"); }
326    if shift { parts.push("Shift"); }
327
328    let key = match ke.code {
329        KeyCode::Char(' ') => "Space".to_string(),
330        KeyCode::Char(c)   => c.to_string(),
331        KeyCode::F(n)      => format!("F{n}"),
332        KeyCode::Esc       => "Esc".into(),
333        KeyCode::Enter     => "Enter".into(),
334        KeyCode::Tab       => "Tab".into(),
335        KeyCode::Backspace => "Backspace".into(),
336        KeyCode::Up        => "\u{2191}".into(),
337        KeyCode::Down      => "\u{2193}".into(),
338        KeyCode::Left      => "\u{2190}".into(),
339        KeyCode::Right     => "\u{2192}".into(),
340        KeyCode::Home      => "Home".into(),
341        KeyCode::End       => "End".into(),
342        KeyCode::PageUp    => "PgUp".into(),
343        KeyCode::PageDown  => "PgDn".into(),
344        other              => format!("{other:?}"),
345    };
346    if parts.is_empty() { key } else { format!("{}-{}", parts.join("-"), key) }
347}
348
349#[cfg(test)]
350mod tests {
351    use super::*;
352    use std::sync::Mutex;
353
354    static HOME_LOCK: Mutex<()> = Mutex::new(());
355
356    #[test]
357    fn parse_empty_file_returns_empty_map() {
358        let m = KeyMap::load_from_str("").unwrap();
359        assert!(m.is_empty());
360    }
361
362    #[test]
363    fn parse_single_binding() {
364        let toml = r#"
365[bindings]
366"j" = "scroll-down"
367"#;
368        let m = KeyMap::load_from_str(toml).unwrap();
369        let key = KeyEvent::new(KeyCode::Char('j'), KeyModifiers::NONE);
370        assert!(matches!(m.lookup(&key), Some(BindingTarget::Command(Command::ScrollLines(1)))));
371    }
372
373    #[test]
374    fn parse_named_special_key() {
375        let toml = r#"
376[bindings]
377"f1" = "toggle-line-numbers"
378"esc" = "cancel"
379"#;
380        let m = KeyMap::load_from_str(toml).unwrap();
381        let f1 = KeyEvent::new(KeyCode::F(1), KeyModifiers::NONE);
382        let esc = KeyEvent::new(KeyCode::Esc, KeyModifiers::NONE);
383        assert!(matches!(m.lookup(&f1), Some(BindingTarget::Command(Command::ToggleLineNumbers))));
384        assert!(matches!(m.lookup(&esc), Some(BindingTarget::Command(Command::Cancel))));
385    }
386
387    #[test]
388    fn parse_modifier_combinations() {
389        let toml = r#"
390[bindings]
391"ctrl-r" = "reload"
392"shift-tab" = "scroll-logical-up"
393"#;
394        let m = KeyMap::load_from_str(toml).unwrap();
395        let ctrl_r = KeyEvent::new(KeyCode::Char('r'), KeyModifiers::CONTROL);
396        let shift_tab = KeyEvent::new(KeyCode::Tab, KeyModifiers::SHIFT);
397        assert!(matches!(m.lookup(&ctrl_r), Some(BindingTarget::Command(Command::Reload))));
398        assert!(matches!(m.lookup(&shift_tab), Some(BindingTarget::Command(Command::ScrollLogicalLines(-1)))));
399    }
400
401    #[test]
402    fn case_letter_resolves_to_shift_prefix() {
403        let toml = r#"
404[bindings]
405"J" = "scroll-logical-down"
406"#;
407        let m = KeyMap::load_from_str(toml).unwrap();
408        let shift_j = KeyEvent::new(KeyCode::Char('j'), KeyModifiers::SHIFT);
409        assert!(matches!(m.lookup(&shift_j), Some(BindingTarget::Command(Command::ScrollLogicalLines(1)))));
410    }
411
412    #[test]
413    fn forbidden_keys_error_at_parse() {
414        for key in &["m", "'", "-", "ctrl-x", "0", "5", "9"] {
415            let toml = format!(r#"
416[bindings]
417"{key}" = "quit"
418"#);
419            let err = KeyMap::load_from_str(&toml).unwrap_err();
420            assert!(err.contains("multi-key sequence"),
421                    "key '{key}' should be forbidden: {err}");
422        }
423    }
424
425    #[test]
426    fn unknown_command_name_errors() {
427        let toml = r#"
428[bindings]
429"j" = "definitely-not-a-real-command"
430"#;
431        let err = KeyMap::load_from_str(toml).unwrap_err();
432        assert!(err.contains("unknown command"));
433    }
434
435    #[test]
436    fn empty_shell_binding_errors() {
437        let toml = r#"
438[bindings]
439"f1" = "!"
440"#;
441        let err = KeyMap::load_from_str(toml).unwrap_err();
442        assert!(err.contains("requires a command"));
443    }
444
445    #[test]
446    fn parse_inline_shell_binding() {
447        let toml = r#"
448[bindings]
449"f2" = "!git status"
450"#;
451        let m = KeyMap::load_from_str(toml).unwrap();
452        let f2 = KeyEvent::new(KeyCode::F(2), KeyModifiers::NONE);
453        match m.lookup(&f2) {
454            Some(BindingTarget::Shell(cmd)) => assert_eq!(cmd, "git status"),
455            other => panic!("expected Shell, got {:?}", other),
456        }
457    }
458
459    #[test]
460    fn lookup_returns_none_for_unbound_key() {
461        let toml = r#"
462[bindings]
463"j" = "scroll-down"
464"#;
465        let m = KeyMap::load_from_str(toml).unwrap();
466        let other = KeyEvent::new(KeyCode::Char('k'), KeyModifiers::NONE);
467        assert!(m.lookup(&other).is_none());
468    }
469
470    #[test]
471    fn ctrl_uppercase_letter_does_not_add_shift() {
472        // "ctrl-J" should be Ctrl + 'j', NOT Ctrl + Shift + 'j'.
473        let toml = r#"
474[bindings]
475"ctrl-J" = "reload"
476"#;
477        let m = KeyMap::load_from_str(toml).unwrap();
478        let ctrl_j = KeyEvent::new(KeyCode::Char('j'), KeyModifiers::CONTROL);
479        assert!(matches!(m.lookup(&ctrl_j), Some(BindingTarget::Command(Command::Reload))),
480                "ctrl-J should resolve to Ctrl+j without Shift");
481        let ctrl_shift_j = KeyEvent::new(KeyCode::Char('j'), KeyModifiers::CONTROL | KeyModifiers::SHIFT);
482        assert!(m.lookup(&ctrl_shift_j).is_none(),
483                "ctrl-J should NOT also match Ctrl+Shift+j");
484    }
485
486    #[test]
487    fn user_remaps_by_command_name_groups_keys() {
488        let toml = r#"
489[bindings]
490"f3" = "scroll-down"
491"f4" = "scroll-down"
492"f5" = "quit"
493"#;
494        let m = KeyMap::load_from_str(toml).unwrap();
495        let groups = m.user_keys_by_command_name();
496        let mut down = groups.get("scroll-down").cloned().unwrap_or_default();
497        down.sort();
498        assert_eq!(down, vec!["F3".to_string(), "F4".to_string()]);
499        assert_eq!(groups.get("quit").cloned().unwrap_or_default(), vec!["F5".to_string()]);
500    }
501
502    #[test]
503    fn dash_with_modifier_is_a_real_key() {
504        // "ctrl--" should resolve to Ctrl + '-' (a valid bind).
505        // (Bare "-" is forbidden by reject_forbidden_key, but Ctrl-- isn't.)
506        let toml = r#"
507[bindings]
508"ctrl--" = "refresh"
509"#;
510        let m = KeyMap::load_from_str(toml).unwrap();
511        let ctrl_dash = KeyEvent::new(KeyCode::Char('-'), KeyModifiers::CONTROL);
512        assert!(matches!(m.lookup(&ctrl_dash), Some(BindingTarget::Command(Command::Refresh))));
513    }
514
515    #[test]
516    fn format_key_event_renders_modifier_combos() {
517        // Ctrl alone
518        assert_eq!(
519            format_key_event(KeyEvent::new(KeyCode::Char('r'), KeyModifiers::CONTROL)),
520            "Ctrl-r",
521        );
522        // Shift alone on a letter → uppercase, no prefix
523        assert_eq!(
524            format_key_event(KeyEvent::new(KeyCode::Char('j'), KeyModifiers::SHIFT)),
525            "J",
526        );
527        // Shift alone on a non-letter (e.g. Tab) → "Shift-Tab"
528        assert_eq!(
529            format_key_event(KeyEvent::new(KeyCode::Tab, KeyModifiers::SHIFT)),
530            "Shift-Tab",
531        );
532        // Plain lowercase letter → "j" (no uppercasing)
533        assert_eq!(
534            format_key_event(KeyEvent::new(KeyCode::Char('j'), KeyModifiers::NONE)),
535            "j",
536        );
537        // Function key
538        assert_eq!(
539            format_key_event(KeyEvent::new(KeyCode::F(3), KeyModifiers::NONE)),
540            "F3",
541        );
542        // Ctrl+Shift+letter — composed prefix, lowercase stored char
543        assert_eq!(
544            format_key_event(KeyEvent::new(KeyCode::Char('x'), KeyModifiers::CONTROL | KeyModifiers::SHIFT)),
545            "Ctrl-Shift-x",
546        );
547    }
548
549    #[test]
550    fn command_kebab_round_trip() {
551        // Every name in command_from_kebab must round-trip through command_to_kebab.
552        let names = [
553            "scroll-down", "scroll-up", "scroll-logical-down", "scroll-logical-up",
554            "page-down", "page-up", "half-page-down", "half-page-up",
555            "quit", "refresh", "reload",
556            "toggle-line-numbers", "toggle-chop", "toggle-follow", "toggle-prettify",
557            "search-forward", "search-backward", "next-match", "previous-match",
558            "option-prefix", "goto-line", "goto-record", "goto-percent",
559            "mark-set", "mark-jump", "ctrl-x-prefix", "jump-previous",
560            "shell-escape", "cancel",
561        ];
562        for name in &names {
563            let cmd = command_from_kebab(name).expect(&format!("from_kebab failed for {name}"));
564            let back = command_to_kebab(&cmd).expect(&format!("to_kebab failed for {name}"));
565            assert_eq!(back, *name, "round-trip mismatch for {name}");
566        }
567    }
568
569    #[test]
570    fn shell_bindings_are_excluded_from_user_keys() {
571        let toml = r#"
572[bindings]
573"f2" = "!git status"
574"f3" = "scroll-down"
575"#;
576        let m = KeyMap::load_from_str(toml).unwrap();
577        let groups = m.user_keys_by_command_name();
578        assert!(!groups.values().any(|v| v.contains(&"F2".to_string())),
579                "shell-bound F2 should not appear: {groups:?}");
580        assert_eq!(groups.get("scroll-down").cloned().unwrap_or_default(), vec!["F3".to_string()]);
581    }
582
583    #[test]
584    fn layered_keys_local_overrides_global_per_binding() {
585        let _guard = HOME_LOCK.lock().unwrap();
586        let prev_home = std::env::var_os("HOME");
587        let prev_global = std::env::var_os("TESS_GLOBAL_CONFIG_DIR");
588
589        let home = tempfile::tempdir().unwrap();
590        let global = tempfile::tempdir().unwrap();
591
592        std::env::set_var("HOME", home.path());
593        std::env::set_var("TESS_GLOBAL_CONFIG_DIR", global.path());
594
595        std::fs::write(
596            global.path().join("keys.toml"),
597            r#"
598[bindings]
599"j" = "scroll-down"
600"k" = "scroll-up"
601"#,
602        )
603        .unwrap();
604
605        let cfg_dir = home.path().join(".config").join("tess");
606        std::fs::create_dir_all(&cfg_dir).unwrap();
607        std::fs::write(
608            cfg_dir.join("keys.toml"),
609            r#"
610[bindings]
611"j" = "page-down"
612"#,
613        )
614        .unwrap();
615
616        let km = KeyMap::load_layered().unwrap();
617
618        // j: local wins (page-down)
619        let j = crossterm::event::KeyEvent::new(
620            crossterm::event::KeyCode::Char('j'),
621            crossterm::event::KeyModifiers::NONE,
622        );
623        match km.lookup(&j) {
624            Some(BindingTarget::Command(cmd)) => {
625                let dbg = format!("{cmd:?}");
626                assert!(dbg.to_lowercase().contains("page"), "got: {dbg}");
627            }
628            other => panic!("expected Command(PageDown), got {other:?}"),
629        }
630
631        // k: global survives (scroll-up)
632        let k = crossterm::event::KeyEvent::new(
633            crossterm::event::KeyCode::Char('k'),
634            crossterm::event::KeyModifiers::NONE,
635        );
636        match km.lookup(&k) {
637            Some(BindingTarget::Command(cmd)) => {
638                assert!(matches!(cmd, Command::ScrollLines(n) if *n < 0), "got: {cmd:?}");
639            }
640            other => panic!("expected Command(ScrollLines(-1)), got {other:?}"),
641        }
642
643        match prev_home {
644            Some(v) => std::env::set_var("HOME", v),
645            None => std::env::remove_var("HOME"),
646        }
647        match prev_global {
648            Some(v) => std::env::set_var("TESS_GLOBAL_CONFIG_DIR", v),
649            None => std::env::remove_var("TESS_GLOBAL_CONFIG_DIR"),
650        }
651    }
652
653    #[test]
654    fn layered_keys_warns_on_bad_global() {
655        let _guard = HOME_LOCK.lock().unwrap();
656        let prev_home = std::env::var_os("HOME");
657        let prev_global = std::env::var_os("TESS_GLOBAL_CONFIG_DIR");
658
659        let home = tempfile::tempdir().unwrap();
660        let global = tempfile::tempdir().unwrap();
661
662        std::env::set_var("HOME", home.path());
663        std::env::set_var("TESS_GLOBAL_CONFIG_DIR", global.path());
664
665        std::fs::write(
666            global.path().join("keys.toml"),
667            "= = not valid",
668        )
669        .unwrap();
670
671        // Local missing — should still succeed.
672        let km = KeyMap::load_layered().unwrap();
673        assert!(km.is_empty());
674
675        match prev_home {
676            Some(v) => std::env::set_var("HOME", v),
677            None => std::env::remove_var("HOME"),
678        }
679        match prev_global {
680            Some(v) => std::env::set_var("TESS_GLOBAL_CONFIG_DIR", v),
681            None => std::env::remove_var("TESS_GLOBAL_CONFIG_DIR"),
682        }
683    }
684}