/*
 * Copyright (c) 2010 Remko Tronçon
 * Licensed under the GNU General Public License v3.
 * See Documentation/Licenses/GPLv3.txt for more information.
 */

#include "Swiften/Avatars/CombinedAvatarProvider.h"

#include <algorithm>
#include <boost/bind.hpp>

namespace Swift {

String CombinedAvatarProvider::getAvatarHash(const JID& jid) const {
	for (size_t i = 0; i < providers.size(); ++i) {
		String hash = providers[i]->getAvatarHash(jid);
		if (!hash.isEmpty()) {
			return hash;
		}
	}
	return String();
}

void CombinedAvatarProvider::addProvider(AvatarProvider* provider) {
	provider->onAvatarChanged.connect(boost::bind(&CombinedAvatarProvider::handleAvatarChanged, this, _1));
	providers.push_back(provider);
}

void CombinedAvatarProvider::removeProvider(AvatarProvider* provider) {
	std::vector<AvatarProvider*>::iterator i = std::remove(providers.begin(), providers.end(), provider);
	for(std::vector<AvatarProvider*>::iterator j = i; j < providers.end(); ++j) {
		provider->onAvatarChanged.disconnect(boost::bind(&CombinedAvatarProvider::handleAvatarChanged, this, _1));
	}
	providers.erase(i, providers.end());
}

void CombinedAvatarProvider::handleAvatarChanged(const JID& jid) {
	String hash = getAvatarHash(jid);
	std::map<JID, String>::iterator i = avatars.find(jid);
	if (i != avatars.end()) {
		if (i->second != hash) {
			if (hash.isEmpty()) {
				avatars.erase(i);
			}
			else {
				avatars.insert(std::make_pair(jid, hash));
			}
			onAvatarChanged(jid);
		}
	}
	else if (!hash.isEmpty()) {
		avatars.insert(std::make_pair(jid, hash));
		onAvatarChanged(jid);
	}
}

}