#include <mdns/MdnsProvider.h>
#include <mdns/MdnsServiceRegister.h>

//Qt includes
#include <QHostInfo>
#include <QThread>

// Utility includes
#include <utils/Logger.h>
#include <HyperionConfig.h>
#include <hyperion/AuthManager.h>

namespace {
	const bool verboseProvider = false;
} //End of constants

MdnsProvider::MdnsProvider(QObject* parent)
	: QObject(parent)
	, _log(Logger::getInstance("MDNS"))
	, _server(nullptr)
	, _hostname(nullptr)
{
}

void MdnsProvider::init()
{
	_server = new QMdnsEngine::Server();
	_hostname = new QMdnsEngine::Hostname(_server);

	connect(_hostname, &QMdnsEngine::Hostname::hostnameChanged, this, &MdnsProvider::onHostnameChanged);
	DebugIf(verboseProvider, _log, "Hostname [%s], isRegistered [%d]", _hostname->hostname().constData(), _hostname->isRegistered());
}

MdnsProvider::~MdnsProvider()
{
	qDeleteAll(_providedServiceTypes);

	_hostname->deleteLater();
	_server->deleteLater();
}

void MdnsProvider::publishService(const QString& serviceType, quint16 servicePort, const QByteArray& serviceName)
{
	QMdnsEngine::Provider* provider(nullptr);

	QByteArray type = MdnsServiceRegister::getServiceType(serviceType);
	if (!type.isEmpty())
	{
		DebugIf(verboseProvider, _log, "Publish new mDNS serviceType [%s], Thread: %s", type.constData(), QSTRING_CSTR(QThread::currentThread()->objectName()));

		if (!_providedServiceTypes.contains(type))
		{
			provider = new QMdnsEngine::Provider(_server, _hostname);
			_providedServiceTypes.insert(type, provider);
		}
		else
		{
			provider = _providedServiceTypes[type];
		}

		QMdnsEngine::Service service;
		service.setType(type);
		service.setPort(servicePort);

		QByteArray name(QHostInfo::localHostName().toUtf8());
		if (!serviceName.isEmpty())
		{
			name.prepend(serviceName + "@");
		}
		service.setName(name);

		QByteArray id = AuthManager::getInstance()->getID().toUtf8();
		const QMap<QByteArray, QByteArray> attributes = { {"id", id}, {"version", HYPERION_VERSION} };
		service.setAttributes(attributes);

		DebugIf(verboseProvider, _log, "[%s], Name: [%s], Port: [%u] ", service.type().constData(), service.name().constData(), service.port());

		provider->update(service);
	}
}

void MdnsProvider::onHostnameChanged(const QByteArray& hostname)
{
	DebugIf(verboseProvider, _log, "mDNS-hostname changed to hostname [%s]", hostname.constData());
}