1 files changed, 10 insertions, 2 deletions
diff --git a/schist_desktop_gui/src/gui/components/transactions_view.rs b/schist_desktop_gui/src/gui/components/transactions_view.rs
index 491049d..11249a1 100644
--- a/schist_desktop_gui/src/gui/components/transactions_view.rs
+++ b/schist_desktop_gui/src/gui/components/transactions_view.rs
@@ -3,7 +3,7 @@ mod view_transactions_view;
 use std::collections::HashMap;
 
 use itertools::Itertools;
-use schist_models::{Account, Transaction};
+use schist_models::{Account, Bucket, Transaction};
 
 use crate::{
     gui::components::{navigation, AccountNavigationEntry, Navigation},
@@ -12,6 +12,7 @@ use crate::{
 
 #[derive(Clone, Debug)]
 pub struct TransactionsView {
+    buckets: Vec<Bucket>,
     navigation: Navigation<AccountNavigationEntry>,
     transactions_by_account_id: HashMap<i32, Vec<Transaction>>,
 }
@@ -22,6 +23,7 @@ pub enum Message {
     SelectNextAccount,
     SelectPrevAccount,
     SetAccounts(Vec<Account>),
+    SetBuckets(Vec<Bucket>),
     SetTransactions(Vec<Transaction>),
 }
 
@@ -30,7 +32,7 @@ pub enum Action {
 }
 
 impl TransactionsView {
-    pub fn new(accounts: &[Account], transactions: &[Transaction]) -> Self {
+    pub fn new(accounts: &[Account], buckets: &[Bucket], transactions: &[Transaction]) -> Self {
         let account_navigation_entries: Vec<AccountNavigationEntry> = accounts
             .iter()
             .cloned()
@@ -41,6 +43,7 @@ impl TransactionsView {
             account_navigation_entries,
         );
         Self {
+            buckets: buckets.to_vec(),
             navigation: navigation,
             transactions_by_account_id: transactions
                 .into_iter()
@@ -63,6 +66,11 @@ impl<'a> Component<'a, Message, Action> for TransactionsView {
                 self.update_navigation(navigation::Message::SetOptions(account_navigation_entries))
             }
 
+            Message::SetBuckets(buckets) => {
+                self.buckets = buckets;
+                Action::None
+            }
+
             Message::SetTransactions(transactions) => {
                 self.transactions_by_account_id =
                     transactions.into_iter().into_group_map_by(|t| t.account_id);