#include <dirent.h>
#include <unistd.h>

#include "FileOperations/fileops.h"
#include "Controls/DeviceHandler.hpp"
#include "wad/nandtitle.h"
#include "system/IosLoader.h"
#include "menu/menus.h"
#include "wpad.h"
#include "usbloader/wbfs.h"
#include "usbloader/GameList.h"
#include "settings/GameTitles.h"
#include "xml/GameTDB.hpp"

static int FindGamePartition()
{
	int partCount = DeviceHandler::GetUSBPartitionCount();

	// Loop through all WBFS partitions first to check them in case IOS249 Rev < 18
	for(int i = 0; i < partCount; ++i)
	{
		if(DeviceHandler::GetFilesystemType(USB1+i) != PART_FS_WBFS)
			continue;

		if (WBFS_OpenPart(i) == 0)
		{
			Settings.partition = i;
			return 0;
		}
	}


	if(IosLoader::IsWaninkokoIOS() && NandTitles.VersionOf(TITLE_ID(1, IOS_GetVersion())) < 18)
		return -1;

	// Loop through FAT/NTFS/EXT partitions, and find the first partition with games on it (if there is one)
	for(int i = 0; i < partCount; ++i)
	{
		if(DeviceHandler::GetFilesystemType(USB1+i) != PART_FS_NTFS &&
		   DeviceHandler::GetFilesystemType(USB1+i) != PART_FS_FAT &&
		   DeviceHandler::GetFilesystemType(USB1+i) != PART_FS_EXT)
		{
			continue;
		}

		if (WBFS_OpenPart(i) != 0)
			continue;

		u32 count;
		// Get the game count...
		WBFS_GetCount(i, &count);

		if (count > 0)
		{
			Settings.partition = i;
			return 0;
		}

		WBFS_Close(i);
	}

	return -1;
}

static int PartitionChoice()
{
	int ret = -1;

	int choice = WindowPrompt(tr( "No WBFS or FAT/NTFS/EXT partition found" ),
			tr( "You need to select or format a partition" ), tr( "Select" ), tr( "Format" ), tr( "Return" ));

	if (choice == 0)
	{
		Sys_LoadMenu();
	}
	else if(choice == 1)
	{
		int part_num = SelectPartitionMenu();
		if(part_num >= 0)
		{
			if(IosLoader::IsWaninkokoIOS() && NandTitles.VersionOf(TITLE_ID(1, IOS_GetVersion())) < 18 &&
			   (DeviceHandler::GetFilesystemType(USB1+part_num) == PART_FS_NTFS ||
				DeviceHandler::GetFilesystemType(USB1+part_num) == PART_FS_FAT ||
				DeviceHandler::GetFilesystemType(USB1+part_num) == PART_FS_EXT))
			{
				WindowPrompt(tr("Warning:"), tr("You are trying to select a FAT32/NTFS/EXT partition with cIOS 249 Rev < 18. This is not supported. Continue on your own risk."), tr("OK"));
			}

			ret = WBFS_OpenPart(part_num);

			Settings.partition = part_num;
			Settings.Save();
		}
	}
	else if(choice == 2)
	{
		while(ret < 0 || ret == -666)
		{
			int part_num = SelectPartitionMenu();
			if(part_num >= 0)
				ret = FormatingPartition(tr( "Formatting, please wait..." ), part_num);
		}
	}

	return ret;
}

/****************************************************************************
 * MountGamePartition
 ***************************************************************************/
int MountGamePartition(bool ShowGUI)
{
	gprintf("MountGamePartition()\n");

	s32 wbfsinit = MountWBFS(ShowGUI);
	if (wbfsinit < 0)
	{
		if(ShowGUI) WindowPrompt(tr( "Error !" ), tr( "USB Device not found" ), tr( "OK" ));
		Sys_LoadMenu();
	}

	s32 ret = -1;

	if(Settings.MultiplePartitions)
		ret = WBFS_OpenAll();
	else
		ret = WBFS_OpenPart(Settings.partition);

	if(ret < 0)
		ret = FindGamePartition();

	if (ret < 0 && ShowGUI)
		ret = PartitionChoice();

	if(ret < 0)
		Sys_LoadMenu();

	gprintf("\tDisc_Init\n");
	ret = Disc_Init();
	if (ret < 0)
	{
		if(ShowGUI)
			WindowPrompt(tr( "Error !" ), tr( "Could not initialize DIP module!" ), tr( "OK" ));
		Sys_LoadMenu();
	}

	gprintf("LoadTitlesFromGameTDB\n");
	//! Clear list if available
	GameTitles.Clear();

	//! gameList is loaded in GameTitles.LoadTitlesFromGameTDB after cache file load
	//! for speed up purpose. If titles override active, load game list here.
	if(Settings.titlesOverride)
		GameTitles.LoadTitlesFromGameTDB(Settings.titlestxt_path);
	else
		gameList.LoadUnfiltered();

	return ret;
}