/*
 * Copyright (c) 2010-2016 Isode Limited.
 * All rights reserved.
 * See the COPYING file for more information.
 */

#include <Slimber/MainController.h>

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

#include <Swiften/LinkLocal/DNSSD/PlatformDNSSDQuerierFactory.h>
#include <Swiften/LinkLocal/LinkLocalService.h>
#include <Swiften/LinkLocal/LinkLocalServiceBrowser.h>

#include <Slimber/FileVCardCollection.h>
#include <Slimber/Menulet.h>
#include <Slimber/MenuletController.h>
#include <Slimber/Server.h>

#include <SwifTools/Application/PlatformApplicationPathProvider.h>

using namespace Swift;

MainController::MainController(Menulet* menulet, EventLoop* eventLoop) {
    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;
    for (const auto& 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 ") + std::to_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 ") + std::to_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();
}