* You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02111-1301  USA

 */

#pragma once

#include <vector>

#include "Swiften/Swiften.h"

#include "Swiften/Server/Server.h"

#include "Swiften/Disco/GetDiscoInfoRequest.h"

#include "Swiften/Disco/EntityCapsManager.h"

#include "Swiften/Disco/CapsManager.h"

#include "Swiften/Disco/CapsMemoryStorage.h"

#include "Swiften/Network/BoostTimerFactory.h"

#include "Swiften/Network/BoostIOServiceThread.h"

#include "Swiften/Server/UserRegistry.h"

#include "Swiften/Base/SafeByteArray.h"

#include "Swiften/Jingle/JingleSessionManager.h"

#include <boost/bind.hpp>

#include "transport/config.h"

#include "transport/factory.h"

#include "transport/presenceoracle.h"

namespace Transport {

	// typedef enum { 	CLIENT_FEATURE_ROSTERX = 2,

	// 				CLIENT_FEATURE_XHTML_IM = 4,

	// 				CLIENT_FEATURE_FILETRANSFER = 8,

	// 				CLIENT_FEATURE_CHATSTATES = 16

	// 				} SpectrumImportantFeatures;

	// 

	class StorageBackend;

	class DiscoInfoResponder;

	class DiscoItemsResponder;

	class Factory;

	class UserRegistry;

	/// Represents one transport instance.

	/// It's used to connect the Jabber server and provides transaction layer

	/// between Jabber server and other classes.

	///

	/// In server mode it represents Jabber server to which users can connect and use

	/// it as transport.

	class Component {

		public:

			/// Creates new Component instance.

			/// \param xml xml stanza

			boost::signal<void (const std::string &xml)> onXMLOut;

			/// This signal is emitted when XML stanza is received from server.

			/// \param xml xml stanza

			boost::signal<void (const std::string &xml)> onXMLIn;

			Config *getConfig() { return m_config; }

			/// This signal is emitted when presence from XMPP user is received.

			/// It's emitted only for presences addressed to transport itself

			/// (for example to="j2j.domain.tld").

			/// \param presence presence data

			boost::signal<void (Swift::Presence::ref presence)> onUserPresenceReceived;

			boost::signal<void (const Swift::JID& jid, boost::shared_ptr<Swift::DiscoInfo> info)> onUserDiscoInfoReceived;

// 			boost::signal<void (boost::shared_ptr<Swift::DiscoInfo> info, Swift::ErrorPayload::ref error, const Swift::JID& jid)> onDiscoInfoResponse;

		private:

			void handleConnected();

			void handleConnectionError(const Swift::ComponentError &error);

			void handlePresence(Swift::Presence::ref presence);

			void handleDataRead(const Swift::SafeByteArray &data);

			void handleDataWritten(const Swift::SafeByteArray &data);

			void handleDiscoInfoResponse(boost::shared_ptr<Swift::DiscoInfo> info, Swift::ErrorPayload::ref error, const Swift::JID& jid);

			void handleCapsChanged(const Swift::JID& jid);

			Swift::NetworkFactories *m_factories;

			Swift::Component *m_component;

			Swift::Server *m_server;

			Swift::Timer::ref m_reconnectTimer;

			Swift::EntityCapsManager *m_entityCapsManager;

			Swift::CapsManager *m_capsManager;

			Swift::CapsMemoryStorage *m_capsMemoryStorage;

			PresenceOracle *m_presenceOracle;

			Swift::StanzaChannel *m_stanzaChannel;

			Swift::IQRouter *m_iqRouter;

			Transport::UserRegistry *m_userRegistry;

			StorageBackend *m_storageBackend;

 			DiscoInfoResponder *m_discoInfoResponder;

			DiscoItemsResponder *m_discoItemsResponder;

			int m_reconnectCount;

			Config* m_config;

void Component::setTransportFeatures(std::list<std::string> &features) {

	m_discoInfoResponder->setTransportFeatures(features);

}

Swift::CapsInfo &Component::getBuddyCapsInfo() {

		return m_discoInfoResponder->getBuddyCapsInfo();

}

void Component::setBuddyFeatures(std::list<std::string> &features) {

	// TODO: handle caps change

	m_discoInfoResponder->setBuddyFeatures(features);

}

void Component::start() {

	if (m_component && !m_component->isAvailable()) {

		LOG4CXX_INFO(logger, "Connecting XMPP server " << CONFIG_STRING(m_config, "service.server") << " port " << CONFIG_INT(m_config, "service.port"));

		m_reconnectCount++;

		m_component->connect(CONFIG_STRING(m_config, "service.server"), CONFIG_INT(m_config, "service.port"));

		m_reconnectTimer->stop();

	}

	else if (m_server) {

		LOG4CXX_INFO(logger, "Starting component in server mode on port " << CONFIG_INT(m_config, "service.port"));

		m_server->start();

		// We're connected right here, because we're in server mode...

		handleConnected();

	}

}

void Component::stop() {

	if (m_component) {

		m_reconnectCount = 0;

		// TODO: Call this once swiften will fix assert(!session_);

// 		m_component->disconnect();

		m_reconnectTimer->stop();

	}

	else if (m_server) {

		LOG4CXX_INFO(logger, "Stopping component in server mode on port " << CONFIG_INT(m_config, "service.port"));

		m_server->stop();

	}

}

void Component::handleConnected() {

	onConnected();

	m_reconnectCount = 0;

}

void Component::handleConnectionError(const ComponentError &error) {

	onConnectionError(error);

// 	if (m_reconnectCount == 2)

// 		Component::instance()->userManager()->removeAllUsers();

	std::string str = "Unknown error";

	switch (error.getType()) {

		case ComponentError::UnknownError: str = "Unknown error"; break;

		case ComponentError::ConnectionError: str = "Connection error"; break;

		case ComponentError::ConnectionReadError: str = "Connection read error"; break;

		case ComponentError::ConnectionWriteError: str = "Connection write error"; break;

		case ComponentError::XMLError: str = "XML Error"; break;

		case ComponentError::AuthenticationFailedError: str = "Authentication failed error"; break;

		case ComponentError::UnexpectedElementError: str = "Unexpected element error"; break;

	}

	LOG4CXX_INFO(logger, "Disconnected from XMPP server. Error: " << str);

	m_reconnectTimer->start();

}

void Component::handleDataRead(const Swift::SafeByteArray &data) {

	std::string d = safeByteArrayToString(data);

	if (!boost::starts_with(d, "<auth")) {

		LOG4CXX_INFO(logger_xml, "XML IN " << d);

	}