#include "../../types.h"
#include <iconv.h>

int selhero_SaveCount = 0;
_uiheroinfo heros[MAX_CHARACTERS];
_uiheroinfo heroInfo;

void RenderStats()
{
	char lvl[4] = "--";
	char str[4] = "--";
	char mag[4] = "--";
	char dex[4] = "--";
	char vit[4] = "--";

	if (heroInfo.heroclass != UI_NUM_CLASSES) {
		sprintf(lvl, "%d", heroInfo.level);
		sprintf(str, "%d", heroInfo.strength);
		sprintf(mag, "%d", heroInfo.magic);
		sprintf(dex, "%d", heroInfo.dexterity);
		sprintf(vit, "%d", heroInfo.vitality);
	}

	DrawArtStr(31, 323, AFT_SMALL, AFC_SILVER, "Level:", JustRight, 118);
	DrawArtStr(149, 323, AFT_SMALL, AFC_SILVER, lvl, JustCentre, 61);
	DrawArtStr(31, 358, AFT_SMALL, AFC_SILVER, "Strength:", JustRight, 118);
	DrawArtStr(149, 358, AFT_SMALL, AFC_SILVER, str, JustCentre, 61);
	DrawArtStr(31, 380, AFT_SMALL, AFC_SILVER, "Magic:", JustRight, 118);
	DrawArtStr(149, 380, AFT_SMALL, AFC_SILVER, mag, JustCentre, 61);
	DrawArtStr(31, 401, AFT_SMALL, AFC_SILVER, "Dexterity:", JustRight, 118);
	DrawArtStr(149, 401, AFT_SMALL, AFC_SILVER, dex, JustCentre, 61);
	DrawArtStr(31, 422, AFT_SMALL, AFC_SILVER, "Vitality:", JustRight, 118);
	DrawArtStr(149, 422, AFT_SMALL, AFC_SILVER, vit, JustCentre, 61);
}

void selhero_Render(bool multiPlayer)
{
	heroInfo.heroclass = UI_NUM_CLASSES;
	if (SelectedItem <= selhero_SaveCount) {
		memcpy(&heroInfo, &heros[SelectedItem - 1], sizeof(heroInfo));
	}

	DrawArt(0, 0, &ArtBackground);
	DrawLogo();

	char *title = "Single Player Characters";
	if (multiPlayer) {
		title = "Multi Player Characters";
	}
	DrawArtStr(0, 161, AFT_BIG, AFC_SILVER, title, JustCentre);

	DrawArt(30, 211, &ArtHero, heroInfo.heroclass);
	RenderStats();

	int w = 368;
	int x = 241;

	DrawArtStr(x, 211, AFT_BIG, AFC_SILVER, "Select Hero", JustCentre, w);

	int spacing = 26;
	int selectorTop = 256;
	int y = selectorTop;
	for (int i = 0; i < selhero_SaveCount; i++) {
		DrawArtStr(x, y, AFT_MED, AFC_GOLD, heros[i].name, JustCentre, w);
		y += spacing;
	}
	DrawArtStr(x, y, AFT_MED, AFC_GOLD, "New Hero", JustCentre, w);

	DrawSelector(x, selectorTop + 3, w, 24, spacing, FOCUS_SMALL);

	DrawArtStr(279, 429, AFT_BIG, AFC_GOLD, "OK");
	DrawArtStr(378, 429, AFT_BIG, AFC_GOLD, "Delete");
	DrawArtStr(501, 429, AFT_BIG, AFC_GOLD, "Cancel");
}

void selhero_Render_Name(bool multiPlayer)
{
	DrawArt(0, 0, &ArtBackground);
	DrawLogo();

	DrawArt(30, 211, &ArtHero, heroInfo.heroclass);
	RenderStats();

	char *title = "New Single Player Hero";
	if (multiPlayer) {
		title = "New Multi Player Hero";
	}

	DrawArtStr(0, 161, AFT_BIG, AFC_SILVER, title, JustCentre);

	int w = 368;
	int x = 241;
	int y = 318;

	DrawArtStr(x, 211, AFT_BIG, AFC_SILVER, "Enter Name", JustCentre, w);

	DrawSelector(x, y - 2, w, 24);

	char lable[17];
	strcpy(lable, heroInfo.name);
	if (GetAnimationFrame(2, 500)) {
		int len = strlen(lable);
		lable[len] = '|';
		lable[len + 1] = '\0';
	}

	DrawArtStr(x + 67, y, AFT_MED, AFC_GOLD, lable); // todo add blinking "|"

	DrawArtStr(329, 429, AFT_BIG, AFC_GOLD, "OK");
	DrawArtStr(451, 429, AFT_BIG, AFC_GOLD, "Cancel");
}

// Have this load the function above and then render it in the main menu.
// Cnacel box is also needed.
void selhero_Render_ClassSelector(bool multiPlayer)
{
	DrawArt(0, 0, &ArtBackground);
	DrawLogo();

	DrawArt(30, 211, &ArtHero, heroInfo.heroclass);
	RenderStats();

	char *title = "New Single Player Hero";
	if (multiPlayer) {
		title = "New Multi Player Hero";
	}
	DrawArtStr(0, 161, AFT_BIG, AFC_SILVER, title, JustCentre);

	int w = 369;
	int x = 241;
	int y = 285;

	DrawArtStr(x, 211, AFT_BIG, AFC_SILVER, "Choose Class", JustCentre, w);

	char *heroclasses[3] = { "Warrior", "Rogue", "Sorcerer" };

	int spacing = 33;
	int selectorTop = y;
	for (int i = 0; i < 3; i++) {
		DrawArtStr(x, y, AFT_MED, AFC_GOLD, heroclasses[i], JustCentre, w);
		y += spacing;
	}

	DrawSelector(x, selectorTop - 2, w, 39, spacing);

	DrawArtStr(329, 429, AFT_BIG, AFC_GOLD, "OK");
	DrawArtStr(451, 429, AFT_BIG, AFC_GOLD, "Cancel");
}

BOOL __stdcall SelHero_GetHeroInfo(_uiheroinfo *pInfo)
{
	heros[selhero_SaveCount] = *pInfo;
	selhero_SaveCount++;

	return TRUE;
}

void selhero_Load(BOOL(__stdcall *fninfo)(BOOL(__stdcall *fninfofunc)(_uiheroinfo *)))
{
	LoadBackgroundArt("ui_art\\selhero.pcx");

	selhero_SaveCount = 0;
	fninfo(SelHero_GetHeroInfo);

	heroInfo.heroclass = UI_NUM_CLASSES;
}

void selhero_Free()
{
	mem_free_dbg(ArtBackground.data);
	ArtBackground.data = NULL;
}

void selhero_setDefaultStats(BOOL(__stdcall *fnstats)(unsigned int, _uidefaultstats *))
{
	_uidefaultstats defaults;
	fnstats(SelectedItem - 1, &defaults);

	heroInfo.heroclass = SelectedItem - 1;
	heroInfo.strength = defaults.strength;
	heroInfo.magic = defaults.magic;
	heroInfo.dexterity = defaults.dexterity;
	heroInfo.vitality = defaults.vitality;
}

bool selhero_Event(bool *aborted)
{
	SDL_Event event;
	while (SDL_PollEvent(&event)) {
		switch (event.type) {
		case SDL_KEYDOWN:
			if (UiFocuseNavigation(&event))
				break;
			switch (event.key.keysym.sym) {
			case SDLK_ESCAPE:
				*aborted = true;
				return true;
			case SDLK_RETURN:
			case SDLK_KP_ENTER:
			case SDLK_SPACE:
				if (SelectedItem == SelectedItemMax) {
					memset(&heroInfo.name, 0, sizeof(heroInfo.name));
					SetMenu(SELHERO_CLASSES);
					break;
				}
				return true;
			}
			break;
		case SDL_QUIT:
			exit(0);
		}
	}

	return false;
}

bool selhero_Event_ClassSelector(bool *aborted)
{
	SDL_Event event;
	while (SDL_PollEvent(&event)) {
		switch (event.type) {
		case SDL_KEYDOWN:
			if (UiFocuseNavigation(&event, true))
				break;
			switch (event.key.keysym.sym) {
			case SDLK_ESCAPE:
				if (selhero_SaveCount) {
					SetMenu(SELHERO_LOAD);
					SelectedItemMax += selhero_SaveCount;
					break;
				}
				*aborted = true;
				return true;
			case SDLK_RETURN:
			case SDLK_KP_ENTER:
			case SDLK_SPACE:
				SetMenu(SELHERO_NAME);
				memset(heroInfo.name, '\0', 16);
				SDL_StartTextInput();
				break;
			}
			break;
		case SDL_QUIT:
			exit(0);
		}
	}

	return false;
}

bool selhero_CatToName(char *in_buf)
{
	iconv_t cd = iconv_open("ISO_8859-1//TRANSLIT//IGNORE", "UTF-8");
	if (cd == (iconv_t)-1) {
		TermMsg("Failed to load iconv!");
	}

	size_t in_left = strlen(in_buf);
	char output[SDL_TEXTINPUTEVENT_TEXT_SIZE] = "";
	char *out_buf = output;
	size_t out_left = sizeof(output) - 1;

	while (in_left && out_left) {
		iconv(cd, &in_buf, &in_left, &out_buf, &out_left);
	}
	iconv_close(cd);
	strncat(heroInfo.name, output, 15 - strlen(heroInfo.name));
}

bool selhero_Event_Name()
{
	SDL_Event event;
	while (SDL_PollEvent(&event)) {

		switch (event.type) {
		case SDL_QUIT:
			exit(0);
			break;
		case SDL_KEYDOWN:
			switch (event.key.keysym.sym) {
			case SDLK_v:
				if (SDL_GetModState() & KMOD_CTRL) {
					selhero_CatToName(SDL_GetClipboardText());
				}
				break;
			case SDLK_ESCAPE:
				SDL_StopTextInput();
				SetMenu(SELHERO_CLASSES);
				break;
			case SDLK_RETURN:
			case SDLK_RETURN2:
			case SDLK_KP_ENTER:
				SDL_StopTextInput();
				return true;
				break;
			case SDLK_BACKSPACE:
			case SDLK_LEFT:
				int nameLen = strlen(heroInfo.name);
				if (nameLen > 0) {
					heroInfo.name[nameLen - 1] = '\0';
				}
				break;
			}
			break;
		case SDL_TEXTINPUT:
			selhero_CatToName(event.text.text);
			break;
		}
	}

	return false;
}

bool UiSelHeroDialog(
    BOOL(__stdcall *fncreate)(_uiheroinfo *),
    BOOL(__stdcall *fnremove)(_uiheroinfo *),
    BOOL(__stdcall *fnstats)(unsigned int, _uidefaultstats *),
    bool multiPlayer)
{
	bool aborted = false;
	bool endMenu = false;

	SelectedItem = 1;
	submenu = SELHERO_LOAD;
	SelectedItemMax = 1 + selhero_SaveCount;
	if (!selhero_SaveCount) {
		submenu = SELHERO_CLASSES;
		SelectedItemMax = 3;
	}

	while (endMenu == false) {
		CapFPS();

		switch (submenu) {
		case SELHERO_LOAD:
			selhero_Render(multiPlayer);
			endMenu = selhero_Event(&aborted);
			break;
		case SELHERO_CLASSES:
			selhero_setDefaultStats(fnstats);
			selhero_Render_ClassSelector(multiPlayer);
			endMenu = selhero_Event_ClassSelector(&aborted);
			break;
		case SELHERO_NAME:
			selhero_Render_Name(multiPlayer);
			endMenu = selhero_Event_Name();
			break;
		}

		DrawMouse();
		UiFadeIn();
	}

	if (!aborted && !heroInfo.hassaved) {
		fncreate(&heroInfo); // todo don't overwrite
	}

	return aborted;
}

BOOL __stdcall UiSelHeroSingDialog(
    BOOL(__stdcall *fninfo)(BOOL(__stdcall *fninfofunc)(_uiheroinfo *)),
    BOOL(__stdcall *fncreate)(_uiheroinfo *),
    BOOL(__stdcall *fnremove)(_uiheroinfo *),
    BOOL(__stdcall *fnstats)(unsigned int, _uidefaultstats *),
    int *dlgresult,
    char *name,
    int *difficulty)
{
	selhero_Load(fninfo);
	if (UiSelHeroDialog(fncreate, fnremove, fnstats, false)) {
		*dlgresult = EXIT_MENU;
	} else {
		strcpy(name, heroInfo.name);

		if (heroInfo.hassaved) {
			*dlgresult = LOAD_GAME;
		}
	}

	BlackPalette();
	selhero_Free();

	return TRUE;
}

//////////////////////////////////////////
//            MULTI PLAYER              //
//////////////////////////////////////////

void selhero_multi_Load()
{
	LoadBackgroundArt("ui_art\\selgame.pcx");
}

void selhero_multi_Free()
{
	mem_free_dbg(ArtBackground.data);
	ArtBackground.data = NULL;
}

void selhero_Render_DifficultySelection()
{
	DrawArt(0, 0, &ArtBackground);
	DrawLogo();

	DrawArtStr(0, 161, AFT_BIG, AFC_SILVER, "Create Game", JustCentre);

	int w = 333;
	int x = 281;
	int y = 282;

	DrawArtStr(x, 211, AFT_BIG, AFC_SILVER, "Select Difficulty", JustCentre, w);

	char *gameOptions[] = { "Normal", "Nightmare", "Hell" };

	DrawArtStr(23, 211, AFT_BIG, AFC_SILVER, gameOptions[SelectedItem - 1], JustCentre, 226);
	if (SelectedItem == 1) {
		DrawArtStr(35, 256, AFT_SMALL, AFC_SILVER, "Normal Difficulty");
		DrawArtStr(35, 272, AFT_SMALL, AFC_SILVER, "This is where a starting");
		DrawArtStr(35, 288, AFT_SMALL, AFC_SILVER, "character should begin");
		DrawArtStr(35, 304, AFT_SMALL, AFC_SILVER, "the quest to defeat");
		DrawArtStr(35, 320, AFT_SMALL, AFC_SILVER, "Diablo.");
	} else if (SelectedItem == 2) {
		DrawArtStr(35, 256, AFT_SMALL, AFC_SILVER, "Nightmare Difficulty");
		DrawArtStr(35, 272, AFT_SMALL, AFC_SILVER, "The denizens of the");
		DrawArtStr(35, 288, AFT_SMALL, AFC_SILVER, "Labyrinth have been");
		DrawArtStr(35, 304, AFT_SMALL, AFC_SILVER, "bolstered and will prove");
		DrawArtStr(35, 320, AFT_SMALL, AFC_SILVER, "to be a greater");
		DrawArtStr(35, 336, AFT_SMALL, AFC_SILVER, "challenge. This is");
		DrawArtStr(35, 352, AFT_SMALL, AFC_SILVER, "recommended for");
		DrawArtStr(35, 368, AFT_SMALL, AFC_SILVER, "experienced characters");
		DrawArtStr(35, 384, AFT_SMALL, AFC_SILVER, "only.");
	} else if (SelectedItem == 3) {
		DrawArtStr(35, 256, AFT_SMALL, AFC_SILVER, "Hell Difficulty");
		DrawArtStr(35, 272, AFT_SMALL, AFC_SILVER, "The most powerful of");
		DrawArtStr(35, 288, AFT_SMALL, AFC_SILVER, "the underworld's");
		DrawArtStr(35, 304, AFT_SMALL, AFC_SILVER, "creatures lurk at the");
		DrawArtStr(35, 320, AFT_SMALL, AFC_SILVER, "gateway into Hell. Only");
		DrawArtStr(35, 336, AFT_SMALL, AFC_SILVER, "the most experienced");
		DrawArtStr(35, 352, AFT_SMALL, AFC_SILVER, "characters should");
		DrawArtStr(35, 368, AFT_SMALL, AFC_SILVER, "venture in this realm.");
	}

	int selectorTop = y;

	int spacing = 26;
	for (int i = 0; i < 3; i++) {
		DrawArtStr(x, y, AFT_MED, AFC_GOLD, gameOptions[i], JustCentre, w);
		y += spacing;
	}

	DrawSelector(x, selectorTop + 3, w, 19, spacing, FOCUS_SMALL);

	DrawArtStr(349, 429, AFT_BIG, AFC_GOLD, "OK");
	DrawArtStr(471, 429, AFT_BIG, AFC_GOLD, "Cancel");
}

bool selhero_Event_GameSelection(int *dlgresult)
{
	SDL_Event event;
	while (SDL_PollEvent(&event)) {
		switch (event.type) {
		case SDL_KEYDOWN:
			if (UiFocuseNavigation(&event))
				break;
			switch (event.key.keysym.sym) {
			case SDLK_ESCAPE:
				*dlgresult = EXIT_MENU;
				return true;
			case SDLK_RETURN:
			case SDLK_KP_ENTER:
			case SDLK_SPACE:
				SetMenu(SELHERO_DIFFICULTY);
				break;
			}
			break;
		case SDL_QUIT:
			exit(0);
		}
	}

	return false;
}

bool selhero_Event_DifficultySelection(int *dlgresult)
{
	SDL_Event event;
	while (SDL_PollEvent(&event)) {
		switch (event.type) {
		case SDL_KEYDOWN:
			if (UiFocuseNavigation(&event, true))
				break;
			switch (event.key.keysym.sym) {
			case SDLK_ESCAPE:
				SetMenu(SELHERO_SELECT_GAME);
				SelectedItemMax += 1;
				break;
			case SDLK_RETURN:
			case SDLK_KP_ENTER:
			case SDLK_SPACE:
				switch (SelectedItem) {
				case 1:
					gnDifficulty = DIFF_NORMAL;
					break;
				case 2:
					gnDifficulty = DIFF_NIGHTMARE;
					break;
				case 3:
					gnDifficulty = DIFF_HELL;
					break;
				}
				*dlgresult = NEW_GAME;
				return true;
			}
			break;
		case SDL_MOUSEBUTTONDOWN:
			if (event.button.button == SDL_BUTTON_LEFT) {
				SDL_Rect CreateHeroCancelBox;
				CreateHeroCancelBox.y = 550;
				CreateHeroCancelBox.x = 675;
				CreateHeroCancelBox.w = 100;
				CreateHeroCancelBox.h = 30;

				SDL_Rect NormalSelectBox;
				NormalSelectBox.x = 280;
				NormalSelectBox.y = 350;
				NormalSelectBox.w = 300;
				NormalSelectBox.h = 30;

				if (IsInsideRect(&event, &NormalSelectBox)) {
					gnDifficulty = DIFF_NORMAL;
					*dlgresult = LOAD_GAME;
				}
				break;
			}
			break;
		case SDL_QUIT:
			exit(0);
		}
	}

	return false;
}

void selhero_Render_GameSelection()
{

	DrawArt(0, 0, &ArtBackground);
	DrawLogo();

	DrawArtStr(0, 161, AFT_BIG, AFC_SILVER, "Join UPD Game", JustCentre);

	int w = 333;
	int x = 281;
	int y = 282;

	DrawArtStr(x, 211, AFT_BIG, AFC_SILVER, "Select Action", JustCentre, w);

	char *gameOptions[] = { "Create Game", "Enter IP", "Localhost" };

	DrawArtStr(35, 211, AFT_MED, AFC_SILVER, "Description:");
	if (SelectedItem == 1) {
		DrawArtStr(35, 256, AFT_SMALL, AFC_SILVER, "Create a new game with");
		DrawArtStr(35, 272, AFT_SMALL, AFC_SILVER, "a difficulty setting of");
		DrawArtStr(35, 288, AFT_SMALL, AFC_SILVER, "your choice.");
	} else if (SelectedItem == 2) {
		DrawArtStr(35, 256, AFT_SMALL, AFC_SILVER, "Join a game directly");
		DrawArtStr(35, 272, AFT_SMALL, AFC_SILVER, "via a know host IP.");
	} else {
		DrawArtStr(35, 256, AFT_SMALL, AFC_SILVER, "Normal Difficulty");
		DrawArtStr(35, 272, AFT_SMALL, AFC_SILVER, "Created by Localhost, A");
		DrawArtStr(35, 288, AFT_SMALL, AFC_SILVER, "level 1 Warrior.");
	}

	int selectorTop = y;

	int spacing = 26;
	for (int i = 0; i < 3; i++) {
		DrawArtStr(x, y, AFT_MED, AFC_GOLD, gameOptions[i], JustCentre, w);
		y += spacing;
	}

	DrawSelector(x, selectorTop + 3, w, 19, spacing, FOCUS_SMALL);

	DrawArtStr(349, 429, AFT_BIG, AFC_GOLD, "OK");
	DrawArtStr(471, 429, AFT_BIG, AFC_GOLD, "Cancel");
}

BOOL __stdcall UiSelHeroMultDialog(
    BOOL(__stdcall *fninfo)(BOOL(__stdcall *fninfofunc)(_uiheroinfo *)),
    BOOL(__stdcall *fncreate)(_uiheroinfo *),
    BOOL(__stdcall *fnremove)(_uiheroinfo *),
    BOOL(__stdcall *fnstats)(unsigned int, _uidefaultstats *),
    int *dlgresult,
    BOOL *hero_is_created,
    char *name)
{
	*hero_is_created = false;

	selhero_Load(fninfo);
	bool abort = UiSelHeroDialog(fncreate, fnremove, fnstats, true);
	BlackPalette();
	selhero_Free();
	if (abort) {
		*dlgresult = EXIT_MENU;
		return TRUE;
	}

	strcpy(name, heroInfo.name);

	selhero_multi_Load();

	submenu = SELHERO_SELECT_GAME;

	SelectedItem = 1;
	SelectedItemMax = 2 + 1;

	bool endMenu = false;
	while (endMenu == false) {
		CapFPS();

		switch (submenu) {
		case SELHERO_SELECT_GAME:
			selhero_Render_GameSelection();
			endMenu = selhero_Event_GameSelection(dlgresult);
			break;
		case SELHERO_DIFFICULTY:
			selhero_Render_DifficultySelection();
			endMenu = selhero_Event_DifficultySelection(dlgresult);
			break;
		}

		DrawMouse();
		UiFadeIn();
	}

	if (*dlgresult != EXIT_MENU) {
		strcpy(name, heroInfo.name);
	}

	BlackPalette();
	selhero_Free();

	return TRUE;
}