Refactors actualbudget importer

10 files changed, 325 insertions, 0 deletions

diff --git a/rust/schist_actualbudget_importer/src/transform_state/is_category_transfer.rs b/rust/schist_actualbudget_importer/src/transform_state/is_category_transfer.rs
new file mode 100644
index 0000000..0f4299a
--- /dev/null
+++ b/rust/schist_actualbudget_importer/src/transform_state/is_category_transfer.rs
@@ -0,0 +1,18 @@
+use actualbudget_models::{actualbudget_account::ActualbudgetAccount, actualbudget_transaction::ActualbudgetTransaction};
+
+pub fn is_category_transfer(
+    transaction: &ActualbudgetTransaction,
+    actualbudget_category_transfers_account: &Option<ActualbudgetAccount>
+) -> bool {
+    match actualbudget_category_transfers_account {
+        None => false,
+        Some(acc) => transaction.account_id.clone() == acc.id.clone(),
+    }
+}
+
+pub fn is_not_category_transfer(
+    transaction: &ActualbudgetTransaction,
+    actualbudget_category_transfers_account: &Option<ActualbudgetAccount>
+) -> bool {
+    !is_category_transfer(transaction, actualbudget_category_transfers_account)
+}
diff --git a/rust/schist_actualbudget_importer/src/transform_state/mod.rs b/rust/schist_actualbudget_importer/src/transform_state/mod.rs
new file mode 100644
index 0000000..d82a59e
--- /dev/null
+++ b/rust/schist_actualbudget_importer/src/transform_state/mod.rs
@@ -0,0 +1,21 @@
+mod is_category_transfer;
+mod transform_accounts;
+mod transform_budget_updates;
+mod transform_categories;
+mod transform_category_transfers;
+mod transform_date;
+mod transform_id;
+mod transform_state;
+mod transform_transactions;
+
+use is_category_transfer::is_category_transfer;
+use is_category_transfer::is_not_category_transfer;
+use transform_accounts::transform_accounts;
+use transform_budget_updates::transform_budget_updates;
+use transform_categories::transform_categories;
+use transform_category_transfers::transform_category_transfers;
+use transform_date::transform_date;
+use transform_id::transform_id;
+use transform_transactions::transform_transactions;
+
+pub use transform_state::*;
diff --git a/rust/schist_actualbudget_importer/src/transform_state/transform_accounts.rs b/rust/schist_actualbudget_importer/src/transform_state/transform_accounts.rs
new file mode 100644
index 0000000..beb1de7
--- /dev/null
+++ b/rust/schist_actualbudget_importer/src/transform_state/transform_accounts.rs
@@ -0,0 +1,27 @@
+use actualbudget_models::actualbudget_account::ActualbudgetAccount;
+use schist_models::{account::Account, datetime_utc::DatetimeUtc};
+
+use super::transform_id;
+
+pub fn transform_accounts(
+    actualbudget_accounts: &[ActualbudgetAccount],
+    first_transaction_date: &DatetimeUtc,
+) -> Vec<Account> {
+    let mut accounts = Vec::<Account>::new();
+    for account in actualbudget_accounts {
+        accounts.push(transform_account(account, first_transaction_date));
+    }
+    accounts
+}
+
+fn transform_account(
+    actualbudget_account: &ActualbudgetAccount,
+    opening_date: &DatetimeUtc,
+) -> Account {
+    Account {
+        id: transform_id(&actualbudget_account.id.clone()),
+        name: actualbudget_account.name.clone(),
+        opening_balance: 0,
+        opening_date: opening_date.clone(),
+    }
+}
diff --git a/rust/schist_actualbudget_importer/src/transform_state/transform_budget_updates.rs b/rust/schist_actualbudget_importer/src/transform_state/transform_budget_updates.rs
new file mode 100644
index 0000000..fc2ee23
--- /dev/null
+++ b/rust/schist_actualbudget_importer/src/transform_state/transform_budget_updates.rs
@@ -0,0 +1,40 @@
+use anyhow::Result;
+use actualbudget_models::actualbudget_zero_budget::ActualbudgetZeroBudget;
+use schist_models::{budget_update::BudgetUpdate, date_utc::DateUtc};
+
+use super::{transform_date, transform_id};
+
+pub fn transform_budget_updates(
+    actualbudget_zero_budgets: &[ActualbudgetZeroBudget],
+) -> Result<Vec<BudgetUpdate>> {
+    let mut budget_updates = Vec::<BudgetUpdate>::new();
+    for zero_budget in actualbudget_zero_budgets {
+        let first_zero_budget_in_category = get_first_zero_budget_in_category(
+            actualbudget_zero_budgets,
+            &zero_budget.category_id.clone(),
+        );
+        if first_zero_budget_in_category.is_some_and(|zb| zb.id == zero_budget.id) {
+            continue;
+        }
+        budget_updates.push(BudgetUpdate {
+            id: transform_id(&zero_budget.id.clone()),
+            category_id: transform_id(&zero_budget.category_id.clone()),
+            date: DateUtc::from(transform_date(&zero_budget.month.clone())?),
+            new_budget: zero_budget.amount.try_into().unwrap(),
+            new_period: 30,
+        })
+    }
+    Ok(budget_updates)
+}
+
+fn get_first_zero_budget_in_category(
+    actualbudget_zero_budgets: &[ActualbudgetZeroBudget],
+    actualbudget_category_id: &str,
+) -> Option<ActualbudgetZeroBudget> {
+    let mut category_zero_budgets = actualbudget_zero_budgets.to_vec();
+    category_zero_budgets.retain(|zb| zb.category_id == actualbudget_category_id);
+    category_zero_budgets
+        .iter()
+        .min_by_key(|zb| zb.month.clone())
+        .cloned()
+}
diff --git a/rust/schist_actualbudget_importer/src/transform_state/transform_categories.rs b/rust/schist_actualbudget_importer/src/transform_state/transform_categories.rs
new file mode 100644
index 0000000..80ff80b
--- /dev/null
+++ b/rust/schist_actualbudget_importer/src/transform_state/transform_categories.rs
@@ -0,0 +1,17 @@
+use actualbudget_models::actualbudget_category::ActualbudgetCategory;
+use schist_models::category::Category;
+
+use super::transform_id;
+
+pub fn transform_categories(
+    actualbudget_categories: &[ActualbudgetCategory],
+) -> Vec<Category> {
+    let mut categories = Vec::<Category>::new();
+    for actualbudget_category in actualbudget_categories {
+        categories.push(Category {
+            id: transform_id(&actualbudget_category.id.clone()),
+            name: actualbudget_category.name.clone(),
+        })
+    }
+    categories
+}
diff --git a/rust/schist_actualbudget_importer/src/transform_state/transform_category_transfers.rs b/rust/schist_actualbudget_importer/src/transform_state/transform_category_transfers.rs
new file mode 100644
index 0000000..86ad498
--- /dev/null
+++ b/rust/schist_actualbudget_importer/src/transform_state/transform_category_transfers.rs
@@ -0,0 +1,65 @@
+use actualbudget_models::{
+    actualbudget_account::ActualbudgetAccount,
+    actualbudget_transaction::ActualbudgetTransaction,
+};
+use schist_models::category_transfer::CategoryTransfer;
+use core::panic;
+
+use super::{is_category_transfer, transform_id};
+
+pub fn transform_category_transfers(
+    actualbudget_transactions: &[ActualbudgetTransaction],
+    actualbudget_category_transfers_account: &Option<ActualbudgetAccount>,
+) -> Vec<CategoryTransfer> {
+    let mut category_transfers = Vec::<CategoryTransfer>::new();
+
+    let mut actualbudget_category_transfers = actualbudget_transactions.to_vec();
+    actualbudget_category_transfers.retain(|transaction|
+        is_category_transfer(transaction, actualbudget_category_transfers_account)
+    );
+    let mut actualbudget_parent_category_transfers = actualbudget_category_transfers.to_vec();
+    actualbudget_parent_category_transfers.retain(|t| t.is_parent);
+    let mut actualbudget_child_category_transfers = actualbudget_category_transfers.to_vec();
+    actualbudget_child_category_transfers.retain(|t| t.is_child);
+
+    for actualbudget_parent_category_transfer in actualbudget_parent_category_transfers {
+        let mut actualbudget_child_category_transfers =
+            actualbudget_child_category_transfers.to_vec();
+        actualbudget_child_category_transfers.retain(|t| {
+            t.parent_id
+                .clone()
+                .is_some_and(|parent_id| parent_id == actualbudget_parent_category_transfer.id)
+        });
+
+        if actualbudget_child_category_transfers.len() != 2 {
+            panic!("Category transfers should always have exactly two children.");
+        }
+
+        let category_1 = actualbudget_child_category_transfers[0].clone();
+        let category_2 = actualbudget_child_category_transfers[1].clone();
+
+        // To avoid duplicates
+        if category_1.amount >= 0 {
+            continue;
+        }
+
+        if actualbudget_child_category_transfers[0].amount
+            != -actualbudget_child_category_transfers[1].amount
+        {
+            panic!("Category transfer children should sum to zero.");
+        }
+
+        category_transfers.push(CategoryTransfer {
+            id: transform_id(&actualbudget_parent_category_transfer.id),
+            description: actualbudget_parent_category_transfer.notes.unwrap_or_else(String::new),
+            quantity: actualbudget_child_category_transfers[0]
+                .amount
+                .try_into()
+                .unwrap(),
+            from_category_id: transform_id(&category_1.id),
+            to_category_id: transform_id(&category_2.id),
+        });
+    }
+
+    category_transfers
+}
diff --git a/rust/schist_actualbudget_importer/src/transform_state/transform_date.rs b/rust/schist_actualbudget_importer/src/transform_state/transform_date.rs
new file mode 100644
index 0000000..803f3ed
--- /dev/null
+++ b/rust/schist_actualbudget_importer/src/transform_state/transform_date.rs
@@ -0,0 +1,7 @@
+use anyhow::Result;
+use actualbudget_models::actualbudget_date::ActualbudgetDate;
+use schist_models::datetime_utc::DatetimeUtc;
+
+pub fn transform_date(actualbudget_date: &ActualbudgetDate) -> Result<DatetimeUtc> {
+    actualbudget_date.to_iso().parse()
+}
diff --git a/rust/schist_actualbudget_importer/src/transform_state/transform_id.rs b/rust/schist_actualbudget_importer/src/transform_state/transform_id.rs
new file mode 100644
index 0000000..f370efe
--- /dev/null
+++ b/rust/schist_actualbudget_importer/src/transform_state/transform_id.rs
@@ -0,0 +1,9 @@
+use std::hash::{DefaultHasher, Hash, Hasher};
+
+pub fn transform_id(actualbudget_id: &str) -> i32 {
+    let mut hasher = DefaultHasher::new();
+    actualbudget_id.hash(&mut hasher);
+    (hasher.finish() % std::convert::TryInto::<u64>::try_into(i32::MAX).unwrap())
+        .try_into()
+        .unwrap()
+}
diff --git a/rust/schist_actualbudget_importer/src/transform_state/transform_state.rs b/rust/schist_actualbudget_importer/src/transform_state/transform_state.rs
new file mode 100644
index 0000000..9236532
--- /dev/null
+++ b/rust/schist_actualbudget_importer/src/transform_state/transform_state.rs
@@ -0,0 +1,36 @@
+use anyhow::Result;
+use schist_models::{account::Account, budget_update::BudgetUpdate, category::Category, category_transfer::CategoryTransfer, transaction::Transaction, transaction_categorisation::TransactionCategorisation};
+
+use crate::{actualbudget_state::ActualbudgetState, schist_state::SchistState};
+
+use super::{transform_accounts, transform_budget_updates, transform_categories, transform_category_transfers, transform_date, transform_transactions};
+
+pub fn transform_state(actualbudget_state: ActualbudgetState) -> Result<SchistState> {
+    let first_transaction_date =
+        transform_date(&actualbudget_state.actualbudget_first_transaction_date)?;
+    let accounts: Vec<Account> =
+        transform_accounts(&actualbudget_state.actualbudget_accounts, &first_transaction_date);
+    let categories: Vec<Category> =
+        transform_categories(&actualbudget_state.actualbudget_categories);
+    let budget_updates: Vec<BudgetUpdate> =
+        transform_budget_updates(&actualbudget_state.actualbudget_zero_budgets)?;
+    let (transactions, transaction_categorisations): (Vec<Transaction>, Vec<TransactionCategorisation>) =
+        transform_transactions(
+            &actualbudget_state.actualbudget_transactions,
+            &actualbudget_state.actualbudget_category_transfers_account,
+        )?;
+    let category_transfers: Vec<CategoryTransfer> = transform_category_transfers(
+        &actualbudget_state.actualbudget_transactions,
+        &actualbudget_state.actualbudget_category_transfers_account,
+    );
+
+    Ok(SchistState {
+        first_transaction_date,
+        accounts,
+        categories,
+        budget_updates,
+        transactions,
+        transaction_categorisations,
+        category_transfers,
+    })
+}
diff --git a/rust/schist_actualbudget_importer/src/transform_state/transform_transactions.rs b/rust/schist_actualbudget_importer/src/transform_state/transform_transactions.rs
new file mode 100644
index 0000000..7bb1de9
--- /dev/null
+++ b/rust/schist_actualbudget_importer/src/transform_state/transform_transactions.rs
@@ -0,0 +1,85 @@
+use anyhow::Result;
+use actualbudget_models::{actualbudget_account::ActualbudgetAccount, actualbudget_transaction::ActualbudgetTransaction};
+use schist_models::{transaction::Transaction, transaction_categorisation::TransactionCategorisation};
+
+use super::{is_not_category_transfer, transform_date, transform_id};
+
+pub fn transform_transactions(
+    actualbudget_transactions: &[ActualbudgetTransaction],
+    actualbudget_category_transfers_account: &Option<ActualbudgetAccount>,
+) -> Result<(Vec<Transaction>, Vec<TransactionCategorisation>)> {
+    let mut transactions = Vec::<Transaction>::new();
+    let mut transaction_categorisations = Vec::<TransactionCategorisation>::new();
+
+    let mut actualbudget_transactions = actualbudget_transactions.to_vec();
+    actualbudget_transactions.retain(|transaction|
+        is_not_category_transfer(transaction, actualbudget_category_transfers_account)
+    );
+    let mut parent_actualbudget_transactions = actualbudget_transactions.to_vec();
+    parent_actualbudget_transactions.retain(|t| t.is_parent);
+    let mut child_actualbudget_transactions = actualbudget_transactions.to_vec();
+    child_actualbudget_transactions.retain(|t| t.is_child);
+    let mut atomic_actualbudget_transactions = actualbudget_transactions.to_vec();
+    atomic_actualbudget_transactions.retain(|t| !t.is_parent && !t.is_child);
+
+    for atomic_actualbudget_transaction in atomic_actualbudget_transactions {
+        let transaction_id = transform_id(&atomic_actualbudget_transaction.id);
+
+        transactions.push(Transaction {
+            id: transaction_id,
+            description: atomic_actualbudget_transaction.notes.unwrap_or_else(String::new),
+            payee: atomic_actualbudget_transaction.payee.unwrap_or_else(String::new),
+            quantity: atomic_actualbudget_transaction.amount.try_into().unwrap(),
+            date: transform_date(&atomic_actualbudget_transaction.date)?,
+            account_id: transform_id(&atomic_actualbudget_transaction.account_id),
+        });
+
+        let actualbudget_category_id = atomic_actualbudget_transaction
+            .category_id
+            .expect("actualbudget transaction should have been atomic, but in fact had no category ID");
+
+        transaction_categorisations.push(TransactionCategorisation {
+            id: transaction_id,
+            description: String::new(),
+            quantity: atomic_actualbudget_transaction.amount.try_into().unwrap(),
+            transaction_id,
+            category_id: transform_id(&actualbudget_category_id),
+        });
+    }
+
+    for parent_actualbudget_transaction in parent_actualbudget_transactions {
+        let mut child_actualbudget_transactions = child_actualbudget_transactions.to_vec();
+        child_actualbudget_transactions.retain(|t| {
+            t.parent_id
+                .clone()
+                .is_some_and(|parent_id| parent_id == parent_actualbudget_transaction.id.clone())
+        });
+
+        let transaction_id = transform_id(&parent_actualbudget_transaction.id);
+
+        transactions.push(Transaction {
+            id: transaction_id,
+            description: parent_actualbudget_transaction.notes.unwrap_or_else(String::new),
+            payee: parent_actualbudget_transaction.payee.unwrap_or_else(String::new),
+            quantity: parent_actualbudget_transaction.amount.try_into().unwrap(),
+            date: transform_date(&parent_actualbudget_transaction.date)?,
+            account_id: transform_id(&parent_actualbudget_transaction.account_id),
+        });
+
+        for child_actualbudget_transaction in child_actualbudget_transactions {
+            let actualbudget_category_id = child_actualbudget_transaction
+                .category_id
+                .expect("actualbudget transaction should have been a child transaction, but in fact had no category ID");
+
+            transaction_categorisations.push(TransactionCategorisation {
+                id: transform_id(&child_actualbudget_transaction.id),
+                description: child_actualbudget_transaction.notes.unwrap_or_else(String::new),
+                quantity: child_actualbudget_transaction.amount.try_into().unwrap(),
+                transaction_id,
+                category_id: transform_id(&actualbudget_category_id),
+            });
+        }
+    }
+
+    Ok((transactions, transaction_categorisations))
+}