/*

* Copyright (c) 2013 Tobias Markmann

* Licensed under the simplified BSD license.

* See Documentation/Licenses/BSD-simplified.txt for more information.

*/

#include <Swift/Controllers/BlockListController.h>

#include <boost/bind.hpp>

#include <Swiften/Client/ClientBlockListManager.h>

#include <Swiften/Base/foreach.h>

#include <Swiften/Base/format.h>

#include <Swift/Controllers/Intl.h>

#include <Swift/Controllers/UIEvents/RequestChangeBlockStateUIEvent.h>

#include <Swift/Controllers/UIEvents/RequestBlockListDialogUIEvent.h>

#include <Swift/Controllers/XMPPEvents/ErrorEvent.h>

#include <Swift/Controllers/UIInterfaces/BlockListEditorWidget.h>

#include <Swift/Controllers/XMPPEvents/EventController.h>

namespace Swift {

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

blockListManager_->getBlockList()->onItemAdded.connect(boost::bind(&BlockListController::handleBlockListChanged, this));

blockListManager_->getBlockList()->onItemRemoved.connect(boost::bind(&BlockListController::handleBlockListChanged, this));

}

BlockListController::~BlockListController() {

blockListManager_->getBlockList()->onItemAdded.disconnect(boost::bind(&BlockListController::handleBlockListChanged, this));

blockListManager_->getBlockList()->onItemRemoved.disconnect(boost::bind(&BlockListController::handleBlockListChanged, this));

}

void BlockListController::blockListDifferences(const std::vector<JID> &newBlockList, std::vector<JID> &jidsToUnblock, std::vector<JID> &jidsToBlock) const {

foreach (const JID& jid, blockListBeforeEdit) {

if (std::find(newBlockList.begin(), newBlockList.end(), jid) == newBlockList.end()) {

jidsToUnblock.push_back(jid);

}

}

foreach (const JID& jid, newBlockList) {

if (std::find(blockListBeforeEdit.begin(), blockListBeforeEdit.end(), jid) == blockListBeforeEdit.end()) {

jidsToBlock.push_back(jid);

}

}

}

void BlockListController::handleUIEvent(boost::shared_ptr<UIEvent> rawEvent) {

// handle UI dialog

boost::shared_ptr<RequestBlockListDialogUIEvent> requestDialogEvent = boost::dynamic_pointer_cast<RequestBlockListDialogUIEvent>(rawEvent);

if (requestDialogEvent != NULL) {

if (blockListEditorWidget_ == NULL) {

blockListEditorWidget_ = blockListEditorWidgetFactory_->createBlockListEditorWidget();

blockListEditorWidget_->onSetNewBlockList.connect(boost::bind(&BlockListController::handleSetNewBlockList, this, _1));

}

blockListBeforeEdit = blockListManager_->getBlockList()->getItems();

blockListEditorWidget_->setCurrentBlockList(blockListBeforeEdit);

blockListEditorWidget_->show();

return;

}

// handle block state change

boost::shared_ptr<RequestChangeBlockStateUIEvent> changeStateEvent = boost::dynamic_pointer_cast<RequestChangeBlockStateUIEvent>(rawEvent);

if (changeStateEvent != NULL) {

if (changeStateEvent->getBlockState() == RequestChangeBlockStateUIEvent::Blocked) {

GenericRequest<BlockPayload>::ref blockRequest = blockListManager_->createBlockJIDRequest(changeStateEvent->getContact());

blockRequest->onResponse.connect(boost::bind(&BlockListController::handleBlockResponse, this, blockRequest, _1, _2, std::vector<JID>(1, changeStateEvent->getContact()), false));

blockRequest->send();

} else if (changeStateEvent->getBlockState() == RequestChangeBlockStateUIEvent::Unblocked) {

GenericRequest<UnblockPayload>::ref unblockRequest = blockListManager_->createUnblockJIDRequest(changeStateEvent->getContact());

unblockRequest->onResponse.connect(boost::bind(&BlockListController::handleUnblockResponse, this, unblockRequest, _1, _2, std::vector<JID>(1, changeStateEvent->getContact()), false));

unblockRequest->send();

}

return;

}

}

void BlockListController::handleBlockResponse(GenericRequest<BlockPayload>::ref request, boost::shared_ptr<BlockPayload>, ErrorPayload::ref error, const std::vector<JID>& jids, bool originEditor) {

if (error) {

std::string errorMessage;

// FIXME: Handle reporting of list of JIDs in a translatable way.

errorMessage = str(format(QT_TRANSLATE_NOOP("", "Failed to block %1%.")) % jids.at(0).toString());

if (!error->getText().empty()) {

errorMessage = str(format(QT_TRANSLATE_NOOP("", "%1%: %2%.")) % errorMessage % error->getText());

}

eventController_->handleIncomingEvent(boost::make_shared<ErrorEvent>(request->getReceiver(), errorMessage));

}

if (originEditor) {

remainingRequests_--;

if (blockListEditorWidget_ && (remainingRequests_ == 0)) {

blockListEditorWidget_->setBusy(false);

}

}

}

void BlockListController::handleUnblockResponse(GenericRequest<UnblockPayload>::ref request, boost::shared_ptr<UnblockPayload>, ErrorPayload::ref error, const std::vector<JID>& jids, bool originEditor) {

if (error) {

std::string errorMessage;

// FIXME: Handle reporting of list of JIDs in a translatable way.

errorMessage = str(format(QT_TRANSLATE_NOOP("", "Failed to unblock %1%.")) % jids.at(0).toString());

if (!error->getText().empty()) {

errorMessage = str(format(QT_TRANSLATE_NOOP("", "%1%: %2%.")) % errorMessage % error->getText());

}

eventController_->handleIncomingEvent(boost::make_shared<ErrorEvent>(request->getReceiver(), errorMessage));

}

if (originEditor) {

remainingRequests_--;

if (blockListEditorWidget_ && (remainingRequests_ == 0)) {

blockListEditorWidget_->setBusy(false);

}

}

}

void BlockListController::handleSetNewBlockList(const std::vector<JID> &newBlockList) {

std::vector<JID> jidsToBlock;

std::vector<JID> jidsToUnblock;

blockListDifferences(newBlockList, jidsToUnblock, jidsToBlock);

if (!jidsToBlock.empty()) {

remainingRequests_++;

GenericRequest<BlockPayload>::ref blockRequest = blockListManager_->createBlockJIDsRequest(jidsToBlock);

blockRequest->onResponse.connect(boost::bind(&BlockListController::handleBlockResponse, this, blockRequest, _1, _2, jidsToBlock, true));

blockRequest->send();

}

if (!jidsToUnblock.empty()) {

remainingRequests_++;

GenericRequest<UnblockPayload>::ref unblockRequest = blockListManager_->createUnblockJIDsRequest(jidsToUnblock);

unblockRequest->onResponse.connect(boost::bind(&BlockListController::handleUnblockResponse, this, unblockRequest, _1, _2, jidsToUnblock, true));

unblockRequest->send();

}

if (!jidsToBlock.empty() || jidsToUnblock.empty()) {

assert(blockListEditorWidget_);

blockListEditorWidget_->setBusy(true);

}

}

void BlockListController::handleBlockListChanged() {

if (blockListEditorWidget_) {

std::vector<JID> jidsToBlock;

std::vector<JID> jidsToUnblock;

blockListDifferences(blockListEditorWidget_->getCurrentBlockList(), jidsToUnblock, jidsToBlock);

blockListBeforeEdit = blockListManager_->getBlockList()->getItems();

foreach (const JID& jid, jidsToBlock) {

if (std::find(blockListBeforeEdit.begin(), blockListBeforeEdit.end(), jid) == blockListBeforeEdit.end()) {

blockListBeforeEdit.push_back(jid);

}

}

foreach (const JID& jid, jidsToUnblock) {

blockListBeforeEdit.erase(std::remove(blockListBeforeEdit.begin(), blockListBeforeEdit.end(), jid), blockListBeforeEdit.end());

}

blockListEditorWidget_->setCurrentBlockList(blockListBeforeEdit);

}

}

}

/*

* Copyright (c) 2013 Tobias Markmann

* Licensed under the simplified BSD license.

* See Documentation/Licenses/BSD-simplified.txt for more information.

*/

#pragma once

#include <boost/shared_ptr.hpp>

#include <Swiften/Queries/GenericRequest.h>

#include <Swift/Controllers/UIEvents/UIEventStream.h>

#include <Swift/Controllers/UIInterfaces/BlockListEditorWidgetFactory.h>

namespace Swift {

class BlockPayload;

class UnblockPayload;

class ClientBlockListManager;

class EventController;

class BlockListController {

public:

BlockListController(ClientBlockListManager* blockListManager, UIEventStream* uiEventStream, BlockListEditorWidgetFactory* blockListEditorWidgetFactory, EventController* eventController);

~BlockListController();

private:

void blockListDifferences(const std::vector<JID> &newBlockList, std::vector<JID>& jidsToUnblock, std::vector<JID>& jidsToBlock) const;

void handleUIEvent(boost::shared_ptr<UIEvent> event);

void handleBlockResponse(GenericRequest<BlockPayload>::ref, boost::shared_ptr<BlockPayload>, ErrorPayload::ref error, const std::vector<JID>& jids, bool originEditor);

void handleUnblockResponse(GenericRequest<UnblockPayload>::ref, boost::shared_ptr<UnblockPayload>, ErrorPayload::ref error, const std::vector<JID>& jids, bool originEditor);

void handleSetNewBlockList(const std::vector<JID>& newBlockList);

void handleBlockListChanged();

private:

ClientBlockListManager* blockListManager_;

BlockListEditorWidgetFactory* blockListEditorWidgetFactory_;

BlockListEditorWidget* blockListEditorWidget_;

EventController* eventController_;

std::vector<JID> blockListBeforeEdit;

int remainingRequests_;

};

}

/*

* Copyright (c) 2010-2014 Kevin Smith and Remko Tronçon

* Licensed under the GNU General Public License v3.

* See Documentation/Licenses/GPLv3.txt for more information.

*/

#include <Swift/Controllers/MainController.h>

#include <cstdlib>

#include <boost/bind.hpp>

#include <boost/lexical_cast.hpp>

#include <boost/shared_ptr.hpp>

#include <boost/smart_ptr/make_shared.hpp>

#include <Swiften/Base/format.h>

#include <Swiften/Base/Algorithm.h>

#include <Swiften/Base/String.h>

#include <Swiften/StringCodecs/Base64.h>

#include <Swiften/Network/TimerFactory.h>

#include <Swiften/Client/Storages.h>

#include <Swiften/VCards/VCardManager.h>

#include <Swiften/Client/NickResolver.h>

#include <Swiften/Base/foreach.h>

#include <Swiften/Client/Client.h>

#include <Swiften/Presence/PresenceSender.h>

#include <Swiften/Elements/ChatState.h>

#include <Swiften/Elements/Presence.h>

#include <Swiften/Elements/VCardUpdate.h>

#include <Swiften/Elements/DiscoInfo.h>

#include <Swiften/Disco/CapsInfoGenerator.h>

#include <Swiften/Disco/GetDiscoInfoRequest.h>

#include <Swiften/Disco/ClientDiscoManager.h>

#include <Swiften/VCards/GetVCardRequest.h>

#include <Swiften/StringCodecs/Hexify.h>

#include <Swiften/Network/NetworkFactories.h>

#include <Swiften/FileTransfer/FileTransferManager.h>

#include <Swiften/Client/ClientXMLTracer.h>

#include <Swiften/Client/StanzaChannel.h>

#include <Swiften/Client/ClientBlockListManager.h>

#include <Swiften/Crypto/CryptoProvider.h>

#include <SwifTools/Dock/Dock.h>

#include <SwifTools/Notifier/TogglableNotifier.h>

#include <SwifTools/Idle/IdleDetector.h>

#include <Swift/Controllers/Intl.h>

#include <Swift/Controllers/UIInterfaces/UIFactory.h>

#include <Swift/Controllers/BuildVersion.h>

#include <Swift/Controllers/Chat/UserSearchController.h>

#include <Swift/Controllers/Chat/ChatsManager.h>

#include <Swift/Controllers/XMPPEvents/EventController.h>

#include <Swift/Controllers/EventWindowController.h>

#include <Swift/Controllers/UIInterfaces/LoginWindow.h>

#include <Swift/Controllers/UIInterfaces/LoginWindowFactory.h>

#include <Swift/Controllers/UIInterfaces/MainWindow.h>

#include <Swift/Controllers/Chat/MUCController.h>

#include <Swift/Controllers/Roster/RosterController.h>

#include <Swift/Controllers/SoundEventController.h>

#include <Swift/Controllers/SoundPlayer.h>

#include <Swift/Controllers/StatusTracker.h>

#include <Swift/Controllers/SystemTray.h>

#include <Swift/Controllers/SystemTrayController.h>

#include <Swift/Controllers/XMLConsoleController.h>

#include <Swift/Controllers/HistoryController.h>

#include <Swift/Controllers/HistoryViewController.h>

#include <Swift/Controllers/FileTransferListController.h>

#include <Swift/Controllers/UIEvents/UIEventStream.h>

#include <Swift/Controllers/PresenceNotifier.h>

#include <Swift/Controllers/EventNotifier.h>

#include <Swift/Controllers/Storages/StoragesFactory.h>

#include <Swift/Controllers/WhiteboardManager.h>

#include <Swift/Controllers/Settings/SettingsProvider.h>

#include <Swift/Controllers/UIEvents/RequestChatUIEvent.h>

#include <Swift/Controllers/UIEvents/JoinMUCUIEvent.h>

#include <Swift/Controllers/Storages/CertificateStorageFactory.h>

#include <Swift/Controllers/Storages/CertificateStorageTrustChecker.h>

#include <Swift/Controllers/ProfileController.h>

#include <Swift/Controllers/ShowProfileController.h>

#include <Swift/Controllers/ContactEditController.h>

#include <Swift/Controllers/XMPPURIController.h>

#include <Swift/Controllers/AdHocManager.h>

#include <Swift/Controllers/FileTransfer/FileTransferOverview.h>

#include <Swift/Controllers/SettingConstants.h>

#include <Swift/Controllers/HighlightManager.h>

#include <Swift/Controllers/HighlightEditorController.h>

#include <Swift/Controllers/BlockListController.h>

#include <Swift/Controllers/ContactSuggester.h>

#include <Swift/Controllers/ContactsFromXMPPRoster.h>

namespace Swift {

static const std::string CLIENT_NAME = "Swift";

static const std::string CLIENT_NODE = "http://swift.im";

MainController::MainController(

EventLoop* eventLoop,

NetworkFactories* networkFactories,

UIFactory* uiFactories,

SettingsProvider* settings,

SystemTray* systemTray,

SoundPlayer* soundPlayer,

StoragesFactory* storagesFactory,

CertificateStorageFactory* certificateStorageFactory,

Dock* dock,

Notifier* notifier,

URIHandler* uriHandler,

IdleDetector* idleDetector,

const std::map<std::string, std::string>& emoticons,

bool useDelayForLatency) :

eventLoop_(eventLoop),

networkFactories_(networkFactories),

uiFactory_(uiFactories),

storagesFactory_(storagesFactory),

certificateStorageFactory_(certificateStorageFactory),

settings_(settings),

uriHandler_(uriHandler),

idleDetector_(idleDetector),

loginWindow_(NULL) ,

useDelayForLatency_(useDelayForLatency),

ftOverview_(NULL),

emoticons_(emoticons) {

storages_ = NULL;

certificateStorage_ = NULL;

statusTracker_ = NULL;

presenceNotifier_ = NULL;

eventNotifier_ = NULL;

rosterController_ = NULL;

chatsManager_ = NULL;

historyController_ = NULL;

historyViewController_ = NULL;

eventWindowController_ = NULL;

profileController_ = NULL;

blockListController_ = NULL;

showProfileController_ = NULL;

contactEditController_ = NULL;

userSearchControllerChat_ = NULL;

userSearchControllerAdd_ = NULL;

userSearchControllerInvite_ = NULL;

contactsFromRosterProvider_ = NULL;

contactSuggesterWithoutRoster_ = NULL;

contactSuggesterWithRoster_ = NULL;

whiteboardManager_ = NULL;

adHocManager_ = NULL;

quitRequested_ = false;

clientInitialized_ = false;

offlineRequested_ = false;

timeBeforeNextReconnect_ = -1;

dock_ = dock;

uiEventStream_ = new UIEventStream();

notifier_ = new TogglableNotifier(notifier);

notifier_->setPersistentEnabled(settings_->getSetting(SettingConstants::SHOW_NOTIFICATIONS));

eventController_ = new EventController();

eventController_->onEventQueueLengthChange.connect(boost::bind(&MainController::handleEventQueueLengthChange, this, _1));

systemTrayController_ = new SystemTrayController(eventController_, systemTray);

loginWindow_ = uiFactory_->createLoginWindow(uiEventStream_);

loginWindow_->setShowNotificationToggle(!notifier->isExternallyConfigured());

highlightManager_ = new HighlightManager(settings_);

highlightEditorController_ = new HighlightEditorController(uiEventStream_, uiFactory_, highlightManager_);

soundEventController_ = new SoundEventController(eventController_, soundPlayer, settings, highlightManager_);

xmppURIController_ = new XMPPURIController(uriHandler_, uiEventStream_);

std::string selectedLoginJID = settings_->getSetting(SettingConstants::LAST_LOGIN_JID);

bool loginAutomatically = settings_->getSetting(SettingConstants::LOGIN_AUTOMATICALLY);

std::string cachedPassword;

std::string cachedCertificate;

ClientOptions cachedOptions;

bool eagle = settings_->getSetting(SettingConstants::FORGET_PASSWORDS);

if (!eagle) {

foreach (std::string profile, settings->getAvailableProfiles()) {

ProfileSettingsProvider profileSettings(profile, settings);

std::string password = profileSettings.getStringSetting("pass");

std::string certificate = profileSettings.getStringSetting("certificate");

std::string jid = profileSettings.getStringSetting("jid");

ClientOptions clientOptions = parseClientOptions(profileSettings.getStringSetting("options"));

loginWindow_->addAvailableAccount(jid, password, certificate, clientOptions);

if (jid == selectedLoginJID) {

cachedPassword = password;

cachedCertificate = certificate;

cachedOptions = clientOptions;

}

}

loginWindow_->selectUser(selectedLoginJID);

loginWindow_->setLoginAutomatically(loginAutomatically);

}

loginWindow_->onLoginRequest.connect(boost::bind(&MainController::handleLoginRequest, this, _1, _2, _3, _4, _5, _6, _7));

loginWindow_->onPurgeSavedLoginRequest.connect(boost::bind(&MainController::handlePurgeSavedLoginRequest, this, _1));

loginWindow_->onCancelLoginRequest.connect(boost::bind(&MainController::handleCancelLoginRequest, this));

loginWindow_->onQuitRequest.connect(boost::bind(&MainController::handleQuitRequest, this));

idleDetector_->setIdleTimeSeconds(settings->getSetting(SettingConstants::IDLE_TIMEOUT));

idleDetector_->onIdleChanged.connect(boost::bind(&MainController::handleInputIdleChanged, this, _1));

xmlConsoleController_ = new XMLConsoleController(uiEventStream_, uiFactory_);

fileTransferListController_ = new FileTransferListController(uiEventStream_, uiFactory_);

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

if (loginAutomatically) {

profileSettings_ = new ProfileSettingsProvider(selectedLoginJID, settings_);

/* FIXME: deal with autologin with a cert*/

handleLoginRequest(selectedLoginJID, cachedPassword, cachedCertificate, CertificateWithKey::ref(), cachedOptions, true, true);

} else {

profileSettings_ = NULL;

}

}

MainController::~MainController() {

idleDetector_->onIdleChanged.disconnect(boost::bind(&MainController::handleInputIdleChanged, this, _1));

purgeCachedCredentials();

//setManagersOffline();

eventController_->disconnectAll();

resetClient();

delete highlightEditorController_;

delete highlightManager_;

delete fileTransferListController_;

delete xmlConsoleController_;

delete xmppURIController_;

delete soundEventController_;

delete systemTrayController_;

delete eventController_;

delete notifier_;

delete uiEventStream_;

}

void MainController::purgeCachedCredentials() {

safeClear(password_);

}

void MainController::resetClient() {

purgeCachedCredentials();

resetCurrentError();

resetPendingReconnects();

vCardPhotoHash_.clear();

delete contactEditController_;

contactEditController_ = NULL;

delete profileController_;

profileController_ = NULL;

delete showProfileController_;

showProfileController_ = NULL;

delete eventWindowController_;

eventWindowController_ = NULL;

delete chatsManager_;

chatsManager_ = NULL;

#ifdef SWIFT_EXPERIMENTAL_HISTORY

delete historyViewController_;

historyViewController_ = NULL;

delete historyController_;

historyController_ = NULL;

#endif

delete ftOverview_;

ftOverview_ = NULL;

delete rosterController_;

rosterController_ = NULL;

delete eventNotifier_;

eventNotifier_ = NULL;

delete presenceNotifier_;

presenceNotifier_ = NULL;

delete certificateStorage_;

certificateStorage_ = NULL;

delete storages_;

storages_ = NULL;

delete statusTracker_;

statusTracker_ = NULL;

delete profileSettings_;

profileSettings_ = NULL;

delete userSearchControllerChat_;

userSearchControllerChat_ = NULL;

delete userSearchControllerAdd_;

userSearchControllerAdd_ = NULL;

delete userSearchControllerInvite_;

userSearchControllerInvite_ = NULL;

delete contactSuggesterWithoutRoster_;

contactSuggesterWithoutRoster_ = NULL;

delete contactSuggesterWithRoster_;

contactSuggesterWithRoster_ = NULL;

delete contactsFromRosterProvider_;

contactsFromRosterProvider_ = NULL;

delete adHocManager_;

adHocManager_ = NULL;

delete whiteboardManager_;

whiteboardManager_ = NULL;

clientInitialized_ = false;

}

void MainController::handleSettingChanged(const std::string& settingPath) {

if (settingPath == SettingConstants::SHOW_NOTIFICATIONS.getKey()) {

notifier_->setPersistentEnabled(settings_->getSetting(SettingConstants::SHOW_NOTIFICATIONS));

}

}

void MainController::resetPendingReconnects() {

timeBeforeNextReconnect_ = -1;

if (reconnectTimer_) {

reconnectTimer_->stop();

reconnectTimer_.reset();

}

resetCurrentError();

}

void MainController::resetCurrentError() {

if (lastDisconnectError_) {

lastDisconnectError_->conclude();

lastDisconnectError_ = boost::shared_ptr<ErrorEvent>();

}

}

void MainController::handleConnected() {

boundJID_ = client_->getJID();

resetCurrentError();

resetPendingReconnects();

if (settings_->getSetting(SettingConstants::FORGET_PASSWORDS)) {

purgeCachedCredentials();

}

bool freshLogin = rosterController_ == NULL;

myStatusLooksOnline_ = true;

if (freshLogin) {

profileController_ = new ProfileController(client_->getVCardManager(), uiFactory_, uiEventStream_);

showProfileController_ = new ShowProfileController(client_->getVCardManager(), uiFactory_, uiEventStream_);

ftOverview_ = new FileTransferOverview(client_->getFileTransferManager());

fileTransferListController_->setFileTransferOverview(ftOverview_);

rosterController_ = new RosterController(boundJID_, client_->getRoster(), client_->getAvatarManager(), uiFactory_, client_->getNickManager(), client_->getNickResolver(), client_->getPresenceOracle(), client_->getSubscriptionManager(), eventController_, uiEventStream_, client_->getIQRouter(), settings_, client_->getEntityCapsProvider(), ftOverview_, client_->getClientBlockListManager(), client_->getVCardManager());

rosterController_->onChangeStatusRequest.connect(boost::bind(&MainController::handleChangeStatusRequest, this, _1, _2));

rosterController_->onSignOutRequest.connect(boost::bind(&MainController::signOut, this));

rosterController_->getWindow()->onShowCertificateRequest.connect(boost::bind(&MainController::handleShowCertificateRequest, this));

blockListController_ = new BlockListController(client_->getClientBlockListManager(), uiEventStream_, uiFactory_, eventController_);

contactEditController_ = new ContactEditController(rosterController_, client_->getVCardManager(), uiFactory_, uiEventStream_);

whiteboardManager_ = new WhiteboardManager(uiFactory_, uiEventStream_, client_->getNickResolver(), client_->getWhiteboardSessionManager());

/* Doing this early as an ordering fix. Various things later will

* want to have the user's nick available and this means it will

* be before they receive stanzas that need it (e.g. bookmarks).*/

client_->getVCardManager()->requestOwnVCard();

contactSuggesterWithoutRoster_ = new ContactSuggester();

contactSuggesterWithRoster_ = new ContactSuggester();

userSearchControllerInvite_ = new UserSearchController(UserSearchController::InviteToChat, jid_, uiEventStream_, client_->getVCardManager(), uiFactory_, client_->getIQRouter(), rosterController_, contactSuggesterWithRoster_, client_->getAvatarManager(), client_->getPresenceOracle());

#ifdef SWIFT_EXPERIMENTAL_HISTORY

historyController_ = new HistoryController(storages_->getHistoryStorage());

historyViewController_ = new HistoryViewController(jid_, uiEventStream_, historyController_, client_->getNickResolver(), client_->getAvatarManager(), client_->getPresenceOracle(), uiFactory_);

chatsManager_ = new ChatsManager(jid_, client_->getStanzaChannel(), client_->getIQRouter(), eventController_, uiFactory_, uiFactory_, client_->getNickResolver(), client_->getPresenceOracle(), client_->getPresenceSender(), uiEventStream_, uiFactory_, useDelayForLatency_, networkFactories_->getTimerFactory(), client_->getMUCRegistry(), client_->getEntityCapsProvider(), client_->getMUCManager(), uiFactory_, profileSettings_, ftOverview_, client_->getRoster(), !settings_->getSetting(SettingConstants::REMEMBER_RECENT_CHATS), settings_, historyController_, whiteboardManager_, highlightManager_, client_->getClientBlockListManager(), emoticons_, userSearchControllerInvite_, client_->getVCardManager());

#else

chatsManager_ = new ChatsManager(jid_, client_->getStanzaChannel(), client_->getIQRouter(), eventController_, uiFactory_, uiFactory_, client_->getNickResolver(), client_->getPresenceOracle(), client_->getPresenceSender(), uiEventStream_, uiFactory_, useDelayForLatency_, networkFactories_->getTimerFactory(), client_->getMUCRegistry(), client_->getEntityCapsProvider(), client_->getMUCManager(), uiFactory_, profileSettings_, ftOverview_, client_->getRoster(), !settings_->getSetting(SettingConstants::REMEMBER_RECENT_CHATS), settings_, NULL, whiteboardManager_, highlightManager_, client_->getClientBlockListManager(), emoticons_, userSearchControllerInvite_, client_->getVCardManager());

#endif

contactsFromRosterProvider_ = new ContactsFromXMPPRoster(client_->getRoster(), client_->getAvatarManager(), client_->getPresenceOracle());

contactSuggesterWithoutRoster_->addContactProvider(chatsManager_);

contactSuggesterWithRoster_->addContactProvider(chatsManager_);

contactSuggesterWithRoster_->addContactProvider(contactsFromRosterProvider_);

highlightEditorController_->setContactSuggester(contactSuggesterWithoutRoster_);

client_->onMessageReceived.connect(boost::bind(&ChatsManager::handleIncomingMessage, chatsManager_, _1));

chatsManager_->setAvatarManager(client_->getAvatarManager());

eventWindowController_ = new EventWindowController(eventController_, uiFactory_);

loginWindow_->morphInto(rosterController_->getWindow());

DiscoInfo discoInfo;

discoInfo.addIdentity(DiscoInfo::Identity(CLIENT_NAME, "client", "pc"));

discoInfo.addFeature(DiscoInfo::ChatStatesFeature);

discoInfo.addFeature(DiscoInfo::SecurityLabelsFeature);

discoInfo.addFeature(DiscoInfo::MessageCorrectionFeature);

#ifdef SWIFT_EXPERIMENTAL_FT

discoInfo.addFeature(DiscoInfo::JingleFeature);

discoInfo.addFeature(DiscoInfo::JingleFTFeature);

discoInfo.addFeature(DiscoInfo::JingleTransportsIBBFeature);

discoInfo.addFeature(DiscoInfo::JingleTransportsS5BFeature);

#endif

#ifdef SWIFT_EXPERIMENTAL_WB

discoInfo.addFeature(DiscoInfo::WhiteboardFeature);

#endif

discoInfo.addFeature(DiscoInfo::MessageDeliveryReceiptsFeature);

client_->getDiscoManager()->setCapsNode(CLIENT_NODE);

client_->getDiscoManager()->setDiscoInfo(discoInfo);

userSearchControllerChat_ = new UserSearchController(UserSearchController::StartChat, jid_, uiEventStream_, client_->getVCardManager(), uiFactory_, client_->getIQRouter(), rosterController_, contactSuggesterWithRoster_, client_->getAvatarManager(), client_->getPresenceOracle());

userSearchControllerAdd_ = new UserSearchController(UserSearchController::AddContact, jid_, uiEventStream_, client_->getVCardManager(), uiFactory_, client_->getIQRouter(), rosterController_, contactSuggesterWithoutRoster_, client_->getAvatarManager(), client_->getPresenceOracle());

adHocManager_ = new AdHocManager(JID(boundJID_.getDomain()), uiFactory_, client_->getIQRouter(), uiEventStream_, rosterController_->getWindow());

chatsManager_->onImpromptuMUCServiceDiscovered.connect(boost::bind(&UserSearchController::setCanInitiateImpromptuMUC, userSearchControllerChat_, _1));

}

loginWindow_->setIsLoggingIn(false);

client_->requestRoster();

GetDiscoInfoRequest::ref discoInfoRequest = GetDiscoInfoRequest::create(JID(boundJID_.getDomain()), client_->getIQRouter());

discoInfoRequest->onResponse.connect(boost::bind(&MainController::handleServerDiscoInfoResponse, this, _1, _2));

discoInfoRequest->send();

client_->getVCardManager()->requestOwnVCard();

rosterController_->setJID(boundJID_);

rosterController_->setEnabled(true);

rosterController_->getWindow()->setStreamEncryptionStatus(client_->isStreamEncrypted());

profileController_->setAvailable(true);

contactEditController_->setAvailable(true);

/* Send presence later to catch all the incoming presences. */

sendPresence(statusTracker_->getNextPresence());

/* Enable chats last of all, so rejoining MUCs has the right sent presence */

assert(chatsManager_);

chatsManager_->setOnline(true);

adHocManager_->setOnline(true);

}

void MainController::handleEventQueueLengthChange(int count) {

dock_->setNumberOfPendingMessages(count);

}

void MainController::reconnectAfterError() {

if (reconnectTimer_) {

reconnectTimer_->stop();

}

performLoginFromCachedCredentials();

}

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

boost::shared_ptr<Presence> presence(new Presence());

if (show == StatusShow::None) {

// Note: this is misleading, None doesn't mean unavailable on the wire.

presence->setType(Presence::Unavailable);

resetPendingReconnects();

myStatusLooksOnline_ = false;

offlineRequested_ = true;

}

else {

offlineRequested_ = false;

presence->setShow(show);

}

presence->setStatus(statusText);

statusTracker_->setRequestedPresence(presence);

if (presence->getType() != Presence::Unavailable) {

profileSettings_->storeInt("lastShow", presence->getShow());

profileSettings_->storeString("lastStatus", presence->getStatus());

}

if (presence->getType() != Presence::Unavailable && !client_->isAvailable()) {

performLoginFromCachedCredentials();

} else {

sendPresence(presence);

}

}

void MainController::sendPresence(boost::shared_ptr<Presence> presence) {

rosterController_->getWindow()->setMyStatusType(presence->getShow());

rosterController_->getWindow()->setMyStatusText(presence->getStatus());

systemTrayController_->setMyStatusType(presence->getShow());

notifier_->setTemporarilyDisabled(presence->getShow() == StatusShow::DND);

// Add information and send

presence->updatePayload(boost::make_shared<VCardUpdate>(vCardPhotoHash_));

client_->getPresenceSender()->sendPresence(presence);

if (presence->getType() == Presence::Unavailable) {

logout();

}

}

void MainController::handleInputIdleChanged(bool idle) {

if (!statusTracker_) {

//Haven't logged in yet.

return;

}

if (settings_->getSetting(SettingConstants::IDLE_GOES_OFFLINE)) {

if (idle) {

logout();

}

}

else {

if (idle) {

if (statusTracker_->goAutoAway(idleDetector_->getIdleTimeSeconds())) {

if (client_ && client_->isAvailable()) {

sendPresence(statusTracker_->getNextPresence());

}

}

} else {

if (statusTracker_->goAutoUnAway()) {

if (client_ && client_->isAvailable()) {

sendPresence(statusTracker_->getNextPresence());

}

}

}

}

}

void MainController::handleShowCertificateRequest() {

std::vector<Certificate::ref> chain = client_->getStanzaChannel()->getPeerCertificateChain();

rosterController_->getWindow()->openCertificateDialog(chain);

}

void MainController::handleLoginRequest(const std::string &username, const std::string &password, const std::string& certificatePath, CertificateWithKey::ref certificate, const ClientOptions& options, bool remember, bool loginAutomatically) {

jid_ = JID(username);

if (!jid_.isValid() || jid_.getNode().empty()) {

loginWindow_->setMessage(QT_TRANSLATE_NOOP("", "User address invalid. User address should be of the form 'alice@wonderland.lit'"));

loginWindow_->setIsLoggingIn(false);

} else {

loginWindow_->setMessage("");

loginWindow_->setIsLoggingIn(true);

profileSettings_ = new ProfileSettingsProvider(username, settings_);

if (!settings_->getSetting(SettingConstants::FORGET_PASSWORDS)) {

profileSettings_->storeString("jid", username);

profileSettings_->storeString("certificate", certificatePath);

profileSettings_->storeString("pass", (remember || loginAutomatically) ? password : "");

std::string optionString = serializeClientOptions(options);

profileSettings_->storeString("options", optionString);

settings_->storeSetting(SettingConstants::LAST_LOGIN_JID, username);

settings_->storeSetting(SettingConstants::LOGIN_AUTOMATICALLY, loginAutomatically);

loginWindow_->addAvailableAccount(profileSettings_->getStringSetting("jid"), profileSettings_->getStringSetting("pass"), profileSettings_->getStringSetting("certificate"), options);

}

password_ = password;

certificate_ = certificate;

clientOptions_ = options;

performLoginFromCachedCredentials();

}

}

void MainController::handlePurgeSavedLoginRequest(const std::string& username) {

settings_->removeProfile(username);

loginWindow_->removeAvailableAccount(username);

}

void MainController::performLoginFromCachedCredentials() {

if (settings_->getSetting(SettingConstants::FORGET_PASSWORDS) && password_.empty()) {

/* Then we can't try to login again. */

return;

}

/* If we logged in with a bare JID, and we have a full bound JID, re-login with the

* bound JID to try and keep dynamically assigned resources */

JID clientJID = jid_;

if (boundJID_.isValid() && jid_.isBare() && boundJID_.toBare() == jid_) {

clientJID = boundJID_;

}

if (!statusTracker_) {

statusTracker_ = new StatusTracker();

}

if (!clientInitialized_) {

storages_ = storagesFactory_->createStorages(jid_.toBare());

certificateStorage_ = certificateStorageFactory_->createCertificateStorage(jid_.toBare());

certificateTrustChecker_ = new CertificateStorageTrustChecker(certificateStorage_);

client_ = boost::make_shared<Swift::Client>(clientJID, createSafeByteArray(password_.c_str()), networkFactories_, storages_);

clientInitialized_ = true;

client_->setCertificateTrustChecker(certificateTrustChecker_);

client_->onDataRead.connect(boost::bind(&XMLConsoleController::handleDataRead, xmlConsoleController_, _1));

client_->onDataWritten.connect(boost::bind(&XMLConsoleController::handleDataWritten, xmlConsoleController_, _1));

client_->onDisconnected.connect(boost::bind(&MainController::handleDisconnected, this, _1));

client_->onConnected.connect(boost::bind(&MainController::handleConnected, this));

client_->setSoftwareVersion(CLIENT_NAME, buildVersion);

client_->getVCardManager()->onVCardChanged.connect(boost::bind(&MainController::handleVCardReceived, this, _1, _2));

presenceNotifier_ = new PresenceNotifier(client_->getStanzaChannel(), notifier_, client_->getMUCRegistry(), client_->getAvatarManager(), client_->getNickResolver(), client_->getPresenceOracle(), networkFactories_->getTimerFactory());

presenceNotifier_->onNotificationActivated.connect(boost::bind(&MainController::handleNotificationClicked, this, _1));

eventNotifier_ = new EventNotifier(eventController_, notifier_, client_->getAvatarManager(), client_->getNickResolver());

eventNotifier_->onNotificationActivated.connect(boost::bind(&MainController::handleNotificationClicked, this, _1));

if (certificate_ && !certificate_->isNull()) {

client_->setCertificate(certificate_);

}

boost::shared_ptr<Presence> presence(new Presence());

presence->setShow(static_cast<StatusShow::Type>(profileSettings_->getIntSetting("lastShow", StatusShow::Online)));

presence->setStatus(profileSettings_->getStringSetting("lastStatus"));

statusTracker_->setRequestedPresence(presence);

} else {

/* In case we're in the middle of another login, make sure they don't overlap */

client_->disconnect();

}

systemTrayController_->setConnecting();

if (rosterController_) {

rosterController_->getWindow()->setConnecting();

}

ClientOptions clientOptions = clientOptions_;

bool eagle = settings_->getSetting(SettingConstants::FORGET_PASSWORDS);

clientOptions.forgetPassword = eagle;

clientOptions.useTLS = eagle ? ClientOptions::RequireTLS : clientOptions_.useTLS;

client_->connect(clientOptions);

}

void MainController::handleDisconnected(const boost::optional<ClientError>& error) {

if (rosterController_) {

rosterController_->getWindow()->setStreamEncryptionStatus(false);

}

if (adHocManager_) {

adHocManager_->setOnline(false);

}

if (settings_->getSetting(SettingConstants::FORGET_PASSWORDS)) {

purgeCachedCredentials();

}

if (quitRequested_) {

resetClient();

loginWindow_->quit();

}

else if (error) {

std::string message;

std::string certificateErrorMessage;

bool forceSignout = false;

switch(error->getType()) {

case ClientError::UnknownError: message = QT_TRANSLATE_NOOP("", "Unknown Error"); break;

case ClientError::DomainNameResolveError: message = QT_TRANSLATE_NOOP("", "Unable to find server"); break;

case ClientError::ConnectionError: message = QT_TRANSLATE_NOOP("", "Error connecting to server"); break;

case ClientError::ConnectionReadError: message = QT_TRANSLATE_NOOP("", "Error while receiving server data"); break;

case ClientError::ConnectionWriteError: message = QT_TRANSLATE_NOOP("", "Error while sending data to the server"); break;

case ClientError::XMLError: message = QT_TRANSLATE_NOOP("", "Error parsing server data"); break;

case ClientError::AuthenticationFailedError: message = QT_TRANSLATE_NOOP("", "Login/password invalid"); break;

case ClientError::CompressionFailedError: message = QT_TRANSLATE_NOOP("", "Error while compressing stream"); break;

case ClientError::ServerVerificationFailedError: message = QT_TRANSLATE_NOOP("", "Server verification failed"); break;

case ClientError::NoSupportedAuthMechanismsError: message = QT_TRANSLATE_NOOP("", "Authentication mechanisms not supported"); break;

case ClientError::UnexpectedElementError: message = QT_TRANSLATE_NOOP("", "Unexpected response"); break;

case ClientError::ResourceBindError: message = QT_TRANSLATE_NOOP("", "Error binding resource"); break;

case ClientError::SessionStartError: message = QT_TRANSLATE_NOOP("", "Error starting session"); break;

case ClientError::StreamError: message = QT_TRANSLATE_NOOP("", "Stream error"); break;

case ClientError::TLSError: message = QT_TRANSLATE_NOOP("", "Encryption error"); break;

case ClientError::ClientCertificateLoadError: message = QT_TRANSLATE_NOOP("", "Error loading certificate (Invalid password?)"); break;

case ClientError::ClientCertificateError: message = QT_TRANSLATE_NOOP("", "Certificate not authorized"); break;

case ClientError::CertificateCardRemoved: message = QT_TRANSLATE_NOOP("", "Certificate card removed"); forceSignout = true; break;

case ClientError::UnknownCertificateError: certificateErrorMessage = QT_TRANSLATE_NOOP("", "Unknown certificate"); break;

case ClientError::CertificateExpiredError: certificateErrorMessage = QT_TRANSLATE_NOOP("", "Certificate has expired"); break;

case ClientError::CertificateNotYetValidError: certificateErrorMessage = QT_TRANSLATE_NOOP("", "Certificate is not yet valid"); break;

case ClientError::CertificateSelfSignedError: certificateErrorMessage = QT_TRANSLATE_NOOP("", "Certificate is self-signed"); break;

case ClientError::CertificateRejectedError: certificateErrorMessage = QT_TRANSLATE_NOOP("", "Certificate has been rejected"); break;

case ClientError::CertificateUntrustedError: certificateErrorMessage = QT_TRANSLATE_NOOP("", "Certificate is not trusted"); break;

case ClientError::InvalidCertificatePurposeError: certificateErrorMessage = QT_TRANSLATE_NOOP("", "Certificate cannot be used for encrypting your connection"); break;

case ClientError::CertificatePathLengthExceededError: certificateErrorMessage = QT_TRANSLATE_NOOP("", "Certificate path length constraint exceeded"); break;

case ClientError::InvalidCertificateSignatureError: certificateErrorMessage = QT_TRANSLATE_NOOP("", "Invalid certificate signature"); break;

case ClientError::InvalidCAError: certificateErrorMessage = QT_TRANSLATE_NOOP("", "Invalid Certificate Authority"); break;

case ClientError::InvalidServerIdentityError: certificateErrorMessage = QT_TRANSLATE_NOOP("", "Certificate does not match the host identity"); break;

case ClientError::RevokedError: certificateErrorMessage = QT_TRANSLATE_NOOP("", "Certificate has been revoked"); break;

case ClientError::RevocationCheckFailedError: certificateErrorMessage = QT_TRANSLATE_NOOP("", "Unable to determine certificate revocation state"); break;

}

bool forceReconnectAfterCertificateTrust = false;

if (!certificateErrorMessage.empty()) {

std::vector<Certificate::ref> certificates = certificateTrustChecker_->getLastCertificateChain();

if (!certificates.empty() && loginWindow_->askUserToTrustCertificatePermanently(certificateErrorMessage, certificates)) {

certificateStorage_->addCertificate(certificates[0]);

forceReconnectAfterCertificateTrust = true;

}

else {

message = QT_TRANSLATE_NOOP("", "Certificate error");

}

}

if (forceReconnectAfterCertificateTrust && settings_->getSetting(SettingConstants::FORGET_PASSWORDS)) {

forceReconnectAfterCertificateTrust = false;

forceSignout = true;

message = QT_TRANSLATE_NOOP("", "Re-enter credentials and retry");

}

if (forceReconnectAfterCertificateTrust) {

performLoginFromCachedCredentials();

}

else if (forceSignout || !rosterController_) { //hasn't been logged in yet or permanent error

signOut();

loginWindow_->setMessage(message);

loginWindow_->setIsLoggingIn(false);

} else {

logout();

if (settings_->getSetting(SettingConstants::FORGET_PASSWORDS)) {

message = str(format(QT_TRANSLATE_NOOP("", "Disconnected from %1%: %2%. To reconnect, Sign Out and provide your password again.")) % jid_.getDomain() % message);

} else {

if (!offlineRequested_) {

setReconnectTimer();

}

if (lastDisconnectError_) {

message = str(format(QT_TRANSLATE_NOOP("", "Reconnect to %1% failed: %2%. Will retry in %3% seconds.")) % jid_.getDomain() % message % boost::lexical_cast<std::string>(timeBeforeNextReconnect_));

lastDisconnectError_->conclude();

} else {

message = str(format(QT_TRANSLATE_NOOP("", "Disconnected from %1%: %2%.")) % jid_.getDomain() % message);

}

lastDisconnectError_ = boost::make_shared<ErrorEvent>(JID(jid_.getDomain()), message);

eventController_->handleIncomingEvent(lastDisconnectError_);

}

}

}

else if (!rosterController_) { //hasn't been logged in yet

loginWindow_->setIsLoggingIn(false);

}

}

void MainController::setReconnectTimer() {

if (timeBeforeNextReconnect_ < 0) {

timeBeforeNextReconnect_ = 1;

} else {

timeBeforeNextReconnect_ = timeBeforeNextReconnect_ >= 150 ? 300 : timeBeforeNextReconnect_ * 2; // Randomly selected by roll of a die, as required by 3920bis

}

if (reconnectTimer_) {

reconnectTimer_->stop();

}

reconnectTimer_ = networkFactories_->getTimerFactory()->createTimer(timeBeforeNextReconnect_ * 1000);

reconnectTimer_->onTick.connect(boost::bind(&MainController::reconnectAfterError, this));

reconnectTimer_->start();

}

void MainController::handleCancelLoginRequest() {

signOut();

}

void MainController::signOut() {

if (settings_->getSetting(SettingConstants::FORGET_PASSWORDS)) {

purgeCachedCredentials();

}

eventController_->clear();

logout();

loginWindow_->loggedOut();

resetClient();

}

void MainController::logout() {

if (settings_->getSetting(SettingConstants::FORGET_PASSWORDS)) {

purgeCachedCredentials();

}

systemTrayController_->setMyStatusType(StatusShow::None);

if (clientInitialized_ /*&& client_->isAvailable()*/) {

client_->disconnect();

}

if (rosterController_ && myStatusLooksOnline_) {

rosterController_->getWindow()->setMyStatusType(StatusShow::None);

rosterController_->getWindow()->setMyStatusText("");

myStatusLooksOnline_ = false;

}

setManagersOffline();

}

void MainController::setManagersOffline() {

if (chatsManager_) {

chatsManager_->setOnline(false);

}

if (rosterController_) {

rosterController_->setEnabled(false);

}

if (profileController_) {

profileController_->setAvailable(false);

}

if (contactEditController_) {

contactEditController_->setAvailable(false);

}

}

void MainController::handleServerDiscoInfoResponse(boost::shared_ptr<DiscoInfo> info, ErrorPayload::ref error) {

if (!error) {

chatsManager_->setServerDiscoInfo(info);

adHocManager_->setServerDiscoInfo(info);

if (info->hasFeature(DiscoInfo::BlockingCommandFeature)) {

rosterController_->getWindow()->setBlockingCommandAvailable(true);

rosterController_->initBlockingCommand();

}

}

}

void MainController::handleVCardReceived(const JID& jid, VCard::ref vCard) {

if (!jid.equals(jid_, JID::WithoutResource) || !vCard) {

return;

}

std::string hash;

if (!vCard->getPhoto().empty()) {

hash = Hexify::hexify(networkFactories_->getCryptoProvider()->getSHA1Hash(vCard->getPhoto()));

}

if (hash != vCardPhotoHash_) {

vCardPhotoHash_ = hash;

if (client_ && client_->isAvailable()) {

sendPresence(statusTracker_->getNextPresence());

}

}

}

void MainController::handleNotificationClicked(const JID& jid) {

assert(chatsManager_);

if (clientInitialized_) {

if (client_->getMUCRegistry()->isMUC(jid)) {

uiEventStream_->send(boost::make_shared<JoinMUCUIEvent>(jid));

}

else {

uiEventStream_->send(boost::shared_ptr<UIEvent>(new RequestChatUIEvent(jid)));

}

}

}

void MainController::handleQuitRequest() {

if (client_ && client_->isActive()) {

quitRequested_ = true;

client_->disconnect();

}

else {

resetClient();

loginWindow_->quit();

}

}

#define SERIALIZE_BOOL(option) result += options.option ? "1" : "0"; result += ",";

#define SERIALIZE_INT(option) result += boost::lexical_cast<std::string>(options.option); result += ",";