source: SMSSender/src/ui/models/grouptablemodel.cpp @ 61:8e5fa8a03259

#include "grouptablemodel.h"

#include <QDebug>

#include "../../business/BCGroupManager.h"

GroupTableModel::GroupTableModel(QObject* parent)
    : QAbstractTableModel(parent)
{
    connect(BCGroupManager::instance(), SIGNAL(groupAdded(IGroup*)),
            this, SLOT(groupUpdated(IGroup*)));
    connect(BCGroupManager::instance(), SIGNAL(groupUpdated(IGroup*)),
            this, SLOT(groupUpdated(IGroup*)));
    connect(BCGroupManager::instance(), SIGNAL(groupRemoved(IGroup*)),
            this, SLOT(groupUpdated(IGroup*)));
}
GroupTableModel::~GroupTableModel() {
    disconnect(BCGroupManager::instance(), SIGNAL(groupAdded(IGroup*)),
               this, SLOT(groupUpdated(IGroup*)));
    disconnect(BCGroupManager::instance(), SIGNAL(groupUpdated(IGroup*)),
               this, SLOT(groupUpdated(IGroup*)));
    disconnect(BCGroupManager::instance(), SIGNAL(groupRemoved(IGroup*)),
               this, SLOT(groupUpdated(IGroup*)));
}

void GroupTableModel::groupUpdated(IGroup* group) {
    int row = BCGroupManager::instance()->getGroupList().values().indexOf(group);

    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 GroupTableModel::headerData(int section, Qt::Orientation orientation, int role) const {
    if (orientation == Qt::Horizontal) {
        if (role == Qt::DisplayRole) {
            switch (section) {
                case ColImage:        return "";
                case ColName:         return tr("Name");
                case ColMembers:      return tr("Members");
            }
        }
        if (role == Qt::TextAlignmentRole) {
            return Qt::AlignLeft + Qt::AlignVCenter;
        }
    }
    return QAbstractTableModel::headerData(section, orientation, role);
}

IGroup* GroupTableModel::dataObject(const QModelIndex& index) const {
    return BCGroupManager::instance()->getGroupList().values().at(index.row());
}

QVariant GroupTableModel::data(const QModelIndex& index, int role) const {
    QList<IGroup*> groups = BCGroupManager::instance()->getGroupList().values();
    if (!index.isValid() || index.row() < 0 || index.row() >= groups.size())
        return QVariant();

    const IGroup* group = groups.at(index.row());

    if (role == Qt::DisplayRole) {
        switch (index.column()) {
            case ColImage:
                return "";
            case ColName:
                return group->name();
            case ColMembers:
                return getMembersStr(group);
        }
    }
    if (role == Qt::DecorationRole) {
        switch (index.column()) {
            case ColImage:
                return group->image();
        }
    }

    return QVariant();
}

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

int GroupTableModel::rowCount(const QModelIndex& parent) const {
    return (parent.isValid()) ? 0 : BCGroupManager::instance()->getGroupList().count();
}

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

    int lastRow = row + count - 1;
    beginRemoveRows(parent, row, lastRow);
    QList<IGroup*> groups = BCGroupManager::instance()->getGroupList().values();
    for (int i = lastRow; i >= row; --i) {
        BCGroupManager::instance()->removeGroup(groups.at(i)->id());
    }
    endRemoveRows();
    return true;
}


QString GroupTableModel::getMembersStr(const IGroup* group) const {
    QStringList members;
    foreach (IContact* contact, group->contacts()) {
        members.append(contact->name());
    }
    members.sort();
    return members.join(", ");
}