1
0
forked from blue/squawk

Merge pull request 'rosterReferences' (#39) from rosterReferences into master

This commit is contained in:
Blue 2020-04-19 13:18:24 +00:00
commit a625ecb47b
18 changed files with 616 additions and 270 deletions

View File

@ -219,6 +219,8 @@ void Core::Account::onClientConnected()
void Core::Account::onClientDisconnected() void Core::Account::onClientDisconnected()
{ {
cancelHistoryRequests();
pendingVCardRequests.clear();
clearConferences(); clearConferences();
if (state != Shared::ConnectionState::disconnected) { if (state != Shared::ConnectionState::disconnected) {
if (reconnectTimes > 0) { if (reconnectTimes > 0) {
@ -854,18 +856,20 @@ void Core::Account::onMamMessageReceived(const QString& queryId, const QXmppMess
{ {
if (msg.id().size() > 0 && (msg.body().size() > 0 || msg.outOfBandUrl().size() > 0)) { if (msg.id().size() > 0 && (msg.body().size() > 0 || msg.outOfBandUrl().size() > 0)) {
std::map<QString, QString>::const_iterator itr = achiveQueries.find(queryId); std::map<QString, QString>::const_iterator itr = achiveQueries.find(queryId);
QString jid = itr->second; if (itr != achiveQueries.end()) {
RosterItem* item = getRosterItem(jid); QString jid = itr->second;
RosterItem* item = getRosterItem(jid);
Shared::Message sMsg(static_cast<Shared::Message::Type>(msg.type())); Shared::Message sMsg(static_cast<Shared::Message::Type>(msg.type()));
initializeMessage(sMsg, msg, false, true, true); initializeMessage(sMsg, msg, false, true, true);
sMsg.setState(Shared::Message::State::sent); sMsg.setState(Shared::Message::State::sent);
QString oId = msg.replaceId(); QString oId = msg.replaceId();
if (oId.size() > 0) { if (oId.size() > 0) {
item->correctMessageInArchive(oId, sMsg); item->correctMessageInArchive(oId, sMsg);
} else { } else {
item->addMessageToArchive(sMsg); item->addMessageToArchive(sMsg);
}
} }
} }
@ -896,10 +900,15 @@ void Core::Account::requestArchive(const QString& jid, int count, const QString&
if (contact == 0) { if (contact == 0) {
qDebug() << "An attempt to request archive for" << jid << "in account" << name << ", but the contact with such id wasn't found, skipping"; qDebug() << "An attempt to request archive for" << jid << "in account" << name << ", but the contact with such id wasn't found, skipping";
emit responseArchive(contact->jid, std::list<Shared::Message>()); emit responseArchive(jid, std::list<Shared::Message>());
return; return;
} }
if (state != Shared::ConnectionState::connected) {
qDebug() << "An attempt to request archive for" << jid << "in account" << name << ", but the account is not online, skipping";
emit responseArchive(contact->jid, std::list<Shared::Message>());
}
if (contact->getArchiveState() == RosterItem::empty && before.size() == 0) { if (contact->getArchiveState() == RosterItem::empty && before.size() == 0) {
QXmppMessage msg(getFullJid(), jid, "", ""); QXmppMessage msg(getFullJid(), jid, "", "");
QString last = Shared::generateUUID(); QString last = Shared::generateUUID();
@ -952,14 +961,16 @@ void Core::Account::onContactNeedHistory(const QString& before, const QString& a
void Core::Account::onMamResultsReceived(const QString& queryId, const QXmppResultSetReply& resultSetReply, bool complete) void Core::Account::onMamResultsReceived(const QString& queryId, const QXmppResultSetReply& resultSetReply, bool complete)
{ {
std::map<QString, QString>::const_iterator itr = achiveQueries.find(queryId); std::map<QString, QString>::const_iterator itr = achiveQueries.find(queryId);
QString jid = itr->second; if (itr != achiveQueries.end()) {
achiveQueries.erase(itr); QString jid = itr->second;
achiveQueries.erase(itr);
RosterItem* ri = getRosterItem(jid); RosterItem* ri = getRosterItem(jid);
if (ri != 0) { if (ri != 0) {
qDebug() << "Flushing messages for" << jid; qDebug() << "Flushing messages for" << jid;
ri->flushMessagesToArchive(complete, resultSetReply.first(), resultSetReply.last()); ri->flushMessagesToArchive(complete, resultSetReply.first(), resultSetReply.last());
}
} }
} }
@ -1743,3 +1754,14 @@ void Core::Account::onReceiptReceived(const QString& jid, const QString& id)
} }
} }
void Core::Account::cancelHistoryRequests()
{
for (const std::pair<QString, Conference*>& pair : conferences) {
pair.second->clearArchiveRequests();
}
for (const std::pair<QString, Contact*>& pair : contacts) {
pair.second->clearArchiveRequests();
}
achiveQueries.clear();
}

View File

@ -236,6 +236,7 @@ private:
void logMessage(const QXmppMessage& msg, const QString& reason = "Message wasn't handled: "); void logMessage(const QXmppMessage& msg, const QString& reason = "Message wasn't handled: ");
void storeConferences(); void storeConferences();
void clearConferences(); void clearConferences();
void cancelHistoryRequests();
void sendMessageWithLocalUploadedFile(Shared::Message msg, const QString& url); void sendMessageWithLocalUploadedFile(Shared::Message msg, const QString& url);
RosterItem* getRosterItem(const QString& jid); RosterItem* getRosterItem(const QString& jid);
}; };

View File

@ -518,3 +518,13 @@ Shared::VCard Core::RosterItem::handleResponseVCard(const QXmppVCardIq& card, co
return vCard; return vCard;
} }
void Core::RosterItem::clearArchiveRequests()
{
syncronizing = false;
requestedCount = 0;
requestedBefore = "";
hisoryCache.clear();
responseCache.clear();
appendCache.clear();
requestCache.clear();
}

View File

@ -76,6 +76,7 @@ public:
virtual void handlePresence(const QXmppPresence& pres) = 0; virtual void handlePresence(const QXmppPresence& pres) = 0;
bool changeMessage(const QString& id, const QMap<QString, QVariant>& data); bool changeMessage(const QString& id, const QMap<QString, QVariant>& data);
void clearArchiveRequests();
signals: signals:
void nameChanged(const QString& name); void nameChanged(const QString& name);

View File

@ -24,6 +24,7 @@ set(squawkUI_SRC
models/room.cpp models/room.cpp
models/abstractparticipant.cpp models/abstractparticipant.cpp
models/participant.cpp models/participant.cpp
models/reference.cpp
utils/messageline.cpp utils/messageline.cpp
utils//message.cpp utils//message.cpp
utils/resizer.cpp utils/resizer.cpp

View File

@ -17,6 +17,8 @@
*/ */
#include "account.h" #include "account.h"
#include "contact.h"
#include "reference.h"
#include <QDebug> #include <QDebug>
Models::Account::Account(const QMap<QString, QVariant>& data, Models::Item* parentItem): Models::Account::Account(const QMap<QString, QVariant>& data, Models::Item* parentItem):
@ -267,4 +269,3 @@ void Models::Account::setPasswordType(unsigned int pt)
{ {
setPasswordType(Shared::Global::fromInt<Shared::AccountPassword>(pt)); setPasswordType(Shared::Global::fromInt<Shared::AccountPassword>(pt));
} }

View File

@ -28,6 +28,9 @@
#include <QIcon> #include <QIcon>
namespace Models { namespace Models {
class Contact;
class Reference;
class Account : public Item { class Account : public Item {
Q_OBJECT Q_OBJECT
public: public:

View File

@ -17,10 +17,11 @@
*/ */
#include "contact.h" #include "contact.h"
#include "account.h"
#include <QDebug> #include <QDebug>
Models::Contact::Contact(const QString& p_jid ,const QMap<QString, QVariant> &data, Item *parentItem): Models::Contact::Contact(const Account* acc, const QString& p_jid ,const QMap<QString, QVariant> &data, Item *parentItem):
Item(Item::contact, data, parentItem), Item(Item::contact, data, parentItem),
jid(p_jid), jid(p_jid),
availability(Shared::Availability::offline), availability(Shared::Availability::offline),
@ -30,7 +31,8 @@ Models::Contact::Contact(const QString& p_jid ,const QMap<QString, QVariant> &da
messages(), messages(),
childMessages(0), childMessages(0),
status(), status(),
avatarPath() avatarPath(),
account(acc)
{ {
QMap<QString, QVariant>::const_iterator itr = data.find("state"); QMap<QString, QVariant>::const_iterator itr = data.find("state");
if (itr != data.end()) { if (itr != data.end()) {
@ -224,9 +226,9 @@ void Models::Contact::_removeChild(int index)
refresh(); refresh();
} }
void Models::Contact::appendChild(Models::Item* child) void Models::Contact::_appendChild(Models::Item* child)
{ {
Item::appendChild(child); Item::_appendChild(child);
connect(child, &Item::childChanged, this, &Contact::refresh); connect(child, &Item::childChanged, this, &Contact::refresh);
refresh(); refresh();
} }
@ -332,17 +334,20 @@ void Models::Contact::getMessages(Models::Contact::Messages& container) const
void Models::Contact::toOfflineState() void Models::Contact::toOfflineState()
{ {
emit childIsAboutToBeRemoved(this, 0, childItems.size()); std::deque<Item*>::size_type size = childItems.size();
for (std::deque<Item*>::size_type i = 0; i < childItems.size(); ++i) { if (size > 0) {
Item* item = childItems[i]; emit childIsAboutToBeRemoved(this, 0, size - 1);
disconnect(item, &Item::childChanged, this, &Contact::refresh); for (std::deque<Item*>::size_type i = 0; i < size; ++i) {
Item::_removeChild(i); Item* item = childItems[0];
item->deleteLater(); disconnect(item, &Item::childChanged, this, &Contact::refresh);
Item::_removeChild(0);
item->deleteLater();
}
childItems.clear();
presences.clear();
emit childRemoved();
refresh();
} }
childItems.clear();
presences.clear();
emit childRemoved();
refresh();
} }
QString Models::Contact::getDisplayedName() const QString Models::Contact::getDisplayedName() const
@ -368,7 +373,8 @@ Models::Contact::Contact(const Models::Contact& other):
state(other.state), state(other.state),
presences(), presences(),
messages(other.messages), messages(other.messages),
childMessages(0) childMessages(0),
account(other.account)
{ {
for (const Presence* pres : other.presences) { for (const Presence* pres : other.presences) {
Presence* pCopy = new Presence(*pres); Presence* pCopy = new Presence(*pres);
@ -415,3 +421,9 @@ void Models::Contact::setAvatarState(unsigned int p_state)
qDebug() << "An attempt to set invalid avatar state" << p_state << "to the contact" << jid << ", skipping"; qDebug() << "An attempt to set invalid avatar state" << p_state << "to the contact" << jid << ", skipping";
} }
} }
const Models::Account * Models::Contact::getParentAccount() const
{
return account;
}

View File

@ -31,13 +31,14 @@
#include <deque> #include <deque>
namespace Models { namespace Models {
class Account;
class Contact : public Item class Contact : public Item
{ {
Q_OBJECT Q_OBJECT
public: public:
typedef std::deque<Shared::Message> Messages; typedef std::deque<Shared::Message> Messages;
Contact(const QString& p_jid, const QMap<QString, QVariant> &data, Item *parentItem = 0); Contact(const Account* acc, const QString& p_jid, const QMap<QString, QVariant> &data, Item *parentItem = 0);
Contact(const Contact& other); Contact(const Contact& other);
~Contact(); ~Contact();
@ -56,7 +57,6 @@ public:
void addPresence(const QString& name, const QMap<QString, QVariant>& data); void addPresence(const QString& name, const QMap<QString, QVariant>& data);
void removePresence(const QString& name); void removePresence(const QString& name);
void appendChild(Models::Item * child) override;
QString getContactName() const; QString getContactName() const;
QString getStatus() const; QString getStatus() const;
@ -71,7 +71,9 @@ public:
protected: protected:
void _removeChild(int index) override; void _removeChild(int index) override;
void _appendChild(Models::Item * child) override;
bool columnInvolvedInDisplay(int col) override; bool columnInvolvedInDisplay(int col) override;
const Account* getParentAccount() const override;
protected slots: protected slots:
void refresh(); void refresh();
@ -98,6 +100,7 @@ private:
unsigned int childMessages; unsigned int childMessages;
QString status; QString status;
QString avatarPath; QString avatarPath;
const Account* account;
}; };
} }

View File

@ -18,6 +18,7 @@
#include "group.h" #include "group.h"
#include "contact.h" #include "contact.h"
#include "reference.h"
Models::Group::Group(const QMap<QString, QVariant>& data, Models::Item* parentItem): Models::Group::Group(const QMap<QString, QVariant>& data, Models::Item* parentItem):
Item(group, data, parentItem), Item(group, data, parentItem),
@ -29,9 +30,9 @@ Models::Group::~Group()
{ {
} }
void Models::Group::appendChild(Models::Item* child) void Models::Group::_appendChild(Models::Item* child)
{ {
Item::appendChild(child); Item::_appendChild(child);
connect(child, &Item::childChanged, this, &Group::refresh); connect(child, &Item::childChanged, this, &Group::refresh);
changed(1); changed(1);
refresh(); refresh();
@ -82,9 +83,14 @@ void Models::Group::refresh()
{ {
unsigned int newAmount(0); unsigned int newAmount(0);
for (std::deque<Models::Item*>::const_iterator itr = childItems.begin(), end = childItems.end(); itr != end; ++itr) { for (Models::Item* item : childItems) {
Models::Contact* cnt = static_cast<Models::Contact*>(*itr); if (item->type == reference) {
newAmount += cnt->getMessagesCount(); item = static_cast<Reference*>(item)->dereference();
}
if (item->type == contact) {
Models::Contact* cnt = static_cast<Models::Contact*>(item);
newAmount += cnt->getMessagesCount();
}
} }
setUnreadMessages(newAmount); setUnreadMessages(newAmount);
@ -94,25 +100,17 @@ unsigned int Models::Group::getOnlineContacts() const
{ {
unsigned int amount(0); unsigned int amount(0);
for (std::deque<Models::Item*>::const_iterator itr = childItems.begin(), end = childItems.end(); itr != end; ++itr) { for (Models::Item* item : childItems) {
Models::Contact* cnt = static_cast<Models::Contact*>(*itr); if (item->type == reference) {
if (cnt->getAvailability() != Shared::Availability::offline) { item = static_cast<Reference*>(item)->dereference();
++amount; }
if (item->type == contact) {
Models::Contact* cnt = static_cast<Models::Contact*>(item);
if (cnt->getAvailability() != Shared::Availability::offline) {
++amount;
}
} }
} }
return amount; return amount;
} }
bool Models::Group::hasContact(const QString& jid) const
{
for (Models::Item* item : childItems) {
if (item->type == Item::contact) {
const Contact* cnt = static_cast<const Contact*>(item);
if (cnt->getJid() == jid) {
return true;
}
}
}
return false;
}

View File

@ -23,6 +23,8 @@
namespace Models { namespace Models {
class Contact;
class Group : public Models::Item class Group : public Models::Item
{ {
Q_OBJECT Q_OBJECT
@ -30,17 +32,15 @@ public:
Group(const QMap<QString, QVariant> &data, Item *parentItem = 0); Group(const QMap<QString, QVariant> &data, Item *parentItem = 0);
~Group(); ~Group();
void appendChild(Models::Item* child) override;
int columnCount() const override; int columnCount() const override;
QVariant data(int column) const override; QVariant data(int column) const override;
unsigned int getUnreadMessages() const; unsigned int getUnreadMessages() const;
unsigned int getOnlineContacts() const; unsigned int getOnlineContacts() const;
bool hasContact(const QString& jid) const;
protected: protected:
void _removeChild(int index) override; void _removeChild(int index) override;
void _appendChild(Models::Item* child) override;
void setUnreadMessages(unsigned int amount); void setUnreadMessages(unsigned int amount);
private slots: private slots:

View File

@ -18,6 +18,8 @@
#include "item.h" #include "item.h"
#include "account.h" #include "account.h"
#include "reference.h"
#include "contact.h"
#include <QDebug> #include <QDebug>
@ -26,7 +28,10 @@ Models::Item::Item(Type p_type, const QMap<QString, QVariant> &p_data, Item *p_p
type(p_type), type(p_type),
name(""), name(""),
childItems(), childItems(),
parent(p_parent) parent(p_parent),
references(),
destroyingByParent(false),
destroyingByOriginal(false)
{ {
QMap<QString, QVariant>::const_iterator itr = p_data.find("name"); QMap<QString, QVariant>::const_iterator itr = p_data.find("name");
if (itr != p_data.end()) { if (itr != p_data.end()) {
@ -45,11 +50,25 @@ Models::Item::Item(const Models::Item& other):
Models::Item::~Item() Models::Item::~Item()
{ {
std::deque<Item*>::const_iterator itr = childItems.begin(); if (!destroyingByParent) {
std::deque<Item*>::const_iterator end = childItems.end(); Item* parent = parentItem();
if (parent != nullptr) {
if (parent->type == reference) {
parent->Item::removeChild(row());
} else {
parent->removeChild(row());
}
}
}
for (;itr != end; ++itr) { for (Reference* ref : references) {
delete (*itr); ref->destroyingByOriginal = true;
delete ref;
}
for (Item* child : childItems) {
child->destroyingByParent = true;
delete child;
} }
} }
@ -62,22 +81,35 @@ void Models::Item::setName(const QString& p_name)
} }
void Models::Item::appendChild(Models::Item* child) void Models::Item::appendChild(Models::Item* child)
{
_appendChild(child);
}
void Models::Item::_appendChild(Models::Item* child)
{ {
bool moving = false; bool moving = false;
int newRow = 0; int newRow = 0;
std::deque<Item*>::const_iterator before = childItems.begin(); std::deque<Item*>::const_iterator before = childItems.begin();
Type ct = child->type;
if (ct == reference) {
ct = static_cast<Reference*>(child)->dereference()->type;
}
while (before != childItems.end()) { while (before != childItems.end()) {
Item* bfr = *before; Item* bfr = *before;
if (bfr->type > child->type) { Type bt = bfr->type;
if (bt == reference) {
bt = static_cast<Reference*>(bfr)->dereference()->type;
}
if (bt > ct) {
break; break;
} else if (bfr->type == child->type && bfr->getDisplayedName() > child->getDisplayedName()) { } else if (bt == ct && bfr->getDisplayedName() > child->getDisplayedName()) {
break; break;
} }
newRow++; newRow++;
before++; before++;
} }
if (child->parent != 0) { if (child->parent != nullptr) {
int oldRow = child->row(); int oldRow = child->row();
moving = true; moving = true;
emit childIsAboutToBeMoved(child->parent, oldRow, oldRow, this, newRow); emit childIsAboutToBeMoved(child->parent, oldRow, oldRow, this, newRow);
@ -115,7 +147,7 @@ int Models::Item::childCount() const
int Models::Item::row() const int Models::Item::row() const
{ {
if (parent != 0) { if (parent != nullptr) {
std::deque<Item*>::const_iterator itr = parent->childItems.begin(); std::deque<Item*>::const_iterator itr = parent->childItems.begin();
std::deque<Item*>::const_iterator end = parent->childItems.end(); std::deque<Item*>::const_iterator end = parent->childItems.end();
@ -126,7 +158,7 @@ int Models::Item::row() const
} }
} }
return 0; //TODO not sure how it helps, i copy-pasted it from the example return -1; //TODO not sure how it helps
} }
Models::Item * Models::Item::parentItem() Models::Item * Models::Item::parentItem()
@ -177,13 +209,13 @@ void Models::Item::_removeChild(int index)
QObject::disconnect(child, &Item::childMoved, this, &Item::childMoved); QObject::disconnect(child, &Item::childMoved, this, &Item::childMoved);
childItems.erase(childItems.begin() + index); childItems.erase(childItems.begin() + index);
child->parent = 0; child->parent = nullptr;
} }
void Models::Item::changed(int col) void Models::Item::changed(int col)
{ {
if (parent != 0) { if (parent != nullptr) {
emit childChanged(this, row(), col); emit childChanged(this, row(), col);
} }
} }
@ -196,21 +228,21 @@ void Models::Item::toOfflineState()
} }
} }
const Models::Item * Models::Item::getParentAccount() const const Models::Account * Models::Item::getParentAccount() const
{ {
const Item* p = this; const Item* p = this;
while (p != 0 && p->type != Item::account) { while (p != nullptr && p->type != Item::account) {
p = p->parentItemConst(); p = p->parentItemConst();
} }
return p; return static_cast<const Account*>(p);
} }
QString Models::Item::getAccountJid() const QString Models::Item::getAccountJid() const
{ {
const Account* acc = static_cast<const Account*>(getParentAccount()); const Account* acc = getParentAccount();
if (acc == 0) { if (acc == nullptr) {
return ""; return "";
} }
return acc->getLogin() + "@" + acc->getServer(); return acc->getLogin() + "@" + acc->getServer();
@ -218,8 +250,8 @@ QString Models::Item::getAccountJid() const
QString Models::Item::getAccountResource() const QString Models::Item::getAccountResource() const
{ {
const Account* acc = static_cast<const Account*>(getParentAccount()); const Account* acc = getParentAccount();
if (acc == 0) { if (acc == nullptr) {
return ""; return "";
} }
return acc->getResource(); return acc->getResource();
@ -227,8 +259,8 @@ QString Models::Item::getAccountResource() const
QString Models::Item::getAccountName() const QString Models::Item::getAccountName() const
{ {
const Account* acc = static_cast<const Account*>(getParentAccount()); const Account* acc = getParentAccount();
if (acc == 0) { if (acc == nullptr) {
return ""; return "";
} }
return acc->getName(); return acc->getName();
@ -236,8 +268,8 @@ QString Models::Item::getAccountName() const
Shared::Availability Models::Item::getAccountAvailability() const Shared::Availability Models::Item::getAccountAvailability() const
{ {
const Account* acc = static_cast<const Account*>(getParentAccount()); const Account* acc = getParentAccount();
if (acc == 0) { if (acc == nullptr) {
return Shared::Availability::offline; return Shared::Availability::offline;
} }
return acc->getAvailability(); return acc->getAvailability();
@ -245,8 +277,8 @@ Shared::Availability Models::Item::getAccountAvailability() const
Shared::ConnectionState Models::Item::getAccountConnectionState() const Shared::ConnectionState Models::Item::getAccountConnectionState() const
{ {
const Account* acc = static_cast<const Account*>(getParentAccount()); const Account* acc = getParentAccount();
if (acc == 0) { if (acc == nullptr) {
return Shared::ConnectionState::disconnected; return Shared::ConnectionState::disconnected;
} }
return acc->getState(); return acc->getState();
@ -260,15 +292,24 @@ QString Models::Item::getDisplayedName() const
void Models::Item::onChildChanged(Models::Item* item, int row, int col) void Models::Item::onChildChanged(Models::Item* item, int row, int col)
{ {
Item* parent = item->parentItem(); Item* parent = item->parentItem();
if (parent != 0 && parent == this) { if (parent != nullptr && parent == this) {
if (item->columnInvolvedInDisplay(col)) { if (item->columnInvolvedInDisplay(col)) {
int newRow = 0; int newRow = 0;
std::deque<Item*>::const_iterator before = childItems.begin(); std::deque<Item*>::const_iterator before = childItems.begin();
Type ct = item->type;
if (ct == reference) {
ct = static_cast<Reference*>(item)->dereference()->type;
}
while (before != childItems.end()) { while (before != childItems.end()) {
Item* bfr = *before; Item* bfr = *before;
if (bfr->type > item->type) { Type bt = bfr->type;
if (bt == reference) {
bt = static_cast<Reference*>(bfr)->dereference()->type;
}
if (bt > ct) {
break; break;
} else if (bfr->type == item->type && bfr->getDisplayedName() > item->getDisplayedName()) { } else if (bt == ct && bfr->getDisplayedName() > item->getDisplayedName()) {
break; break;
} }
newRow++; newRow++;
@ -292,3 +333,42 @@ bool Models::Item::columnInvolvedInDisplay(int col)
{ {
return col == 0; return col == 0;
} }
void Models::Item::addReference(Models::Reference* ref)
{
references.insert(ref);
}
void Models::Item::removeReference(Models::Reference* ref)
{
std::set<Reference*>::const_iterator itr = references.find(ref);
if (itr != references.end()) {
references.erase(itr);
}
}
int Models::Item::getContact(const QString& jid) const
{
int index = -1;
for (std::deque<Item*>::size_type i = 0; i < childItems.size(); ++i) {
const Models::Item* item = childItems[i];
const Contact* cnt = nullptr;
if (item->type == Item::reference) {
item = static_cast<const Reference*>(item)->dereferenceConst();
}
if (item->type == Item::contact) {
cnt = static_cast<const Contact*>(item);
if (cnt->getJid() == jid) {
index = i;
break;
}
}
}
return index;
}
std::set<Models::Reference *>::size_type Models::Item::referencesCount() const
{
return references.size();
}

View File

@ -24,12 +24,17 @@
#include <QVariant> #include <QVariant>
#include <deque> #include <deque>
#include <set>
#include "shared/enums.h" #include "shared/enums.h"
namespace Models { namespace Models {
class Reference;
class Contact;
class Account;
class Item : public QObject{ class Item : public QObject{
friend class Reference;
Q_OBJECT Q_OBJECT
public: public:
enum Type { enum Type {
@ -39,7 +44,8 @@ class Item : public QObject{
room, room,
presence, presence,
participant, participant,
root root,
reference
}; };
explicit Item(Type p_type, const QMap<QString, QVariant> &data, Item *parentItem = 0); explicit Item(Type p_type, const QMap<QString, QVariant> &data, Item *parentItem = 0);
@ -62,8 +68,8 @@ class Item : public QObject{
QString getName() const; QString getName() const;
void setName(const QString& name); void setName(const QString& name);
Item *child(int row); virtual Item *child(int row);
int childCount() const; virtual int childCount() const;
virtual int columnCount() const; virtual int columnCount() const;
virtual QVariant data(int column) const; virtual QVariant data(int column) const;
int row() const; int row() const;
@ -79,23 +85,29 @@ class Item : public QObject{
const Type type; const Type type;
int getContact(const QString& jid) const;
std::set<Reference*>::size_type referencesCount() const;
protected: protected:
virtual void changed(int col); virtual void changed(int col);
virtual void _removeChild(int index); virtual void _removeChild(int index);
virtual void _appendChild(Item *child);
virtual bool columnInvolvedInDisplay(int col); virtual bool columnInvolvedInDisplay(int col);
const Item* getParentAccount() const; virtual const Account* getParentAccount() const;
void addReference(Reference* ref);
void removeReference(Reference* ref);
protected slots: protected slots:
void onChildChanged(Models::Item* item, int row, int col); void onChildChanged(Models::Item* item, int row, int col);
virtual void toOfflineState();
protected: protected:
QString name; QString name;
std::deque<Item*> childItems; std::deque<Item*> childItems;
Item* parent; Item* parent;
std::set<Reference*> references;
protected slots: bool destroyingByParent;
virtual void toOfflineState(); bool destroyingByOriginal;
}; };
} }

177
ui/models/reference.cpp Normal file
View File

@ -0,0 +1,177 @@
/*
* 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 "reference.h"
#include <QDebug>
using namespace Models;
Models::Reference::Reference(Models::Item* original, Models::Item* parent):
Models::Item(reference, {}, parent),
original(original),
ax(-1),
bx(-1),
cx(-1),
c(false)
{
connect(original, &Item::childChanged, this, &Reference::onChildChanged, Qt::QueuedConnection);
connect(original, &Item::childIsAboutToBeInserted, this, &Reference::onChildIsAboutToBeInserted);
connect(original, &Item::childInserted, this, &Reference::onChildInserted);
connect(original, &Item::childIsAboutToBeRemoved, this, &Reference::onChildIsAboutToBeRemoved);
connect(original, &Item::childRemoved, this, &Reference::onChildRemoved);
connect(original, &Item::childIsAboutToBeMoved, this, &Reference::onChildIsAboutToBeMoved);
connect(original, &Item::childMoved, this, &Reference::onChildMoved);
original->addReference(this);
for (int i = 0; i < original->childCount(); i++) {
Reference* ref = new Reference(original->child(i));
Item::appendChild(ref);
}
}
Models::Reference::~Reference()
{
if (!destroyingByOriginal) {
original->removeReference(this);
}
disconnect(original, &Item::childChanged, this, &Reference::onChildChanged);
disconnect(original, &Item::childIsAboutToBeInserted, this, &Reference::onChildIsAboutToBeInserted);
disconnect(original, &Item::childInserted, this, &Reference::onChildInserted);
disconnect(original, &Item::childIsAboutToBeRemoved, this, &Reference::onChildIsAboutToBeRemoved);
disconnect(original, &Item::childRemoved, this, &Reference::onChildRemoved);
disconnect(original, &Item::childIsAboutToBeMoved, this, &Reference::onChildIsAboutToBeMoved);
disconnect(original, &Item::childMoved, this, &Reference::onChildMoved);
}
int Models::Reference::columnCount() const
{
return original->columnCount();
}
QVariant Models::Reference::data(int column) const
{
return original->data(column);
}
QString Models::Reference::getDisplayedName() const
{
return original->getDisplayedName();
}
Models::Item * Models::Reference::dereference()
{
return original;
}
const Models::Item * Models::Reference::dereferenceConst() const
{
return original;
}
void Models::Reference::appendChild(Models::Item* child)
{
original->appendChild(child);
}
void Models::Reference::removeChild(int index)
{
original->removeChild(index);
}
void Models::Reference::toOfflineState()
{
original->toOfflineState();
}
void Models::Reference::onChildChanged(Models::Item* item, int row, int col)
{
if (item == original) {
emit childChanged(this, row, col);
}
}
void Models::Reference::onChildIsAboutToBeInserted(Models::Item* parent, int first, int last)
{
if (parent == original) {
ax = first;
bx = last;
}
}
void Models::Reference::onChildInserted()
{
if (ax != -1 && bx != -1) {
for (int i = ax; i <= bx; ++i) {
Reference* ref = new Reference(original->child(i));
Item::appendChild(ref);
}
ax = -1;
bx = -1;
}
}
void Models::Reference::onChildIsAboutToBeRemoved(Models::Item* parent, int first, int last)
{
if (parent == original) {
emit childIsAboutToBeRemoved(this, first, last);
for (int i = first; i <= last; ++i) {
Reference* ref = static_cast<Reference*>(childItems[first]);
Item::_removeChild(first);
delete ref;
}
childRemoved();
}
}
void Models::Reference::onChildRemoved()
{
}
void Models::Reference::onChildIsAboutToBeMoved(Models::Item* source, int first, int last, Models::Item* destination, int newIndex)
{
if (destination == original) {
if (source != original) {
c = true;
ax = first;
bx = last;
cx = newIndex;
emit childIsAboutToBeMoved(source, first, last, destination, newIndex);
} else {
ax = newIndex;
bx = newIndex + last - first + 1;
}
}
}
void Models::Reference::onChildMoved()
{
if (c) {
std::deque<Item*>::const_iterator beg = childItems.begin() + ax;
std::deque<Item*>::const_iterator end = childItems.begin() + bx + 1;
std::deque<Item*>::const_iterator tgt = childItems.begin() + cx;
std::deque<Item*> temp;
temp.insert(temp.end(), beg, end);
childItems.erase(beg, end);
childItems.insert(tgt, temp.begin(), temp.end());
emit childMoved();
} else {
onChildInserted();
}
}

66
ui/models/reference.h Normal file
View File

@ -0,0 +1,66 @@
/*
* 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/>.
*/
#ifndef MODELS_REFERENCE_H
#define MODELS_REFERENCE_H
#include "item.h"
namespace Models {
/**
* @todo write docs
*/
class Reference : public Models::Item
{
Q_OBJECT
public:
Reference(Models::Item* original, Models::Item* parent = 0);
~Reference();
int columnCount() const override;
QVariant data(int column) const override;
QString getDisplayedName() const override;
void appendChild(Models::Item * child) override;
void removeChild(int index) override;
Item* dereference();
const Item* dereferenceConst() const;
protected slots:
void toOfflineState() override;
private slots:
void onChildChanged(Models::Item* item, int row, int col);
void onChildIsAboutToBeInserted(Item* parent, int first, int last);
void onChildInserted();
void onChildIsAboutToBeRemoved(Item* parent, int first, int last);
void onChildRemoved();
void onChildIsAboutToBeMoved(Item* source, int first, int last, Item* destination, int newIndex);
void onChildMoved();
private:
Models::Item* original;
int ax;
int bx;
int cx;
bool c;
};
}
#endif // MODELS_REFERENCE_H

View File

@ -62,6 +62,9 @@ QVariant Models::Roster::data (const QModelIndex& index, int role) const
QVariant result; QVariant result;
Item *item = static_cast<Item*>(index.internalPointer()); Item *item = static_cast<Item*>(index.internalPointer());
if (item->type == Item::reference) {
item = static_cast<Reference*>(item)->dereference();
}
switch (role) { switch (role) {
case Qt::DisplayRole: case Qt::DisplayRole:
{ {
@ -430,7 +433,8 @@ void Models::Roster::addContact(const QString& account, const QString& jid, cons
{ {
Item* parent; Item* parent;
Account* acc; Account* acc;
Contact* sample = 0; Contact* contact;
Reference* ref = 0;
ElId id(account, jid); ElId id(account, jid);
{ {
@ -442,13 +446,18 @@ void Models::Roster::addContact(const QString& account, const QString& jid, cons
acc = itr->second; acc = itr->second;
} }
for (std::multimap<ElId, Contact*>::iterator itr = contacts.lower_bound(id), eItr = contacts.upper_bound(id); itr != eItr; ++itr) { {
sample = itr->second; //need to find if this contact is already added somewhere std::map<ElId, Contact*>::iterator itr = contacts.find(id);
break; //so one iteration is enough if (itr == contacts.end()) {
contact = new Contact(acc, jid, data);
contacts.insert(std::make_pair(id, contact));
} else {
contact = itr->second;
}
} }
if (group == "") { //this means this contact is already added somewhere and there is no sense to add it ungrouped if (group == "") {
if (sample != 0) { 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"; qDebug() << "An attempt to add a contact" << jid << "to the ungrouped contact set of account" << account << "for the second time, skipping";
return; return;
} else { } else {
@ -464,39 +473,26 @@ void Models::Roster::addContact(const QString& account, const QString& jid, cons
parent = itr->second; parent = itr->second;
for (int i = 0; i < parent->childCount(); ++i) { //checking if the contact is already added to that group if (parent->getContact(jid) != -1) {
Item* item = parent->child(i); qDebug() << "An attempt to add a contact" << jid << "to the group" << group << "for the second time, skipping";
if (item->type == Item::contact) { return;
Contact* ca = static_cast<Contact*>(item);
if (ca->getJid() == jid) {
qDebug() << "An attempt to add a contact" << jid << "to the group" << group << "for the second time, skipping";
return;
}
}
} }
for (int i = 0; i < acc->childCount(); ++i) { //checking if that contact is among ugrouped int refIndex = acc->getContact(jid);
Item* item = acc->child(i); if (refIndex != -1) { //checking if that contact is among ugrouped
if (item->type == Item::contact) { qDebug() << "An attempt to add a already existing contact " << jid
Contact* ca = static_cast<Contact*>(item); << " to the group " << group
if (ca->getJid() == jid) { << ", contact will be moved from ungrouped contacts of " << account;
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);
parent->appendChild(ca);
return;
}
}
} }
} }
Contact* contact;
if (sample == 0) { if (ref == 0) {
contact = new Contact(jid, data); ref = new Reference(contact);
} else {
contact = sample->copy();
} }
contacts.insert(std::make_pair(id, contact)); parent->appendChild(ref);
parent->appendChild(contact);
} }
void Models::Roster::removeGroup(const QString& account, const QString& name) void Models::Roster::removeGroup(const QString& account, const QString& name)
@ -513,33 +509,23 @@ void Models::Roster::removeGroup(const QString& account, const QString& name)
parent->removeChild(row); parent->removeChild(row);
std::deque<Contact*> toInsert; std::deque<Reference*> toInsert;
for (int i = 0; item->childCount() > 0; ++i) { for (int i = 0; item->childCount() > 0; ++i) {
Contact* cont = static_cast<Contact*>(item->child(0)); Reference* ref = static_cast<Reference*>(item->child(0));
item->removeChild(0); item->removeChild(0);
std::multimap<ElId, Contact*>::iterator cBeg = contacts.lower_bound({account, cont->getJid()}); Contact* cont = static_cast<Contact*>(ref->dereference());
std::multimap<ElId, Contact*>::iterator cEnd = contacts.upper_bound({account, cont->getJid()}); if (cont->referencesCount() == 1) {
toInsert.push_back(ref);
int count = std::distance(cBeg, cEnd);
if (count > 1) {
for (; cBeg != cEnd; ++count, ++cBeg) {
if (cBeg->second == cont) {
delete cont;
contacts.erase(cBeg);
break;
}
}
} else { } else {
toInsert.push_back(cont); delete ref;
} }
} }
if (toInsert.size() > 0) { if (toInsert.size() > 0) {
Account* acc = accounts.find("account")->second; Account* acc = accounts.find("account")->second;
for (std::deque<Contact*>::size_type i = 0; i < toInsert.size(); ++i) { for (std::deque<Contact*>::size_type i = 0; i < toInsert.size(); ++i) {
Contact* cont = toInsert[i]; acc->appendChild(toInsert[i]); //TODO optimisation
acc->appendChild(cont); //TODO optimisation
} }
} }
@ -550,19 +536,18 @@ void Models::Roster::removeGroup(const QString& account, const QString& name)
void Models::Roster::changeContact(const QString& account, const QString& jid, const QMap<QString, QVariant>& data) void Models::Roster::changeContact(const QString& account, const QString& jid, const QMap<QString, QVariant>& data)
{ {
ElId id(account, jid); ElId id(account, jid);
std::multimap<ElId, Contact*>::iterator cBeg = contacts.lower_bound(id); std::map<ElId, Contact*>::iterator cItr = contacts.find(id);
std::multimap<ElId, Contact*>::iterator cEnd = contacts.upper_bound(id);
for (; cBeg != cEnd; ++cBeg) { if (cItr != contacts.end()) {
for (QMap<QString, QVariant>::const_iterator itr = data.begin(), end = data.end(); itr != end; ++itr) { for (QMap<QString, QVariant>::const_iterator itr = data.begin(), end = data.end(); itr != end; ++itr) {
cBeg->second->update(itr.key(), itr.value()); cItr->second->update(itr.key(), itr.value());
} }
} } else {
std::map<ElId, Room*>::iterator rItr = rooms.find(id);
std::map<ElId, Room*>::iterator rItr = rooms.find(id); if (rItr != rooms.end()) {
if (rItr != rooms.end()) { for (QMap<QString, QVariant>::const_iterator itr = data.begin(), end = data.end(); itr != end; ++itr) {
for (QMap<QString, QVariant>::const_iterator itr = data.begin(), end = data.end(); itr != end; ++itr) { rItr->second->update(itr.key(), itr.value());
rItr->second->update(itr.key(), itr.value()); }
} }
} }
} }
@ -570,18 +555,13 @@ void Models::Roster::changeContact(const QString& account, const QString& jid, c
void Models::Roster::changeMessage(const QString& account, const QString& jid, const QString& id, const QMap<QString, QVariant>& data) void Models::Roster::changeMessage(const QString& account, const QString& jid, const QString& id, const QMap<QString, QVariant>& data)
{ {
ElId elid(account, jid); ElId elid(account, jid);
std::multimap<ElId, Contact*>::iterator cBeg = contacts.lower_bound(elid); std::map<ElId, Contact*>::iterator cItr = contacts.find(elid);
std::multimap<ElId, Contact*>::iterator cEnd = contacts.upper_bound(elid);
for (; cBeg != cEnd; ++cBeg) { if (cItr != contacts.end()) {
for (QMap<QString, QVariant>::const_iterator itr = data.begin(), end = data.end(); itr != end; ++itr) { cItr->second->changeMessage(id, data);
cBeg->second->changeMessage(id, data); } else {
} std::map<ElId, Room*>::iterator rItr = rooms.find(elid);
} if (rItr != rooms.end()) {
std::map<ElId, Room*>::iterator rItr = rooms.find(elid);
if (rItr != rooms.end()) {
for (QMap<QString, QVariant>::const_iterator itr = data.begin(), end = data.end(); itr != end; ++itr) {
rItr->second->changeMessage(id, data); rItr->second->changeMessage(id, data);
} }
} }
@ -590,25 +570,26 @@ void Models::Roster::changeMessage(const QString& account, const QString& jid, c
void Models::Roster::removeContact(const QString& account, const QString& jid) void Models::Roster::removeContact(const QString& account, const QString& jid)
{ {
ElId id(account, jid); ElId id(account, jid);
std::multimap<ElId, Contact*>::iterator cBeg = contacts.lower_bound(id); std::map<ElId, Contact*>::iterator itr = contacts.find(id);
std::multimap<ElId, Contact*>::iterator cEnd = contacts.upper_bound(id);
std::multimap<ElId, Contact*>::iterator cpBeg = cBeg;
QSet<QString> toRemove; if (itr != contacts.end()) {
for (; cBeg != cEnd; ++cBeg) { Contact* contact = itr->second;
Contact* contact = cBeg->second;
Item* parent = contact->parentItem(); contacts.erase(itr);
if (parent->type == Item::group && parent->childCount() == 1) { delete contact;
toRemove.insert(parent->getName());
std::set<ElId> toRemove;
for (std::pair<ElId, Group*> pair : groups) {
if (pair.second->childCount() == 0) {
toRemove.insert(pair.first);
}
} }
parent->removeChild(contact->row()); for (const ElId& elId : toRemove) {
contact->deleteLater(); removeGroup(elId.account, elId.name);
} }
contacts.erase(cpBeg, cEnd); } else {
qDebug() << "An attempt to remove contact " << jid << " from account " << account <<" which doesn't exist there, skipping";
for (QSet<QString>::const_iterator itr = toRemove.begin(), end = toRemove.end(); itr != end; ++itr) {
removeGroup(account, *itr);
} }
} }
@ -617,6 +598,12 @@ void Models::Roster::removeContact(const QString& account, const QString& jid, c
ElId contactId(account, jid); ElId contactId(account, jid);
ElId groupId(account, group); 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); std::map<ElId, Group*>::iterator gItr = groups.find(groupId);
if (gItr == groups.end()) { if (gItr == groups.end()) {
qDebug() << "An attempt to remove contact " << jid << " from non existing group " << group << " of account " << account <<", skipping"; qDebug() << "An attempt to remove contact " << jid << " from non existing group " << group << " of account " << account <<", skipping";
@ -624,57 +611,25 @@ void Models::Roster::removeContact(const QString& account, const QString& jid, c
} }
Account* acc = accounts.find(account)->second; //I assume the account is found, otherwise there will be no groups with that ElId; 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; Group* gr = gItr->second;
Contact* cont = 0; Contact* cont = cItr->second;
unsigned int entries(0); int contRow = gr->getContact(jid);
unsigned int ungroupped(0); if (contRow == -1) {
for (std::multimap<ElId, Contact*>::iterator cBeg = contacts.lower_bound(contactId), cEnd = contacts.upper_bound(contactId); cBeg != cEnd; ++cBeg) { qDebug() << "An attempt to remove contact " << jid << " of account " << account << " from group " << group <<", but there is no such contact in that group, skipping";
++entries; return;
Contact* elem = cBeg->second; }
if (elem->parentItem() == acc) { Reference* ref = static_cast<Reference*>(gr->child(contRow));
++ungroupped; 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 (ungroupped == 0 && entries == 1) { if (gr->childCount() == 0) {
for (std::multimap<ElId, Contact*>::iterator cBeg = contacts.lower_bound(contactId), cEnd = contacts.upper_bound(contactId); cBeg != cEnd; ++cBeg) { removeGroup(account, group);
if (cBeg->second->parentItem() == gr) {
cont = cBeg->second;
break;
}
}
if (cont == 0) {
qDebug() << "An attempt to remove contact " << jid << " of account " << account << " from group " << group <<", but there is no such contact in that group, skipping";
return;
}
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(cont);
if (gr->childCount() == 0) {
removeGroup(account, group);
}
} else {
for (std::multimap<ElId, Contact*>::iterator cBeg = contacts.lower_bound(contactId), cEnd = contacts.upper_bound(contactId); cBeg != cEnd; ++cBeg) {
if (cBeg->second->parentItem() == gr) {
cont = cBeg->second;
contacts.erase(cBeg);
break;
}
}
if (cont == 0) {
qDebug() << "An attempt to remove contact" << jid << "of account" << account << "from group" << group <<", but there is no such contact in that group, skipping";
return;
}
gr->removeChild(cont->row());
cont->deleteLater();
if (gr->childCount() == 0) {
removeGroup(account, group);
}
} }
} }
@ -736,50 +691,46 @@ void Models::Roster::onChildRemoved()
void Models::Roster::addPresence(const QString& account, const QString& jid, const QString& name, const QMap<QString, QVariant>& data) void Models::Roster::addPresence(const QString& account, const QString& jid, const QString& name, const QMap<QString, QVariant>& data)
{ {
ElId contactId(account, jid); ElId contactId(account, jid);
std::multimap<ElId, Contact*>::iterator cBeg = contacts.lower_bound(contactId); std::map<ElId, Contact*>::iterator itr = contacts.find(contactId);
std::multimap<ElId, Contact*>::iterator cEnd = contacts.upper_bound(contactId); if (itr != contacts.end()) {
for (;cBeg != cEnd; ++cBeg) { itr->second->addPresence(name, data);
cBeg->second->addPresence(name, data);
} }
} }
void Models::Roster::removePresence(const QString& account, const QString& jid, const QString& name) void Models::Roster::removePresence(const QString& account, const QString& jid, const QString& name)
{ {
ElId contactId(account, jid); ElId contactId(account, jid);
std::multimap<ElId, Contact*>::iterator cBeg = contacts.lower_bound(contactId); std::map<ElId, Contact*>::iterator itr = contacts.find(contactId);
std::multimap<ElId, Contact*>::iterator cEnd = contacts.upper_bound(contactId); if (itr != contacts.end()) {
for (;cBeg != cEnd; ++cBeg) { itr->second->removePresence(name);
cBeg->second->removePresence(name);
} }
} }
void Models::Roster::addMessage(const QString& account, const Shared::Message& data) void Models::Roster::addMessage(const QString& account, const Shared::Message& data)
{ {
ElId id(account, data.getPenPalJid()); ElId id(account, data.getPenPalJid());
std::map<ElId, Contact*>::iterator itr = contacts.find(id);
std::multimap<ElId, Contact*>::iterator cBeg = contacts.lower_bound(id); if (itr != contacts.end()) {
std::multimap<ElId, Contact*>::iterator cEnd = contacts.upper_bound(id); itr->second->addMessage(data);
} else {
for (;cBeg != cEnd; ++cBeg) { std::map<ElId, Room*>::const_iterator rItr = rooms.find(id);
cBeg->second->addMessage(data); if (rItr != rooms.end()) {
} rItr->second->addMessage(data);
}
std::map<ElId, Room*>::const_iterator rItr = rooms.find(id);
if (rItr != rooms.end()) {
rItr->second->addMessage(data);
} }
} }
void Models::Roster::dropMessages(const QString& account, const QString& jid) void Models::Roster::dropMessages(const QString& account, const QString& jid)
{ {
ElId id(account, jid); ElId id(account, jid);
for (std::multimap<ElId, Contact*>::iterator cBeg = contacts.lower_bound(id), cEnd = contacts.upper_bound(id) ;cBeg != cEnd; ++cBeg) { std::map<ElId, Contact*>::iterator itr = contacts.find(id);
cBeg->second->dropMessages(); if (itr != contacts.end()) {
} itr->second->dropMessages();
} else {
std::map<ElId, Room*>::const_iterator rItr = rooms.find(id); std::map<ElId, Room*>::const_iterator rItr = rooms.find(id);
if (rItr != rooms.end()) { if (rItr != rooms.end()) {
rItr->second->dropMessages(); rItr->second->dropMessages();
}
} }
} }
@ -797,10 +748,10 @@ void Models::Roster::removeAccount(const QString& account)
accountsModel->removeAccount(index); accountsModel->removeAccount(index);
accounts.erase(itr); accounts.erase(itr);
std::multimap<ElId, Contact*>::const_iterator cItr = contacts.begin(); std::map<ElId, Contact*>::const_iterator cItr = contacts.begin();
while (cItr != contacts.end()) { while (cItr != contacts.end()) {
if (cItr->first.account == account) { if (cItr->first.account == account) {
std::multimap<ElId, Contact*>::const_iterator lItr = cItr; std::map<ElId, Contact*>::const_iterator lItr = cItr;
++cItr; ++cItr;
contacts.erase(lItr); contacts.erase(lItr);
} else { } else {
@ -836,7 +787,7 @@ void Models::Roster::removeAccount(const QString& account)
QString Models::Roster::getContactName(const QString& account, const QString& jid) QString Models::Roster::getContactName(const QString& account, const QString& jid)
{ {
ElId id(account, jid); ElId id(account, jid);
std::multimap<ElId, Contact*>::const_iterator cItr = contacts.find(id); std::map<ElId, Contact*>::const_iterator cItr = contacts.find(id);
QString name = ""; QString name = "";
if (cItr == contacts.end()) { if (cItr == contacts.end()) {
std::map<ElId, Room*>::const_iterator rItr = rooms.find(id); std::map<ElId, Room*>::const_iterator rItr = rooms.find(id);
@ -969,15 +920,14 @@ bool Models::Roster::groupHasContact(const QString& account, const QString& grou
if (gItr == groups.end()) { if (gItr == groups.end()) {
return false; return false;
} else { } else {
const Group* gr = gItr->second; return gItr->second->getContact(contact) != -1;
return gr->hasContact(contact);
} }
} }
QString Models::Roster::getContactIconPath(const QString& account, const QString& jid, const QString& resource) QString Models::Roster::getContactIconPath(const QString& account, const QString& jid, const QString& resource)
{ {
ElId id(account, jid); ElId id(account, jid);
std::multimap<ElId, Contact*>::const_iterator cItr = contacts.find(id); std::map<ElId, Contact*>::const_iterator cItr = contacts.find(id);
QString path = ""; QString path = "";
if (cItr == contacts.end()) { if (cItr == contacts.end()) {
std::map<ElId, Room*>::const_iterator rItr = rooms.find(id); std::map<ElId, Room*>::const_iterator rItr = rooms.find(id);

View File

@ -32,6 +32,7 @@
#include "contact.h" #include "contact.h"
#include "group.h" #include "group.h"
#include "room.h" #include "room.h"
#include "reference.h"
namespace Models namespace Models
{ {
@ -87,7 +88,7 @@ private:
Item* root; Item* root;
std::map<QString, Account*> accounts; std::map<QString, Account*> accounts;
std::map<ElId, Group*> groups; std::map<ElId, Group*> groups;
std::multimap<ElId, Contact*> contacts; std::map<ElId, Contact*> contacts;
std::map<ElId, Room*> rooms; std::map<ElId, Room*> rooms;
private slots: private slots:

View File

@ -305,6 +305,9 @@ void Squawk::onRosterItemDoubleClicked(const QModelIndex& item)
{ {
if (item.isValid()) { if (item.isValid()) {
Models::Item* node = static_cast<Models::Item*>(item.internalPointer()); Models::Item* node = static_cast<Models::Item*>(item.internalPointer());
if (node->type == Models::Item::reference) {
node = static_cast<Models::Reference*>(node)->dereference();
}
Models::Contact* contact = 0; Models::Contact* contact = 0;
Models::Room* room = 0; Models::Room* room = 0;
QString res; QString res;
@ -681,7 +684,9 @@ void Squawk::onRosterContextMenu(const QPoint& point)
QModelIndex index = m_ui->roster->indexAt(point); QModelIndex index = m_ui->roster->indexAt(point);
if (index.isValid()) { if (index.isValid()) {
Models::Item* item = static_cast<Models::Item*>(index.internalPointer()); Models::Item* item = static_cast<Models::Item*>(index.internalPointer());
if (item->type == Models::Item::reference) {
item = static_cast<Models::Reference*>(item)->dereference();
}
contextMenu->clear(); contextMenu->clear();
bool hasMenu = false; bool hasMenu = false;
bool active = item->getAccountConnectionState() == Shared::ConnectionState::connected; bool active = item->getAccountConnectionState() == Shared::ConnectionState::connected;
@ -1105,6 +1110,9 @@ void Squawk::onRosterSelectionChanged(const QModelIndex& current, const QModelIn
if (current.isValid()) { if (current.isValid()) {
Models::Item* node = static_cast<Models::Item*>(current.internalPointer()); Models::Item* node = static_cast<Models::Item*>(current.internalPointer());
if (node->type == Models::Item::reference) {
node = static_cast<Models::Reference*>(node)->dereference();
}
Models::Contact* contact = 0; Models::Contact* contact = 0;
Models::Room* room = 0; Models::Room* room = 0;
QString res; QString res;