/****************************************************************************
 * SettingsPrompts
 * USB Loader GX 2009
 *
 * Backgroundmusic
 ***************************************************************************/
#include <sys/dir.h>
#include "gui_bgm.h"
#include "menu.h"

GuiBGM::GuiBGM(const u8 *s, int l, int v)
    :GuiSound(s, l, v)
{
    loop = 0;
    loopMode = ONCE;
    currentPath = NULL;
    currentPlaying = 0;

    //shouldn't be needed but
    //fixes some kind of weird bug in ogg system
    GuiSound::Load(s, l, v);
}

GuiBGM::~GuiBGM()
{
    if(currentPath)
        delete [] currentPath;

    ClearList();
};

void GuiBGM::SetLoop(bool l)
{
}

void GuiBGM::SetLoop(int l)
{
    loop = false;
    loopMode = ONCE;

    if(l == LOOP)
    {
        loop = true;
    }
    else
        loopMode = l;
}

bool GuiBGM::Load(const char *path)
{
    if(!path)
    {
        LoadStandard();
        return false;
    }
    if(strcmp(path, "") == 0)
    {
        LoadStandard();
        return false;
    }

    if(!GuiSound::Load(path))
    {
        LoadStandard();
        return false;
    }

    return ParsePath(path);
}

bool GuiBGM::LoadStandard()
{
    ClearList();
    if(currentPath)
    {
        delete [] currentPath;
        currentPath = NULL;
    }

    strcpy(Settings.ogg_path, "");

    bool ret = GuiSound::Load(bg_music_ogg, bg_music_ogg_size, true);

    if(ret)
        Play();

    return ret;
}

bool GuiBGM::ParsePath(const char * folderpath)
{
    ClearList();

    if(currentPath)
        delete [] currentPath;

    currentPath = new char[strlen(folderpath)+1];
    sprintf(currentPath, "%s", folderpath);

    char * isdirpath = strrchr(folderpath, '.');
    if(isdirpath)
    {
        char * pathptr = strrchr(currentPath, '/');
        if(pathptr)
        {
            pathptr++;
            pathptr[0] = 0;
        }
    }

    char * LoadedFilename = strrchr(folderpath, '/')+1;

    char filename[1024];
    struct stat st;

    DIR_ITER * dir = diropen(currentPath);
    if (dir == NULL)
    {
        LoadStandard();
        return false;
    }
    u32 counter = 0;

    while (dirnext(dir,filename,&st) == 0)
    {
        char * fileext = strrchr(filename, '.');
        if(fileext)
        {
            if(strcasecmp(fileext, ".mp3") == 0 || strcasecmp(fileext, ".ogg") == 0
                || strcasecmp(fileext, ".wav") == 0)
            {
                AddEntrie(filename);

                if(strcmp(LoadedFilename, filename) == 0)
                    currentPlaying = counter;

                counter++;
            }
        }
    }

    dirclose(dir);

    snprintf(Settings.ogg_path, sizeof(Settings.ogg_path), "%s", folderpath);

    return true;
}

void GuiBGM::AddEntrie(const char * filename)
{
    if(!filename)
        return;

    char * NewEntrie = new char[strlen(filename)+1];
    sprintf(NewEntrie, "%s", filename);

    PlayList.push_back(NewEntrie);
}

void GuiBGM::ClearList()
{
    for(u32 i = 0; i < PlayList.size(); i++)
    {
        if(PlayList.at(i) != NULL)
        {
            delete [] PlayList.at(i);
            PlayList.at(i) = NULL;
        }
    }

    PlayList.clear();
}

bool GuiBGM::PlayNext()
{
    if(!currentPath)
        return false;

    currentPlaying++;
    if(currentPlaying >= (int) PlayList.size())
        currentPlaying = 0;

    snprintf(Settings.ogg_path, sizeof(Settings.ogg_path), "%s%s", currentPath, PlayList.at(currentPlaying));

    if(!GuiSound::Load(Settings.ogg_path))
        return false;

    Play();

    return true;
}

bool GuiBGM::PlayPrevious()
{
    if(!currentPath)
        return false;

    currentPlaying--;
    if(currentPlaying < 0)
        currentPlaying = PlayList.size()-1;

    snprintf(Settings.ogg_path, sizeof(Settings.ogg_path), "%s%s", currentPath, PlayList.at(currentPlaying));

    if(!GuiSound::Load(Settings.ogg_path))
        return false;

    Play();

    return true;
}

bool GuiBGM::PlayRandom()
{
    if(!currentPath)
        return false;

    srand (time(NULL));

    currentPlaying = rand() % PlayList.size();

    //just in case
    if(currentPlaying < 0)
        currentPlaying = PlayList.size()-1;
    else if(currentPlaying >= (int) PlayList.size())
        currentPlaying = 0;

    snprintf(Settings.ogg_path, sizeof(Settings.ogg_path), "%s%s", currentPath, PlayList.at(currentPlaying));

    if(!GuiSound::Load(Settings.ogg_path))
        return false;

    Play();

    return true;
}

void GuiBGM::UpdateState()
{
    if(!IsPlaying())
    {
        if(loopMode == DIR_LOOP)
        {
            PlayNext();
        }
        else if(loopMode == RANDOM_BGM)
        {
            PlayRandom();
        }
    }
}