#include <stdio.h>
#include <math.h>
#include <baselib/system.h>
#include <baselib/keys.h>
#include <baselib/mouse.h>
#include <baselib/network.h>

#include "interface.h"
#include "ui.h"
#include "versions.h"

#include "game/mapres_image.h"
#include "game/mapres_tilemap.h"
#include "game/mapres.h"

using namespace baselib;

/********************************************************
 MENU                                                  
*********************************************************/

struct pretty_font
{
    char m_CharStartTable[256];
    char m_CharEndTable[256];
    int font_texture;
};  

extern pretty_font *current_font;
void gfx_pretty_text(float x, float y, float size, const char *text);
float gfx_pretty_text_width(float size, const char *text);

void draw_scrolling_background(int id, float w, float h, float t)
{
	float tx = w/256.0f;
	float ty = h/256.0f;

	float start_x = fmod(t, 1.0f);
	float start_y = 1.0f - fmod(t*0.8f, 1.0f);

    gfx_blend_normal();
    gfx_texture_set(id);
    gfx_quads_begin();
    gfx_quads_setcolor(1,1,1,1);
	gfx_quads_setsubset(
		start_x, // startx
		start_y, // starty
		start_x+tx, // endx
		start_y+ty); // endy								
    gfx_quads_drawTL(0.0f,0.0f,w,h);
    gfx_quads_end();
}

int background_texture;
int not_empty_item_texture;
int empty_item_texture;
int active_item_texture;
int selected_item_texture;
int join_button_texture;
int join_button_hot_texture;
int join_button_active_texture;
int join_button_grey_texture;
int quit_button_texture;
int quit_button_hot_texture;
int quit_button_active_texture;
int up_button_texture;
int up_button_active_texture;
int down_button_texture;
int down_button_active_texture;
int teewars_banner_texture;
int scroll_indicator_texture;
int connect_localhost_texture;
int refresh_button_texture;
int refresh_button_hot_texture;
int refresh_button_active_texture;
int input_box_texture;

int music_menu;
int music_menu_id = -1;

struct button_textures
{
	int *normal_texture;
	int *hot_texture;
	int *active_texture;
};

button_textures connect_localhost_button = { &connect_localhost_texture, &connect_localhost_texture, &connect_localhost_texture };
button_textures join_button = { &join_button_texture, &join_button_hot_texture, &join_button_active_texture };
button_textures quit_button = { &quit_button_texture, &quit_button_hot_texture, &quit_button_active_texture };
button_textures scroll_up_button = { &up_button_texture, &up_button_texture, &up_button_active_texture };
button_textures scroll_down_button = { &down_button_texture, &down_button_texture, &down_button_active_texture };
button_textures list_item_button = { &not_empty_item_texture, &active_item_texture, &active_item_texture };
button_textures selected_item_button = { &selected_item_texture, &selected_item_texture, &selected_item_texture };
button_textures refresh_button = { &refresh_button_texture, &refresh_button_hot_texture, &refresh_button_active_texture };

void draw_menu_button(void *id, const char *text, int checked, float x, float y, float w, float h, void *extra)
{
	button_textures *tx = (button_textures *)extra;

    gfx_blend_normal();
	
	if (ui_active_item() == id && ui_hot_item() == id)
		gfx_texture_set(*tx->active_texture);
	else if (ui_hot_item() == id)
		gfx_texture_set(*tx->hot_texture);
	else
		gfx_texture_set(*tx->normal_texture);
	
    gfx_quads_begin();

    gfx_quads_setcolor(1,1,1,1);

    gfx_quads_drawTL(x,y,w,h);
    gfx_quads_end();

    gfx_texture_set(current_font->font_texture);
    gfx_pretty_text(x + 4, y - 3.5f, 18.f, text);
}

void draw_image_button(void *id, const char *text, int checked, float x, float y, float w, float h, void *extra)
{
	ui_do_image(*(int *)id, x, y, w, h);
}

struct server_info
{
	int version;
    int players;
	int max_players;
	netaddr4 address;
	char name[129];
	char map[65];
};

struct server_list
{   
    server_info infos[10];
	int active_count, info_count;
	int scroll_index;
	int selected_index;
};
#include <string.h>

int ui_do_edit_box(void *id, float x, float y, float w, float h, char *str, int str_size)
{
    int inside = ui_mouse_inside(x, y, w, h);
	int r = 0;

	if(inside)
	{
		ui_set_hot_item(id);

		if(ui_mouse_button(0))
			ui_set_active_item(id);
	}

	if (ui_active_item() == id)
	{
		char c = keys::last_char();
		int k = keys::last_key();
		int len = strlen(str);
	
		if (c >= 32 && c < 128)
		{
			if (len < str_size - 1)
			{
				str[len] = c;
				str[len+1] = 0;
			}
		}

		if (k == keys::backspace)
		{
			if (len > 0)
				str[len-1] = 0;
		}
		r = 1;
	}

	ui_do_label(x + 4, y - 3.5f, str);

	if (ui_active_item() == id)
	{
		float w = gfx_pretty_text_width(18.0f, str);
		ui_do_label(x + 4 + w, y - 3.5f, "_");
	}

	return r;
}

int do_scroll_bar(void *id, float x, float y, float height, int steps, int last_index)
{
	int r = last_index;

    if (ui_do_button(&up_button_texture, "", 0, x, y, 8, 8, draw_menu_button, &scroll_up_button))
	{
		if (r > 0)
			--r;
	}
    else if (ui_do_button(&down_button_texture, "", 0, x, y + height - 8, 8, 8, draw_menu_button, &scroll_down_button))
	{
		if (r < steps)
			++r;
	}
	else if (steps > 0) // only if there's actually stuff to scroll through
	{
		int inside = ui_mouse_inside(x, y + 8, 8, height - 16);
        if (inside && (!ui_active_item() || ui_active_item() == id))
			ui_set_hot_item(id);

		if(ui_active_item() == id)
		{
			if (ui_mouse_button(0))
			{
				float pos = ui_mouse_y() - y - 8;
				float perc = pos / (height - 16);

				r = (steps + 1) * perc;
				if (r < 0)
					r = 0;
				else if (r > steps)
					r = steps;
			}
			else
				ui_set_active_item(0);
		}
		else if (ui_hot_item() == id && ui_mouse_button(0))
			ui_set_active_item(id);
		else if (inside && (!ui_active_item() || ui_active_item() == id))
			ui_set_hot_item(id);
	}

	ui_do_image(scroll_indicator_texture, x, y + 8 + r * ((height - 32) / steps), 8, 16);
	
	return r;
}

int do_server_list(server_list *list, float x, float y, int visible_items)
{
	const float spacing = 1.5f;
	const float item_height = 14;
	const float item_width = 364;
	const float real_width = item_width + 10;
	const float real_height = item_height * visible_items + spacing * (visible_items - 1);

	int r = -1;

	for (int i = 0; i < visible_items; i++)
	{
		int item_index = i + list->scroll_index;
		if (item_index >= list->active_count)
			ui_do_image(empty_item_texture, x, y + i * item_height + i * spacing, item_width, item_height);
		else
		{
			server_info *item = &list->infos[item_index];

			bool clicked = false;
			if (list->selected_index == item_index)
				clicked = ui_do_button(item, item->name, 0, x, y + i * item_height + i * spacing, item_width, item_height, draw_menu_button, &selected_item_button);
			else
				clicked = ui_do_button(item, item->name, 0, x, y + i * item_height + i * spacing, item_width, item_height, draw_menu_button, &list_item_button);

			char temp[64]; // plenty of extra room so we don't get sad :o
			sprintf(temp, "%i/%i", item->players, item->max_players);

			gfx_texture_set(current_font->font_texture);
			gfx_pretty_text(x + 300, y + i * item_height + i * spacing - 3.5f, 18.f, temp);
            gfx_pretty_text(x + 180, y + i * item_height + i * spacing - 3.5f, 18.f, item->map);

			if (clicked)
			{
				r = item_index;
				list->selected_index = item_index;
			}
		}
	}

	list->scroll_index = do_scroll_bar(&list->scroll_index, x + real_width - 8, y, real_height, list->active_count - visible_items, list->scroll_index);
	
	return r;
}

#include <cstring>

char *read_int(char *buffer, int *value)
{
    *value = buffer[0] << 24;
    *value |= buffer[1] << 16;
    *value |= buffer[2] << 8;
    *value |= buffer[3];

	return buffer + 4;
}

char *read_netaddr(char *buffer, netaddr4 *addr)
{
	addr->ip[0] = *buffer++;
	addr->ip[1] = *buffer++;
	addr->ip[2] = *buffer++;
	addr->ip[3] = *buffer++;

	int port;
	buffer = read_int(buffer, &port);

	addr->port = port;

	return buffer;
}

void refresh_list(server_list *list)
{
	netaddr4 addr;
	netaddr4 me(0, 0, 0, 0, 0);

	list->selected_index = -1;
	
	if (net_host_lookup(MASTER_SERVER_ADDRESS, MASTER_SERVER_PORT, &addr) == 0)
    {
        socket_tcp4 sock;
        sock.open(&me);

		//sock.set_non_blocking();

		// try and connect with a timeout of 1 second
        if (sock.connect_non_blocking(&addr))
        {
            char data[256];
            int total_received = 0;
            int pointer = 0;
            int received;

            int master_server_version = -1;
            int server_count = -1;

            // read header
            while (total_received < 12 && (received = sock.recv(data + total_received, 12 - total_received)) > 0)
                total_received += received;

            // see if we have the header
            if (total_received == 12)
            {
                int signature;
                read_int(data, &signature);
    
                // check signature
                if (signature == 'TWSL')
                {
                    read_int(data + 4, &master_server_version);
                    read_int(data + 8, &server_count);

                    // TODO: handle master server version O.o
                        
                    const int server_info_size = 212;
                    const int wanted_data_count = server_count * server_info_size;

                    list->active_count = 0;
    
                    for (int i = 0; i < server_count; i++)
                    { 
                         total_received = 0;

                        // read data for a server
                        while (sock.is_connected() && total_received < server_info_size && (received = sock.recv(data + total_received, server_info_size - total_received)) > 0)
                            total_received += received;

                        // check if we got enough data
                        if (total_received == server_info_size)
                        {
                            char *d = data;

                            server_info *info = &list->infos[i];

                            d = read_int(d, &info->version);
                            d = read_netaddr(d, &info->address);

							//dbg_msg("menu/got_serverinfo", "IP: %i.%i.%i.%i:%i", (int)info->address.ip[0], (int)info->address.ip[1], (int)info->address.ip[2], (int)info->address.ip[3], info->address.port);

                            d = read_int(d, &info->players);
                            d = read_int(d, &info->max_players);
                            memcpy(info->name, d, 128);
                            d += 128;
							memcpy(info->map, d, 64);

                            // let's be safe.
                            info->name[128] = 0;
                            info->map[64] = 0;

                            ++list->active_count;
                        }
                        else
                            break;
                    }

                    if (list->scroll_index >= list->active_count)
                        list->scroll_index = list->active_count - 1;

                    if (list->scroll_index < 0)
                        list->scroll_index = 0;
                }
            }

            sock.close();
        }
	}
}

static int menu_render(netaddr4 *server_address, char *str, int max_len)
{
	// background color
	gfx_clear(89/255.f,122/255.f,0.0);

	// GRADIENT: top to bottom
	// top color: 60, 80, 0
	// bottom color: 90, 120, 0
	
	// world coordsys
	float zoom = 3.0f;
	gfx_mapscreen(0,0,400.0f*zoom,300.0f*zoom);

	// GUI coordsys
	gfx_mapscreen(0,0,400.0f,300.0f);

	static server_list list;
	static bool inited = false;

	if (!inited)
	{
		list.info_count = 256;

		list.scroll_index = 0;
		list.selected_index = -1;

		inited = true;

		refresh_list(&list);
	}

	static int64 start = time_get();

	float t = double(time_get() - start) / double(time_freq());
	draw_scrolling_background(background_texture, 400, 300, t * 0.01);

	ui_do_image(teewars_banner_texture, 70, 10, 256, 64);

	do_server_list(&list, 10, 80, 8);

	/*
    if (ui_do_button(&connect_localhost_button, "", 0, 15, 250, 64, 24, draw_menu_button, &connect_localhost_button))
    {
        *server_address = netaddr4(127, 0, 0, 1, 8303);
        return 1;
    }*/	

	if (ui_do_button(&refresh_button, "", 0, 220, 210, 64, 24, draw_menu_button, &refresh_button))
	{
		refresh_list(&list);
	} 

	if (list.selected_index == -1)
	{
		ui_do_image(join_button_grey_texture, 290, 210, 64, 24);
	}
	else if (ui_do_button(&join_button, "", 0, 290, 210, 64, 24, draw_menu_button, &join_button))
	{
		*server_address = list.infos[list.selected_index].address;

		dbg_msg("menu/join_button", "IP: %i.%i.%i.%i:%i", (int)server_address->ip[0], (int)server_address->ip[1], (int)server_address->ip[2], (int)server_address->ip[3], server_address->port);

		return 1;
	}

	const float name_x = 10, name_y = 215;

	ui_do_label(name_x + 4, name_y - 3.5f, "Name:");
	ui_do_image(input_box_texture, name_x + 50 - 5, name_y - 5, 150 + 10, 14 + 10);
	ui_do_edit_box(str, name_x + 50, name_y, 150, 14, str, max_len);

	if (ui_do_button(&quit_button, "", 0, 290, 250, 69, 25, draw_menu_button, &quit_button))
		return -1;

	ui_do_label(10.0f, 300.0f-20.0f, "Version: " TEEWARS_VERSION);
	
	return 0;
}

void modmenu_init()
{
	keys::enable_char_cache();
	keys::enable_key_cache();

    current_font->font_texture = gfx_load_texture_tga("data/big_font.tga");
	background_texture = gfx_load_texture_tga("data/gui_bg.tga");
    not_empty_item_texture = gfx_load_texture_tga("data/gui/game_list_item_not_empty.tga");
    empty_item_texture = gfx_load_texture_tga("data/gui/game_list_item_empty.tga");
    active_item_texture = gfx_load_texture_tga("data/gui/game_list_item_active.tga");
	selected_item_texture = gfx_load_texture_tga("data/gui/game_list_item_selected.tga");

	join_button_texture = gfx_load_texture_tga("data/gui/join_button.tga");
	join_button_hot_texture = gfx_load_texture_tga("data/gui/join_button_hot.tga");
	join_button_active_texture = gfx_load_texture_tga("data/gui/join_button_active.tga");
	join_button_grey_texture = gfx_load_texture_tga("data/gui/join_button_greyed.tga");


//    button_not_hilighted_texture = gfx_load_texture_tga("data/gui/game_list_join_button.tga");
//	button_hilighted_texture = gfx_load_texture_tga("data/gui/button_hilighted.tga");
//	button_active_texture = gfx_load_texture_tga("data/gui/button_active.tga");

    quit_button_texture = gfx_load_texture_tga("data/gui/quit_button.tga");
	quit_button_hot_texture = gfx_load_texture_tga("data/gui/quit_button_hot.tga");
	quit_button_active_texture = gfx_load_texture_tga("data/gui/quit_button_active.tga");

    up_button_texture = gfx_load_texture_tga("data/gui/scroll_arrow_up.tga");
	up_button_active_texture = gfx_load_texture_tga("data/gui/scroll_arrow_up_active.tga");

    down_button_texture = gfx_load_texture_tga("data/gui/scroll_arrow_down.tga");
	down_button_active_texture = gfx_load_texture_tga("data/gui/scroll_arrow_down_active.tga");

    teewars_banner_texture = gfx_load_texture_tga("data/gui_logo.tga");
    scroll_indicator_texture = gfx_load_texture_tga("data/gui/scroll_drag.tga");
	connect_localhost_texture = gfx_load_texture_tga("data/gui/game_list_connect_localhost.tga");

	refresh_button_texture = gfx_load_texture_tga("data/gui/refresh_button.tga");
	refresh_button_hot_texture = gfx_load_texture_tga("data/gui/refresh_button_hot.tga");
	refresh_button_active_texture = gfx_load_texture_tga("data/gui/refresh_button_active.tga");

	input_box_texture = gfx_load_texture_tga("data/gui/input_box.tga");

	music_menu = snd_load_wav("data/audio/Music_Menu.wav");
}

void modmenu_shutdown()
{
}

int modmenu_render(void *ptr, char *str, int max_len)
{
	static int mouse_x = 0;
	static int mouse_y = 0;

	if (music_menu_id == -1)
	{
		dbg_msg("menu", "no music is playing, so let's play some tunes!");
		music_menu_id = snd_play(music_menu, SND_LOOP);
	}

	netaddr4 *server_address = (netaddr4 *)ptr;	

    // handle mouse movement
    float mx, my, mwx, mwy;
    {
        int rx, ry;
        inp_mouse_relative(&rx, &ry);
        mouse_x += rx;
        mouse_y += ry;
        if(mouse_x < 0) mouse_x = 0;
        if(mouse_y < 0) mouse_y = 0;
        if(mouse_x > gfx_screenwidth()) mouse_x = gfx_screenwidth();
        if(mouse_y > gfx_screenheight()) mouse_y = gfx_screenheight();
            
        // update the ui
        mx = (mouse_x/(float)gfx_screenwidth())*400.0f;
        my = (mouse_y/(float)gfx_screenheight())*300.0f;
        mwx = mx*3.0f; // adjust to zoom and offset
        mwy = mx*3.0f; // adjust to zoom and offset
            
        int buttons = 0;
        if(inp_mouse_button_pressed(0)) buttons |= 1;
        if(inp_mouse_button_pressed(1)) buttons |= 2;
        if(inp_mouse_button_pressed(2)) buttons |= 4;
            
        ui_update(mx,my,mx*3.0f,my*3.0f,buttons);
    }

    int r = menu_render(server_address, str, max_len);

    // render butt ugly mouse cursor
    gfx_texture_set(-1);
    gfx_quads_begin();
    gfx_quads_setcolor(0,0,0,1);
    gfx_quads_draw_freeform(mx,my,mx,my,
                                mx+7,my,
                                mx,my+7);
    gfx_quads_setcolor(1,1,1,1);
    gfx_quads_draw_freeform(mx+1,my+1,mx+1,my+1,
                                mx+5,my+1,
                                mx+1,my+5);
    gfx_quads_end();

	if (r)
	{
		snd_stop(music_menu_id);
		music_menu_id = -1;
	}

	keys::clear_char();
	keys::clear_key();

	return r;
}