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

#include "base/main.h"
#include "base/list.h"
#include "base/textFile.h"
#include "base/homeDirector.h"
#include "base/net_multiplayer.h"

#include "client/language.h"
#include "client/interface.h"
#include "client/screen.h"
#include "client/image.h"

#ifdef SUPPORT_NET_UNIX_UDP
#include <sys/socket.h>
#include <sys/select.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <unistd.h>
#include <fcntl.h>
#include <errno.h>

#include "net_unix/udp.h"
#include "net_unix/dns.h"

extern int errno;
#endif

#ifndef NO_SOUND
#include "audio/music.h"
#endif

#include "screen/screen_gameType.h"
#include "screen/screen_browser.h"

#include "widget/widget_image.h"
#include "widget/widget_label.h"
#include "widget/widget_button.h"
#include "widget/widget_select.h"

#define SERVER_TIMEOUT	200

static widget_image_t *image_backgorund;

static widget_button_t *button_play;
static widget_button_t *button_back;
static widget_button_t *button_refresh;

static widget_label_t *label_server;

static widget_select_t *select_server;

static server_t server_list;

/* function prototypes */
static int LoadServers ();
static int RefreshServers ();
static server_t *server_getcurr ();


void startScreenBrowser()
{
#ifndef NO_SOUND
	playMusic("menu", MUSIC_GROUP_BASE);
#endif

	LoadServers ();
}

void drawScreenBrowser()
{
	drawWidgetImage(image_backgorund);

	drawWidgetLabel(label_server);

	drawWidgetSelect(select_server);

	drawWidgetButton(button_play);
	drawWidgetButton(button_back);
	drawWidgetButton(button_refresh);
}

void eventScreenBrowser()
{
	eventWidgetSelect(select_server);

	eventWidgetButton(button_play);
	eventWidgetButton(button_back);
	eventWidgetButton(button_refresh);
}

void stopScreenBrowser()
{
	unsigned i = 0;
	server_t *server;

	//destroyListItem(select_server->list, free);
	//select_server->list = newList();
	removeAllFromWidgetSelect(select_server);

	while (1) {
		i = 0;
		for (server = server_list.next; server != &server_list; server = server->next) {
			server->next->prev = server->prev;
			server->prev->next = server->next;

			if (server->name)
				free (server->name);

			if (server->version)
				free (server->version);

			if (server->arena)
				free (server->arena);

			free (server);

			i ++;
			break;
		}

		if (!i)
			return;
	}
}

static void eventWidget(void *p)
{
	widget_button_t *button;
	
	button = (widget_button_t *)(p);

	if( button == button_play )
	{
		server_t *server = server_getcurr ();

		if (!server)
			return;

		/* server je offline */
		if (!server->state)
			return;

		struct in_addr srv;
		srv.s_addr = server->ip;

		char *address = (char *) inet_ntoa (srv);

		setSettingGameType (NET_GAME_TYPE_CLIENT);
		setSettingIP(address);
		setSettingPort(server->port);

		setScreen("world");
	}

	if( button == button_back )
	{
		setScreen("gameType");
	}

	if( button == button_refresh )
	{
		RefreshServers ();
	}
}

server_t *server_getcurr ()
{
	unsigned i = 0;
	server_t *server;

	for (server = server_list.next; server != &server_list; server = server->next) {
		if (i == select_server->select)
			return server;

		i ++;
	}

	return 0;
}

int server_getinfo (server_t *server)
{
#ifdef SUPPORT_NET_UNIX_UDP
	struct sockaddr_in srv;
	int mySocket;

	if ((mySocket = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) == -1) {
	    return -1;
	}

	// Lets make socket non-blocking
	int oldFlag = fcntl (mySocket, F_GETFL, 0);
	if (fcntl (mySocket, F_SETFL, oldFlag | O_NONBLOCK) == -1) {
		return -1;
	}

	struct timeval tv;

	fd_set myset;
	FD_ZERO(&myset);
	FD_SET(mySocket, &myset);

	tv.tv_sec = 2;
	tv.tv_usec = 0;

	srv.sin_family = AF_INET;
	srv.sin_port = htons (server->port);
	srv.sin_addr.s_addr = server->ip;

	if (connect (mySocket, (struct sockaddr *) &srv, sizeof (srv)) == -1) {
		if (errno != EINPROGRESS)
		    return -1;
	}

	int ret = select (mySocket+1, NULL, &myset, NULL, &tv);

	if (ret == -1)
		return 0;

	if (ret == 0)
		return -2;

	char request[8];
	memcpy (request, "status\n", 7);

	int r = send (mySocket, request, 7, 0);

	if (r == -1)
		return -2;

	FD_ZERO(&myset);
	FD_SET(mySocket, &myset);
	tv.tv_sec = 1;
	tv.tv_usec = 0;

	char *str = (char *) malloc (sizeof (char) * 257);

	if (!str)
		return -1;

	while (1) {
		int sel = select (mySocket + 1, &myset, NULL, NULL, &tv);

		if (sel == 0) {
			free (str);
			close (mySocket);
			return -2;
		}

		if (sel == -1) {
			free (str);
			close (mySocket);
			return -2;
		}
  
		if ((ret = recv(mySocket, str, 256, 0)) == -1) {
			free (str);
			close (mySocket);
			return -1;
		}
    
		if (ret > 0)
			break;
	}

	close (mySocket);

	server->ping = 1000 - (tv.tv_usec/1000);
#else
	return 0;
#endif

	/*
		["name: F\n"] >= svn82
		"version: svnX\n"
		"clients: N\n"
		"maxclients: M\n"
		"uptime: D\n"
	*/

	unsigned i = 0;

	while (i < ret) {
		if (str[i] == '\n')
			str[i] = '\0';

		i ++;
	}

	unsigned name = 0;

	if (strstr (str, "name: "))
		name = 1;

	int len = 0;

	if (name) {
		len = strlen (str);
		if (!strncmp (str, "name: ", 6)) {
			unsigned name_len = strlen (str+6);
			server->name = (char *) malloc (sizeof (char) * (name_len + 2));
	  
			if (!server->name)
				return 0;
	
			memcpy (server->name, str+6, name_len+1);
			server->name[name_len+1] = '\0';
		}
	}

	/* HACK */
	if (!name)
		len = -1;

	if (!strncmp (str+len+1, "version: ", 9)) {
		unsigned ver_len = strlen (str+len+10);
		server->version = (char *) malloc (sizeof (char) * (ver_len + 2));
	 
		if (!server->version)
			return 0;
	
		memcpy (server->version, str+len+10, ver_len+1);
		server->version[ver_len+1] = '\0';
	}

	len += strlen (str+len+1) + 1;


	if (!strncmp (str+len+1, "clients: ", 9))
		server->clients = atoi (str+len+10);

	len += strlen (str+len+1) + 1;


	if (!strncmp (str+len+1, "maxclients: ", 12))
		server->maxclients = atoi (str+len+13);

	len += strlen (str+len+1) + 1;
	

	if (!strncmp (str+len+1, "uptime: ", 8))
		server->uptime = atoi (str+len+9);

	len += strlen (str+len+1) + 1;
	

	if (!strncmp (str+len+1, "arena: ", 7)) {
		unsigned arena_len = strlen (str+len+9);
		server->arena = (char *) malloc (sizeof (char) * (arena_len + 2));
	  
		if (!server->arena)
			return 0;
	
		memcpy (server->arena, str+len+8, arena_len+1);
		server->arena[arena_len+1] = '\0';
	}


	free (str);

	server->state = 1;

	return 1;
}

static int LoadServers ()
// Get server list
{
	int s;
	char buf[1025];

	/* TODO: dodelat TCP makro */
#ifdef SUPPORT_NET_UNIX_UDP
	struct sockaddr_in server;
	char *master_server_ip;

	master_server_ip = getIPFormDNS(NET_MASTER_SERVER_DOMAIN);

	//printf("master_server_ip = %s\n", master_server_ip);

	if( master_server_ip == NULL ) // master server is down
	{
		return -1;
	}
	
	server.sin_family = AF_INET;
	server.sin_port = htons (NET_MASTER_PORT);
	server.sin_addr.s_addr = inet_addr (master_server_ip);
	
	free(master_server_ip);

	if ((s = socket (AF_INET, SOCK_STREAM, 0)) < 0) {
		close (s);
		return 0;
	}

	int oldFlag = fcntl (s, F_GETFL, 0);
	if (fcntl (s, F_SETFL, oldFlag | O_NONBLOCK) == -1) {
		return -1;
	}

	if (connect (s, (struct sockaddr *) &server, sizeof (server)) == -1) {
		if (errno != EINPROGRESS)
		    return -1;
	}

	struct timeval tv;
	// Mno�ina
	fd_set myset;
	// Mno�ina obsahuje n�hodn� data. Odstran�m je.
	FD_ZERO(&myset);
	// Zapln�n� mno�iny sokety
	FD_SET(s, &myset);
	// Vypln�m �asov� �daj (nap��klad na 3 minuty)
	tv.tv_sec = 3;// Po�et sekund
	tv.tv_usec = 0;// Po�et mikrosekund
	// Zavol�m select (V Linuxu mus�m m�t nastavenou prom�nnou max.)
	int ret = select (s+1, NULL, &myset, NULL, &tv);

	if (ret == -1)
		return 0;

	if (ret == 0)
		return -2;

	/* send request for server list */
	send (s, "l", 1, 0);

	// Mno�ina obsahuje n�hodn� data. Odstran�m je.
	FD_ZERO(&myset);
	// Zapln�n� mno�iny sokety
	FD_SET(s, &myset);

	ret = select (s+1, &myset, NULL, NULL, &tv);

	if (ret == -1)
		return 0;

	if (ret == 0)
		return -2;

	int len = recv (s, buf, 1024, 0);

	close (s);
#else
	return -1;
#endif

	if (len <= 0)
		return 0;

	unsigned i = 0;
	char list[256];

	struct in_addr srv;

	typedef struct {
		unsigned port;
		unsigned ip;
	} response_head;

	while (1) {
		response_head *header = (response_head *) ((char *) buf + i);

		server_t *ctx;
			
		// alloc and init context
		ctx = (server_t *) malloc (sizeof (server_t));
		
		ctx->ip = header->ip;
		ctx->port = header->port;
		ctx->name = 0;
		ctx->version = 0;
		ctx->arena = 0;

		int ret = server_getinfo (ctx);

		srv.s_addr = ctx->ip;

		char *address = (char *) inet_ntoa (srv);

		if (ret == 1) {
			sprintf (list, "%s (%s) - %s:%d - %s - %d/%d - %dms", ctx->name ? ctx->name : "Unknown", ctx->version, address, ctx->port,
				ctx->arena ? ctx->arena : "Unknown", ctx->clients, ctx->maxclients, ctx->ping);
		} else if (ret == -2)
			sprintf (list, "%s:%d - Timeout", address, ctx->port);
		else
			sprintf (list, "%s:%d - Offline", address, ctx->port);


		addToWidgetSelect(select_server, list);

		// add into list
		ctx->next = &server_list;
		ctx->prev = server_list.prev;
		ctx->prev->next = ctx;
		ctx->next->prev = ctx;

		i += sizeof (response_head) + 1;

		if (buf[i-1] != '\n' || (i+1) > len)
			break;
	}

	return 1;
}

static int RefreshServers ()
{
	destroyListItem(select_server->list, free);
	select_server->list = newList();

	struct in_addr srv;

	server_t *server;
	for (server = server_list.next; server != &server_list; server = server->next) {
		if (server->name)
			free (server->name);

		if (server->version)
			free (server->version);

		if (server->arena)
			free (server->arena);

		server->name = 0;
		server->version = 0;
		server->arena = 0;

		char list[256];
	
		int ret = server_getinfo (server);

		srv.s_addr = server->ip;

		char *address = (char *) inet_ntoa (srv);
	
		if (ret == 1) {
			sprintf (list, "%s (%s) - %s:%d - %s - %d/%d - %dms", server->name ? server->name : "Unknown", server->version, 
				address, server->port, server->arena ? server->arena : "Unknown", server->clients, server->maxclients, server->ping);
		} else if (ret == -2)
			sprintf (list, "%s:%d - Timeout", address, server->port);
		else
			sprintf (list, "%s:%d - Offline", address, server->port);

		addToWidgetSelect(select_server, list);

	}

	return 1;
}

void initScreenBrowser()
{
	SDL_Surface *image;

	image = getImage(IMAGE_GROUP_BASE, "screen_main");
	image_backgorund  = newWidgetImage(0, 0, image);

	button_back = newWidgetButton(getMyText("BACK"), 100, WINDOW_SIZE_Y-100, eventWidget);
	button_play = newWidgetButton(getMyText("PLAY"), WINDOW_SIZE_X-200, WINDOW_SIZE_Y-100, eventWidget);
	button_refresh = newWidgetButton(getMyText("REFRESH"), WINDOW_SIZE_X/2-50, WINDOW_SIZE_Y-100, eventWidget);

	label_server = newWidgetLabel("Server list", 50, 145, WIDGET_LABEL_LEFT);

	select_server = newWidgetSelect(label_server->x, label_server->y+40, eventWidget);

	registerScreen( newScreen("browser", startScreenBrowser, eventScreenBrowser,
		drawScreenBrowser, stopScreenBrowser) );

	server_list.next = &server_list;
	server_list.prev = &server_list;
}

void quitScreenBrowser()
{
	destroyWidgetImage(image_backgorund);

	destroyWidgetButton(button_play);
	destroyWidgetButton(button_back);
	destroyWidgetButton(button_refresh);
	destroyWidgetLabel(label_server);
	destroyWidgetSelect(select_server);
}