1 files changed, 31 insertions, 2 deletions
diff --git a/schist_desktop_gui/src/gui.rs b/schist_desktop_gui/src/gui.rs
index 26f147b..78fea11 100644
--- a/schist_desktop_gui/src/gui.rs
+++ b/schist_desktop_gui/src/gui.rs
@@ -7,8 +7,11 @@ use std::{path::PathBuf, str::FromStr};
 
 use diesel::SqliteConnection;
 use iced::{Element, Task};
-use schist_models::{Account, Bucket};
-use schist_queries::{accounts::get_all_accounts, buckets::get_all_buckets};
+use schist_models::{Account, AccountTransfer, Bucket, Transaction};
+use schist_queries::{
+    account_transfers::get_all_account_transfers, accounts::get_all_accounts,
+    buckets::get_all_buckets, transactions::get_all_transactions,
+};
 
 use crate::{
     cache::{put_to_cache, remove_from_cache, CACHE_KEYS},
@@ -31,7 +34,9 @@ pub enum Message {
     FilesScreenMessage(files_screen::Message),
     MainScreenMessage(main_screen::Message),
     RefetchedAccounts(FetchResult<Vec<Account>>),
+    RefetchedAccountTransfers(FetchResult<Vec<AccountTransfer>>),
     RefetchedBuckets(FetchResult<Vec<Bucket>>),
+    RefetchedTransactions(FetchResult<Vec<Transaction>>),
 }
 
 pub type FetchResult<T> = std::result::Result<T, FetchError>;
@@ -185,6 +190,26 @@ impl Gui {
                 }
                 iced::Task::none()
             }
+            Message::RefetchedAccountTransfers(account_transfers) => {
+                match account_transfers {
+                    Ok(account_transfers) => {
+                        self.main_screen
+                            .update(main_screen::Message::SetAccountTransfers(account_transfers));
+                    }
+                    Err(err) => println!("Failed to get account transfers: {:?}", err.message),
+                }
+                iced::Task::none()
+            }
+            Message::RefetchedTransactions(transactions) => {
+                match transactions {
+                    Ok(transactions) => {
+                        self.main_screen
+                            .update(main_screen::Message::SetTransactions(transactions));
+                    }
+                    Err(err) => println!("Failed to get transactions: {:?}", err.message),
+                };
+                Task::none()
+            }
             Message::FilesScreenMessage(message) => self.update_files_screen(message),
             Message::FailedToCreateFile(err) => {
                 self.update_files_screen(files_screen::Message::ReportFailedToCreateFile(err))
@@ -226,6 +251,8 @@ impl Gui {
     ) -> Task<Message> {
         let buckets = get_all_buckets(&mut connection).map_err(FetchError::new);
         let accounts = get_all_accounts(&mut connection).map_err(FetchError::new);
+        let account_transfers = get_all_account_transfers(&mut connection).map_err(FetchError::new);
+        let transactions = get_all_transactions(&mut connection).map_err(FetchError::new);
         self.connection = Some(connection);
         self.active_screen = Screen::MainScreen;
         if let Some(path) = path.to_str() {
@@ -236,6 +263,8 @@ impl Gui {
         iced::Task::batch(vec![
             self.update(Message::RefetchedBuckets(buckets)),
             self.update(Message::RefetchedAccounts(accounts)),
+            self.update(Message::RefetchedAccountTransfers(account_transfers)),
+            self.update(Message::RefetchedTransactions(transactions)),
         ])
     }