#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include "../src/ipc.h"

#define MSG "coucou"
#define SERVICE_NAME "pong"

#define PRINTERR(ret,msg) {\
	const char * err = ipc_errors_get (ret.error_code);\
	fprintf(stderr, "error while %s: %s\n", msg, err);\
}

void chomp (char *str, ssize_t len)
{
	if (str[len - 1] == '\n') {
		str[len - 1] = '\0';
	}
	if (str[len - 2] == '\n') {
		str[len - 2] = '\0';
	}
}

struct ipc_ctx *ctx = NULL;

void non_interactive (int verbosity, size_t nb_msg, char *msg_str)
{
	SECURE_DECLARATION (struct ipc_message, m);

	// init service
	TEST_IPC_QUIT_ON_ERROR (ipc_connection (ctx, SERVICE_NAME, NULL), EXIT_FAILURE);

	if (verbosity > 1) {
		printf ("msg to send (%ld): %.*s\n", (ssize_t) strlen (MSG) + 1, (int)strlen (MSG), MSG);
	}

	for (size_t i = 0 ; i < nb_msg ; i++) {
		TEST_IPC_QUIT_ON_ERROR (ipc_message_format_data (&m, 42, msg_str, (ssize_t) strlen (msg_str) + 1), EXIT_FAILURE);
		TEST_IPC_QUIT_ON_ERROR (ipc_write_fd (ctx->pollfd[0].fd, &m), EXIT_FAILURE);
		if (verbosity > 1) {
			printf ("msg written (type: %u): %s\n", m.user_type, m.payload);
		}
		ipc_message_empty (&m);
		TEST_IPC_QUIT_ON_ERROR (ipc_read (ctx, 0 /* read from the only valid index */, &m), EXIT_FAILURE);

		if (verbosity > 1) {
			printf ("msg recv (type: %u): %s\n", m.user_type, m.payload);
		}
		ipc_message_empty (&m);
	}

	TEST_IPC_QUIT_ON_ERROR (ipc_close_all (ctx), EXIT_FAILURE);
}

void interactive ()
{
	// init service
	TEST_IPC_QUIT_ON_ERROR (ipc_connection (ctx, SERVICE_NAME, NULL), EXIT_FAILURE);

	SECURE_DECLARATION (struct ipc_error, ret);
	SECURE_DECLARATION (struct ipc_event, event);

	ipc_add_fd (ctx, 0);	// add STDIN

	ipc_ctx_print (ctx);

	int timer = 10000;

	while (1) {
		printf ("msg to send: ");
		fflush (stdout);

		TEST_IPC_WAIT_EVENT_Q (ipc_wait_event (ctx, &event, &timer), EXIT_FAILURE);
		switch (event.type) {
		case IPC_EVENT_TYPE_EXTRA_SOCKET:
			{
				// structure not read, should read the message here
				ssize_t len;
				char buf[4096];
				memset (buf, 0, 4096);

				len = read (event.origin, buf, 4096);

				buf[len - 1] = '\0';
				chomp (buf, len);

#if 0
				printf ("\n");
				printf ("message to send: %.*s\n", (int)len, buf);
#endif

				// in case we want to quit the program
				if (len == 0 || strncmp (buf, "quit", 4) == 0 || strncmp (buf, "exit", 4) == 0) {

					struct ipc_error ret = ipc_close_all (ctx);
					if (ret.error_code != IPC_ERROR_NONE) {
						fprintf (stderr, "%s", ret.error_message);
						exit (EXIT_FAILURE);
					}

					ipc_ctx_free (ctx);

					exit (EXIT_SUCCESS);
				}
				// send the message read on STDIN
				struct ipc_message *m = NULL;
				SECURE_BUFFER_HEAP_ALLOCATION (m, sizeof (struct ipc_message),, return; );

				struct ipc_error ret = ipc_message_format_data (m, 42, buf, len);
				if (ret.error_code != IPC_ERROR_NONE) {
					fprintf (stderr, "%s", ret.error_message);
					exit (EXIT_FAILURE);
				}
#if 1
				printf ("\n");
				printf ("right before sending a message\n");
#endif
				ret = ipc_write_fd (ctx->pollfd[1].fd, m);
				if (ret.error_code != IPC_ERROR_NONE) {
					fprintf (stderr, "%s", ret.error_message);
					exit (EXIT_FAILURE);
				}
#if 1
				printf ("right after sending a message\n");
#endif

				ipc_message_empty (m);
				free (m);
			}
			break;
		case IPC_EVENT_TYPE_MESSAGE:
			{
				struct ipc_message *m = event.m;
				printf ("\rmsg recv: %.*s\n", m->length, m->payload);
			};
			break;
		case IPC_EVENT_TYPE_DISCONNECTION:
		case IPC_EVENT_TYPE_NOT_SET:
		case IPC_EVENT_TYPE_CONNECTION:
		case IPC_EVENT_TYPE_ERROR:
		default:
			fprintf (stderr, "should not happen, event type %d\n", event.type);
		}
	}
}

int main (int argc, char *argv[])
{
	printf("usage: %s [verbosity #messages message]\n", argv[0]);

	ctx = malloc (sizeof (struct ipc_ctx));
	memset (ctx, 0, sizeof (struct ipc_ctx));

	if (argc == 4)
		non_interactive (atoi(argv[1]), (size_t) atoi(argv[2]), argv[3]);
	else
		interactive ();

	return EXIT_SUCCESS;
}