-#include "Swiften/Controllers/MainController.h"

-

-#include <boost/bind.hpp>

-#include <boost/lexical_cast.hpp>

-#include <boost/shared_ptr.hpp>

-#include <stdlib.h>

-

-#include "Swiften/Application/Application.h"

-#include "Swiften/Application/ApplicationMessageDisplay.h"

-#include "Swiften/Controllers/ChatController.h"

-#include "Swiften/Controllers/ChatWindowFactory.h"

-#include "Swiften/Controllers/EventController.h"

-#include "Swiften/Controllers/LoginWindow.h"

-#include "Swiften/Controllers/LoginWindowFactory.h"

-#include "Swiften/Controllers/MainWindow.h"

-#include "Swiften/Controllers/MainWindowFactory.h"

-#include "Swiften/Controllers/MUCController.h"

-#include "Swiften/Controllers/NickResolver.h"

-#include "Swiften/Controllers/RosterController.h"

-#include "Swiften/Controllers/XMPPRosterController.h"

-#include "Swiften/Base/foreach.h"

-#include "Swiften/Base/String.h"

-#include "Swiften/Client/Client.h"

-#include "Swiften/Elements/Presence.h"

-#include "Swiften/Roster/XMPPRoster.h"

-#include "Swiften/Queries/Responders/SoftwareVersionResponder.h"

-#include "Swiften/Roster/TreeWidgetFactory.h"

-#include "Swiften/Settings/SettingsProvider.h"

-#include "Swiften/Elements/DiscoInfo.h"

-#include "Swiften/Queries/Responders/DiscoInfoResponder.h"

-#include "Swiften/Disco/CapsInfoGenerator.h"

-#include "Swiften/Queries/Requests/GetDiscoInfoRequest.h"

-

-namespace Swift {

-

-static const String CLIENT_NAME = "Swift";

-static const String CLIENT_VERSION = "0.3";

-static const String CLIENT_NODE = "http://swift.im";

-

-typedef std::pair<JID, ChatController*> JIDChatControllerPair;

-typedef std::pair<JID, MUCController*> JIDMUCControllerPair;

-

-MainController::MainController(ChatWindowFactory* chatWindowFactory, MainWindowFactory *mainWindowFactory, LoginWindowFactory *loginWindowFactory, TreeWidgetFactory *treeWidgetFactory, SettingsProvider *settings, Application* application)

- : client_(NULL), chatWindowFactory_(chatWindowFactory), mainWindowFactory_(mainWindowFactory), loginWindowFactory_(loginWindowFactory), treeWidgetFactory_(treeWidgetFactory), settings_(settings),

- xmppRosterController_(NULL), rosterController_(NULL), loginWindow_(NULL), clientVersionResponder_(NULL), nickResolver_(NULL), discoResponder_(NULL),

- serverDiscoInfo_(new DiscoInfo()), presenceOracle_(NULL) {

- application_ = application;

- eventController_ = new EventController();

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

- loginWindow_ = loginWindowFactory_->createLoginWindow(settings->getStringSetting("jid"), settings->getStringSetting("pass"), settings->getStringSetting("certificate"));

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

-}

-

-MainController::~MainController() {

- delete discoResponder_;

- delete clientVersionResponder_;

- delete xmppRosterController_;

- delete rosterController_;

- foreach (JIDChatControllerPair controllerPair, chatControllers_) {

- delete controllerPair.second;

- }

- foreach (JIDMUCControllerPair controllerPair, mucControllers_) {

- delete controllerPair.second;

- }

- delete presenceOracle_;

- delete nickResolver_;

- delete client_;

-}

-

-void MainController::handleConnected() {

- delete presenceOracle_;

- presenceOracle_ = new PresenceOracle(client_);

-

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

-

- boost::shared_ptr<XMPPRoster> xmppRoster(new XMPPRoster());

-

-

- delete nickResolver_;

- nickResolver_ = new NickResolver(xmppRoster);

-

- delete rosterController_;

- rosterController_ = new RosterController(xmppRoster, mainWindowFactory_, treeWidgetFactory_);

- rosterController_->onStartChatRequest.connect(boost::bind(&MainController::handleChatRequest, this, _1));

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

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

-

- delete xmppRosterController_;

- xmppRosterController_ = new XMPPRosterController(client_, xmppRoster);

- xmppRosterController_->requestRoster();

-

- delete clientVersionResponder_;

- clientVersionResponder_ = new SoftwareVersionResponder(CLIENT_NAME, CLIENT_VERSION, client_);

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

-

- DiscoInfo discoInfo;

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

- capsInfo_ = boost::shared_ptr<CapsInfo>(new CapsInfo(CapsInfoGenerator(CLIENT_NODE).generateCapsInfo(discoInfo)));

- discoResponder_ = new DiscoInfoResponder(client_);

- discoResponder_->setDiscoInfo(discoInfo);

- discoResponder_->setDiscoInfo(capsInfo_->getNode() + "#" + capsInfo_->getVersion(), discoInfo);

-

- serverDiscoInfo_ = boost::shared_ptr<DiscoInfo>(new DiscoInfo());

- GetDiscoInfoRequest* discoInfoRequest = new GetDiscoInfoRequest(JID(), client_, Request::AutoDeleteAfterResponse);

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

- discoInfoRequest->send();

-

- //Send presence last to catch all the incoming presences.

- boost::shared_ptr<Presence> initialPresence(new Presence());

- initialPresence->addPayload(capsInfo_);

- client_->sendPresence(initialPresence);

-}

-

-void MainController::handleEventQueueLengthChange(int count) {

- application_->getApplicationMessageDisplay()->setMessage(count == 0 ? "" : boost::lexical_cast<std::string>(count).c_str());

-}

-

-void MainController::handleChangeStatusRequest(StatusShow::Type show, const String &statusText) {

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

- presence->addPayload(capsInfo_);

- presence->setShow(show);

- presence->setStatus(statusText);

- // FIXME: This is wrong. None doesn't mean unavailable

- if (show == StatusShow::None) {

- presence->setType(Presence::Unavailable);

- }

- client_->sendPresence(presence);

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

- logout();

- }

-}

-

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

- rosterController_->handleIncomingPresence(presence);

-}

-

-void MainController::handleLoginRequest(const String &username, const String &password, const String& certificateFile, bool remember) {

- loginWindow_->setMessage("");

-

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

- settings_->storeString("certificate", certificateFile);

- settings_->storeString("pass", remember ? password : "");

-

- delete client_;

- client_ = new Swift::Client(JID(username), password);

- if (!certificateFile.isEmpty()) {

- client_->setCertificate(certificateFile);

- }

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

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

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

- client_->connect();

-}

-

-void MainController::handleError(const ClientError& error) {

- String message;

- switch(error.getType()) {

- case ClientError::NoError: assert(false); break;

- case ClientError::DomainNameResolveError: message = "Unable to find server"; break;

- case ClientError::ConnectionError: message = "Error connecting to server"; break;

- case ClientError::ConnectionReadError: message = "Error while receiving server data"; break;

- case ClientError::XMLError: message = "Error parsing server data"; break;

- case ClientError::AuthenticationFailedError: message = "Login/password invalid"; break;

- case ClientError::NoSupportedAuthMechanismsError: message = "Authentication mechanisms not supported"; break;

- case ClientError::UnexpectedElementError: message = "Unexpected response"; break;

- case ClientError::ResourceBindError: message = "Error binding resource"; break;

- case ClientError::SessionStartError: message = "Error starting session"; break;

- case ClientError::TLSError: message = "Encryption error"; break;

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

- case ClientError::ClientCertificateError: message = "Certificate not authorized"; break;

- }

- loginWindow_->setMessage(message);

- logout();

-}

-

-void MainController::logout() {

- loginWindow_->loggedOut();

-

- delete discoResponder_;

- discoResponder_ = NULL;

- delete clientVersionResponder_;

- clientVersionResponder_ = NULL;

- foreach (JIDChatControllerPair controllerPair, chatControllers_) {

- delete controllerPair.second;

- }

- client_->disconnect();

- chatControllers_.clear();

- foreach (JIDMUCControllerPair controllerPair, mucControllers_) {

- delete controllerPair.second;

- }

- mucControllers_.clear();

-}

-

-

-void MainController::handleChatRequest(const String &contact) {

- getChatController(JID(contact))->showChatWindow();

-}

-

-ChatController* MainController::getChatController(const JID &contact) {

- JID lookupContact(contact);

- if (chatControllers_.find(lookupContact) == chatControllers_.end()) {

- lookupContact = JID(contact.toBare());

- }

- if (chatControllers_.find(lookupContact) == chatControllers_.end()) {

- chatControllers_[contact] = new ChatController(client_, client_, chatWindowFactory_, contact, nickResolver_, presenceOracle_);

- chatControllers_[contact]->setAvailableServerFeatures(serverDiscoInfo_);

- lookupContact = contact;

- }

- return chatControllers_[lookupContact];

-}

-

-void MainController::handleChatControllerJIDChanged(const JID& from, const JID& to) {

- chatControllers_[to] = chatControllers_[from];

- chatControllers_.erase(from);

-}

-

-void MainController::handleJoinMUCRequest(const JID &muc, const String &nick) {

- mucControllers_[muc] = new MUCController(muc, nick, client_, client_, chatWindowFactory_, treeWidgetFactory_, presenceOracle_);

- mucControllers_[muc]->setAvailableServerFeatures(serverDiscoInfo_);

-}

-

-void MainController::handleIncomingMessage(boost::shared_ptr<Message> message) {

- JID jid = message->getFrom();

- boost::shared_ptr<MessageEvent> event(new MessageEvent(message));

-

- // Try to deliver it to a MUC

- if (message->getType() == Message::Groupchat || message->getType() == Message::Error) {

- std::map<JID, MUCController*>::iterator i = mucControllers_.find(jid.toBare());

- if (i != mucControllers_.end()) {

- i->second->handleIncomingMessage(event);

- return;

- }

- else if (message->getType() == Message::Groupchat) {

- //FIXME: Error handling - groupchat messages from an unknown muc.

- return;

- }

- }

-

- //if not a mucroom

- eventController_->handleIncomingEvent(event);

-

- // FIXME: This logic should go into a chat manager

- if (event->isReadable()) {

- getChatController(jid)->handleIncomingMessage(event);

- }

-}

-

-void MainController::handleServerDiscoInfoResponse(boost::shared_ptr<DiscoInfo> info, const boost::optional<Error>& error) {

- if (!error) {

- serverDiscoInfo_ = info;

- foreach (JIDChatControllerPair pair, chatControllers_) {

- pair.second->setAvailableServerFeatures(info);

- }

- foreach (JIDMUCControllerPair pair, mucControllers_) {

- pair.second->setAvailableServerFeatures(info);

- }

- }

-}

-

-}