/****************************************************************************
* 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/CSettings.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)
{
    if(char2 == 0)
        return true;
    if(char1 == 0)
        return false;

    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);
}