#include <poll.h>

bool send_command(Client *client, RemoteCommand cmd)
{
	pthread_mutex_lock(&client->mtx);
	bool ret = write_u32(client->fd, cmd);
	pthread_mutex_unlock(&client->mtx);
	return ret;
}

static bool handle_packets(Client *client) {
	while (client->state != CS_DISCONNECTED || ! interrupted) {
		struct pollfd pfd = {
			.fd = client->fd,
			.events = POLLIN,
			.revents = 0,
		};

		int pstate = poll(&pfd, 1, 0);

		if (pstate == -1) {
			perror("poll");
			break;
		}

		if (pstate == 0) {
			sched_yield();
			continue;
		}

		if (! (pfd.revents & POLLIN))
			return false;

		HostCommand command;
		if (! read_u32(client->fd, &command))
			break;

		CommandHandler *handler = NULL;

		if (command < HOST_COMMAND_COUNT)
			handler = &command_handlers[command];

		if (handler && handler->func) {
			bool good = client->state & handler->state_flags;
			if (! good)
				printf("Received %s command, but client is in invalid state: %d\n", handler->name, client->state);
			if (! handler->func(client, good))
				break;
		} else {
			printf("Received invalid command %d\n", command);
		}
	}

	return client->state == CS_DISCONNECTED || errno == EINTR;
}