From 61bc6da8217be26010ff13dbfa0f0496445da920 Mon Sep 17 00:00:00 2001
From: Diego Frias <96022404+dzfrias@users.noreply.github.com>
Date: Mon, 20 Mar 2023 09:26:51 -0700
Subject: [PATCH] fix: panic on key up/down (#16)

---
 src/lib.rs | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index 0b5db27..fb11983 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -122,7 +122,10 @@ impl TreeState {
         let new_index = current_index.map_or(0, |current_index| {
             current_index.saturating_sub(1).min(visible.len() - 1)
         });
-        let new_identifier = visible[new_index].identifier.clone();
+        let new_identifier = visible
+            .get(new_index)
+            .map(|o| o.identifier.clone())
+            .unwrap_or_default();
         self.select(new_identifier);
     }
 
@@ -137,7 +140,10 @@ impl TreeState {
         let new_index = current_index.map_or(0, |current_index| {
             current_index.saturating_add(1).min(visible.len() - 1)
         });
-        let new_identifier = visible[new_index].identifier.clone();
+        let new_identifier = visible
+            .get(new_index)
+            .map(|o| o.identifier.clone())
+            .unwrap_or_default();
         self.select(new_identifier);
     }