source: SMSSender/src/ui/models/accounttreemodel.cpp @ 56:e63ceb8090cd

#include "accounttreemodel.h"

#include <QDebug>

AccountTreeModel::AccountTreeModel(QObject* parent)
    : QAbstractTableModel(parent)
{
    connect(BCAccountManager::instance(), SIGNAL(accountAdded(IAccount*)),
            this, SLOT(accountUpdated(IAccount*)));
    connect(BCAccountManager::instance(), SIGNAL(accountUpdated(IAccount*)),
            this, SLOT(accountUpdated(IAccount*)));
    connect(BCAccountManager::instance(), SIGNAL(accountRemoved(IAccount*)),
            this, SLOT(accountUpdated(IAccount*)));
}
AccountTreeModel::~AccountTreeModel() {
    disconnect(BCAccountManager::instance(), SIGNAL(accountAdded(IAccount*)),
               this, SLOT(accountUpdated(IAccount*)));
    disconnect(BCAccountManager::instance(), SIGNAL(accountUpdated(IAccount*)),
               this, SLOT(accountUpdated(IAccount*)));
    disconnect(BCAccountManager::instance(), SIGNAL(accountRemoved(IAccount*)),
               this, SLOT(accountUpdated(IAccount*)));
}

void AccountTreeModel::accountUpdated(IAccount* account) {
    int row = BCAccountManager::instance()->getAccountList().values().indexOf(account);

    QModelIndex idx_start = index(row, 0);
    QModelIndex idx_end   = index(row, MaxCol);

    if (idx_start.isValid() && idx_end.isValid()) {
        emit dataChanged(idx_start, idx_end);
    } else {
        reset();
    }
}

QVariant AccountTreeModel::headerData(int section, Qt::Orientation orientation, int role) const {
    if (orientation == Qt::Horizontal) {
        if (role == Qt::DisplayRole) {
            switch (section) {
                case ColName:         return tr("Name");
                case ColGatewayName:  return tr("Gateway");
                case ColEnabled:      return tr("Enabled");
            }
        }
        if (role == Qt::TextAlignmentRole) {
            return Qt::AlignLeft + Qt::AlignVCenter;
        }
    }
    return QAbstractTableModel::headerData(section, orientation, role);
}

IAccount* AccountTreeModel::dataObject(const QModelIndex& index) const {
    return BCAccountManager::instance()->getAccountList().values().at(index.row());
}

QVariant AccountTreeModel::data(const QModelIndex& index, int role) const {
    QList<IAccount*> accounts = BCAccountManager::instance()->getAccountList().values();
    if (!index.isValid() || index.row() < 0 || index.row() >= accounts.size())
        return QVariant();

    const IAccount* account = accounts.at(index.row());

    if (role == Qt::DisplayRole) {
        switch (index.column()) {
            case ColName:
                return account->name();
            case ColGatewayName:
                return account->gateway()->name();
        }
    }
    if (role == Qt::CheckStateRole) {
        switch (index.column()) {
            case ColEnabled:
                return (account->isEnabled()) ? Qt::Checked : Qt::Unchecked;
        }
    }
    if (role == Qt::DecorationRole) {
        switch (index.column()) {
            case ColGatewayName:
                return account->gateway()->icon();
        }
    }

    if (role == Qt::TextAlignmentRole) {
        switch (index.column()) {
            case ColName:
            case ColGatewayName:
                return Qt::AlignLeft + Qt::AlignVCenter;
            case ColEnabled:
                return Qt::AlignCenter + Qt::AlignVCenter;
        }
    }

    return QVariant();
}

Qt::ItemFlags AccountTreeModel::flags(const QModelIndex& index) const {
    if (!index.isValid())
        return Qt::ItemIsEnabled;

    Qt::ItemFlags flags = QAbstractTableModel::flags(index);
    if (index.column() == ColEnabled)
        flags |= Qt::ItemIsUserCheckable;
    return flags;
}

bool AccountTreeModel::setData(const QModelIndex& index, const QVariant& value, int role /* = Qt::EditRole */) {
    if (!index.isValid()) {
        return false;
    }

    if (role == Qt::CheckStateRole) {
        if (index.column() == ColEnabled) {
            IAccount* account = dataObject(index);
            Q_ASSERT(account);

            const Qt::CheckState cs = static_cast<Qt::CheckState>(value.toInt());
            account->setEnabled(cs != Qt::Unchecked);
            BCAccountManager::instance()->saveAccount(account);

            return true;
        }
    }

    return false;
}

int AccountTreeModel::columnCount(const QModelIndex& parent) const {
    return (parent.isValid()) ? 0 : MaxCol+1;
}

int AccountTreeModel::rowCount(const QModelIndex& parent) const {
    return (parent.isValid()) ? 0 : BCAccountManager::instance()->getAccountList().count();
}

bool AccountTreeModel::removeRows(int row, int count, const QModelIndex& parent) {
    if (parent.isValid())
        return false;

    int lastRow = row + count - 1;
    beginRemoveRows(parent, row, lastRow);
    QList<IAccount*> accounts = BCAccountManager::instance()->getAccountList().values();
    for (int i = lastRow; i >= row; --i) {
        BCAccountManager::instance()->removeAccount(accounts.at(i)->id());
    }
    endRemoveRows();
    return true;
}