From b2bf10f260cc03f47e29db2df0114687c7409611 Mon Sep 17 00:00:00 2001
From: Joseph Ferano <joseph@ferano.io>
Date: Thu, 8 Jun 2023 10:31:54 +0700
Subject: [PATCH] Scroll columns with a lot of tasks

---
 src/app.rs | 2 +-
 src/ui.rs  | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/app.rs b/src/app.rs
index 6f8f979..1cd895b 100644
--- a/src/app.rs
+++ b/src/app.rs
@@ -110,7 +110,7 @@ impl<'a> Column {
 
     pub fn remove_task(&mut self) {
         self.tasks.remove(self.selected_task_idx);
-        self.select_previous_task();
+        self.select_next_task();
     }
 
     pub fn get_selected_task(&self) -> Option<&Task> {
diff --git a/src/ui.rs b/src/ui.rs
index 71fc4d5..2931e24 100644
--- a/src/ui.rs
+++ b/src/ui.rs
@@ -49,7 +49,9 @@ fn draw_tasks<B: Backend>(f: &mut Frame<B>, area: &Rect, state: &AppState) {
             .fg(Color::White);
         let block = Block::default().style(style).title(s).borders(Borders::ALL);
         let list = List::new(items).block(block);
-        f.render_widget(list, columns[i])
+        let mut list_state = ListState::default();
+        list_state.select(Some(column.selected_task_idx + 1));
+        f.render_stateful_widget(list, columns[i], &mut list_state);
     }
 }