#include <cec/CECHandler.h>
#include <utils/Logger.h>

#include <algorithm>

#include <libcec/cecloader.h>

#include <QJsonArray>
#include <QJsonDocument>
#include <QJsonObject>

/* Enable to turn on detailed CEC logs */
// #define VERBOSE_CEC

CECHandler::CECHandler()
{
	qRegisterMetaType<CECEvent>("CECEvent");

	 _logger = Logger::getInstance("CEC");

	_cecCallbacks            = getCallbacks();
	_cecConfig               = getConfig();
	_cecConfig.callbacks     = &_cecCallbacks;
	_cecConfig.callbackParam = this;
}

CECHandler::~CECHandler()
{
	stop();
}

bool CECHandler::start()
{
	if (_cecAdapter)
		return true;

	Info(_logger, "Starting CEC handler");

	_cecAdapter = LibCecInitialise(&_cecConfig);
	if(!_cecAdapter)
	{
		Error(_logger, "Failed loading libcec.so");
		return false;
	}

	auto adapters = getAdapters();
	if (adapters.isEmpty())
	{
		Error(_logger, "Failed to find CEC adapter");
		UnloadLibCec(_cecAdapter);
		_cecAdapter = nullptr;

		return false;
	}

	Info(_logger, "Auto detecting CEC adapter");
	bool opened = false;
	for (const auto & adapter : adapters)
	{
		printAdapter(adapter);

		if (!opened && openAdapter(adapter))
		{
			Info(_logger, "CEC Handler initialized with adapter : %s", adapter.strComName);

			opened = true;
		}
	}

	if (!opened)
	{
		UnloadLibCec(_cecAdapter);
		_cecAdapter = nullptr;
	}

	return opened;
}

void CECHandler::stop()
{
	if (_cecAdapter)
	{
		Info(_logger, "Stopping CEC handler");

		_cecAdapter->Close();
		UnloadLibCec(_cecAdapter);
		_cecAdapter = nullptr;
	}
}

CECConfig CECHandler::getConfig() const
{
	CECConfig configuration;

	const std::string name("HyperionCEC");
	name.copy(configuration.strDeviceName, std::min(name.size(), sizeof(configuration.strDeviceName)));

	configuration.deviceTypes.Add(CEC::CEC_DEVICE_TYPE_RECORDING_DEVICE);
	configuration.clientVersion = CEC::LIBCEC_VERSION_CURRENT;

	return configuration;
}

CECCallbacks CECHandler::getCallbacks() const
{
	CECCallbacks callbacks;

	callbacks.sourceActivated      = onCecSourceActivated;
	callbacks.commandReceived      = onCecCommandReceived;
	callbacks.alert                = onCecAlert;
	callbacks.logMessage           = onCecLogMessage;
	callbacks.keyPress             = onCecKeyPress;
	callbacks.configurationChanged = onCecConfigurationChanged;
	callbacks.menuStateChanged     = onCecMenuStateChanged;

	return callbacks;
}

QVector<CECAdapterDescriptor> CECHandler::getAdapters() const
{
	if (!_cecAdapter)
		return {};

	QVector<CECAdapterDescriptor> descriptors(16);
	int8_t size = _cecAdapter->DetectAdapters(descriptors.data(), descriptors.size(), nullptr, true /*quickscan*/);
	descriptors.resize(size);

	return descriptors;
}

bool CECHandler::openAdapter(const CECAdapterDescriptor & descriptor)
{
	if (!_cecAdapter)
		return false;

	if(!_cecAdapter->Open(descriptor.strComName))
	{
		Error(_logger, QString("Failed to open the CEC adaper on port %1")
			.arg(descriptor.strComName)
				.toLocal8Bit());

		return false;
	}
	return true;
}

void CECHandler::printAdapter(const CECAdapterDescriptor & descriptor) const
{
	Info(_logger, QString("CEC Adapter:").toLocal8Bit());
	Info(_logger, QString("\tName   : %1").arg(descriptor.strComName).toLocal8Bit());
	Info(_logger, QString("\tPath   : %1").arg(descriptor.strComPath).toLocal8Bit());
}

QString CECHandler::scan() const
{
	if (!_cecAdapter)
		return {};

	Info(_logger, "Starting CEC scan");

	QJsonArray devices;
	CECLogicalAddresses addresses = _cecAdapter->GetActiveDevices();
	for (int address = CEC::CECDEVICE_TV; address <= CEC::CECDEVICE_BROADCAST; ++address)
	{
		if (addresses[address])
		{
			CECLogicalAddress logicalAddress = (CECLogicalAddress)address;

			QJsonObject device;
			CECVendorId vendor = (CECVendorId)_cecAdapter->GetDeviceVendorId(logicalAddress);
			CECPowerStatus power = _cecAdapter->GetDevicePowerStatus(logicalAddress);

			device["name"    ] = _cecAdapter->GetDeviceOSDName(logicalAddress).c_str();
			device["vendor"  ] = _cecAdapter->ToString(vendor);
			device["address" ] = _cecAdapter->ToString(logicalAddress);
			device["power"   ] = _cecAdapter->ToString(power);

			devices << device;

			Info(_logger, QString("\tCECDevice: %1 / %2 / %3 / %4")
				.arg(device["name"].toString())
				.arg(device["vendor"].toString())
				.arg(device["address"].toString())
				.arg(device["power"].toString())
				.toLocal8Bit());
		}
	}

	return QJsonDocument(devices).toJson(QJsonDocument::Compact);
}

void CECHandler::onCecLogMessage(void * context, const CECLogMessage * message)
{
#ifdef VERBOSE_CEC
	CECHandler * handler = qobject_cast<CECHandler*>(static_cast<QObject*>(context));
	if (!handler)
		return;

	switch (message->level)
	{
	case CEC::CEC_LOG_ERROR:
		Error(handler->_logger, QString("%1")
			.arg(message->message)
				.toLocal8Bit());
		break;
	case CEC::CEC_LOG_WARNING:
		Warning(handler->_logger, QString("%1")
			.arg(message->message)
				.toLocal8Bit());
		break;
	case CEC::CEC_LOG_TRAFFIC:
	case CEC::CEC_LOG_NOTICE:
		Info(handler->_logger, QString("%1")
			.arg(message->message)
				.toLocal8Bit());
		break;
	case CEC::CEC_LOG_DEBUG:
		Debug(handler->_logger, QString("%1")
			.arg(message->message)
				.toLocal8Bit());
		break;
	default:
		break;
	}
#endif
}

void CECHandler::onCecKeyPress(void * context, const CECKeyPress * key)
{
#ifdef VERBOSE_CEC
	CECHandler * handler = qobject_cast<CECHandler*>(static_cast<QObject*>(context));
	if (!handler)
		return;

	CECAdapter * adapter = handler->_cecAdapter;

	Debug(handler->_logger, QString("CECHandler::onCecKeyPress: %1")
		.arg(adapter->ToString(key->keycode))
			.toLocal8Bit());
#endif
}

void CECHandler::onCecAlert(void * context, const CECAlert alert, const CECParameter data)
{
#ifdef VERBOSE_CEC
	CECHandler * handler = qobject_cast<CECHandler*>(static_cast<QObject*>(context));
	if (!handler)
		return;

	Error(handler->_logger, QString("CECHandler::onCecAlert: %1")
		.arg(alert)
			.toLocal8Bit());
#endif
}

void CECHandler::onCecConfigurationChanged(void * context, const CECConfig * configuration)
{
#ifdef VERBOSE_CEC
	CECHandler * handler = qobject_cast<CECHandler*>(static_cast<QObject*>(context));
	if (!handler)
		return;

	Debug(handler->_logger, QString("CECHandler::onCecConfigurationChanged: %1")
		.arg(configuration->strDeviceName)
			.toLocal8Bit());
#endif
}

int CECHandler::onCecMenuStateChanged(void * context, const CECMenuState state)
{
#ifdef VERBOSE_CEC
	CECHandler * handler = qobject_cast<CECHandler*>(static_cast<QObject*>(context));
	if (!handler)
		return 0;

	CECAdapter * adapter = handler->_cecAdapter;

	Debug(handler->_logger, QString("CECHandler::onCecMenuStateChanged: %1")
		.arg(adapter->ToString(state))
			.toLocal8Bit());
#endif
	return 0;
}

void CECHandler::onCecCommandReceived(void * context, const CECCommand * command)
{
	CECHandler * handler = qobject_cast<CECHandler*>(static_cast<QObject*>(context));
	if (!handler)
		return;

	CECAdapter * adapter = handler->_cecAdapter;

#ifdef VERBOSE_CEC
	Debug(handler->_logger, QString("CECHandler::onCecCommandReceived: %1 (%2 > %3)")
		.arg(adapter->ToString(command->opcode))
		.arg(adapter->ToString(command->initiator))
		.arg(adapter->ToString(command->destination))
			.toLocal8Bit());
#endif
	/* We do NOT check sender */
	// if (address == CEC::CECDEVICE_TV)
	{
		if (command->opcode == CEC::CEC_OPCODE_SET_STREAM_PATH)
		{
			Info(handler->_logger, QString("CEC source activated: %1")
				.arg(adapter->ToString(command->initiator))
					.toLocal8Bit());
			emit handler->cecEvent(CECEvent::On);
		}
		if (command->opcode == CEC::CEC_OPCODE_STANDBY)
		{
			Info(handler->_logger, QString("CEC source deactivated: %1")
				.arg(adapter->ToString(command->initiator))
					.toLocal8Bit());
			emit handler->cecEvent(CECEvent::Off);
		}
	}
}

void CECHandler::onCecSourceActivated(void * context, const CECLogicalAddress address, const uint8_t activated)
{
	/* We use CECHandler::onCecCommandReceived for
	 * source activated/deactivated notifications. */

#ifdef VERBOSE_CEC
	CECHandler * handler = qobject_cast<CECHandler*>(static_cast<QObject*>(context));
	if (!handler)
		return;

	CECAdapter * adapter = handler->_cecAdapter;

	Debug(handler->_logger, QString("CEC source %1 : %2")
		.arg(activated ? "activated" : "deactivated")
		.arg(adapter->ToString(address))
			.toLocal8Bit());
#endif
}