/**

 * libtransport -- C++ library for easy XMPP Transports development

 *

 * Copyright (C) 2011, Jan Kaluza <hanzz.k@gmail.com>

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; either version 2 of the License, or

 * (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * 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 <string>

#include <map>

#include "Swiften/Swiften.h"

#include "transport/userregistry.h"

namespace Transport {

class User;

class Component;

class StorageBackend;

class StorageResponder;

class RosterResponder;

/// Manages online XMPP Users.

/// This class handles presences and creates User classes when new user connects.

/// It also removes the User class once the last user's resource disconnected.

class UserManager {

	public:

		/// Creates new UserManager.

		/// \param component Component which's presence will be handled

		/// \param storageBackend Storage backend used to fetch UserInfos

		UserManager(Component *component, UserRegistry *userRegistry, StorageBackend *storageBackend = NULL);

		/// Destroys UserManager.

		~UserManager();

		/// Returns user according to his bare JID.

		/// \param barejid bare JID of user

		/// \return User class associated with this user

		User *getUser(const std::string &barejid);

		const std::map<std::string, User *> &getUsers() {

			return m_users;

		}

		/// Returns number of online users.

		/// \return number of online users

		int getUserCount();

		/// Removes user. This function disconnects user and safely removes

		/// User class. This does *not* remove user from database.

		/// \param user User class to remove

		void removeUser(User *user);

		/// Called when new User class is created.

		/// \param user newly created User class

		boost::signal<void (User *user)> onUserCreated;

		/// Called when User class is going to be removed

		/// \param user removed User class

		boost::signal<void (User *user)> onUserDestroyed;

		bool isUserConnected(const std::string &barejid) const {

			return m_users.find(barejid) != m_users.end();

		}

		void connectUser(const Swift::JID &user);

	private:

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

		void handleMessageReceived(Swift::Message::ref message);

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

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

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

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

		void addUser(User *user);

		long m_onlineBuddies;

		User *m_cachedUser;

		std::map<std::string, User *> m_users;

		Component *m_component;

		StorageBackend *m_storageBackend;

		StorageResponder *m_storageResponder;

		UserRegistry *m_userRegistry;

		friend class RosterResponder;

};

}

[service]

jid = localhost

password = secret

server = 127.0.0.1

port = 5222

server_mode = 1

backend_host=localhost # < this option doesn't work yet

backend_port=10001

admin_username=admin

admin_password=test

#cert= #patch to PKCS#12 certificate

#cert_password= #password to that certificate if any

users_per_backend=2

backend=../../backends/libpurple/spectrum_libpurple_backend

#backend=../../backends/libircclient-qt/spectrum_libircclient-qt_backend

#protocol=prpl-icq

[logging]

#config=logging.cfg # log4cxx/log4j logging configuration file

#backend_config=backend_logging.cfg # log4cxx/log4j logging configuration file for backends

[database]

type = sqlite3 # or "none" without database backend

database = test.sql

prefix=icq

	if (!m_connected) {

		// we are not connected to legacy network, so we should do it when disco#info arrive :)

		if (m_readyForConnect == false) {

			// Forward status message to legacy network, but only if it's sent from active resource

// 					if (m_activeResource == presence->getFrom().getResource().getUTF8String()) {

// 						forwardStatus(presenceShow, stanzaStatus);

// 					}

			boost::shared_ptr<Swift::CapsInfo> capsInfo = presence->getPayload<Swift::CapsInfo>();

			if (capsInfo && capsInfo->getHash() == "sha-1") {

				if (m_entityCapsManager->getCaps(presence->getFrom()) != Swift::DiscoInfo::ref()) {

					LOG4CXX_INFO(logger, m_jid.toString() << ": Ready to be connected to legacy network");

					m_readyForConnect = true;

					onReadyToConnect();

				}

			}

			else if (m_component->inServerMode()) {

				LOG4CXX_INFO(logger, m_jid.toString() << ": Ready to be connected to legacy network");

				m_readyForConnect = true;

				onReadyToConnect();

			}

			else {

				m_reconnectTimer->start();

			}

		}

	}

	bool isMUC = presence->getPayload<Swift::MUCPayload>() != NULL || *presence->getTo().getNode().c_str() == '#';

	if (isMUC) {

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

			LOG4CXX_INFO(logger, m_jid.toString() << ": Going to left room " << presence->getTo().getNode());

			onRoomLeft(presence->getTo().getNode());

		}

		else {

			// force connection to legacy network to let backend to handle auto-join on connect.

			if (!m_readyForConnect) {

				LOG4CXX_INFO(logger, m_jid.toString() << ": Ready to be connected to legacy network");

				m_readyForConnect = true;

				onReadyToConnect();

			}

			LOG4CXX_INFO(logger, m_jid.toString() << ": Going to join room " << presence->getTo().getNode() << " as " << presence->getTo().getResource());

			onRoomJoined(presence->getTo().getNode(), presence->getTo().getResource(), "");

		}

		return;

	}

	Swift::Presence::ref highest = m_presenceOracle->getHighestPriorityPresence(m_jid.toBare());

	if (highest) {

		Swift::Presence::ref response = Swift::Presence::create(highest);

		response->setTo(presence->getFrom().toBare());

		response->setFrom(m_component->getJID());

		m_component->getStanzaChannel()->sendPresence(response);

		LOG4CXX_INFO(logger, m_jid.toString() << ": Changing legacy network presence to " << response->getType());

		onPresenceChanged(response);

	}

	else {

		Swift::Presence::ref response = Swift::Presence::create();

		response->setTo(m_jid.toBare());

		response->setFrom(m_component->getJID());

		response->setType(Swift::Presence::Unavailable);

		m_component->getStanzaChannel()->sendPresence(response);

		onPresenceChanged(response);

	}

}

void User::handleSubscription(Swift::Presence::ref presence) {

	m_rosterManager->handleSubscription(presence);

}

void User::onConnectingTimeout() {

	if (m_connected || m_readyForConnect)

		return;

	m_reconnectTimer->stop();

	m_readyForConnect = true;

	onReadyToConnect();

}

void User::handleDisconnected(const std::string &error) {

	if (error.empty()) {

		LOG4CXX_INFO(logger, m_jid.toString() << ": Disconnected from legacy network");

	}

	else {

		LOG4CXX_INFO(logger, m_jid.toString() << ": Disconnected from legacy network with error " << error);

	}

	onDisconnected();

	boost::shared_ptr<Swift::Message> msg(new Swift::Message());

	msg->setBody(error);

	msg->setTo(m_jid.toBare());

	msg->setFrom(m_component->getJID());

	m_component->getStanzaChannel()->sendMessage(msg);

	// In server mode, server finishes the session and pass unavailable session to userManager,

	// so we can't removeUser() in server mode, because it would be removed twice.

	// Once in finishSession and once in m_userManager->removeUser.

	if (m_component->inServerMode()) {

		dynamic_cast<Swift::ServerStanzaChannel *>(m_component->getStanzaChannel())->finishSession(m_jid, boost::shared_ptr<Swift::Element>(new Swift::StreamError()));

			m_userManager->removeUser(this);

		}

	else {

		m_userManager->removeUser(this);

	}

}

}