/*
 * Copyright (c) 2012 Yoann Blein
 * Licensed under the simplified BSD license.
 * See Documentation/Licenses/BSD-simplified.txt for more information.
 */

#include <Swiften/ScreenSharing/ScreenSharing.h>

#include <Swiften/Base/foreach.h>
#include <Swiften/Elements/JingleRawUDPTransportPayload.h>
#include <Swiften/Network/PlatformNetworkEnvironment.h>
#include <Swiften/Network/UDPSocket.h>
#include <Swiften/Network/UDPSocketFactory.h>
#include <Swiften/ScreenSharing/RTPSession.h>
#include <Swiften/Jingle/JingleSession.h>

#include <boost/bind.hpp>

namespace Swift {

ScreenSharing::ScreenSharing(boost::shared_ptr<JingleSession> session, UDPSocketFactory* udpSocketFactory)
	: rtpSession(0), jingleSession(session), udpSocketFactory(udpSocketFactory)
{
	jingleSession->onSessionTerminateReceived.connect(boost::bind(&ScreenSharing::handleSessionTerminateReceived, this, _1));
}

ScreenSharing::~ScreenSharing()
{
	jingleSession->onSessionTerminateReceived.disconnect(boost::bind(&ScreenSharing::handleSessionTerminateReceived, this, _1));
}

void ScreenSharing::stop()
{
	jingleSession->sendTerminate(JinglePayload::Reason::Success);
	if (rtpSession)
		rtpSession->stop();
	onStateChange(ScreenSharing::Finished);
	onFinished();
}

bool ScreenSharing::addBestCandidate(boost::shared_ptr<JingleRawUDPTransportPayload> transport)
{
	// TODO: NAT traversal

	JingleRawUDPTransportPayload::Candidate candidate;
	candidate.cid = idGenerator.generateID();
	candidate.component = 1;
	candidate.generation = 0;

	PlatformNetworkEnvironment env;
	std::vector<NetworkInterface> interfaces = env.getNetworkInterfaces();

	serverSocket = udpSocketFactory->createUDPSocket();

	// Find the first ip which is not loopback
	/*foreach (const NetworkInterface& interface, interfaces) {
		if (!interface.isLoopback() && !interface.getAddresses().empty()) { // exclude loopback
			serverSocket->bindOnAvailablePort(interface.getAddresses().front());

			candidate.hostAddressPort = serverSocket->getLocalAddress();
			candidate.type = JingleRawUDPTransportPayload::Candidate::Host;
			transport->addCandidate(candidate);

			return true;
		}
	}*/

	foreach (const NetworkInterface& interface, interfaces) {
		if (!interface.isLoopback()) { // exclude loopback
			foreach (const HostAddress& addr, interface.getAddresses()) {
				int port = serverSocket->bindOnAvailablePort(addr);
				if (!port)
					continue;

				candidate.hostAddressPort = serverSocket->getLocalAddress();
				candidate.type = JingleRawUDPTransportPayload::Candidate::Host;
				transport->addCandidate(candidate);

				return true;
			}
		}
	}

	// else loopback for self sharing
	/*if (!interfaces.empty() && !interfaces.front().getAddresses().empty()) {
		int port = serverSocket->bindOnAvailablePort(interfaces.front().getAddresses().front());
		if (port) {
			candidate.hostAddressPort = serverSocket->getLocalAddress();
			candidate.type = JingleRawUDPTransportPayload::Candidate::Host;
			transport->addCandidate(candidate);
			return true;
		}
	}*/

	return false;
}

void ScreenSharing::handleSessionTerminateReceived(boost::optional<JinglePayload::Reason> reason)
{
	if (rtpSession)
		rtpSession->stop();

	if (reason.is_initialized() && reason.get().type == JinglePayload::Reason::Cancel) {
		onStateChange(ScreenSharing::Canceled);
	} else if (reason.is_initialized() && reason.get().type == JinglePayload::Reason::Success) {
		onStateChange(ScreenSharing::Finished);
	} else {
		onStateChange(ScreenSharing::Failed);
	}
	onFinished();
}

}