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

 */

#include "transport/rostermanager.h"

#include "transport/rosterstorage.h"

#include "transport/storagebackend.h"

#include "transport/buddy.h"

#include "transport/usermanager.h"

#include "transport/buddy.h"

#include "transport/user.h"

#include "Swiften/Roster/SetRosterRequest.h"

#include "Swiften/Elements/RosterPayload.h"

#include "Swiften/Elements/RosterItemPayload.h"

#include "Swiften/Elements/RosterItemExchangePayload.h"

#include "log4cxx/logger.h"

#include <boost/foreach.hpp>

#include <map>

#include <iterator>

using namespace log4cxx;

namespace Transport {

static LoggerPtr logger = Logger::getLogger("RosterManager");

RosterManager::RosterManager(User *user, Component *component){

	m_rosterStorage = NULL;

	m_user = user;

	m_component = component;

	m_setBuddyTimer = m_component->getNetworkFactories()->getTimerFactory()->createTimer(1000);

	m_RIETimer = m_component->getNetworkFactories()->getTimerFactory()->createTimer(5000);

	m_RIETimer->onTick.connect(boost::bind(&RosterManager::sendRIE, this));

	m_supportRemoteRoster = false;

	if (!m_component->inServerMode()) {

		AddressedRosterRequest::ref request = AddressedRosterRequest::ref(new AddressedRosterRequest(m_component->getIQRouter(), m_user->getJID().toBare()));

		request->onResponse.connect(boost::bind(&RosterManager::handleRemoteRosterResponse, this, _1, _2));

		request->send();

	}

}

RosterManager::~RosterManager() {

	m_setBuddyTimer->stop();

	m_RIETimer->stop();

	if (m_rosterStorage) {

		m_rosterStorage->storeBuddies();

	}

	sendUnavailablePresences(m_user->getJID().toBare());

	for (std::map<std::string, Buddy *, std::less<std::string>, boost::pool_allocator< std::pair<std::string, Buddy *> > >::iterator it = m_buddies.begin(); it != m_buddies.end(); it++) {

		Buddy *buddy = (*it).second;

		if (!buddy) {

			continue;

		}

		delete buddy;

	}

	if (m_requests.size() != 0) {

		LOG4CXX_INFO(logger, m_user->getJID().toString() <<  ": Removing " << m_requests.size() << " unresponded IQs");

		BOOST_FOREACH(Swift::SetRosterRequest::ref request, m_requests) {

			request->onResponse.disconnect_all_slots();

			m_component->getIQRouter()->removeHandler(request);

		}

		m_requests.clear();

	}

	boost::singleton_pool<boost::pool_allocator_tag, sizeof(unsigned int)>::release_memory();

	if (m_rosterStorage)

		delete m_rosterStorage;

}

void RosterManager::setBuddy(Buddy *buddy) {

// 	m_setBuddyTimer->onTick.connect(boost::bind(&RosterManager::setBuddyCallback, this, buddy));

// 	m_setBuddyTimer->start();

	setBuddyCallback(buddy);

}

void RosterManager::sendBuddyRosterPush(Buddy *buddy) {

	// user can't receive anything in server mode if he's not logged in.

	// He will ask for roster later (handled in rosterreponsder.cpp)

	if (m_component->inServerMode() && !m_user->isConnected())

		return;

	Swift::RosterPayload::ref payload = Swift::RosterPayload::ref(new Swift::RosterPayload());

	Swift::RosterItemPayload item;

	item.setJID(buddy->getJID().toBare());

	item.setName(buddy->getAlias());

	item.setGroups(buddy->getGroups());

	item.setSubscription(Swift::RosterItemPayload::Both);

	payload->addItem(item);

		request->onResponse.connect(boost::bind(&RosterManager::handleBuddyRosterPushResponse, this, _1, request, buddy->getName()));

		request->send();

		m_requests.push_back(request);

	}

void RosterManager::sendBuddySubscribePresence(Buddy *buddy) {

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

	response->setTo(m_user->getJID());

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

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

// 	TODO: NICKNAME

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

}

void RosterManager::handleBuddyChanged(Buddy *buddy) {

	if (m_rosterStorage) {

		m_rosterStorage->storeBuddy(buddy);

	}

}

void RosterManager::setBuddyCallback(Buddy *buddy) {

	m_setBuddyTimer->onTick.disconnect(boost::bind(&RosterManager::setBuddyCallback, this, buddy));

	LOG4CXX_INFO(logger, "Associating buddy " << buddy->getName() << " with " << m_user->getJID().toString());

	m_buddies[buddy->getName()] = buddy;

	onBuddySet(buddy);

	// In server mode the only way is to send jabber:iq:roster push.

	// In component mode we send RIE or Subscribe presences, based on features.

	if (m_component->inServerMode()) {

		sendBuddyRosterPush(buddy);

	}

	else {

		if (m_setBuddyTimer->onTick.empty()) {

			m_setBuddyTimer->stop();

			if (m_supportRemoteRoster) {

				sendBuddyRosterPush(buddy);

			}

			else {

				// Send RIE only if there's resource which supports it.

				Swift::JID jidWithRIE = m_user->getJIDWithFeature("http://jabber.org/protocol/rosterx");

				if (jidWithRIE.isValid()) {

					m_RIETimer->start();

				}

				else {

					sendBuddySubscribePresence(buddy);

				}

			}

		}

	}

	if (m_rosterStorage)

		m_rosterStorage->storeBuddy(buddy);

}

void RosterManager::unsetBuddy(Buddy *buddy) {

	m_buddies.erase(buddy->getName());

	if (m_rosterStorage)

		m_rosterStorage->removeBuddyFromQueue(buddy);

	onBuddyUnset(buddy);

}

void RosterManager::storeBuddy(Buddy *buddy) {

	if (m_rosterStorage) {

		m_rosterStorage->storeBuddy(buddy);

	}

}

void RosterManager::handleBuddyRosterPushResponse(Swift::ErrorPayload::ref error, Swift::SetRosterRequest::ref request, const std::string &key) {

	LOG4CXX_INFO(logger, "handleBuddyRosterPushResponse called for buddy " << key);

	if (m_buddies[key] != NULL) {

		m_buddies[key]->handleBuddyChanged();

	}

	else {

		LOG4CXX_WARN(logger, "handleBuddyRosterPushResponse called for unknown buddy " << key);

	}

	m_requests.remove(request);

	request->onResponse.disconnect_all_slots();

}

void RosterManager::handleRemoteRosterResponse(boost::shared_ptr<Swift::RosterPayload> payload, Swift::ErrorPayload::ref error) {

	if (error) {

		m_supportRemoteRoster = false;

		LOG4CXX_INFO(logger, m_user->getJID().toString() << ": This server does not support remote roster protoXEP");

		return;

	}

	LOG4CXX_INFO(logger, m_user->getJID().toString() << ": This server supports remote roster protoXEP");

	m_supportRemoteRoster = true;

	return;

	BOOST_FOREACH(const Swift::RosterItemPayload &item, payload->getItems()) {

		std::string legacyName = Buddy::JIDToLegacyName(item.getJID());

		if (m_buddies.find(legacyName) != m_buddies.end()) {

			continue;

		}

		std::cout << "LEGACYNAME " << legacyName << "\n";