database.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

#ifndef DATABASE_H
#define DATABASE_H

#include <QSqlDatabase>
#include <QSqlQuery>
#include <QSqlError>
#include <QString>
#include <QList>
#include <QDate>
#include "transaction.h"

class Database {
public:
    Database();
    ~Database();
    
    bool open(const QString &path);
    bool createTables();
    
    // Transaction operations
    bool addTransaction(const Transaction &transaction);
    bool updateTransaction(const Transaction &transaction);
    bool deleteTransaction(int id);
    QList<Transaction> getTransactions(const QDate &startDate, const QDate &endDate);
    QList<Transaction> getAllTransactions();
    
    // Recurring rule operations
    bool addRecurringRule(const RecurringRule &rule);
    bool updateRecurringRule(const RecurringRule &rule);
    bool deleteRecurringRule(int id);
    QList<RecurringRule> getAllRecurringRules();
    void regenerateProjectionsForRule(const RecurringRule &rule);
    void deleteProjectionsForRule(int ruleId);
    void regenerateAllProjections();
    
    // Reconciliation
    bool convertToActual(int transactionId, double actualAmount);
    
    // Settings
    QString getSetting(const QString &name, const QString &defaultValue = QString());
    bool setSetting(const QString &name, const QString &value);
    
    QString lastError() const;
    
private:
    QSqlDatabase db;
    QString errorMsg;
    
    Transaction queryToTransaction(QSqlQuery &query);
    RecurringRule queryToRecurringRule(QSqlQuery &query);
};

#endif // DATABASE_H