#include "../src/ipc.h"
#include <signal.h>

#include <sys/socket.h>
#include <sys/un.h>
#include <unistd.h>

#define PONGD_SERVICE_NAME "pong"
#define PONGD_VERBOSE

#define PRINTERR(ret,msg) {\
	fprintf(stderr, "error while %s: %s\n", msg, ret.error_message);\
}

int cpt = 0;

int verbosity = 1;

struct ipc_connection_info *srv = NULL;
struct ipc_connection_infos *clients = NULL;

void main_loop ()
{
	double base_timer = 0;
	double timer = base_timer;
	SECURE_DECLARATION (struct ipc_error, ret);

	clients = malloc (sizeof (struct ipc_connection_infos));
	memset (clients, 0, sizeof (struct ipc_connection_infos));

	SECURE_DECLARATION (struct ipc_event, event);
	event.type = IPC_EVENT_TYPE_NOT_SET;

	while (1) {
		// ipc_service_poll_event provides one event at a time
		// warning: event->m is free'ed if not NULL
		TEST_IPC_WAIT_EVENT_Q (ipc_wait_event (clients, srv, &event, &timer), EXIT_FAILURE);

		switch (event.type) {
		case IPC_EVENT_TYPE_CONNECTION:
			{
				cpt++;
				if (verbosity > 1) {
					printf ("connection: %d clients connected, new client is %d\n", cpt, (event.origin)->fd);
				}
			};
			break;
		case IPC_EVENT_TYPE_DISCONNECTION:
			{
				cpt--;
				if (verbosity > 1) {
					printf ("disconnection: %d clients remaining\n", cpt);
				}

				// free the ipc_client structure
				free (event.origin);
			};
			break;
		case IPC_EVENT_TYPE_MESSAGE:
			{
				struct ipc_message *m = event.m;
				if (verbosity > 1) {
					if (m->length > 0) {
						printf ("message received (type %d, user type %d, size %u bytes): %.*s\n",
							m->type, m->user_type, m->length, m->length, m->payload);
					} else {
						printf ("message with a 0-byte size :(\n");
					}
				}

				ret = ipc_write (event.origin, m);
				if (ret.error_code != IPC_ERROR_NONE) {
					PRINTERR (ret, "server write");
				}
				if (verbosity > 1) {
					printf ("message sent\n");
				}
			};
			break;
		case IPC_EVENT_TYPE_TIMER:{
				printf ("timer\n");

				timer = base_timer;
			};
			break;
		case IPC_EVENT_TYPE_ERROR:
			{
				cpt--;
				fprintf (stderr, "a problem happened with client %d (now disconnected)", (event.origin)->fd);
				fprintf (stderr, ", %d clients remaining\n", cpt);

				// free the ipc_client structure
				free (event.origin);
			};
			break;
		default:
			{
				fprintf (stderr, "there must be a problem, event not set\n");
			};
		}
	}

	// should never go there
	exit (EXIT_FAILURE);
}

void exit_program (int signal)
{
	printf ("Quitting, signal: %d\n", signal);

	// free remaining clients
	for (size_t i = 0; i < clients->size; i++) {
		struct ipc_connection_info *cli = clients->cinfos[i];
		if (cli != NULL) {
			free (cli);
		}
		clients->cinfos[i] = NULL;
	}

	ipc_connections_free (clients);
	free (clients);

	// the application will shut down, and close the service
	struct ipc_error ret = ipc_server_close (srv);
	if (ret.error_code != IPC_ERROR_NONE) {
		PRINTERR (ret, "server close");
	}
	free (srv);

	exit (EXIT_SUCCESS);
}

/*
 * service ping-pong: send back everything sent by the clients
 * stop the program on SIG{TERM,INT,ALRM,USR{1,2},HUP} signals
 */

int main (int argc, char *argv[], char **env)
{
	argc = argc;		// warnings
	argv = argv;		// warnings

	if (argc > 1) {
		verbosity = atoi(argv[1]);
	}

	printf ("pid = %d\n", getpid ());

	srv = malloc (sizeof (struct ipc_connection_info));
	if (srv == NULL) {
		exit (EXIT_FAILURE);
	}
	memset (srv, 0, sizeof (struct ipc_connection_info));
	srv->type = '\0';
	srv->index = 0;
	srv->version = 0;
	srv->fd = 0;
	srv->spath = NULL;

	struct ipc_error ret = ipc_server_init (env, srv, PONGD_SERVICE_NAME);
	if (ret.error_code != IPC_ERROR_NONE) {
		PRINTERR (ret, "server init");
		return EXIT_FAILURE;
	}
	printf ("Listening on %s.\n", srv->spath);

	printf ("MAIN: server created\n");

	signal (SIGHUP, exit_program);
	signal (SIGALRM, exit_program);
	signal (SIGUSR1, exit_program);
	signal (SIGUSR2, exit_program);
	signal (SIGTERM, exit_program);
	signal (SIGINT, exit_program);

	// the service will loop until the end of time, or a signal
	main_loop ();

	// main_loop should not return
	return EXIT_FAILURE;
}