squawk/ui/models/roster.cpp

1098 lines
39 KiB
C++

/*
* Squawk messenger.
* Copyright (C) 2019 Yury Gubich <blue@macaw.me>
*
* 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 "roster.h"
#include <QDebug>
#include <QIcon>
#include <QFont>
#include "shared/defines.h"
Models::Roster::Roster(QObject* parent):
QAbstractItemModel(parent),
accountsModel(new Accounts()),
root(new Item(Item::root, {{"name", "root"}})),
accounts(),
groups(),
contacts()
{
connect(accountsModel, &Accounts::dataChanged, this, &Roster::onAccountDataChanged);
connect(root, &Item::childChanged, this, &Roster::onChildChanged);
connect(root, &Item::childIsAboutToBeInserted, this, &Roster::onChildIsAboutToBeInserted);
connect(root, &Item::childInserted, this, &Roster::onChildInserted);
connect(root, &Item::childIsAboutToBeRemoved, this, &Roster::onChildIsAboutToBeRemoved);
connect(root, &Item::childRemoved, this, &Roster::onChildRemoved);
connect(root, &Item::childIsAboutToBeMoved, this, &Roster::onChildIsAboutToBeMoved);
connect(root, &Item::childMoved, this, &Roster::onChildMoved);
}
Models::Roster::~Roster() {
delete accountsModel;
delete root;
}
void Models::Roster::addAccount(const QMap<QString, QVariant>& data) {
Account* acc = new Account(data);
connect(acc, &Account::reconnected, this, &Roster::onAccountReconnected);
root->appendChild(acc);
accounts.insert(std::make_pair(acc->getName(), acc));
accountsModel->addAccount(acc);
emit addedElement({acc->getId()});
}
QVariant Models::Roster::data (const QModelIndex& index, int role) const {
if (!index.isValid())
return QVariant();
QVariant result;
Item *item = static_cast<Item*>(index.internalPointer());
if (item->type == Item::reference)
item = static_cast<Reference*>(item)->dereference();
switch (role) {
case Qt::DisplayRole: {
if (index.column() != 0)
break;
switch (item->type) {
case Item::group: {
Group* gr = static_cast<Group*>(item);
QString str("");
str += gr->getName();
unsigned int amount = gr->getUnreadMessages();
if (amount > 0)
str += QString(" (") + tr("New messages") + ")";
result = str;
}
break;
default:
result = item->data(index.column());
break;
}
}
break;
case Qt::DecorationRole:
switch (item->type) {
case Item::account: {
quint8 col = index.column();
Account* acc = static_cast<Account*>(item);
if (col == 0) {
result = acc->getStatusIcon(false);
} else if (col == 1) {
QString path = acc->getAvatarPath();
if (path.size() > 0)
result = QIcon(path);
}
}
break;
case Item::contact: {
Contact* contact = static_cast<Contact*>(item);
quint8 col = index.column();
if (col == 0) {
result = contact->getStatusIcon(false);
} else if (col == 1) {
if (contact->getAvatarState() != Shared::Avatar::empty)
result = QIcon(contact->getAvatarPath());
}
}
break;
case Item::presence: {
if (index.column() != 0)
break;
Presence* presence = static_cast<Presence*>(item);
result = presence->getStatusIcon(false);
}
break;
case Item::room: {
quint8 col = index.column();
Room* room = static_cast<Room*>(item);
if (col == 0) {
result = room->getStatusIcon(false);
} else if (col == 1) {
QString path = room->getAvatarPath();
if (path.size() > 0)
result = QIcon(path);
}
}
break;
case Item::participant: {
quint8 col = index.column();
Participant* p = static_cast<Participant*>(item);
if (col == 0) {
result = p->getStatusIcon(false);
} else if (col == 1) {
QString path = p->getAvatarPath();
if (path.size() > 0)
result = QIcon(path);
}
if (index.column() != 0)
break;
}
break;
default:
break;
}
break;
case Qt::FontRole:
switch (item->type) {
case Item::account:{
QFont font;
font.setBold(true);
result = font;
}
break;
case Item::group:{
QFont font;
font.setItalic(true);
result = font;
}
break;
default:
break;
}
break;
case Qt::ToolTipRole:
switch (item->type) {
case Item::account: {
Account* acc = static_cast<Account*>(item);
result = Shared::Global::getName(acc->getAvailability());
}
break;
case Item::contact: {
Contact* contact = static_cast<Contact*>(item);
QString str("");
int mc = contact->getMessagesCount();
if (mc > 0)
str += QString(tr("New messages: ")) + std::to_string(mc).c_str() + "\n";
str += tr("Jabber ID: ") + contact->getJid() + "\n";
Shared::SubscriptionState ss = contact->getState();
if (ss == Shared::SubscriptionState::both || ss == Shared::SubscriptionState::to) {
Shared::Availability av = contact->getAvailability();
str += tr("Availability: ") + Shared::Global::getName(av);
if (av != Shared::Availability::offline) {
QString s = contact->getStatus();
if (s.size() > 0)
str += "\n" + tr("Status: ") + s;
}
str += "\n" + tr("Subscription: ") + Shared::Global::getName(ss);
} else {
str += tr("Subscription: ") + Shared::Global::getName(ss);
}
result = str;
}
break;
case Item::presence: {
Presence* contact = static_cast<Presence*>(item);
QString str;
Shared::Availability av = contact->getAvailability();
str += tr("Availability: ") + Shared::Global::getName(av);
QString s = contact->getStatus();
if (s.size() > 0)
str += "\n" + tr("Status: ") + s;
str += "\n" + tr("Client: ") + contact->getClientNode();
result = str;
}
break;
case Item::participant: {
Participant* p = static_cast<Participant*>(item);
QString str;
Shared::Availability av = p->getAvailability();
str += tr("Availability: ") + Shared::Global::getName(av) + "\n";
QString s = p->getStatus();
if (s.size() > 0)
str += tr("Status: ") + s + "\n";
str += tr("Affiliation: ") + Shared::Global::getName(p->getAffiliation()) + "\n";
str += tr("Role: ") + Shared::Global::getName(p->getRole()) + "\n";
str += tr("Client: ") + p->getClientNode();
result = str;
}
break;
case Item::group: {
Group* gr = static_cast<Group*>(item);
unsigned int count = gr->getUnreadMessages();
QString str("");
if (count > 0)
str += tr("New messages: ") + std::to_string(count).c_str() + "\n";
str += tr("Online contacts: ") + std::to_string(gr->getOnlineContacts()).c_str() + "\n";
str += tr("Total contacts: ") + std::to_string(gr->childCount()).c_str();
result = str;
}
break;
case Item::room: {
Room* rm = static_cast<Room*>(item);
unsigned int count = rm->getMessagesCount();
QString str("");
if (count > 0)
str += tr("New messages: ") + std::to_string(count).c_str() + "\n";
str += tr("Jabber ID: ") + rm->getJid() + "\n";
str += tr("Subscription: ") + rm->getStatusText();
if (rm->getJoined())
str += QString("\n") + tr("Members: ") + std::to_string(rm->childCount()).c_str();
result = str;
}
break;
default:
result = "";
break;
}
break;
case Qt::ForegroundRole:
switch (item->type) {
case Item::account: {
Account* acc = static_cast<Account*>(item);
if (!acc->getActive())
result = qApp->palette().brush(QPalette::Disabled, QPalette::Text);
}
break;
default:
break;
}
default:
break;
}
return result;
}
int Models::Roster::columnCount (const QModelIndex& parent) const {
if (parent.isValid())
return static_cast<Item*>(parent.internalPointer())->columnCount();
else
return root->columnCount();
}
void Models::Roster::updateAccount(const QString& account, const QString& field, const QVariant& value) {
std::map<QString, Account*>::iterator itr = accounts.find(account);
if (itr != accounts.end()) {
Account* acc = itr->second;
acc->update(field, value);
}
}
Qt::ItemFlags Models::Roster::flags(const QModelIndex& index) const {
if (!index.isValid())
return Qt::ItemFlags();
return QAbstractItemModel::flags(index);
}
int Models::Roster::rowCount (const QModelIndex& parent) const{
Item *parentItem;
if (!parent.isValid())
parentItem = root;
else
parentItem = static_cast<Item*>(parent.internalPointer());
return parentItem->childCount();
}
QVariant Models::Roster::headerData(int section, Qt::Orientation orientation, int role) const {
SHARED_UNUSED(section);
SHARED_UNUSED(orientation);
SHARED_UNUSED(role);
return QVariant();
}
QModelIndex Models::Roster::parent (const QModelIndex& child) const {
if (!child.isValid())
return QModelIndex();
Item *childItem = static_cast<Item*>(child.internalPointer());
if (childItem == root)
return QModelIndex();
Item *parentItem = childItem->parentItem();
if (parentItem == root)
return createIndex(0, 0, parentItem);
return createIndex(parentItem->row(), 0, parentItem);
}
QModelIndex Models::Roster::index (int row, int column, const QModelIndex& parent) const {
if (!hasIndex(row, column, parent))
return QModelIndex();
Item *parentItem;
if (!parent.isValid())
parentItem = root;
else
parentItem = static_cast<Item*>(parent.internalPointer());
Item *childItem = parentItem->child(row);
if (childItem)
return createIndex(row, column, childItem);
else
return QModelIndex();
}
Models::Roster::ElId::ElId(const QString& p_account, const QString& p_name):
account(p_account),
name(p_name)
{}
bool Models::Roster::ElId::operator <(const Models::Roster::ElId& other) const {
if (account == other.account)
return name < other.name;
else
return account < other.account;
}
bool Models::Roster::ElId::operator!=(const Models::Roster::ElId& other) const {
return !(operator == (other));
}
bool Models::Roster::ElId::operator==(const Models::Roster::ElId& other) const {
return (account == other.account) && (name == other.name);
}
void Models::Roster::onAccountDataChanged(const QModelIndex& tl, const QModelIndex& br, const QVector<int>& roles) {
if (tl.column() == 0) {
emit dataChanged(tl, br, roles);
} else if (tl.column() == 2) {
int row = tl.row();
Account* acc = accountsModel->getAccount(row);
emit dataChanged(createIndex(row, 0, acc), createIndex(br.row(), 0, acc), roles);
}
}
void Models::Roster::addGroup(const QString& account, const QString& name) {
ElId id(account, name);
std::map<ElId, Group*>::const_iterator gItr = groups.find(id);
if (gItr != groups.end()) {
qDebug() << "An attempt to add group " << name << " to account " << account <<" which already exists there, skipping";
return;
}
std::map<QString, Account*>::iterator itr = accounts.find(account);
if (itr != accounts.end()) {
Account* acc = itr->second;
Group* group = new Group({{"name", name}});
groups.insert(std::make_pair(id, group));
acc->appendChild(group);
emit addedElement({acc->getId(), group->getId()});
} else {
qDebug() << "An attempt to add group " << name << " to non existing account " << account << ", skipping";
}
}
void Models::Roster::addContact(const QString& account, const QString& jid, const QString& group, const QMap<QString, QVariant>& data) {
Item* parent;
Account* acc;
Contact* contact;
Reference* ref = 0;
ElId id(account, jid);
{
std::map<QString, Account*>::iterator itr = accounts.find(account);
if (itr == accounts.end()) {
qDebug() << "An attempt to add a contact" << jid << "to non existing account" << account << ", skipping";
return;
}
acc = itr->second;
}
{
std::map<ElId, Contact*>::iterator itr = contacts.find(id);
if (itr == contacts.end()) {
contact = new Contact(acc, jid, data);
connect(contact, &Contact::requestArchive, this, &Roster::onElementRequestArchive);
connect(contact, &Contact::fileDownloadRequest, this, &Roster::fileDownloadRequest);
connect(contact, &Contact::unnoticedMessage, this, &Roster::unnoticedMessage);
connect(contact, &Contact::localPathInvalid, this, &Roster::localPathInvalid);
connect(contact, &Contact::unreadMessagesCountChanged, this, &Roster::recalculateUnreadMessages);
contacts.insert(std::make_pair(id, contact));
} else {
contact = itr->second;
}
}
std::list<QString> path = {acc->getId()};
if (group == "") {
if (acc->getContact(jid) != -1) {
qDebug() << "An attempt to add a contact" << jid << "to the ungrouped contact set of account" << account << "for the second time, skipping";
return;
} else {
parent = acc;
}
} else {
std::map<ElId, Group*>::iterator itr = groups.find({account, group});
if (itr == groups.end()) {
qDebug() << "An attempt to add a contact" << jid << "to non existing group" << group << ", adding group";
addGroup(account, group);
itr = groups.find({account, group});
}
parent = itr->second;
path.push_back(parent->getId());
if (parent->getContact(jid) != -1) {
qDebug() << "An attempt to add a contact" << jid << "to the group" << group << "for the second time, skipping";
return;
}
int refIndex = acc->getContact(jid);
if (refIndex != -1) { //checking if that contact is among ugrouped
qDebug() << "An attempt to add a already existing contact " << jid
<< " to the group " << group
<< ", contact will be moved from ungrouped contacts of " << account;
ref = static_cast<Reference*>(acc->child(refIndex));
acc->removeChild(refIndex);
}
}
path.push_back(contact->getId());
if (ref == 0)
ref = new Reference(contact);
parent->appendChild(ref);
emit addedElement(path);
}
void Models::Roster::removeGroup(const QString& account, const QString& name) {
ElId id(account, name);
std::map<ElId, Group*>::const_iterator gItr = groups.find(id);
if (gItr == groups.end()) {
qDebug() << "An attempt to remove group " << name << " from account " << account <<" which doesn't exist there, skipping";
return;
}
Group* item = gItr->second;
Item* parent = item->parentItem();
int row = item->row();
parent->removeChild(row);
std::deque<Reference*> toInsert;
for (int i = 0; item->childCount() > 0; ++i) {
Reference* ref = static_cast<Reference*>(item->child(0));
item->removeChild(0);
Contact* cont = static_cast<Contact*>(ref->dereference());
if (cont->referencesCount() == 1)
toInsert.push_back(ref);
else
delete ref;
}
if (toInsert.size() > 0) {
Account* acc = accounts.find("account")->second;
for (std::deque<Contact*>::size_type i = 0; i < toInsert.size(); ++i) {
acc->appendChild(toInsert[i]); //TODO optimisation
}
}
item->deleteLater();
groups.erase(gItr);
}
void Models::Roster::changeContact(const QString& account, const QString& jid, const QMap<QString, QVariant>& data) {
Element* el = getElement(ElId(account, jid));
if (el != nullptr) {
for (QMap<QString, QVariant>::const_iterator itr = data.begin(), end = data.end(); itr != end; ++itr) {
el->update(itr.key(), itr.value());
}
}
}
void Models::Roster::changeMessage(const QString& account, const QString& jid, const QString& id, const QMap<QString, QVariant>& data) {
Element* el = getElement(ElId(account, jid));
if (el != nullptr)
el->changeMessage(id, data);
else
qDebug() << "A request to change a message of the contact " << jid << " in the account " << account << " but it wasn't found";
}
void Models::Roster::removeContact(const QString& account, const QString& jid) {
ElId id(account, jid);
std::map<ElId, Contact*>::iterator itr = contacts.find(id);
if (itr != contacts.end()) {
Contact* contact = itr->second;
contacts.erase(itr);
delete contact;
std::set<ElId> toRemove;
for (std::pair<ElId, Group*> pair : groups) {
if (pair.second->childCount() == 0)
toRemove.insert(pair.first);
}
for (const ElId& elId : toRemove)
removeGroup(elId.account, elId.name);
} else {
qDebug() << "An attempt to remove contact " << jid << " from account " << account <<" which doesn't exist there, skipping";
}
}
void Models::Roster::removeContact(const QString& account, const QString& jid, const QString& group) {
ElId contactId(account, jid);
ElId groupId(account, group);
std::map<ElId, Contact*>::iterator cItr = contacts.find(contactId);
if (cItr == contacts.end()) {
qDebug() << "An attempt to remove non existing contact " << jid << " from group " << group << " of account " << account <<", skipping";
return;
}
std::map<ElId, Group*>::iterator gItr = groups.find(groupId);
if (gItr == groups.end()) {
qDebug() << "An attempt to remove contact " << jid << " from non existing group " << group << " of account " << account <<", skipping";
return;
}
Account* acc = accounts.find(account)->second; //I assume the account is found, otherwise there will be no groups with that ElId;
Group* gr = gItr->second;
Contact* cont = cItr->second;
int contRow = gr->getContact(jid);
if (contRow == -1) {
qDebug() << "An attempt to remove contact " << jid << " of account " << account << " from group " << group <<", but there is no such contact in that group, skipping";
return;
}
Reference* ref = static_cast<Reference*>(gr->child(contRow));
gr->removeChild(contRow);
if (cont->referencesCount() == 1) {
qDebug() << "An attempt to remove last instance of contact" << jid << "from the group" << group << ", contact will be moved to ungrouped contacts of" << account;
acc->appendChild(ref);
} else {
delete ref;
}
if (gr->childCount() == 0)
removeGroup(account, group);
}
void Models::Roster::onChildChanged(Models::Item* item, int row, int col) {
SHARED_UNUSED(col);
QModelIndex index = createIndex(row, 0, item);
QModelIndex index2 = createIndex(row, 1, item);
emit dataChanged(index, index2);
}
void Models::Roster::onChildIsAboutToBeInserted(Models::Item* parent, int first, int last) {
int row = 0;
if (parent != root) {
row = parent->row();
beginInsertRows(createIndex(row, 0, parent), first, last);
} else {
beginInsertRows(QModelIndex(), first, last);
}
}
void Models::Roster::onChildIsAboutToBeMoved(Models::Item* source, int first, int last, Models::Item* destination, int newIndex) {
int oldRow = 0;
if (source != root)
oldRow = source->row();
int newRow = 0;
if (destination != root)
newRow = destination->row();
beginMoveRows(createIndex(oldRow, 0, source), first, last, createIndex(newRow, 0, destination), newIndex);
}
void Models::Roster::onChildIsAboutToBeRemoved(Models::Item* parent, int first, int last) {
int row = 0;
if (parent != root)
row = parent->row();
beginRemoveRows(createIndex(row, 0, parent), first, last);
}
void Models::Roster::onChildInserted() {
endInsertRows();
}
void Models::Roster::onChildMoved() {
endMoveRows();
}
void Models::Roster::onChildRemoved() {
endRemoveRows();
}
void Models::Roster::addPresence(const QString& account, const QString& jid, const QString& name, const QMap<QString, QVariant>& data) {
ElId contactId(account, jid);
std::map<ElId, Contact*>::iterator itr = contacts.find(contactId);
if (itr != contacts.end())
itr->second->addPresence(name, data);
else
qDebug() << "Received a presence" << jid + "/" + name << "don't know what to do with it";
}
void Models::Roster::removePresence(const QString& account, const QString& jid, const QString& name) {
ElId contactId(account, jid);
std::map<ElId, Contact*>::iterator itr = contacts.find(contactId);
if (itr != contacts.end()) {
itr->second->removePresence(name);
}
}
void Models::Roster::addMessage(const QString& account, const Shared::Message& data) {
Element* el = getElement(ElId(account, data.getPenPalJid()));
if (el != nullptr)
el->addMessage(data);
}
void Models::Roster::removeAccount(const QString& account) {
std::map<QString, Account*>::const_iterator itr = accounts.find(account);
if (itr == accounts.end()) {
qDebug() << "An attempt to remove non existing account " << account << ", skipping";
return;
}
Account* acc = itr->second;
int index = acc->row();
root->removeChild(index);
accountsModel->removeAccount(index);
accounts.erase(itr);
std::map<ElId, Contact*>::const_iterator cItr = contacts.begin();
while (cItr != contacts.end()) {
if (cItr->first.account == account) {
std::map<ElId, Contact*>::const_iterator lItr = cItr;
++cItr;
contacts.erase(lItr);
} else {
++cItr;
}
}
std::map<ElId, Group*>::const_iterator gItr = groups.begin();
while (gItr != groups.end()) {
if (gItr->first.account == account) {
std::map<ElId, Group*>::const_iterator lItr = gItr;
++gItr;
groups.erase(lItr);
} else {
++gItr;
}
}
std::map<ElId, Room*>::const_iterator rItr = rooms.begin();
while (rItr != rooms.end()) {
if (rItr->first.account == account) {
std::map<ElId, Room*>::const_iterator lItr = rItr;
++rItr;
rooms.erase(lItr);
} else {
++rItr;
}
}
disconnect(acc, &Account::reconnected, this, &Roster::onAccountReconnected);
acc->deleteLater();
}
QString Models::Roster::getContactName(const QString& account, const QString& jid) const {
ElId id(account, jid);
std::map<ElId, Contact*>::const_iterator cItr = contacts.find(id);
QString name = "";
if (cItr == contacts.end()) {
std::map<ElId, Room*>::const_iterator rItr = rooms.find(id);
if (rItr == rooms.end())
qDebug() << "An attempt to get a name of non existing contact/room " << account << ":" << jid << ", skipping";
else
name = rItr->second->getRoomName();
} else {
name = cItr->second->getContactName();
}
return name;
}
void Models::Roster::addRoom(const QString& account, const QString jid, const QMap<QString, QVariant>& data) {
Account* acc;
{
std::map<QString, Account*>::iterator itr = accounts.find(account);
if (itr == accounts.end()) {
qDebug() << "An attempt to add a room " << jid << " to non existing account " << account << ", skipping";
return;
}
acc = itr->second;
}
ElId id = {account, jid};
std::map<ElId, Room*>::const_iterator itr = rooms.find(id);
if (itr != rooms.end()) {
qDebug() << "An attempt to add already existing room" << jid << ", skipping";
return;
}
Room* room = new Room(acc, jid, data);
connect(room, &Room::requestArchive, this, &Roster::onElementRequestArchive);
connect(room, &Room::fileDownloadRequest, this, &Roster::fileDownloadRequest);
connect(room, &Room::unnoticedMessage, this, &Roster::unnoticedMessage);
connect(room, &Room::localPathInvalid, this, &Roster::localPathInvalid);
connect(room, &Room::unreadMessagesCountChanged, this, &Roster::recalculateUnreadMessages);
rooms.insert(std::make_pair(id, room));
acc->appendChild(room);
emit addedElement({acc->getId(), room->getId()});
}
void Models::Roster::changeRoom(const QString& account, const QString jid, const QMap<QString, QVariant>& data) {
ElId id = {account, jid};
std::map<ElId, Room*>::const_iterator itr = rooms.find(id);
if (itr == rooms.end()) {
qDebug() << "An attempt to change non existing room" << jid << ", skipping";
return;
}
Room* room = itr->second;
for (QMap<QString, QVariant>::const_iterator dItr = data.begin(), dEnd = data.end(); dItr != dEnd; ++dItr) {
room->update(dItr.key(), dItr.value());
}
}
void Models::Roster::removeRoom(const QString& account, const QString jid) {
Account* acc;
{
std::map<QString, Account*>::iterator itr = accounts.find(account);
if (itr == accounts.end()) {
qDebug() << "An attempt to remove a room " << jid << " from non existing account " << account << ", skipping";
return;
}
acc = itr->second;
}
ElId id = {account, jid};
std::map<ElId, Room*>::const_iterator itr = rooms.find(id);
if (itr == rooms.end()) {
qDebug() << "An attempt to remove non existing room" << jid << ", skipping";
return;
}
Room* room = itr->second;
acc->removeChild(room->row());
room->deleteLater();
rooms.erase(itr);
}
void Models::Roster::addRoomParticipant(const QString& account, const QString& jid, const QString& name, const QMap<QString, QVariant>& data) {
ElId id = {account, jid};
std::map<ElId, Room*>::const_iterator itr = rooms.find(id);
if (itr == rooms.end()) {
qDebug() << "An attempt to add participant" << name << "non existing room" << jid << "of an account" << account << ", skipping";
return;
} else {
itr->second->addParticipant(name, data);
}
}
void Models::Roster::changeRoomParticipant(const QString& account, const QString& jid, const QString& name, const QMap<QString, QVariant>& data) {
ElId id = {account, jid};
std::map<ElId, Room*>::const_iterator itr = rooms.find(id);
if (itr == rooms.end()) {
qDebug() << "An attempt change participant" << name << "of non existing room" << jid << "of an account" << account << ", skipping";
return;
} else {
itr->second->changeParticipant(name, data);
}
}
void Models::Roster::removeRoomParticipant(const QString& account, const QString& jid, const QString& name) {
ElId id = {account, jid};
std::map<ElId, Room*>::const_iterator itr = rooms.find(id);
if (itr == rooms.end()) {
qDebug() << "An attempt remove participant" << name << "from non existing room" << jid << "of an account" << account << ", skipping";
return;
} else {
itr->second->removeParticipant(name);
}
}
std::deque<QString> Models::Roster::groupList(const QString& account) const {
std::deque<QString> answer;
for (std::pair<ElId, Group*> pair : groups) {
if (pair.first.account == account)
answer.push_back(pair.first.name);
}
return answer;
}
bool Models::Roster::groupHasContact(const QString& account, const QString& group, const QString& contact) const {
ElId grId({account, group});
std::map<ElId, Group*>::const_iterator gItr = groups.find(grId);
if (gItr == groups.end()) {
return false;
} else {
return gItr->second->getContact(contact) != -1;
}
}
QString Models::Roster::getContactIconPath(const QString& account, const QString& jid, const QString& resource) const {
ElId id(account, jid);
std::map<ElId, Contact*>::const_iterator cItr = contacts.find(id);
QString path = "";
if (cItr == contacts.end()) {
std::map<ElId, Room*>::const_iterator rItr = rooms.find(id);
if (rItr == rooms.end())
qDebug() << "An attempt to get an icon path of non existing contact" << account << ":" << jid << ", returning empty value";
else
path = rItr->second->getParticipantIconPath(resource);
} else {
if (cItr->second->getAvatarState() != Shared::Avatar::empty)
path = cItr->second->getAvatarPath();
}
return path;
}
Models::Account * Models::Roster::getAccount(const QString& name) {
return const_cast<Models::Account*>(getAccountConst(name));}
const Models::Account * Models::Roster::getAccountConst(const QString& name) const {
return accounts.at(name);}
const Models::Element * Models::Roster::getElementConst(const Models::Roster::ElId& id) const {
std::map<ElId, Contact*>::const_iterator cItr = contacts.find(id);
if (cItr != contacts.end()) {
return cItr->second;
} else {
std::map<ElId, Room*>::const_iterator rItr = rooms.find(id);
if (rItr != rooms.end())
return rItr->second;
}
return nullptr;
}
bool Models::Roster::markMessageAsRead(const Models::Roster::ElId& elementId, const QString& messageId) {
const Element* el = getElementConst(elementId);
if (el != nullptr)
return el->markMessageAsRead(messageId);
else
return false;
}
QModelIndex Models::Roster::getAccountIndex(const QString& name) const {
std::map<QString, Account*>::const_iterator itr = accounts.find(name);
if (itr == accounts.end()) {
return QModelIndex();
} else {
return index(itr->second->row(), 0, QModelIndex());
}
}
QModelIndex Models::Roster::getGroupIndex(const QString& account, const QString& name) const {
std::map<QString, Account*>::const_iterator itr = accounts.find(account);
if (itr == accounts.end()) {
return QModelIndex();
} else {
std::map<ElId, Group*>::const_iterator gItr = groups.find(ElId(account, name));
if (gItr == groups.end()) {
return QModelIndex();
} else {
QModelIndex accIndex = index(itr->second->row(), 0, QModelIndex());
return index(gItr->second->row(), 0, accIndex);
}
}
}
QModelIndex Models::Roster::getContactIndex(const QString& account, const QString& jid, const QString& resource) const {
std::map<QString, Account*>::const_iterator itr = accounts.find(account);
if (itr == accounts.end()) {
return QModelIndex();
} else {
Account* acc = itr->second;
QModelIndex accIndex = index(acc->row(), 0, QModelIndex());
std::map<ElId, Contact*>::const_iterator cItr = contacts.find(ElId(account, jid));
if (cItr != contacts.end()) {
QModelIndex contactIndex = index(acc->getContact(jid), 0, accIndex);
if (resource.size() == 0) {
return contactIndex;
} else {
Presence* pres = cItr->second->getPresence(resource);
if (pres != nullptr)
return index(pres->row(), 0, contactIndex);
else
return contactIndex;
}
} else {
std::map<ElId, Room*>::const_iterator rItr = rooms.find(ElId(account, jid));
if (rItr != rooms.end()) {
QModelIndex roomIndex = index(rItr->second->row(), 0, accIndex);
if (resource.size() == 0) {
return roomIndex;
} else {
Participant* part = rItr->second->getParticipant(resource);
if (part != nullptr)
return index(part->row(), 0, roomIndex);
else
return roomIndex;
}
} else {
return QModelIndex();
}
}
}
}
void Models::Roster::onElementRequestArchive(const QString& before) {
Element* el = static_cast<Element*>(sender());
emit requestArchive(el->getAccountName(), el->getJid(), before);
}
void Models::Roster::responseArchive(const QString& account, const QString& jid, const std::list<Shared::Message>& list, bool last) {
ElId id(account, jid);
Element* el = getElement(id);
if (el != nullptr)
el->responseArchive(list, last);
}
void Models::Roster::fileProgress(const std::list<Shared::MessageInfo>& msgs, qreal value, bool up) {
for (const Shared::MessageInfo& info : msgs) {
Element* el = getElement(ElId(info.account, info.jid));
if (el != nullptr)
el->fileProgress(info.messageId, value, up);
}
}
void Models::Roster::fileComplete(const std::list<Shared::MessageInfo>& msgs, bool up) {
for (const Shared::MessageInfo& info : msgs) {
Element* el = getElement(ElId(info.account, info.jid));
if (el != nullptr)
el->fileComplete(info.messageId, up);
}
}
void Models::Roster::fileError(const std::list<Shared::MessageInfo>& msgs, const QString& err, bool up) {
for (const Shared::MessageInfo& info : msgs) {
Element* el = getElement(ElId(info.account, info.jid));
if (el != nullptr)
el->fileError(info.messageId, err, up);
}
}
Models::Element * Models::Roster::getElement(const Models::Roster::ElId& id) {
return const_cast<Models::Element*>(getElementConst(id));
}
Models::Item::Type Models::Roster::getContactType(const Models::Roster::ElId& id) const {
const Models::Element* el = getElementConst(id);
if (el == nullptr)
return Item::root;
return el->type;
}
void Models::Roster::onAccountReconnected() {
Account* acc = static_cast<Account*>(sender());
QString accName = acc->getName();
for (const std::pair<const ElId, Contact*>& pair : contacts) {
if (pair.first.account == accName)
pair.second->handleRecconnect();
}
}
void Models::Roster::recalculateUnreadMessages() {
int count(0);
for (const std::pair<const ElId, Contact*>& pair : contacts)
count += pair.second->getMessagesCount();
for (const std::pair<const ElId, Room*>& pair : rooms)
count += pair.second->getMessagesCount();
emit unreadMessagesCountChanged(count);
}
std::list<QString> Models::Roster::getItemPath(const QModelIndex& index) const {
std::list<QString> result;
if (index.isValid() && index.model() == this) {
Item* item = static_cast<Item*>(index.internalPointer());
while (item->type != Item::root) {
result.push_front(item->getId());
item = item->parentItem();
}
}
return result;
}
QModelIndex Models::Roster::getIndexByPath(const std::list<QString>& path) const {
if (path.empty())
return QModelIndex();
QModelIndex current;
for (const QString& hop : path) {
int rows = rowCount(current);
bool found = false;
for (int i = 0; i < rows; ++i) {
QModelIndex el = index(i, 0, current);
Item* item = static_cast<Item*>(el.internalPointer());
if (item->getId() == hop) {
found = true;
current = el;
break;
}
}
if (!found)
break;
}
return current; //this way I will return the last matching model index, may be it's logically incorrect
// std::list<QString>::const_iterator pathItr = path.begin();
// QString accName = *pathItr;
// QModelIndex accIndex = getAccountIndex(accName);
// if (path.size() == 1)
// return accIndex;
//
// if (!accIndex.isValid())
// return QModelIndex();
//
// ++pathItr;
// ElId id{accName, *pathItr};
// QModelIndex contactIndex = getContactIndex(id.account, id.name);
// if (!contactIndex.isValid())
// contactIndex = getGroupIndex(id.account, id.name);
//
// if (path.size() == 2)
// return contactIndex;
//
// if (!contactIndex.isValid())
// return QModelIndex();
//
// ++pathItr;
}