/****************************************************************************
 * 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();
        }
    }
}