#ifndef CASHFLOW_H
#define CASHFLOW_H

#include <QMainWindow>
#include <QTableWidget>
#include <QDate>
#include "database.h"
#include "transaction.h"

QT_BEGIN_NAMESPACE
namespace Ui { class CashFlow; }
QT_END_NAMESPACE

class CashFlow : public QMainWindow
{
    Q_OBJECT

public:
    CashFlow(QWidget *parent = nullptr);
    ~CashFlow();

private slots:
    void onDateRangeChanged();
    void onTransactionSelected();
    void onSaveTransaction();
    void onTransactionFieldChanged();
    void onNewTransaction();
    void onDeleteTransaction();
    void onRecurringSelected();
    void onSaveRecurring();
    void onNewRecurring();
    void onDeleteRecurring();
    void onPeriodChanged();
    
private:
    Ui::CashFlow *ui;
    Database *database;
    int currentTransactionId;
    int currentRecurringId;
    double startingBalance;
    QFont currentAmountFont;
    
    enum PeriodType {
        Daily,
        Weekly,
        Monthly,
        Quarterly
    };
    
    void setupConnections();
    void refreshView();
    void refreshTransactionTable();
    void refreshRecurringTable();
    void calculateAndDisplayBalance();
    QList<Transaction> getAllTransactionsInRange();
    QList<Transaction> generateProjectedTransactions();
    void clearTransactionEntry();
    void loadTransactionToEntry(const Transaction &t);
    void clearRecurringEntry();
    void loadRecurringToEntry(const RecurringRule &r);
    QDate getPeriodEnd(const QDate &date, PeriodType periodType);
    QDate getPeriodStart(const QDate &date, PeriodType periodType);
    QString getPeriodLabel(const QDate &date, PeriodType periodType, int count);
    void insertPeriodEndRow(const QString &label, double balance, const QMap<QString, double> &accountBalances);
    void updateAmountColors();
    void loadSettings();
    void applySettings();
    void onSaveSettings();
    void onChooseAmountFont();
    QString formatCurrency(double amount) const;
};
#endif // CASHFLOW_H