/****************************************************************************
 * libwiigui
 *
 * gui_customoptionbrowser.cpp
 *
 * GUI class definitions
 ***************************************************************************/

#include "gui.h"
#include "../wpad.h"
#include "../main.h"
#include "../gecko.h"
#include "../settings/CSettings.h"
#include "gui_customoptionbrowser.h"
#include "themes/CTheme.h"

#include <unistd.h>

#define GAMESELECTSIZE      30
#define OPTION_LIST_PADDING PAGESIZE

customOptionList::customOptionList(int Size)
{
    name = value = NULL;
    size = 0;
    SetSize(Size == 0 ? PAGESIZE : Size);
    length = Size;
    changed = false;
}
customOptionList::~customOptionList()
{
    for (int i = 0; i < size; i++)
    {
        free(name[i]);
        free(value[i]);
    }
    delete[] name;
    delete[] value;
}
void customOptionList::SetLength(int Length) //set number of lines
{
    if (Length < 0 || Length == length) return;

    if (Length > size) SetSize(Length + OPTION_LIST_PADDING);
    if (Length < length)
    {
        for (int i = Length; i < length; i++)
        {
            free(name[i]); // clear unused
            name[i] = NULL;
            free(value[i]);
            value[i] = NULL;
        }
    }
    length = Length;
    changed = true;
}
void customOptionList::SetSize(int Size) //set number of lines
{
    if (Size < 0 || Size == size) return;

    if (Size > size)
    {
        char **newName = new char *[Size];
        char **newValue = new char *[Size];
        int i;
        for (i = 0; i < size; i++)
        {
            newName[i] = name[i]; // copy
            newValue[i] = value[i];
        }
        for (; i < Size; i++)
        {
            newName[i] = NULL; // fill rest with NULL
            newValue[i] = NULL;
        }
        delete[] name;
        name = newName; // set new
        delete[] value;
        value = newValue;
        size = Size;
    }
}
void customOptionList::SetName(int i, const char *format, ...)
{
    if (i >= length) SetLength(i + 1);

    if (i >= 0 && i < length)
    {
        if (name[i]) free(name[i]);
        name[i] = 0;
        va_list va;
        va_start( va, format );
        vasprintf(&name[i], format, va);
        va_end( va );
        changed = true;
    }
    //gprintf("customOptionList::SetName( %d, %s )\n", i, name[i] );
}
void customOptionList::SetValue(int i, const char *format, ...)
{
    if (i >= length) SetLength(i + 1);

    if (i >= 0 && i < length)
    {
        char *tmp = 0;
        va_list va;
        va_start( va, format );
        vasprintf(&tmp, format, va);
        va_end( va );

        if (tmp)
        {
            if (value[i] && !strcmp(tmp, value[i]))
                free(tmp);
            else
            {
                free(value[i]);
                value[i] = tmp;
                changed = true;
            }
        }
    }
    //gprintf("customOptionList::SetValue( %d, %s )\n", i, value[i] );
}
void customOptionList::Clear(bool OnlyValue/*=false*/)
{
    for (int i = 0; i < size; i++)
    {
        if (!OnlyValue)
        {
            free(name[i]);
            name[i] = NULL;
        }
        free(value[i]);
        value[i] = NULL;
    }
    changed = true;
}

/**
 * Constructor for the GuiCustomOptionBrowser class.
 */
GuiCustomOptionBrowser::GuiCustomOptionBrowser(int w, int h, customOptionList * l, const char *themePath,
        const char *custombg, const u8 *imagebg, int scrollon, int col2)
{
    width = w;
    height = h;
    options = l;
    size = PAGESIZE;
    scrollbaron = scrollon;
    selectable = true;
    listOffset = this->FindMenuItem(-1, 1);
    selectedItem = 0;
    focus = 1; // allow focus
    coL2 = col2;
    char imgPath[100];

    trigA = new GuiTrigger;
    trigA->SetSimpleTrigger(-1, WPAD_BUTTON_A | WPAD_CLASSIC_BUTTON_A, PAD_BUTTON_A);
    trigHeldA = new GuiTrigger;
    trigHeldA->SetHeldTrigger(-1, WPAD_BUTTON_A, PAD_BUTTON_A);
    btnSoundClick = new GuiSound(button_click_pcm, button_click_pcm_size, Settings.sfxvolume);

    snprintf(imgPath, sizeof(imgPath), "%s%s", themePath, custombg);
    bgOptions = new GuiImageData(imgPath, imagebg);

    bgOptionsImg = new GuiImage(bgOptions);
    bgOptionsImg->SetParent(this);
    bgOptionsImg->SetAlignment(ALIGN_LEFT, ALIGN_MIDDLE);

    snprintf(imgPath, sizeof(imgPath), "%sbg_options_entry.png", themePath);
    bgOptionsEntry = new GuiImageData(imgPath, bg_options_entry_png);

    snprintf(imgPath, sizeof(imgPath), "%sscrollbar.png", themePath);
    scrollbar = new GuiImageData(imgPath, scrollbar_png);
    scrollbarImg = new GuiImage(scrollbar);
    scrollbarImg->SetParent(this);
    scrollbarImg->SetAlignment(ALIGN_RIGHT, ALIGN_TOP);
    scrollbarImg->SetPosition(0, 4);

    snprintf(imgPath, sizeof(imgPath), "%sscrollbar_arrowdown.png", themePath);
    arrowDown = new GuiImageData(imgPath, scrollbar_arrowdown_png);
    arrowDownImg = new GuiImage(arrowDown);
    arrowDownOver = new GuiImageData(imgPath, scrollbar_arrowdown_png);
    arrowDownOverImg = new GuiImage(arrowDownOver);
    snprintf(imgPath, sizeof(imgPath), "%sscrollbar_arrowup.png", themePath);
    arrowUp = new GuiImageData(imgPath, scrollbar_arrowup_png);
    arrowUpImg = new GuiImage(arrowUp);
    arrowUpOver = new GuiImageData(imgPath, scrollbar_arrowup_png);
    arrowUpOverImg = new GuiImage(arrowUpOver);
    snprintf(imgPath, sizeof(imgPath), "%sscrollbar_box.png", themePath);
    scrollbarBox = new GuiImageData(imgPath, scrollbar_box_png);
    scrollbarBoxImg = new GuiImage(scrollbarBox);
    scrollbarBoxOver = new GuiImageData(imgPath, scrollbar_box_png);
    scrollbarBoxOverImg = new GuiImage(scrollbarBoxOver);

    arrowUpBtn = new GuiButton(arrowUpImg->GetWidth(), arrowUpImg->GetHeight());
    arrowUpBtn->SetParent(this);
    arrowUpBtn->SetImage(arrowUpImg);
    arrowUpBtn->SetImageOver(arrowUpOverImg);
    arrowUpBtn->SetImageHold(arrowUpOverImg);
    arrowUpBtn->SetAlignment(ALIGN_CENTRE, ALIGN_TOP);
    arrowUpBtn->SetPosition(width / 2 - 18 + 7, -18);
    arrowUpBtn->SetSelectable(false);
    arrowUpBtn->SetTrigger(trigA);
    arrowUpBtn->SetEffectOnOver(EFFECT_SCALE, 50, 130);
    arrowUpBtn->SetSoundClick(btnSoundClick);

    arrowDownBtn = new GuiButton(arrowDownImg->GetWidth(), arrowDownImg->GetHeight());
    arrowDownBtn->SetParent(this);
    arrowDownBtn->SetImage(arrowDownImg);
    arrowDownBtn->SetImageOver(arrowDownOverImg);
    arrowDownBtn->SetImageHold(arrowDownOverImg);
    arrowDownBtn->SetAlignment(ALIGN_CENTRE, ALIGN_BOTTOM);
    arrowDownBtn->SetPosition(width / 2 - 18 + 7, 18);
    arrowDownBtn->SetSelectable(false);
    arrowDownBtn->SetTrigger(trigA);
    arrowDownBtn->SetEffectOnOver(EFFECT_SCALE, 50, 130);
    arrowDownBtn->SetSoundClick(btnSoundClick);

    scrollbarBoxBtn = new GuiButton(scrollbarBoxImg->GetWidth(), scrollbarBoxImg->GetHeight());
    scrollbarBoxBtn->SetParent(this);
    scrollbarBoxBtn->SetImage(scrollbarBoxImg);
    scrollbarBoxBtn->SetImageOver(scrollbarBoxOverImg);
    scrollbarBoxBtn->SetImageHold(scrollbarBoxOverImg);
    scrollbarBoxBtn->SetAlignment(ALIGN_CENTRE, ALIGN_TOP);
    scrollbarBoxBtn->SetSelectable(false);
    scrollbarBoxBtn->SetEffectOnOver(EFFECT_SCALE, 50, 120);
    scrollbarBoxBtn->SetMinY(0);
    scrollbarBoxBtn->SetMaxY(height - 30);
    scrollbarBoxBtn->SetHoldable(true);
    scrollbarBoxBtn->SetTrigger(trigHeldA);

    optionIndex = new int[size];
    optionVal = new GuiText *[size];
    optionValOver = new GuiText *[size];
    optionBtn = new GuiButton *[size];
    optionTxt = new GuiText *[size];
    optionBg = new GuiImage *[size];

    for (int i = 0; i < size; i++)
    {
        optionTxt[i] = new GuiText(options->GetName(i), 20, Theme.settingstext);
        optionTxt[i]->SetAlignment(ALIGN_LEFT, ALIGN_MIDDLE);
        optionTxt[i]->SetPosition(24, 0);
        optionTxt[i]->SetMaxWidth(bgOptionsImg->GetWidth() - (coL2 + 24), DOTTED);

        optionBg[i] = new GuiImage(bgOptionsEntry);

        optionVal[i] = new GuiText((char *) NULL, 20, Theme.settingstext);
        optionVal[i]->SetAlignment(ALIGN_LEFT, ALIGN_MIDDLE);

        optionValOver[i] = new GuiText((char *) NULL, 20, Theme.settingstext);
        optionValOver[i]->SetAlignment(ALIGN_LEFT, ALIGN_MIDDLE);

        optionBtn[i] = new GuiButton(width - 28, GAMESELECTSIZE);
        optionBtn[i]->SetParent(this);
        optionBtn[i]->SetLabel(optionTxt[i], 0);
        optionBtn[i]->SetLabel(optionVal[i], 1);
        optionBtn[i]->SetLabelOver(optionValOver[i], 1);
        optionBtn[i]->SetImageOver(optionBg[i]);
        optionBtn[i]->SetPosition(10, GAMESELECTSIZE * i + 4);
        optionBtn[i]->SetRumble(false);
        optionBtn[i]->SetTrigger(trigA);
        optionBtn[i]->SetSoundClick(btnSoundClick);

    }
    UpdateListEntries();
}

/**
 * Destructor for the GuiCustomOptionBrowser class.
 */
GuiCustomOptionBrowser::~GuiCustomOptionBrowser()
{
    delete arrowUpBtn;
    delete arrowDownBtn;
    delete scrollbarBoxBtn;
    delete scrollbarImg;
    delete arrowDownImg;
    delete arrowDownOverImg;
    delete arrowUpImg;
    delete arrowUpOverImg;
    delete scrollbarBoxImg;
    delete scrollbarBoxOverImg;
    delete scrollbar;
    delete arrowDown;
    delete arrowDownOver;
    delete arrowUp;
    delete arrowUpOver;
    delete scrollbarBox;
    delete scrollbarBoxOver;

    delete bgOptionsImg;
    delete bgOptions;
    delete bgOptionsEntry;

    delete trigA;
    delete trigHeldA;
    delete btnSoundClick;

    for (int i = 0; i < size; i++)
    {
        delete optionTxt[i];
        delete optionVal[i];
        delete optionValOver[i];
        delete optionBg[i];
        delete optionBtn[i];
    }
    delete[] optionIndex;
    delete[] optionVal;
    delete[] optionValOver;
    delete[] optionBtn;
    delete[] optionTxt;
    delete[] optionBg;
}

void GuiCustomOptionBrowser::SetFocus(int f)
{
    LOCK( this );
    focus = f;

    for (int i = 0; i < size; i++)
        optionBtn[i]->ResetState();

    if (f == 1) optionBtn[selectedItem]->SetState(STATE_SELECTED);
}

void GuiCustomOptionBrowser::ResetState()
{
    LOCK( this );
    if (state != STATE_DISABLED)
    {
        state = STATE_DEFAULT;
        stateChan = -1;
    }

    for (int i = 0; i < size; i++)
    {
        optionBtn[i]->ResetState();
    }
}

int GuiCustomOptionBrowser::GetClickedOption()
{
    int found = -1;
    for (int i = 0; i < size; i++)
    {
        if (optionBtn[i]->GetState() == STATE_CLICKED)
        {
            optionBtn[i]->SetState(STATE_SELECTED);
            found = optionIndex[i];
            break;
        }
    }
    return found;
}

int GuiCustomOptionBrowser::GetSelectedOption()
{
    int found = -1;
    for (int i = 0; i < size; i++)
    {
        if (optionBtn[i]->GetState() == STATE_SELECTED)
        {
            found = optionIndex[i];
            break;
        }
    }
    return found;
}

void GuiCustomOptionBrowser::SetClickable(bool enable)
{
    for (int i = 0; i < size; i++)
    {
        optionBtn[i]->SetClickable(enable);
    }
}

void GuiCustomOptionBrowser::SetScrollbar(int enable)
{
    scrollbaron = enable;
}

void GuiCustomOptionBrowser::SetOffset(int optionnumber)
{
    listOffset = optionnumber;
    selectedItem = optionnumber;
}

/****************************************************************************
 * FindMenuItem
 *
 * Help function to find the next visible menu item on the list
 ***************************************************************************/

int GuiCustomOptionBrowser::FindMenuItem(int currentItem, int direction)
{
    int nextItem = currentItem + direction;

    if (nextItem < 0 || nextItem >= options->GetLength()) return -1;

    if (strlen(options->GetName(nextItem)) > 0)
        return nextItem;
    else return FindMenuItem(nextItem, direction);
}

/**
 * Draw the button on screen
 */
void GuiCustomOptionBrowser::Draw()
{
    LOCK( this );
    if (!this->IsVisible()) return;

    bgOptionsImg->Draw();

    int next = listOffset;

    for (int i = 0; i < size; i++)
    {
        if (next >= 0)
        {
            optionBtn[i]->Draw();
            next = this->FindMenuItem(next, 1);
        }
        else break;
    }

    if (scrollbaron == 1)
    {
        scrollbarImg->Draw();
        arrowUpBtn->Draw();
        arrowDownBtn->Draw();
        scrollbarBoxBtn->Draw();
    }
    this->UpdateEffects();
}

void GuiCustomOptionBrowser::UpdateListEntries()
{
    scrollbaron = options->GetLength() > size;
    if (listOffset < 0) listOffset = this->FindMenuItem(-1, 1);
    int next = listOffset;

    int maxNameWidth = 0;
    for (int i = 0; i < size; i++)
    {
        if (next >= 0)
        {
            if (optionBtn[i]->GetState() == STATE_DISABLED)
            {
                optionBtn[i]->SetVisible(true);
                optionBtn[i]->SetState(STATE_DEFAULT);
            }

            optionTxt[i]->SetText(options->GetName(next));
            if (maxNameWidth < optionTxt[i]->GetTextWidth()) maxNameWidth = optionTxt[i]->GetTextWidth();
            optionVal[i]->SetText(options->GetValue(next));
            optionValOver[i]->SetText(options->GetValue(next));

            optionIndex[i] = next;
            next = this->FindMenuItem(next, 1);
        }
        else
        {
            optionBtn[i]->SetVisible(false);
            optionBtn[i]->SetState(STATE_DISABLED);
        }
    }
    if (coL2 < (24 + maxNameWidth + 16)) coL2 = 24 + maxNameWidth + 16;
    for (int i = 0; i < size; i++)
    {
        if (optionBtn[i]->GetState() != STATE_DISABLED)
        {
            optionVal[i]->SetPosition(coL2, 0);
            optionVal[i]->SetMaxWidth(bgOptionsImg->GetWidth() - (coL2 + 24), DOTTED);

            optionValOver[i]->SetPosition(coL2, 0);
            optionValOver[i]->SetMaxWidth(bgOptionsImg->GetWidth() - (coL2 + 24), SCROLL_HORIZONTAL);
        }
    }
}

void GuiCustomOptionBrowser::Update(GuiTrigger * t)
{
    LOCK( this );
    int next, prev, lang = options->GetLength();

    if (state == STATE_DISABLED || !t) return;

    if (options->IsChanged())
    {
        coL2 = 0;
        UpdateListEntries();
    }
    int old_listOffset = listOffset;

    if (scrollbaron == 1)
    {
        // update the location of the scroll box based on the position in the option list
        arrowUpBtn->Update(t);
        arrowDownBtn->Update(t);
        scrollbarBoxBtn->Update(t);
    }

    next = listOffset;

    u32 buttonshold = ButtonsHold();

    if (buttonshold != WPAD_BUTTON_UP && buttonshold != WPAD_BUTTON_DOWN)
    {
        for (int i = 0; i < size; i++)
        {
            if (next >= 0) next = this->FindMenuItem(next, 1);

            if (focus)
            {
                if (i != selectedItem && optionBtn[i]->GetState() == STATE_SELECTED)
                {
                    optionBtn[i]->ResetState();
                }
                else if (i == selectedItem && optionBtn[i]->GetState() == STATE_DEFAULT)
                {
                    optionBtn[selectedItem]->SetState(STATE_SELECTED);
                }
            }

            optionBtn[i]->Update(t);

            if (optionBtn[i]->GetState() == STATE_SELECTED)
            {
                selectedItem = i;
            }
        }
    }

    // pad/joystick navigation
    if (!focus) return; // skip navigation

    if (t->Down())
    {
        next = this->FindMenuItem(optionIndex[selectedItem], 1);

        if (next >= 0)
        {
            if (selectedItem == size - 1)
            {
                // move list down by 1
                listOffset = this->FindMenuItem(listOffset, 1);
            }
            else if (optionBtn[selectedItem + 1]->IsVisible())
            {
                optionBtn[selectedItem]->ResetState();
                optionBtn[selectedItem + 1]->SetState(STATE_SELECTED, t->chan);
                selectedItem++;
            }
        }
    }
    else if (t->Up())
    {
        prev = this->FindMenuItem(optionIndex[selectedItem], -1);

        if (prev >= 0)
        {
            if (selectedItem == 0)
            {
                // move list up by 1
                listOffset = prev;
            }
            else
            {
                optionBtn[selectedItem]->ResetState();
                optionBtn[selectedItem - 1]->SetState(STATE_SELECTED, t->chan);
                selectedItem--;
            }
        }
    }

    if (scrollbaron == 1)
    {
        if (arrowDownBtn->GetState() == STATE_CLICKED || arrowDownBtn->GetState() == STATE_HELD)
        {

            next = this->FindMenuItem(optionIndex[selectedItem], 1);

            if (next >= 0)
            {
                if (selectedItem == size - 1)
                {
                    // move list down by 1
                    listOffset = this->FindMenuItem(listOffset, 1);
                }
                else if (optionBtn[selectedItem + 1]->IsVisible())
                {
                    optionBtn[selectedItem]->ResetState();
                    optionBtn[selectedItem + 1]->SetState(STATE_SELECTED, t->chan);
                    selectedItem++;
                }
                scrollbarBoxBtn->Draw();
                usleep(35000);
            }
            if (buttonshold != WPAD_BUTTON_A)
            {
                arrowDownBtn->ResetState();
            }
        }
        else if (arrowUpBtn->GetState() == STATE_CLICKED || arrowUpBtn->GetState() == STATE_HELD)
        {
            prev = this->FindMenuItem(optionIndex[selectedItem], -1);

            if (prev >= 0)
            {
                if (selectedItem == 0)
                {
                    // move list up by 1
                    listOffset = prev;
                }
                else
                {
                    optionBtn[selectedItem]->ResetState();
                    optionBtn[selectedItem - 1]->SetState(STATE_SELECTED, t->chan);
                    selectedItem--;
                }
                scrollbarBoxBtn->Draw();
                usleep(35000);
            }
            if (buttonshold != WPAD_BUTTON_A)
            {
                arrowUpBtn->ResetState();
            }
        }

        if (scrollbarBoxBtn->GetState() == STATE_HELD && scrollbarBoxBtn->GetStateChan() == t->chan && t->wpad.ir.valid
                && options->GetLength() > size)
        {
            scrollbarBoxBtn->SetPosition(width / 2 - 18 + 7, 0);

            int position = t->wpad.ir.y - 50 - scrollbarBoxBtn->GetTop();

            listOffset = (position * lang) / 180 - selectedItem;

            if (listOffset <= 0)
            {
                listOffset = 0;
                selectedItem = 0;
            }
            else if (listOffset + size >= lang)
            {
                listOffset = lang - size;
                selectedItem = size - 1;
            }
        }
        int positionbar = 237 * (listOffset + selectedItem) / lang;

        if (positionbar > 216) positionbar = 216;
        scrollbarBoxBtn->SetPosition(width / 2 - 18 + 7, positionbar + 8);

        if (t->Right())
        {
            if (listOffset < lang && lang > size)
            {
                listOffset = listOffset + size;
                if (listOffset + size >= lang) listOffset = lang - size;
            }
        }
        else if (t->Left())
        {
            if (listOffset > 0)
            {
                listOffset = listOffset - size;
                if (listOffset < 0) listOffset = 0;
            }
        }
    }

    if (old_listOffset != listOffset) UpdateListEntries();

    if (updateCB) updateCB(this);
}