Know about focus in marker pane

1 files changed, 10 insertions, 0 deletions

diff --git a/src/interactive/widgets/mark.rs b/src/interactive/widgets/mark.rs
index 67596bd..047c71a 100644
--- a/src/interactive/widgets/mark.rs
+++ b/src/interactive/widgets/mark.rs
@@ -18,6 +18,7 @@ pub struct MarkPane {
     selected: Option<usize>,
     marked: EntryMarkMap,
     list: List,
+    has_focus: bool,
 }
 
 pub struct MarkPaneProps {
@@ -25,7 +26,16 @@ pub struct MarkPaneProps {
 }
 
 impl MarkPane {
+    pub fn set_focus(&mut self, has_focus: bool) {
+        self.has_focus = has_focus;
+        if has_focus {
+            self.selected = Some(self.marked.len().saturating_sub(1));
+        } else {
+            self.selected = None
+        }
+    }
     pub fn toggle_index(mut self, index: TreeIndex, tree: &Tree) -> Option<Self> {
+        // TODO: use HashMapEntry (Vacant/Occupied)
         if self.marked.get(&index).is_some() {
             self.marked.remove(&index);
         } else {