#include "global.h"
#include "HidDevice.h"
#include "RageLog.h"

HidDevice::HidDevice(int vid, const std::vector<int> pids, int interfaceNum, bool autoReconnection, bool nonBlockingRead) :
	vid{ vid },
	pids{ pids },
	interfaceNum{ interfaceNum },
	autoReconnection{ autoReconnection },
	nonBlockingRead{ nonBlockingRead }
{
	bool result = TryConnect();

	if (!result)
	{
		LOG->Warn("HidDevice with vendor_id %04x and pids %s - %d not found", vid, GetPidsString(pids).c_str(), interfaceNum);
	}
	else
		foundOnce = true;
}

HidDevice::HidDevice(int vid, int pid, int interfaceNum, bool autoReconnection, bool nonBlockingRead) :
	HidDevice(vid, make_pids(pid, 1), interfaceNum, autoReconnection, nonBlockingRead)
{
}

HidDevice::~HidDevice()
{
	Close();
	hid_exit();
}

void HidDevice::Close()
{
	hid_close(handle);
	handle = nullptr;

	foundDeviceInfo.vid = 0;
	foundDeviceInfo.pid = 0;
	foundDeviceInfo.interfaceNum = 0;
	foundDeviceInfo.path = nullptr;
}

bool HidDevice::Open(const char* path)
{
	handle = hid_open_path(path);

	if(nonBlockingRead)
		hid_set_nonblocking(handle, 1);

	if(handle)
		LOG->Info("HidDevice %04x:%04x: %d opened by path %s", foundDeviceInfo.vid, foundDeviceInfo.pid, foundDeviceInfo.interfaceNum, foundDeviceInfo.path);

	return handle != nullptr;
}

bool HidDevice::TryConnect()
{
	GetDeviceInfo(vid, pids, interfaceNum, &foundDeviceInfo);

	if (foundDeviceInfo.path == nullptr)
		return false;

	return Open(foundDeviceInfo.path);
}

bool HidDevice::CheckConnection()
{
	if (IsConnected())
		return true;

	if (!autoReconnection || !foundOnce)
		return false;

	return TryConnect();
}

const wchar_t* HidDevice::GetError()
{
	return hid_read_error(handle);
}

bool HidDevice::IsConnected() {
	return handle != nullptr;
}

bool HidDevice::FoundOnce()
{
	return foundOnce;
}

const RString HidDevice::GetPidsString(const std::vector<int> pids)
{
	RString pidsString;
	char pid[5] = { 0 };
	size_t size = pids.size();

	for (size_t i = 0; i < size; ++i)
	{
		sprintf(pid, "%04X", pids[i]);
		pidsString += pid;

		if (i != size - 1)
			pidsString += ",";
	}

	return pidsString;
}

void HidDevice::GetDeviceInfo(int vid, const std::vector<int> pids, int interfaceNumber, HidDeviceInfo* device_info)
{
	bool found{ false };
	struct hid_device_info* devs, * cur_dev;
	size_t size = pids.size();

	devs = hid_enumerate(vid, 0);
	cur_dev = devs;

	if (devs && cur_dev)
	{
		// Look for the desired devices by iterating connected ones
		while (cur_dev)
		{
			for (int i = 0; i < size; i++)
			{
				if (cur_dev->vendor_id == vid &&
					cur_dev->product_id == pids[i])
				{
					if (interfaceNumber == -1)
					{
						found = true;
						break;
					}
					else
					{
						if (cur_dev->interface_number == interfaceNumber)
						{
							found = true;
							break;
						}
					}
				}
			}

			if (found)
				break;

			cur_dev = cur_dev->next;
		}
	}

	if (found && cur_dev != nullptr)
	{
		device_info->vid = cur_dev->vendor_id;
		device_info->pid = cur_dev->product_id;
		device_info->interfaceNum = cur_dev->interface_number;
		device_info->path = cur_dev->path;
	}
}

int HidDevice::Read(unsigned char* data, size_t length)
{
	if (!CheckConnection())
		return NotConnected;

	int result = hid_read(handle, data, length);

	if (result == -1)
	{
		LOG->Warn("HidDevice %04x:%04x: %d read failed. Fail reason %ls", foundDeviceInfo.vid, foundDeviceInfo.pid, foundDeviceInfo.interfaceNum, GetError());
		Close();
	}

	return result;
}

HidResults HidDevice::Write(const unsigned char* data, size_t length)
{
	if (!CheckConnection())
		return NotConnected;

	int result = hid_write(handle, data, length);

	if (result != length)
	{
		LOG->Warn("HidDevice %04x:%04x: %d write failed. Fail reason %ls", foundDeviceInfo.vid, foundDeviceInfo.pid, foundDeviceInfo.interfaceNum, GetError());
		Close();
		return OperationFailed;
	}

	return Success;
}