#include "Swiften/Client/Client.h"

#include <boost/bind.hpp>

#include "Swiften/Network/MainBoostIOServiceThread.h"

#include "Swiften/Network/BoostIOServiceThread.h"

#include "Swiften/Client/ClientSession.h"

#include "Swiften/StreamStack/PlatformTLSLayerFactory.h"

#include "Swiften/Network/Connector.h"

#include "Swiften/Network/BoostConnectionFactory.h"

#include "Swiften/Network/BoostTimerFactory.h"

#include "Swiften/TLS/PKCS12Certificate.h"

#include "Swiften/Session/BasicSessionStream.h"

namespace Swift {

Client::Client(const JID& jid, const String& password) :

IQRouter(this), jid_(jid), password_(password) {

connectionFactory_ = new BoostConnectionFactory(&MainBoostIOServiceThread::getInstance().getIOService());

timerFactory_ = new BoostTimerFactory(&MainBoostIOServiceThread::getInstance().getIOService());

tlsLayerFactory_ = new PlatformTLSLayerFactory();

}

Client::~Client() {

if (session_ || connection_) {

std::cerr << "Warning: Client not disconnected properly" << std::endl;

}

delete tlsLayerFactory_;

delete timerFactory_;

delete connectionFactory_;

}

bool Client::isAvailable() {

return session_;

}

void Client::connect() {

assert(!connector_);

connector_ = boost::shared_ptr<Connector>(new Connector(jid_.getDomain(), &resolver_, connectionFactory_));

connector_->onConnectFinished.connect(boost::bind(&Client::handleConnectorFinished, this, _1));

connector_->start();

}

void Client::handleConnectorFinished(boost::shared_ptr<Connection> connection) {

// TODO: Add domain name resolver error

connector_.reset();

if (!connection) {

onError(ClientError::ConnectionError);

}

else {

assert(!connection_);

connection_ = connection;

assert(!sessionStream_);

sessionStream_ = boost::shared_ptr<BasicSessionStream>(new BasicSessionStream(connection_, &payloadParserFactories_, &payloadSerializers_, tlsLayerFactory_, timerFactory_));

if (!certificate_.isEmpty()) {

sessionStream_->setTLSCertificate(PKCS12Certificate(certificate_, password_));

}

sessionStream_->onDataRead.connect(boost::bind(&Client::handleDataRead, this, _1));