| /* |
| * Copyright (C) 2019-2020 by Savoir-faire Linux |
| * Author: Isa Nanic <isa.nanic@savoirfairelinux.com> |
| * Author: Yang Wang <yang.wang@savoirfairelinux.com> |
| * |
| * This program is free software; you can redistribute it and/or modify |
| * it under the terms of the GNU General Public License as published by |
| * the Free Software Foundation; either version 3 of the License, or |
| * (at your option) any later version. |
| * |
| * This program is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| * GNU General Public License for more details. |
| * |
| * You should have received a copy of the GNU General Public License |
| * along with this program. If not, see <http://www.gnu.org/licenses/>. |
| */ |
| |
| #include "bannedlistmodel.h" |
| #include "lrcinstance.h" |
| |
| BannedListModel::BannedListModel(QObject *parent) |
| : QAbstractListModel(parent) |
| {} |
| |
| BannedListModel::~BannedListModel() {} |
| |
| int |
| BannedListModel::rowCount(const QModelIndex &parent) const |
| { |
| if (!parent.isValid()) { |
| return LRCInstance::getCurrentAccountInfo().contactModel->getBannedContacts().size(); |
| } |
| return 0; |
| } |
| |
| int |
| BannedListModel::columnCount(const QModelIndex &parent) const |
| { |
| Q_UNUSED(parent); |
| /* |
| * Only need one column. |
| */ |
| return 1; |
| } |
| |
| QVariant |
| BannedListModel::data(const QModelIndex &index, int role) const |
| { |
| auto contactList = LRCInstance::getCurrentAccountInfo().contactModel->getBannedContacts(); |
| if (!index.isValid() || contactList.size() <= index.row()) { |
| return QVariant(); |
| } |
| |
| auto contactInfo = LRCInstance::getCurrentAccountInfo().contactModel->getContact( |
| contactList.at(index.row())); |
| |
| switch (role) { |
| case Role::ContactName: |
| return QVariant(contactInfo.registeredName); |
| case Role::ContactID: |
| return QVariant(contactInfo.profileInfo.uri); |
| case Role::ContactPicture: |
| QImage avatarImage = Utils::fallbackAvatar(QSize(48, 48), |
| contactInfo.profileInfo.uri, |
| contactInfo.registeredName); |
| |
| return QString::fromLatin1(Utils::QImageToByteArray(avatarImage).toBase64().data()); |
| } |
| return QVariant(); |
| } |
| |
| QHash<int, QByteArray> |
| BannedListModel::roleNames() const |
| { |
| QHash<int, QByteArray> roles; |
| roles[ContactName] = "ContactName"; |
| roles[ContactID] = "ContactID"; |
| roles[ContactPicture] = "ContactPicture"; |
| return roles; |
| } |
| |
| QModelIndex |
| BannedListModel::index(int row, int column, const QModelIndex &parent) const |
| { |
| Q_UNUSED(parent); |
| if (column != 0) { |
| return QModelIndex(); |
| } |
| |
| if (row >= 0 && row < rowCount()) { |
| return createIndex(row, column); |
| } |
| return QModelIndex(); |
| } |
| |
| QModelIndex |
| BannedListModel::parent(const QModelIndex &child) const |
| { |
| Q_UNUSED(child); |
| return QModelIndex(); |
| } |
| |
| Qt::ItemFlags |
| BannedListModel::flags(const QModelIndex &index) const |
| { |
| auto flags = QAbstractItemModel::flags(index) | Qt::ItemNeverHasChildren | Qt::ItemIsSelectable; |
| if (!index.isValid()) { |
| return QAbstractItemModel::flags(index); |
| } |
| return flags; |
| } |
| |
| void |
| BannedListModel::reset() |
| { |
| beginResetModel(); |
| endResetModel(); |
| } |