dbmanager.cpp

#include "dbmanager.h"
#include <QtSql>
#include <iostream>
#include <QMessageBox>

dbManager::dbManager()
{

}

void dbManager::createNewDatabase(QString filename)
{
    if (sqlDatabase.isOpen()) sqlDatabase.close();
    sqlDatabase = QSqlDatabase::addDatabase("QSQLITE");
    sqlDatabase.setDatabaseName(filename);
    if (!sqlDatabase.open())
    {
        QMessageBox errorBox;
        errorBox.setText("Failed to create database file!");
        errorBox.setInformativeText(sqlDatabase.lastError().text());
        errorBox.setDefaultButton(QMessageBox::Ok);
        errorBox.exec();

    } else {
        QSqlQuery query;
        query.exec("CREATE TABLE \"expenses\""
                "(\"name\"	TEXT NOT NULL,"
                "\"value\"	REAL NOT NULL,"
                "\"category\"	TEXT NOT NULL,"
                "\"date\"	TEXT NOT NULL,"
                "\"person\"	TEXT NOT NULL,"
                "PRIMARY KEY(\"name\"))");
        query.exec("CREATE TABLE \"incomes\" "
                "(\"name\"	TEXT NOT NULL,"
                "\"value\"	REAL NOT NULL,"
                "\"category\"	TEXT NOT NULL,"
                "\"date\"	TEXT NOT NULL,"
                "\"person\"	TEXT NOT NULL,"
                "PRIMARY KEY(\"name\"))");
        expensesTableModel = getTableModel("expenses");
        incomesTableModel = getTableModel("incomes");
    }
}

void dbManager::openExistingDatabase(QString filename)
{
    if (sqlDatabase.isOpen()) sqlDatabase.close();
    sqlDatabase = QSqlDatabase::addDatabase("QSQLITE");
    sqlDatabase.setDatabaseName(filename);
    if (!sqlDatabase.open())
    {
        QMessageBox errorBox;
        errorBox.setText("Failed to open database file!");
        errorBox.setInformativeText(sqlDatabase.lastError().text());
        errorBox.setDefaultButton(QMessageBox::Ok);
        errorBox.exec();
    } else {
        expensesTableModel = getTableModel("expenses");
        incomesTableModel = getTableModel("incomes");
    }
}

bool dbManager::isOpen()
{
    return sqlDatabase.isOpen();
}

std::shared_ptr<QSqlTableModel> dbManager::getTableModel(const QString &tableName)
{
    std::shared_ptr<QSqlTableModel> model(new QSqlTableModel(nullptr, sqlDatabase));
    model->setTable(tableName);
    model->setEditStrategy(QSqlTableModel::OnManualSubmit);
    model->select();
    return model;
}

QStringList dbManager::getDataArray(std::shared_ptr<QSqlTableModel> model, QString value)
{
    QStringList toReturn;
    for (int i = 0; i < model->rowCount(); i++)
    {
        toReturn.append(model->record(i).value(value).toString());
    }
    return toReturn;
}

void dbManager::closeDatabase()
{
    sqlDatabase.removeDatabase("QSQLITE");
    sqlDatabase.close();
}

QSqlDatabase dbManager::getDatabase() {
    return sqlDatabase.database("QSQLITE");
}