#include <assert.h>
#include <stdio.h>
#include <stdlib.h>

#include <fcntl.h>
#include <sys/types.h>
#include <sys/time.h>

#ifndef __WIN32__
#include <sys/ioctl.h>
#include <sys/socket.h>
#include <sys/select.h>
#else
#include <io.h>
#include <winsock2.h>
#endif

#include "main.h"
#include "list.h"
#include "tux.h"
#include "proto.h"

#include "client.h"
#include "language.h"

#include "screen_analyze.h"
#include "screen_world.h"

#include "udp.h"

static int protocolType;

static sock_udp_t *sock_server_udp;

static list_t *clientBuffer;
static my_time_t lastPing;
static my_time_t lastPingServerAlive;

typedef struct proto_cmd_client_struct
{
	char *name;
	int len;
	void (*fce_proto)(char *msg);
} proto_cmd_client_t;

static proto_cmd_client_t proto_cmd_list[] =
{
	{ .name = "error",	.len = 5,	.fce_proto = proto_recv_error_client },
	{ .name = "init",	.len = 4,	.fce_proto = proto_recv_init_client },
	{ .name = "event",	.len = 5,	.fce_proto = proto_recv_event_client },
	{ .name = "newtux",	.len = 6,	.fce_proto = proto_recv_newtux_client },
	{ .name = "del",	.len = 3,	.fce_proto = proto_recv_del_client },
	{ .name = "additem",	.len = 7,	.fce_proto = proto_recv_additem_client },
	{ .name = "shot",	.len = 4,	.fce_proto = proto_recv_shot_client },
	{ .name = "kill",	.len = 4,	.fce_proto = proto_recv_kill_client },
	{ .name = "module",	.len = 6,	.fce_proto = proto_recv_module_client },
	{ .name = "chat",	.len = 4,	.fce_proto = proto_recv_chat_client },
	{ .name = "ping",	.len = 4,	.fce_proto = proto_recv_ping_client },
	{ .name = "end",	.len = 3,	.fce_proto = proto_recv_end_client },
	{ .name = "",		.len = 0,	.fce_proto = NULL },
};

static proto_cmd_client_t* findCmdProto(char *msg)
{
	int len;
	int i;

	len = strlen(msg);

	for( i = 0 ; proto_cmd_list[i].len != 0 ; i++ )
	{
		proto_cmd_client_t *thisCmd;
		
		thisCmd = &proto_cmd_list[i];

		if( len >= thisCmd->len && strncmp(msg, thisCmd->name, thisCmd->len) == 0 )
		{
			return thisCmd;
		}
	}

	return NULL;
}

static void initClient()
{
	char name[STR_NAME_SIZE];

	clientBuffer = newList();
	lastPing = getMyTime();
	lastPingServerAlive = getMyTime();

	getSettingNameRight(name);
	proto_send_hello_client(name);
}

int initUdpClient(char *ip, int port, int proto)
{
	sock_server_udp = connectUdpSocket(ip, port, proto);

	if( sock_server_udp == NULL )
	{
		return -1;
	}

	printf("connect UDP %s %d\n", ip, port);

	protocolType = NET_PROTOCOL_TYPE_UDP;
	initClient();

	return 0;
}

void sendServer(char *msg)
{
	int ret;

	assert( msg != NULL );

#ifndef PUBLIC_SERVER
	if( isParamFlag("--send") )
	{
		printf("send -> %s", msg);
	}
#endif

	ret = writeUdpSocket(sock_server_udp, sock_server_udp, msg, strlen(msg));
}

static int eventServerSelect()
{
	char buffer[STR_PROTO_SIZE];
	int ret;

	memset(buffer, 0, STR_PROTO_SIZE);

	ret = readUdpSocket(sock_server_udp, sock_server_udp, buffer, STR_PROTO_SIZE-1);

	if( ret < 0 )
	{
		return ret;
	}

	addList(clientBuffer, strdup(buffer) );
	
	return ret;
}

static void eventServerBuffer()
{
	proto_cmd_client_t *protoCmd;
	char *line;
	int i;

	/* obsluha udalosti od servera */
	
	assert( clientBuffer != NULL );

	for( i = 0 ; i < clientBuffer->count ; i++ )
	{
		line = (char *)clientBuffer->list[i];
		protoCmd = findCmdProto(line);

#ifndef PUBLIC_SERVER
		if( isParamFlag("--recv") )
		{
			printf("recv -> %s", line);
		}
#endif
		if( protoCmd != NULL )
		{
			protoCmd->fce_proto(line);
			lastPingServerAlive = getMyTime();
		}
	}

	destroyListItem(clientBuffer, free);
	clientBuffer = newList();
}

static void eventPingServer()
{
	my_time_t currentTime;

 	currentTime = getMyTime();

	if( currentTime - lastPing > CLIENT_TIMEOUT )
	{
		proto_send_ping_client();
		lastPing = getMyTime();
	}
}

static bool_t isServerAlive()
{
	my_time_t currentTime;

 	currentTime = getMyTime();

	if( currentTime - lastPingServerAlive > SERVER_TIMEOUT_ALIVE )
	{
		return FALSE;
	}

	return TRUE;
}

static void selectClientUdpSocket()
{
	fd_set readfds;
	struct timeval tv;
	int max_fd;
	bool_t isNext;

	if( isServerAlive() == FALSE )
	{
		fprintf(stderr, "Server does not respond!\n");
		setMsgToAnalyze(getMyText("ERROR_SERVER_DONT_ALIVE"));
		setWorldEnd();
		return;
	}

	do{
		isNext = FALSE;

		tv.tv_sec = 0;
		tv.tv_usec = 0;
		
		FD_ZERO(&readfds);
		FD_SET(sock_server_udp->sock, &readfds);
		max_fd = sock_server_udp->sock;
	
		select(max_fd+1, &readfds, (fd_set *)NULL, (fd_set *)NULL, &tv);
	
		if( FD_ISSET(sock_server_udp->sock, &readfds) )
		{
			eventServerSelect();
			isNext = TRUE;
		}

	}while( isNext == TRUE );
}

void eventClient()
{
	eventPingServer();
	selectClientUdpSocket();
	eventServerBuffer();
}

static void quitClient()
{
	proto_send_end_client();
	assert( clientBuffer != NULL );
	destroyListItem(clientBuffer, free);
}

void quitUdpClient()
{
	quitClient();

	assert( sock_server_udp != NULL );
	closeUdpSocket(sock_server_udp);

	printf("quit UDP conenct\n");
}