From 27b95c9186bd45edd4c982a7cc89cf69cf2c5d46 Mon Sep 17 00:00:00 2001
From: Joe Carstairs <me@joeac.net>
Date: Wed, 16 Oct 2024 12:51:21 +0100
Subject: All backend stuff lives in backend folder

---
 .../app/src/queries/category_balances/get_all.rs   | 43 ++++++++++++++++++++++
 backend/app/src/queries/category_balances/mod.rs   |  3 ++
 2 files changed, 46 insertions(+)
 create mode 100644 backend/app/src/queries/category_balances/get_all.rs
 create mode 100644 backend/app/src/queries/category_balances/mod.rs

(limited to 'backend/app/src/queries/category_balances')

diff --git a/backend/app/src/queries/category_balances/get_all.rs b/backend/app/src/queries/category_balances/get_all.rs
new file mode 100644
index 0000000..713fbeb
--- /dev/null
+++ b/backend/app/src/queries/category_balances/get_all.rs
@@ -0,0 +1,43 @@
+use crate::{
+    models::category_balance::CategoryBalance,
+    queries::{
+        budget_updates::get_all_budget_updates,
+        categories::get_all_categories,
+        category_transfers::{
+            sum_category_transfer_quantity_per_from_category_id,
+            sum_category_transfer_quantity_per_to_category_id,
+        },
+        transaction_categorisations::sum_transaction_categorisation_quantity_per_category_id,
+    },
+    utils::calculate_category_balance,
+};
+use diesel::{result::Error, SqliteConnection};
+
+pub fn get_all_category_balances(
+    connection: &mut SqliteConnection,
+) -> Result<Vec<CategoryBalance>, Error> {
+    let categories = get_all_categories(connection)?;
+    let transaction_sums = sum_transaction_categorisation_quantity_per_category_id(connection)?;
+    let category_transfer_from_sums =
+        sum_category_transfer_quantity_per_from_category_id(connection)?;
+    let category_transfer_to_sums = sum_category_transfer_quantity_per_to_category_id(connection)?;
+    let budget_updates = get_all_budget_updates(connection)?;
+
+    let mut balances = Vec::<CategoryBalance>::new();
+    for category in categories {
+        let balance = calculate_category_balance(
+            &category,
+            &transaction_sums,
+            &category_transfer_from_sums,
+            &category_transfer_to_sums,
+            &budget_updates,
+        );
+
+        balances.push(CategoryBalance {
+            category_id: category.id,
+            balance: balance.try_into().unwrap(),
+        })
+    }
+
+    Ok(balances)
+}
diff --git a/backend/app/src/queries/category_balances/mod.rs b/backend/app/src/queries/category_balances/mod.rs
new file mode 100644
index 0000000..3fc1be2
--- /dev/null
+++ b/backend/app/src/queries/category_balances/mod.rs
@@ -0,0 +1,3 @@
+mod get_all;
+
+pub use get_all::get_all_category_balances;
-- 
cgit v1.2.3