1 files changed, 86 insertions, 0 deletions
diff --git a/src/ynab/budget.rs b/src/ynab/budget.rs
new file mode 100644
index 0000000..85eab22
--- /dev/null
+++ b/src/ynab/budget.rs
@@ -0,0 +1,86 @@
+pub struct Budget<'a> {
+    api: &'a ynab_api::apis::client::APIClient,
+    budget: ynab_api::models::BudgetDetail,
+}
+
+impl<'a> Budget<'a> {
+    pub fn new(
+        api: &'a ynab_api::apis::client::APIClient,
+        budget: ynab_api::models::BudgetDetail,
+    ) -> Self {
+        Self { api, budget }
+    }
+
+    pub fn name(&self) -> String {
+        self.budget.name.clone()
+    }
+
+    pub fn id(&self) -> String {
+        self.budget.id.clone()
+    }
+
+    pub fn reimbursables(&self) -> Vec<super::transaction::Transaction> {
+        let reimbursables_id = self
+            .api
+            .categories_api()
+            .get_categories(&self.budget.id, 0)
+            .unwrap()
+            .data
+            .category_groups
+            .iter()
+            .map(|group| {
+                group
+                    .categories
+                    .iter()
+                    .map(|c| (c.id.clone(), c.name.clone()))
+            })
+            .flat_map(|cs| cs)
+            .find(|(_, name)| name == "Reimbursables")
+            .map(|(id, _)| id)
+            .unwrap();
+
+        let mut reimbursables = vec![];
+        if let Some(transactions) = &self.budget.transactions {
+            if let Some(payees) = &self.budget.payees {
+                let mut payee_map = std::collections::HashMap::new();
+                for p in payees {
+                    payee_map.insert(p.id.clone(), p.name.clone());
+                }
+                let payee_map = payee_map;
+
+                for t in transactions {
+                    if let Some(category_id) = &t.category_id {
+                        if category_id != &reimbursables_id {
+                            continue;
+                        }
+                    } else {
+                        continue;
+                    }
+
+                    let payee = t
+                        .payee_id
+                        .iter()
+                        .flat_map(|payee_id| payee_map.get(payee_id).cloned())
+                        .next()
+                        .unwrap_or_else(|| "(none)".to_string());
+                    let reimbursed = if let Some(color) = &t.flag_color {
+                        color == "green"
+                    } else {
+                        false
+                    };
+
+                    reimbursables.push(super::transaction::Transaction {
+                        date: t.date.clone(),
+                        payee,
+                        amount: t.amount,
+                        reimbursed,
+                    })
+                }
+            } else {
+                panic!("no payees?");
+            }
+        }
+
+        reimbursables
+    }
+}