/*
 * Copyright (c) 2012 Mateusz Piękos
 * Licensed under the simplified BSD license.
 * See Documentation/Licenses/BSD-simplified.txt for more information.
 */

#include <Swift/Controllers/WhiteboardManager.h>

#include <boost/bind.hpp>

#include <Swiften/Base/foreach.h>
#include <Swift/Controllers/UIEvents/RequestWhiteboardUIEvent.h>
#include <Swift/Controllers/UIEvents/AcceptWhiteboardSessionUIEvent.h>
#include <Swift/Controllers/UIEvents/CancelWhiteboardSessionUIEvent.h>
#include <Swiften/Client/StanzaChannel.h>
#include <Swiften/Whiteboard/WhiteboardSessionManager.h>

namespace Swift {
	typedef std::pair<JID, WhiteboardWindow*> JIDWhiteboardWindowPair;

	WhiteboardManager::WhiteboardManager(WhiteboardWindowFactory* whiteboardWindowFactory, UIEventStream* uiEventStream, WhiteboardSessionManager* whiteboardSessionManager) : whiteboardWindowFactory_(whiteboardWindowFactory), uiEventStream_(uiEventStream), whiteboardSessionManager_(whiteboardSessionManager) {
		uiEventConnection_ = uiEventStream_->onUIEvent.connect(boost::bind(&WhiteboardManager::handleUIEvent, this, _1));
		whiteboardSessionManager_->onRequestAccepted.connect(boost::bind(&WhiteboardManager::handleAcceptedRequest, this, _1, _2));
	}

	WhiteboardManager::~WhiteboardManager() {
		foreach (JIDWhiteboardWindowPair whiteboardWindowPair, whiteboardWindows_) {
			delete whiteboardWindowPair.second;
		}
	}

	WhiteboardWindow* WhiteboardManager::createNewWhiteboardWindow(const JID& contact, WhiteboardSession* session) {
		WhiteboardWindow *window = whiteboardWindowFactory_->createWhiteboardWindow(session);
		whiteboardWindows_[contact] = window;
		return window;
	}

	WhiteboardWindow* WhiteboardManager::findWhiteboardWindow(const JID& contact) {
		if (whiteboardWindows_.find(contact) == whiteboardWindows_.end()) {
			return NULL;
		}
		return whiteboardWindows_[contact];
	}

	void WhiteboardManager::handleUIEvent(boost::shared_ptr<UIEvent> event) {
		boost::shared_ptr<RequestWhiteboardUIEvent> whiteboardEvent = boost::dynamic_pointer_cast<RequestWhiteboardUIEvent>(event);
		if (whiteboardEvent) {
			whiteboardSessionManager_->requestSession(whiteboardEvent->getContact());
		}
		boost::shared_ptr<AcceptWhiteboardSessionUIEvent> sessionAcceptEvent = boost::dynamic_pointer_cast<AcceptWhiteboardSessionUIEvent>(event);
		if (sessionAcceptEvent) {
			acceptSession(sessionAcceptEvent->getContact());
		}
		boost::shared_ptr<CancelWhiteboardSessionUIEvent> sessionCancelEvent = boost::dynamic_pointer_cast<CancelWhiteboardSessionUIEvent>(event);
		if (sessionCancelEvent) {
			whiteboardSessionManager_->cancelSession(sessionCancelEvent->getContact());
		}
	}

	void WhiteboardManager::acceptSession(const JID& from) {
		WhiteboardSession* session = whiteboardSessionManager_->acceptSession(from);
		WhiteboardWindow* window = findWhiteboardWindow(from);
		if (window == NULL) {
			window = createNewWhiteboardWindow(from, session);
		}
		window->show();
	}

	void WhiteboardManager::handleAcceptedRequest(const JID& from, WhiteboardSession* session) {
		WhiteboardWindow* window = findWhiteboardWindow(from);
		if (window == NULL) {
			window = createNewWhiteboardWindow(from, session);
		}
		window->show();
	}
}