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

#include "Slimber/MainController.h"

#include <boost/bind.hpp>
#include <boost/lexical_cast.hpp>
#include <iostream>

#include "Swiften/Base/foreach.h"
#include "SwifTools/Application/PlatformApplicationPathProvider.h"
#include "Swiften/LinkLocal/LinkLocalService.h"
#include "Swiften/LinkLocal/LinkLocalServiceBrowser.h"
#include "Swiften/LinkLocal/DNSSD/PlatformDNSSDQuerierFactory.h"
#include "Slimber/Server.h"
#include "Slimber/FileVCardCollection.h"
#include "Slimber/MenuletController.h"
#include "Slimber/Menulet.h"

using namespace Swift;

MainController::MainController(Menulet* menulet, EventLoop* eventLoop) : menulet(menulet) {
	dnsSDQuerier = PlatformDNSSDQuerierFactory(eventLoop).createQuerier();
	assert(dnsSDQuerier);

	linkLocalServiceBrowser = new LinkLocalServiceBrowser(dnsSDQuerier);
	linkLocalServiceBrowser->onServiceAdded.connect(
			boost::bind(&MainController::handleServicesChanged, this));
	linkLocalServiceBrowser->onServiceRemoved.connect(
			boost::bind(&MainController::handleServicesChanged, this));
	linkLocalServiceBrowser->onServiceChanged.connect(
			boost::bind(&MainController::handleServicesChanged, this));

	vCardCollection = new FileVCardCollection(
			PlatformApplicationPathProvider("Slimber").getDataDir());

	server = new Server(5222, 5562, linkLocalServiceBrowser, vCardCollection, eventLoop);
	server->onStopped.connect(
			boost::bind(&MainController::handleServerStopped, this, _1));
	server->onSelfConnected.connect(
			boost::bind(&MainController::handleSelfConnected, this, _1));

	menuletController = new MenuletController(menulet);
	menuletController->onRestartRequested.connect(boost::bind(
			&MainController::handleRestartRequested, this));


	start();
}

MainController::~MainController() {
	delete server;
	delete menuletController;
	delete vCardCollection;
	linkLocalServiceBrowser->stop();
	delete linkLocalServiceBrowser;
	dnsSDQuerier->stop();
}

void MainController::start() {
	dnsSDQuerier->start();
	linkLocalServiceBrowser->start();

	handleSelfConnected(false);
	handleServicesChanged();

	server->start();
}

void MainController::stop() {
	server->stop();
	linkLocalServiceBrowser->stop();
	dnsSDQuerier->stop();
}

void MainController::handleSelfConnected(bool b) {
	if (b) {
		menuletController->setXMPPStatus("You are logged in", MenuletController::Online);
	}
	else {
		menuletController->setXMPPStatus("You are not logged in", MenuletController::Offline);
	}
}

void MainController::handleServicesChanged() {
	std::vector<std::string> names;
	foreach(const LinkLocalService& service, linkLocalServiceBrowser->getServices()) {
		std::string description = service.getDescription();
		if (description != service.getName()) {
			description += " (" + service.getName() + ")";
		}
		names.push_back(description);
	}
	menuletController->setUserNames(names);
}

void MainController::handleServerStopped(boost::optional<ServerError> error) {
	if (error) {
		std::string message;
		switch (error->getType()) {
			case ServerError::C2SPortConflict: 
				message = std::string("Error: Port ") + boost::lexical_cast<std::string>(server->getClientToServerPort()) + std::string(" in use");
				break;
			case ServerError::C2SError:
				message = std::string("Local connection server error");
			 break;
			case ServerError::LinkLocalPortConflict: 
				message = std::string("Error: Port ") + boost::lexical_cast<std::string>(server->getLinkLocalPort()) + std::string(" in use");
				break;
			case ServerError::LinkLocalError: 
				message = std::string("External connection server error");
				break;
		}
		menuletController->setXMPPStatus(message, MenuletController::Offline);
	}
	else {
		menuletController->setXMPPStatus("XMPP Server Not Running", MenuletController::Offline);
	}
}

void MainController::handleRestartRequested() {
	stop();
	start();
}