source: SMSSender/lib/libdatatypes/src/abstract/persistence/abstractaccountstorageofficer.cpp @ 71:415b45c71663

#include "abstractaccountstorageofficer.h"

#include <QVariant>
#include <QSqlQuery>
#include <QDebug>
#include <QSqlError>

//#include <typeconvert.h>
#include <exceptions/persistence/storageexceptions.h>


AbstractAccountStorageOfficer::AbstractAccountStorageOfficer(IAccount* account, IStorage* storage)
    : AbstractStorageOfficer(storage)
    , account_(account)
{
    QSqlQuery query = storage->createQuery();
    query.prepare("CREATE TABLE IF NOT EXISTS t_accounts ("
                    "accountId INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT, "
                    "name      TEXT    NOT NULL UNIQUE, "
                    "type      TEXT    NOT NULL "
                  "); ");
    if (!query.exec()) {
        throw Storage::EWriteException(tr("Could not create the account table in the database."));
    }
}


void AbstractAccountStorageOfficer::readFromStorage(){
    QSqlQuery query = storage()->createQuery();
    query.exec("SELECT name "
                 "FROM t_accounts "
                 "WHERE (accountId = '" + QString::number(account()->id()) + "'); ");
    if (!query.next()) {
        throw Storage::EReadException(tr("No such account in the database! [accountId: %1]").arg(account()->id()));
    }

    account()->setName(query.value(0).toString());
    account()->setEnabled(storage()->readBool(getNamespace(), "enabled", true));
}

void AbstractAccountStorageOfficer::writeToStorage() const {
    QSqlQuery query = storage()->createQuery();
    query.exec("SELECT accountId "
                 "FROM t_accounts "
                 "WHERE (accountId = '" + QString::number(account()->id()) + "'); ");

    if (!query.next()){
        // Insert

        query.prepare("INSERT INTO t_accounts (name, type) "
                        "VALUES ("
                          ":accountName, "
                          ":gatewayName"
                        "); ");
        query.bindValue(":accountName", account()->name());
        query.bindValue(":gatewayName", account()->gateway()->name());
        if (!query.exec()) {
            throw Storage::EWriteException(tr("The account could not have been written to the database."));
        }

        // Set the accountId since it is undefined if the account was not saved before.
        query.exec("SELECT accountId "
                     "FROM t_accounts "
                     "WHERE (name = '" + account()->name() + "'); ");
        if (!query.next()) {
            throw Storage::EWriteException(tr("The account could not have been written to the database."));
        }
        account()->setId(query.value(0).toInt());
    }else{
        // Update
        query.prepare("UPDATE t_accounts SET "
                        "name=:accountName, "
                        "type=:gatewayName "
                      "WHERE (accountId = :accountId); ");
        query.bindValue(":accountName", account()->name());
        query.bindValue(":gatewayName", account()->gateway()->name());
        query.bindValue(":accountId", account()->id());

        if (!query.exec()) {
            qCritical() << query.lastError().text();
            throw Storage::EWriteException(tr("The account could not have been written to the database."));
        }
    }
    storage()->writeBool(getNamespace(), "enabled", account()->isEnabled());
}

void AbstractAccountStorageOfficer::removeFromStorage() {
    storage()->removeValues(getNamespace());

    QSqlQuery query = storage()->createQuery();
    query.prepare("DELETE FROM t_accounts "
                    "WHERE (accountId = '" + QString::number(account()->id()) + "'); ");
    if (!query.exec()) {
        throw Storage::EWriteException(tr("Could not remove the account from the database."));
    }

    account()->setId(-1); // Invalidate
}



QString AbstractAccountStorageOfficer::getNamespace() const {
    return "account_" + QString::number(account()->id());
}


IAccount* AbstractAccountStorageOfficer::account() const {
    return account_;
}