/****************************************************************************
 * Copyright (C) 2010
 * by Dimok
 *
 * This software is provided 'as-is', without any express or implied
 * warranty. In no event will the authors be held liable for any
 * damages arising from the use of this software.
 *
 * Permission is granted to anyone to use this software for any
 * purpose, including commercial applications, and to alter it and
 * redistribute it freely, subject to the following restrictions:
 *
 * 1. The origin of this software must not be misrepresented; you
 * must not claim that you wrote the original software. If you use
 * this software in a product, an acknowledgment in the product
 * documentation would be appreciated but is not required.
 *
 * 2. Altered source versions must be plainly marked as such, and
 * must not be misrepresented as being the original software.
 *
 * 3. This notice may not be removed or altered from any source
 * distribution.
 ***************************************************************************/
#include <algorithm>
#include <string>
#include <malloc.h>
#include "usbloader/wbfs.h"
#include "settings/newtitles.h"
#include "settings/cfg.h"
#include "xml/xml.h"
#include "FreeTypeGX.h"
#include "GameList.h"
#include "memory.h"

GameList gameList;

GameList::GameList()
{

}

void GameList::clear()
{
    FullGameList.clear();
    FilteredList.clear();
    //! Clear memory of the vector completely
    std::vector<struct discHdr *>().swap(FilteredList);
    std::vector<struct discHdr>().swap(FullGameList);
}

struct discHdr * GameList::at(int i)
{
    if(i < 0 || i >= (int) FilteredList.size())
        return NULL;

    return FilteredList[i];
}

int GameList::ReadGameList()
{
    FullGameList.clear();
    FilteredList.clear();

    // Retrieve all stuff from WBFS
    u32 cnt;

    int ret = WBFS_GetCount(&cnt);
    if (ret < 0)
        return -1;

    /* Buffer length */
    u32 len = sizeof(struct discHdr) * cnt;

    /* Allocate memory */
    struct discHdr *buffer = (struct discHdr *) allocate_memory(len);
    if (!buffer)
        return -1;

    /* Clear buffer */
    memset(buffer, 0, len);

    /* Get header list */
    ret = WBFS_GetHeaders(buffer, cnt, sizeof(struct discHdr));
    if (ret < 0)
    {
        if (buffer)
            free(buffer);
        return -1;
    }

    FullGameList.resize(cnt);
    memcpy(&FullGameList[0], buffer, len);

    free(buffer);

	return LoadUnfiltered();
}

static bool WCharSortCallback(const wchar_t char1, const wchar_t char2)
{
    return char2 > char1;
}

int GameList::FilterList(const wchar_t * gameFilter)
{
    if(FullGameList.size() == 0)
        ReadGameList();
    if(gameFilter)
        GameFilter.assign(gameFilter);

    FilteredList.clear();
    AvailableSearchChars.clear();

    for (u32 i = 0; i < FullGameList.size(); ++i)
    {
		struct discHdr *header = &FullGameList[i];

		/* Register game */
		NewTitles::Instance()->CheckGame(header->id);

		/* Filters */
		if (Settings.fave)
		{
			struct Game_NUM* game_num = CFG_get_game_num(header->id);
			if (!game_num || game_num->favorite==0)
				continue;
		}

		//ignore uLoader cfg "iso".  i was told it is "__CFG_"  but not confirmed
		if (strncasecmp((char*)header->id, "__CFG_", 6) == 0)
			continue;

		if (Settings.parentalcontrol && !Settings.godmode)
			if (get_block(header) >= Settings.parentalcontrol)
				continue;

		/* Other parental control method */
		if (Settings.parentalcontrol == 0 && Settings.godmode == 0 && Settings.parental.enabled == 1)
		{
			// Check game rating in WiiTDB, since the default Wii parental control setting is enabled
			s32 rating = GetRatingForGame((char *) header->id);
			if ((rating != -1 && rating > Settings.parental.rating) ||
			    (rating == -1 && get_pegi_block(header) > Settings.parental.rating))
			{
				continue;
			}
		}

        wchar_t *gameName = charToWideChar(get_title(header));

		if(gameName && *GameFilter.c_str())
		{
			if(wcsnicmp(gameName, GameFilter.c_str(), GameFilter.size()) != 0)
			{
				delete [] gameName;
				continue;
			}
		}

		if(gameName)
		{
		    if(wcslen(gameName) > GameFilter.size() && AvailableSearchChars.find(gameName[GameFilter.size()]) == std::string::npos)
                AvailableSearchChars.push_back(gameName[GameFilter.size()]);
            delete [] gameName;
		}

		FilteredList.push_back(header);
	}

	NewTitles::Instance()->Save();

    AvailableSearchChars.push_back(L'\0');

    if(FilteredList.size() < 2)
        AvailableSearchChars.clear();

	SortList();

	return FilteredList.size();
}

int GameList::LoadUnfiltered()
{
    if(FullGameList.size() == 0)
        ReadGameList();

    GameFilter.clear();
    AvailableSearchChars.clear();
    FilteredList.clear();

    for (u32 i = 0; i < FullGameList.size(); ++i)
    {
		struct discHdr *header = &FullGameList[i];

		/* Register game */
		NewTitles::Instance()->CheckGame(header->id);

        wchar_t *gameName = charToWideChar(get_title(header));
		if(gameName)
		{
		    if(wcslen(gameName) > GameFilter.size() && AvailableSearchChars.find(gameName[GameFilter.size()]) == std::string::npos)
                AvailableSearchChars.push_back(gameName[GameFilter.size()]);
            delete [] gameName;
		}

		FilteredList.push_back(header);
	}

	NewTitles::Instance()->Save();

    AvailableSearchChars.push_back(L'\0');

    if(FilteredList.size() < 2)
        AvailableSearchChars.clear();

    SortList();

	return FilteredList.size();
}

void GameList::SortList()
{
	if(FilteredList.size() < 2)
        return;

	if (Settings.sort == PLAYCOUNT)
	{
        std::sort(FilteredList.begin(), FilteredList.end(), PlaycountSortCallback);
	}
	else if (Settings.fave)
	{
        std::sort(FilteredList.begin(), FilteredList.end(), FavoriteSortCallback);
	}
	else
	{
        std::sort(FilteredList.begin(), FilteredList.end(), NameSortCallback);
	}

    if(AvailableSearchChars.size() > 1)
        std::sort(AvailableSearchChars.begin(), AvailableSearchChars.end(), WCharSortCallback);

}

bool GameList::NameSortCallback(const struct discHdr *a, const struct discHdr *b)
{
    return (strcasecmp(get_title((struct discHdr *) a), get_title((struct discHdr *) b)) < 0);
}

bool GameList::PlaycountSortCallback(const struct discHdr *a, const struct discHdr *b)
{
    struct Game_NUM* game_num1 = CFG_get_game_num(a->id);
    struct Game_NUM* game_num2 = CFG_get_game_num(b->id);
    int count1 = 0, count2 = 0;

    if(game_num1)
        count1 = game_num1->count;
    if(game_num2)
        count2 = game_num2->count;

    if(count1 == count2)
        return NameSortCallback(a, b);

    return (count1 > count2);
}

bool GameList::FavoriteSortCallback(const struct discHdr *a, const struct discHdr *b)
{
    struct Game_NUM* game_num1 = CFG_get_game_num(a->id);
    struct Game_NUM* game_num2 = CFG_get_game_num(b->id);
    int fav1 = 0, fav2 = 0;

    if(game_num1)
        fav1 = game_num1->favorite;
    if(game_num2)
        fav2 = game_num2->favorite;

    if(fav1 == fav2);
        return NameSortCallback(a, b);

    return (fav1 > fav2);
}