diff --git a/lib/Tasks/generate_transactions_task.dart b/lib/Tasks/generate_transactions_task.dart index 13da41c..2115b1b 100644 --- a/lib/Tasks/generate_transactions_task.dart +++ b/lib/Tasks/generate_transactions_task.dart @@ -1,4 +1,5 @@ import 'package:drift/drift.dart'; +import 'package:flutter/material.dart'; import 'package:logger/logger.dart'; import '../Entities/drift_database.dart'; @@ -24,7 +25,7 @@ class GenerateTransactionsTask extends Task { await _recurringTransactionRepository.findBy(); for (final recurringTransaction in recurringTransactions) { - _logger.i('Generating Transactions of $recurringTransaction...'); + _logger.i('Generating Transactions of ${recurringTransaction.name}...'); final List transactions = await _transactionRepository .findBy( @@ -33,42 +34,76 @@ class GenerateTransactionsTask extends Task { ); final Transaction? transaction = transactions.firstOrNull; - final DateTime date = (transaction != null) - ? transaction.date! - : recurringTransaction.startDate!; - final DateTime nextTransactionDate; - - switch (recurringTransaction.timeFrame) { - case TimeFrameEnum.daily: - nextTransactionDate = date.add(const Duration(days: 1)); - case TimeFrameEnum.weekly: - nextTransactionDate = date.add(const Duration(days: 7)); - case TimeFrameEnum.monthly: - nextTransactionDate = DateTime(date.year, date.month + 1, date.day); - case TimeFrameEnum.yearly: - nextTransactionDate = DateTime(date.year + 1, date.month, date.day); - } - - if (DateTime.now().compareTo(nextTransactionDate) <= 0) { - // TODO: Nicht mit NOW, sondern Ende dieses Monats Vergleichen - final TransactionsCompanion transaction = TransactionsCompanion( - name: Value(recurringTransaction.name), - date: Value(nextTransactionDate), - amount: Value(recurringTransaction.amount), - checked: const Value(false), - accountId: Value(recurringTransaction.accountId), - recurringTransactionId: Value(recurringTransaction.id), - ); - - _logger.i( - 'Adding transaction ${transaction.name} on ${transaction.date}', - ); - - await _transactionRepository.add(transaction); - } + await _generateTransactions(recurringTransaction, transaction?.date); } _logger.i('Generating transactions completed.'); return true; } + + Future _generateTransactions( + final RecurringTransaction recurringTransaction, + final DateTime? lastTransactionDate, { + final int number = 0, + }) async { + final DateTime newTransactionDate; + + if (lastTransactionDate != null) { + switch (recurringTransaction.timeFrame) { + case TimeFrameEnum.daily: + newTransactionDate = lastTransactionDate.add(const Duration(days: 1)); + case TimeFrameEnum.weekly: + newTransactionDate = lastTransactionDate.add(const Duration(days: 7)); + case TimeFrameEnum.monthly: + newTransactionDate = DateUtils.addMonthsToMonthDate( + lastTransactionDate, + 1, + ); + case TimeFrameEnum.yearly: + newTransactionDate = DateUtils.addMonthsToMonthDate( + lastTransactionDate, + 12, + ); + } + } else { + newTransactionDate = recurringTransaction.startDate!; + } + + _logger.i('New transaction-date is $newTransactionDate'); + + final DateTime endOfMonth = DateTime( + DateTime.now().year, + DateTime.now().month + 1, + 0, + ); + + if (endOfMonth.compareTo(newTransactionDate) >= 0) { + _logger.i('$newTransactionDate is before $endOfMonth'); + + final TransactionsCompanion transaction = TransactionsCompanion( + name: Value('${recurringTransaction.name} #$number'), + date: Value(newTransactionDate), + amount: Value(recurringTransaction.amount), + checked: const Value(false), + accountId: Value(recurringTransaction.accountId), + recurringTransactionId: Value(recurringTransaction.id), + ); + + _logger.i( + 'Adding transaction ${transaction.name.value}' + ' on ${transaction.date.value}', + ); + + await _transactionRepository.add(transaction); + + return _generateTransactions( + recurringTransaction, + newTransactionDate, + number: number + 1, + ); + } else { + _logger.i('$newTransactionDate is after $endOfMonth'); + return true; + } + } }