Implemented Transaction#getSourceAccount, `Transaction#getDestinati…

…onAccount`, `Transaction#retrieveSourceAccount` & `Transaction#retrieveDestinationAccount`
financrr · Mar 24, 2024 · 34f5ec7 · 34f5ec7
1 parent 02c8496
commit 34f5ec7
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -6,7 +6,7 @@
   - Added `Restrr` methods: `createAccount`, `getAccounts`, `retrieveAccountById`, `retrieveAllAccounts`
 - Added `Transaction`
   - Added `TransactionRoutes`
-  - Added `Transaction` methods: `delete`, `update`
+  - Added `Transaction` methods: `delete`, `update`, `getSourceAccount`, `getDestinationAccount`, `retrieveSourceAccount`, `retrieveDestinationAccount`
   - Added `Restrr` methods: `createTransaction`, `retrieveTransactionById`, `retrieveAllTransactions`
 - Added `Restrr#getCurrencies`
 - Fixed `Session#delete` using a wrong route

diff --git a/lib/src/api/entities/transaction/transaction.dart b/lib/src/api/entities/transaction/transaction.dart
@@ -23,4 +23,12 @@ abstract class Transaction extends RestrrEntity {
     Id? budget,
     DateTime? executedAt,
   });
+
+  Account? getSourceAccount();
+
+  Future<Account>? retrieveSourceAccount({bool forceRetrieve = false});
+
+  Account? getDestinationAccount();
+
+  Future<Account>? retrieveDestinationAccount({bool forceRetrieve = false});
 }
diff --git a/lib/src/internal/entities/transaction_impl.dart b/lib/src/internal/entities/transaction_impl.dart
@@ -84,4 +84,26 @@ class TransactionImpl extends RestrrEntityImpl implements Transaction {
         });
     return response.data!;
   }
+
+  @override
+  Account? getSourceAccount() => source == null ? null : api.accountCache.get(source!);
+
+  @override
+  Future<Account>? retrieveSourceAccount({bool forceRetrieve = false}) {
+    if (source == null) {
+      return null;
+    }
+    return api.retrieveAccountById(source!, forceRetrieve: forceRetrieve);
+  }
+
+  @override
+  Account? getDestinationAccount() => destination == null ? null : api.accountCache.get(destination!);
+
+  @override
+  Future<Account>? retrieveDestinationAccount({bool forceRetrieve = false}) {
+    if (destination == null) {
+      return null;
+    }
+    return api.retrieveAccountById(destination!, forceRetrieve: forceRetrieve);
+  }
 }