BucketsView expands and collapses groups

1 files changed, 38 insertions, 0 deletions

diff --git a/schist_desktop_gui/src/gui/screens/main_screen/main_screen.rs b/schist_desktop_gui/src/gui/screens/main_screen/main_screen.rs
index 6347914..cddec32 100644
--- a/schist_desktop_gui/src/gui/screens/main_screen/main_screen.rs
+++ b/schist_desktop_gui/src/gui/screens/main_screen/main_screen.rs
@@ -28,6 +28,8 @@ pub struct MainScreen {
 pub enum Message {
     BalancesViewMessage(balances_view::Message),
     BucketsViewMessage(buckets_view::Message),
+    ExpandItem,
+    CollapseItem,
     NextItem,
     PrevItem,
     TransactionsViewMessage(transactions_view::Message),
@@ -148,6 +150,42 @@ impl<'a> Component<'a, Message, Action> for MainScreen {
                     Action::None
                 }
             },
+
+            Message::ExpandItem => match self.active_view {
+                View::Balances => Action::None,
+                View::Buckets => match self.buckets_view.active_group() {
+                    Some(active_group) => {
+                        match self.buckets_view.update(buckets_view::Message::ExpandGroup(
+                            active_group.clone(),
+                            self.buckets
+                                .iter()
+                                .filter(|bucket| bucket.group.eq(&active_group))
+                                .cloned()
+                                .collect(),
+                        )) {
+                            buckets_view::Action::None => Action::None,
+                        }
+                    }
+                    None => Action::None,
+                },
+                View::Transactions => Action::None,
+            },
+
+            Message::CollapseItem => match self.active_view {
+                View::Balances => Action::None,
+                View::Buckets => match self.buckets_view.active_group() {
+                    Some(active_group) => {
+                        match self
+                            .buckets_view
+                            .update(buckets_view::Message::CollapseGroup(active_group))
+                        {
+                            buckets_view::Action::None => Action::None,
+                        }
+                    }
+                    None => Action::None,
+                },
+                View::Transactions => Action::None,
+            },
         }
     }
 }