#include "Swift/Controllers/Roster/AppearOffline.h"

#include "Swift/Controllers/Roster/SetAvatar.h"

#include "Swift/Controllers/Roster/SetName.h"

#include "Swift/Controllers/Roster/OfflineRosterFilter.h"

#include "Swift/Controllers/Roster/GroupRosterItem.h"

#include "Swiften/Roster/XMPPRoster.h"

#include "Swiften/Roster/XMPPRosterItem.h"

#include "Swift/Controllers/UIEvents/AddContactUIEvent.h"

#include "Swift/Controllers/UIEvents/RemoveRosterItemUIEvent.h"

#include "Swift/Controllers/UIEvents/RenameRosterItemUIEvent.h"

#include "Swift/Controllers/UIEvents/RenameGroupUIEvent.h"

#include "Swift/Controllers/UIEvents/SendFileUIEvent.h"

#include <Swiften/FileTransfer/FileTransferManager.h>

#include <Swiften/Client/NickManager.h>

#include <Swift/Controllers/Intl.h>

#include <Swiften/Base/format.h>

#include <Swiften/Elements/DiscoInfo.h>

#include <Swiften/Disco/EntityCapsManager.h>

#include <Swiften/Jingle/JingleSessionManager.h>

#include <Swift/Controllers/SettingConstants.h>

namespace Swift {

/**

* The controller does not gain ownership of these parameters.

*/

RosterController::RosterController(const JID& jid, XMPPRoster* xmppRoster, AvatarManager* avatarManager, MainWindowFactory* mainWindowFactory, NickManager* nickManager, NickResolver* nickResolver, PresenceOracle* presenceOracle, SubscriptionManager* subscriptionManager, EventController* eventController, UIEventStream* uiEventStream, IQRouter* iqRouter, SettingsProvider* settings, EntityCapsProvider* entityCapsManager, FileTransferOverview* fileTransferOverview)

: myJID_(jid), xmppRoster_(xmppRoster), mainWindowFactory_(mainWindowFactory), mainWindow_(mainWindowFactory_->createMainWindow(uiEventStream)), roster_(new Roster()), offlineFilter_(new OfflineRosterFilter()), nickManager_(nickManager), nickResolver_(nickResolver), uiEventStream_(uiEventStream), entityCapsManager_(entityCapsManager), ftOverview_(fileTransferOverview) {

assert(fileTransferOverview);

iqRouter_ = iqRouter;

presenceOracle_ = presenceOracle;

subscriptionManager_ = subscriptionManager;

eventController_ = eventController;

settings_ = settings;

expandiness_ = new RosterGroupExpandinessPersister(roster_, settings);

mainWindow_->setRosterModel(roster_);

changeStatusConnection_ = mainWindow_->onChangeStatusRequest.connect(boost::bind(&RosterController::handleChangeStatusRequest, this, _1, _2));

signOutConnection_ = mainWindow_->onSignOutRequest.connect(boost::bind(boost::ref(onSignOutRequest)));

xmppRoster_->onJIDAdded.connect(boost::bind(&RosterController::handleOnJIDAdded, this, _1));

xmppRoster_->onJIDUpdated.connect(boost::bind(&RosterController::handleOnJIDUpdated, this, _1, _2, _3));

xmppRoster_->onJIDRemoved.connect(boost::bind(&RosterController::handleOnJIDRemoved, this, _1));

xmppRoster_->onRosterCleared.connect(boost::bind(&RosterController::handleRosterCleared, this));

subscriptionManager_->onPresenceSubscriptionRequest.connect(boost::bind(&RosterController::handleSubscriptionRequest, this, _1, _2));

presenceOracle_->onPresenceChange.connect(boost::bind(&RosterController::handleIncomingPresence, this, _1));

uiEventConnection_ = uiEventStream->onUIEvent.connect(boost::bind(&RosterController::handleUIEvent, this, _1));

avatarManager_ = avatarManager;

avatarManager_->onAvatarChanged.connect(boost::bind(&RosterController::handleAvatarChanged, this, _1));

mainWindow_->setMyAvatarPath(avatarManager_->getAvatarPath(myJID_).string());

nickManager_->onOwnNickChanged.connect(boost::bind(&MainWindow::setMyNick, mainWindow_, _1));

mainWindow_->setMyJID(jid);

mainWindow_->setMyNick(nickManager_->getOwnNick());

entityCapsManager_->onCapsChanged.connect(boost::bind(&RosterController::handleOnCapsChanged, this, _1));

settings_->onSettingChanged.connect(boost::bind(&RosterController::handleSettingChanged, this, _1));

}

RosterController::~RosterController() {

settings_->onSettingChanged.disconnect(boost::bind(&RosterController::handleSettingChanged, this, _1));

nickManager_->onOwnNickChanged.disconnect(boost::bind(&MainWindow::setMyNick, mainWindow_, _1));

delete offlineFilter_;

delete expandiness_;

mainWindow_->setRosterModel(NULL);

if (mainWindow_->canDelete()) {

delete mainWindow_;

}

delete roster_;

}

void RosterController::setEnabled(bool enabled) {

if (!enabled) {

roster_->applyOnItems(AppearOffline());

}

}

void RosterController::handleShowOfflineToggled(bool state) {

if (state) {

roster_->removeFilter(offlineFilter_);

} else {

roster_->addFilter(offlineFilter_);

}

}

void RosterController::handleChangeStatusRequest(StatusShow::Type show, const std::string &statusText) {

onChangeStatusRequest(show, statusText);

}

void RosterController::handleOnJIDAdded(const JID& jid) {

std::vector<std::string> groups = xmppRoster_->getGroupsForJID(jid);

std::string name = nickResolver_->jidToNick(jid);

if (!groups.empty()) {

foreach(const std::string& group, groups) {

roster_->addContact(jid, jid, name, group, avatarManager_->getAvatarPath(jid).string());

}

}

else {

roster_->addContact(jid, jid, name, QT_TRANSLATE_NOOP("", "Contacts"), avatarManager_->getAvatarPath(jid).string());

}

applyAllPresenceTo(jid);

}

void RosterController::applyAllPresenceTo(const JID& jid) {

foreach (Presence::ref presence, presenceOracle_->getAllPresence(jid)) {

roster_->applyOnItems(SetPresence(presence));

}

}

void RosterController::handleRosterCleared() {

roster_->removeAll();

}