source: SMSSender/src/ui/models/contactgroupmodel.cpp @ 55:33dc8614a999

#include "contactgroupmodel.h"

#include <QImage>
#include <QPixmap>
#include <QDebug>

ContactGroupListModel::ContactGroupListModel(QObject* parent)
    : QAbstractListModel(parent)
{
    connect(BCContactManager::instance(), SIGNAL(contactAdded(IContact*)),
            this, SLOT(dataUpdated(IContact*)));
    connect(BCContactManager::instance(), SIGNAL(contactUpdated(IContact*)),
            this, SLOT(dataUpdated(IContact*)));
    connect(BCContactManager::instance(), SIGNAL(contactRemoved(IContact*)),
            this, SLOT(dataUpdated(IContact*)));

    connect(BCGroupManager::instance(), SIGNAL(groupAdded(IGroup*)),
            this, SLOT(dataUpdated(IGroup*)));
    connect(BCGroupManager::instance(), SIGNAL(groupUpdated(IGroup*)),
            this, SLOT(dataUpdated(IGroup*)));
    connect(BCGroupManager::instance(), SIGNAL(groupRemoved(IGroup*)),
            this, SLOT(dataUpdated(IGroup*)));

    reloadList();
}
ContactGroupListModel::~ContactGroupListModel() {
    disconnect(BCContactManager::instance(), SIGNAL(contactAdded(IContact*)),
               this, SLOT(dataUpdated(IContact*)));
    disconnect(BCContactManager::instance(), SIGNAL(contactUpdated(IContact*)),
               this, SLOT(dataUpdated(IContact*)));
    disconnect(BCContactManager::instance(), SIGNAL(contactRemoved(IContact*)),
               this, SLOT(dataUpdated(IContact*)));

    disconnect(BCGroupManager::instance(), SIGNAL(groupAdded(IGroup*)),
               this, SLOT(dataUpdated(IGroup*)));
    disconnect(BCGroupManager::instance(), SIGNAL(groupUpdated(IGroup*)),
               this, SLOT(dataUpdated(IGroup*)));
    disconnect(BCGroupManager::instance(), SIGNAL(groupRemoved(IGroup*)),
               this, SLOT(dataUpdated(IGroup*)));
}

void ContactGroupListModel::dataUpdated(IContact* contact) {
    dataUpdated(static_cast<IInterface*>(contact));
}
void ContactGroupListModel::dataUpdated(IGroup* group) {
    dataUpdated(static_cast<IInterface*>(group));
}
void ContactGroupListModel::dataUpdated(IInterface* contactOrGroup) {
    int row = items_.indexOf(contactOrGroup);
    QModelIndex idx1 = index(row, 0);

    row = items_.indexOf(contactOrGroup);

    reloadList();

    if (row < 0) { // New contact/group added
        reset();
    } else {
        row = items_.indexOf(contactOrGroup);

        if (row < 0) { // Contact/group removed
            reset();
        } else {
            QModelIndex idx2 = index(row, 0);

            emit dataChanged(idx1, idx1);
            if (idx1 != idx2) {
                emit dataChanged(idx2, idx2);
            }
        }
    }
}

void ContactGroupListModel::reloadList() {
    items_.clear();
    foreach(IContact* contact, BCContactManager::instance()->getContactList()) {
        items_.append(contact);
    }
    foreach(IGroup* group, BCGroupManager::instance()->getGroupList()) {
        items_.append(group);
    }
}

void ContactGroupListModel::addFilteredContact(IContact* contact) {
    filteredItems_.insert(contact);
    items_.removeAll(contact);

    /* Filter groups, from whose all contacts are filtered */
    foreach (IGroup* group, groups()) {
        bool hasUnfilteredContacts = false;
        foreach (IContact* contact, group->contacts()) {
            if (items_.contains(contact)) {
                hasUnfilteredContacts = true;
                break;
            }
        }
        if (!hasUnfilteredContacts) {
            /* Add filtered Group */
            filteredItems_.insert(group);
            items_.removeAll(group);
        }
    }

    reset();
}
void ContactGroupListModel::removeFilteredContact(IContact* contact) {
    items_.append(contact);
    filteredItems_.remove(contact);

    /* Remove filter from groups, when all its contacts are unfiltered */
    foreach (IGroup* group, filteredGroups()) {
        bool hasFilteredContacts = false;
        foreach (IContact* contact, group->contacts()) {
            if (filteredItems_.contains(contact)) {
                hasFilteredContacts = true;
                break;
            }
        }
        if (!hasFilteredContacts) {
            /* Remove filtered group */
            items_.append(group);
            filteredItems_.remove(group);
        }
    }

    reset();
}

IInterface* ContactGroupListModel::dataObject(const QModelIndex& index) const {
    return items_.at(index.row());
}

QVariant ContactGroupListModel::data(const QModelIndex& index, int role) const {
    if (!index.isValid() || index.row() < 0 || index.row() >= items_.size())
        return QVariant();

    IInterface* contactOrGroup = items_.at(index.row());
    IContact*   contact        = dynamic_cast<IContact*>(contactOrGroup);
    IGroup*     group          = dynamic_cast<IGroup*>(contactOrGroup);

    switch (role) {
    case Qt::DisplayRole:
        if (contact != NULL) {
            QString contactStr = contact->name();
            QString aliasStr   = "";
            foreach(QString alias, contact->aliases()) {
                aliasStr += alias + ", ";
            }
            aliasStr.remove(QRegExp(", $"));
            if (aliasStr != "") {
                contactStr += " (" + aliasStr + ")";
            }
            return contactStr;
        }
        if (group != NULL) {
            return "<" + group->name() + ">";
        }
        break;

    case Qt::EditRole:
        if (contact != NULL)
            return contact->name();
        if (group != NULL)
            return group->name();
        break;

    case Qt::DecorationRole:
        QImage img;
        if (contact != NULL) {
            img = contact->image();
        }
        if (group != NULL) {
            img = group->image();
        }
        return QPixmap::fromImage(img.scaled(32, 32, Qt::KeepAspectRatio));
    }

/*    if (role == Qt::TextAlignmentRole) {
        switch (index.column()) {
            case COL_NAME:
            case COL_GATEWAY_NAME:
                return Qt::AlignLeft + Qt::AlignVCenter;
            case COL_ENABLED:
                return Qt::AlignCenter + Qt::AlignVCenter;
        }
    }*/

    return QVariant();
}

QSet<IGroup*> ContactGroupListModel::groups() const {
    QSet<IGroup*> result;
    foreach (IInterface* i, items_) {
        IGroup* g = dynamic_cast<IGroup*>(i);
        if (g != NULL)
            result.insert(g);
    }
    return result;
}
QSet<IContact*> ContactGroupListModel::contacts() const {
    QSet<IContact*> result;
    foreach (IInterface* i, items_) {
        IContact* c = dynamic_cast<IContact*>(i);
        if (c != NULL)
            result.insert(c);
    }
    return result;
}
QSet<IGroup*> ContactGroupListModel::filteredGroups() const {
    QSet<IGroup*> result;
    foreach (IInterface* i, filteredItems_) {
        IGroup* g = dynamic_cast<IGroup*>(i);
        if (g != NULL)
            result.insert(g);
    }
    return result;
}
QSet<IContact*> ContactGroupListModel::filteredContacts() const {
    QSet<IContact*> result;
    foreach (IInterface* i, filteredItems_) {
        IContact* c = dynamic_cast<IContact*>(i);
        if (c != NULL)
            result.insert(c);
    }
    return result;
}


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

int ContactGroupListModel::rowCount(const QModelIndex& parent) const {
    return (parent.isValid()) ? 0 : items_.count();
}