diff options
Diffstat (limited to 'Swift/Controllers/HistoryViewController.cpp')
-rw-r--r-- | Swift/Controllers/HistoryViewController.cpp | 64 |
1 files changed, 20 insertions, 44 deletions
diff --git a/Swift/Controllers/HistoryViewController.cpp b/Swift/Controllers/HistoryViewController.cpp index baa7ebf..5f200f7 100644 --- a/Swift/Controllers/HistoryViewController.cpp +++ b/Swift/Controllers/HistoryViewController.cpp @@ -15,6 +15,7 @@ #include <Swiften/Avatars/AvatarManager.h> namespace Swift { + static const std::string category[] = { "Contacts", "MUC", "Contacts" }; HistoryViewController::HistoryViewController( const JID& selfJID, @@ -91,20 +92,14 @@ void HistoryViewController::handleSelectedContactChanged(RosterItem* newContact) JID contactJID = contact->getJID(); std::vector<HistoryMessage> messages; - if (roomPrivateContacts_.count(contactJID)) { - currentResultDate_ = *roomPrivateContacts_[contactJID].rbegin(); - messages = historyController_->getMessagesFromDate(selfJID_, contactJID, HistoryMessage::PrivateMessage, currentResultDate_); - currentItemType_ = HistoryMessage::PrivateMessage; - } - else if (contacts_.count(contactJID)) { - currentResultDate_ = *contacts_[contactJID].rbegin(); - messages = historyController_->getMessagesFromDate(selfJID_, contactJID, HistoryMessage::Chat, currentResultDate_); - currentItemType_ = HistoryMessage::Chat; - } - else { - currentResultDate_ = *rooms_[contactJID].rbegin(); - messages = historyController_->getMessagesFromDate(selfJID_, contactJID, HistoryMessage::Groupchat, currentResultDate_); - currentItemType_ = HistoryMessage::Groupchat; + for (int it = HistoryMessage::Chat; it <= HistoryMessage::PrivateMessage; it++) { + HistoryMessage::Type type = static_cast<HistoryMessage::Type>(it); + + if (contacts_[type].count(contactJID)) { + currentResultDate_ = *contacts_[type][contactJID].rbegin(); + currentItemType_ = type; + messages = historyController_->getMessagesFromDate(selfJID_, contactJID, type, currentResultDate_); + } } historyWindow_->setDate(currentResultDate_); @@ -129,20 +124,13 @@ void HistoryViewController::handleNewMessage(const HistoryMessage& message) { if (selectedItem_ && selectedItem_->getJID() == displayJID) { addNewMessage(message, false); } + // add new contact - else if (message.getType() == HistoryMessage::Groupchat && !rooms_.count(displayJID)) { - roster_->addContact(displayJID, displayJID, nickResolver_->jidToNick(displayJID), "MUC", avatarManager_->getAvatarPath(displayJID).string()); + else if (contacts_[message.getType()].count(displayJID)) { + roster_->addContact(displayJID, displayJID, nickResolver_->jidToNick(displayJID), category[message.getType()], avatarManager_->getAvatarPath(displayJID).string()); // // TODO add message date here - rooms_[displayJID] = std::set<boost::gregorian::date>(); - } - else if (message.getType() == HistoryMessage::Chat && !contacts_.count(displayJID)) { - roster_->addContact(displayJID, displayJID, nickResolver_->jidToNick(displayJID), "Contacts", avatarManager_->getAvatarPath(displayJID).string()); - contacts_[displayJID] = std::set<boost::gregorian::date>(); - } - else if (message.getType() == HistoryMessage::PrivateMessage && !roomPrivateContacts_.count(displayJID)) { - roster_->addContact(displayJID, displayJID, nickResolver_->jidToNick(displayJID), "Contacts", avatarManager_->getAvatarPath(displayJID).string()); - roomPrivateContacts_[displayJID] = std::set<boost::gregorian::date>(); + contacts_[message.getType()][displayJID] = std::set<boost::gregorian::date>(); } } @@ -155,25 +143,15 @@ void HistoryViewController::addNewMessage(const HistoryMessage& message, bool ad void HistoryViewController::handleReturnPressed(const std::string& keyword) { reset(); - // MUCs - rooms_ = historyController_->getContacts(selfJID_, HistoryMessage::Groupchat, keyword); - for (ContactsMap::const_iterator room = rooms_.begin(); room != rooms_.end(); room++) { - const JID& jid = room->first; - roster_->addContact(jid, jid, nickResolver_->jidToNick(jid), "MUC", avatarManager_->getAvatarPath(jid).string()); - } + for (int it = HistoryMessage::Chat; it <= HistoryMessage::PrivateMessage; it++) { + HistoryMessage::Type type = static_cast<HistoryMessage::Type>(it); - // Contacts - contacts_ = historyController_->getContacts(selfJID_, HistoryMessage::Chat, keyword); - for (ContactsMap::const_iterator contact = contacts_.begin(); contact != contacts_.end(); contact++) { - const JID& jid = contact->first; - roster_->addContact(jid, jid, nickResolver_->jidToNick(jid), "Contacts", avatarManager_->getAvatarPath(jid).string()); - } + contacts_[type] = historyController_->getContacts(selfJID_, type, keyword); - // MUC contacts - roomPrivateContacts_ = historyController_->getContacts(selfJID_, HistoryMessage::PrivateMessage, keyword); - for (ContactsMap::const_iterator contact = roomPrivateContacts_.begin(); contact != roomPrivateContacts_.end(); contact++) { - const JID& jid = contact->first; - roster_->addContact(jid, jid, nickResolver_->jidToNick(jid), "Contacts", avatarManager_->getAvatarPath(jid).string()); + for (ContactsMap::const_iterator contact = contacts_[type].begin(); contact != contacts_[type].end(); contact++) { + const JID& jid = contact->first; + roster_->addContact(jid, jid, nickResolver_->jidToNick(jid), category[type], avatarManager_->getAvatarPath(jid).string()); + } } } @@ -203,9 +181,7 @@ void HistoryViewController::handleNextButtonClicked() { void HistoryViewController::reset() { roster_->removeAll(); - rooms_.clear(); contacts_.clear(); - roomPrivateContacts_.clear(); selectedItem_ = NULL; historyWindow_->resetConversationView(); } |