#include "InstallerMain.h"
#include "InstallerUtil.h"
#include "SelectGame.h"
#include "GamesList.h"
#include "ChooseMethod.h"
#include "PerformInstall.h"
#include "LocalCopyMethod.h"

int selected_game_index = -1;
mod_info_t **game_sel_list = NULL;
unsigned int game_sel_count = 0;

void AppendSelectableGame(mod_info_t *mod)
{
	if (game_sel_list == NULL)
	{
		game_sel_list = 
			(mod_info_t **)malloc(sizeof(mod_info_t *) * (game_sel_count + 1));
	}
	else
	{
		game_sel_list = 
			(mod_info_t **)realloc(game_sel_list, 
			sizeof(mod_info_t *) * (game_sel_count + 1));
	}

	game_sel_list[game_sel_count++] = mod;
}

INT_PTR CALLBACK ChooseGameHandler(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)
{
	switch (message)
	{
	case WM_INITDIALOG:
		{
			HWND lbox = GetDlgItem(hDlg, IDC_SELGAME_LIST);

			for (unsigned int i = 0; i < g_mod_count; i++)
			{
				LRESULT res = SendMessage(lbox,
					LB_ADDSTRING,
					0,
					(LPARAM)g_mod_list[i].name);

				if (res == LB_ERR || res == LB_ERRSPACE)
				{
					continue;
				}

				AppendSelectableGame(&g_mod_list[i]);
			}

			UpdateWindow(lbox);
			
			return (INT_PTR)TRUE;
		}
	case WM_COMMAND:
		{
			if (LOWORD(wParam) == ID_SELGAME_EXIT
				|| LOWORD(wParam) == ID_CLOSE)
			{
				return AskToExit(hDlg);
			}
			else if (LOWORD(wParam) == ID_SELGAME_BACK)
			{
				EndDialog(hDlg, (INT_PTR)DisplayChooseMethod);
				return (INT_PTR)TRUE;
			}
			else if (LOWORD(wParam) == IDC_SELGAME_LIST)
			{
				if (HIWORD(wParam) == LBN_SELCHANGE)
				{
					HWND lbox = (HWND)lParam;
					LRESULT cursel = SendMessage(lbox, LB_GETCURSEL, 0, 0);

					selected_game_index = -1;

					if (cursel == LB_ERR)
					{
						break;
					}

					if (cursel >= (LRESULT)g_mod_count)
					{
						break;
					}

					selected_game_index = (int)cursel;

					HWND button = GetDlgItem(hDlg, ID_SELGAME_NEXT);
					EnableWindow(button, TRUE);
				}
			}
			else if (LOWORD(wParam) == ID_SELGAME_NEXT)
			{
				if (selected_game_index == -1)
				{
					break;
				}

				g_LocalCopier.SetOutputPath(game_sel_list[selected_game_index]->mod_path);
				SetInstallMethod(&g_LocalCopier);

				EndDialog(hDlg, (INT_PTR)DisplayPerformInstall);

				return (INT_PTR)TRUE;
			}
			break;
		}
	case WM_DESTROY:
		{
			ReleaseGamesList();
			free(game_sel_list);
			game_sel_list = NULL;
			game_sel_count = 0;
			break;
		}
	}

	return (INT_PTR)FALSE;
}

void *DisplaySelectGame(HWND hWnd)
{
	INT_PTR val;
	
	if ((val = DialogBox(
		g_hInstance, 
		MAKEINTRESOURCE(IDD_SELECT_GAME),
		hWnd,
		ChooseGameHandler)) == -1)
	{
		return NULL;
	}

	return (void *)val;
}