@@ -269,41 +269,11 @@ pub enum KeymapResult {
269269 Cancelled(Vec<KeyEvent>),
270270}
271271
272−#[derive(Debug, Clone, PartialEq, Deserialize)]
273−#[serde(transparent)]
274−pub struct Keymap(KeyTrie);
275−
276272/// A map of command names to keybinds that will execute the command.
277273pub type ReverseKeymap = HashMap<String, Vec<Vec<KeyEvent>>>;
278274
279−impl Keymap {
280− pub fn new(root: KeyTrie) -> Self {
281− Keymap(root)
282− }
283−}
284−
285−impl Deref for Keymap {
286− type Target = KeyTrie;
287−
288− fn deref(&self) -> &Self::Target {
289− &self.0
290− }
291−}
292−
293−impl DerefMut for Keymap {
294− fn deref_mut(&mut self) -> &mut Self::Target {
295− &mut self.0
296− }
297−}
298−
299−impl Default for Keymap {
300− fn default() -> Self {
301− Self(KeyTrie::Node(KeyTrieNode::default()))
302− }
303−}
304−
305275pub struct Keymaps {
306− pub map: Box<dyn DynAccess<HashMap<Mode, Keymap>>>,
276+ pub map: Box<dyn DynAccess<HashMap<Mode, KeyTrie>>>,
307277 /// Stores pending keys waiting for the next key. This is relative to a
308278 /// sticky node if one is in use.
309279 state: Vec<KeyEvent>,
@@ -312,7 +282,7 @@ pub struct Keymaps {
312282}
313283
314284impl Keymaps {
315− pub fn new(map: Box<dyn DynAccess<HashMap<Mode, Keymap>>>) -> Self {
285+ pub fn new(map: Box<dyn DynAccess<HashMap<Mode, KeyTrie>>>) -> Self {
316286 Self {
317287 map,
318288 state: Vec::new(),
@@ -320,7 +290,7 @@ impl Keymaps {
320290 }
321291 }
322292
323− pub fn map(&self) -> DynGuard<HashMap<Mode, Keymap>> {
293+ pub fn map(&self) -> DynGuard<HashMap<Mode, KeyTrie>> {
324294 self.map.load()
325295 }
326296
@@ -352,7 +322,7 @@ impl Keymaps {
352322 let first = self.state.get(0).unwrap_or(&key);
353323 let trie_node = match self.sticky {
354324 Some(ref trie) => Cow::Owned(KeyTrie::Node(trie.clone())),
355− None => Cow::Borrowed(&keymap.0),
325+ None => Cow::Borrowed(keymap),
356326 };
357327
358328 let trie = match trie_node.search(&[*first]) {
@@ -395,9 +365,13 @@ impl Default for Keymaps {
395365}
396366
397367/// Merge default config keys with user overwritten keys for custom user config.
398−pub fn merge_keys(dst: &mut HashMap<Mode, Keymap>, mut delta: HashMap<Mode, Keymap>) {
368+pub fn merge_keys(dst: &mut HashMap<Mode, KeyTrie>, mut delta: HashMap<Mode, KeyTrie>) {
399369 for (mode, keys) in dst {
400− keys.merge_nodes(delta.remove(mode).unwrap_or_default().0)
370+ keys.merge_nodes(
371+ delta
372+ .remove(mode)
373+ .unwrap_or_else(|| KeyTrie::Node(KeyTrieNode::default())),
374+ )
401375 }
402376}
403377
@@ -426,8 +400,7 @@ mod tests {
426400 #[test]
427401 fn merge_partial_keys() {
428402 let keymap = hashmap! {
429− Mode::Normal => Keymap::new(
430− keymap!({ "Normal mode"
403+ Mode::Normal => keymap!({ "Normal mode"
431404 "i" => normal_mode,
432405 "无" => insert_mode,
433406 "z" => jump_backward,
@@ -436,7 +409,6 @@ mod tests {
436409 "g" => delete_char_forward,
437410 },
438411 })
439− )
440412 };
441413 let mut merged_keyamp = default();
442414 merge_keys(&mut merged_keyamp, keymap.clone());
@@ -501,8 +473,7 @@ mod tests {
501473 #[test]
502474 fn order_should_be_set() {
503475 let keymap = hashmap! {
504− Mode::Normal => Keymap::new(
505− keymap!({ "Normal mode"
476+ Mode::Normal => keymap!({ "Normal mode"
506477 "space" => { ""
507478 "s" => { ""
508479 "v" => vsplit,
@@ -510,7 +481,6 @@ mod tests {
510481 },
511482 },
512483 })
513− )
514484 };
515485 let mut merged_keyamp = default();
516486 merge_keys(&mut merged_keyamp, keymap.clone());
@@ -553,7 +523,7 @@ mod tests {
553523 },
554524 "j" | "k" => move_line_down,
555525 });
556− let keymap = Keymap::new(normal_mode);
526+ let keymap = normal_mode;
557527 let mut reverse_map = keymap.reverse_map();
558528
559529 // sort keybindings in order to have consistent tests
@@ -601,7 +571,7 @@ mod tests {
601571 modifiers: KeyModifiers::NONE,
602572 };
603573
604− let expectation = Keymap::new(KeyTrie::Node(KeyTrieNode::new(
574+ let expectation = KeyTrie::Node(KeyTrieNode::new(
605575 "",
606576 hashmap! {
607577 key => KeyTrie::Sequence(vec!{
@@ -618,7 +588,7 @@ mod tests {
618588 })
619589 },
620590 vec![key],
621− )));
591+ ));
622592
623593 assert_eq!(toml::from_str(keys), Ok(expectation));
624594 }