/****************************************************************************
 * 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 <unistd.h>
#include <gccore.h>
#include "settings/CSettings.h"
#include "themes/CTheme.h"
#include "prompts/PromptWindows.h"
#include "prompts/DiscBrowser.h"
#include "language/gettext.h"
#include "wad/nandtitle.h"
#include "IndGameLoadSM.hpp"

static const char * OnOffText[MAX_ON_OFF] =
{
    trNOOP( "OFF" ),
    trNOOP( "ON" )
};

static const char * VideoModeText[VIDEO_MODE_MAX] =
{
    trNOOP( "Disc Default" ),
    trNOOP( "System Default" ),
    trNOOP( "AutoPatch" ),
    trNOOP( "Force PAL50" ),
    trNOOP( "Force PAL60" ),
    trNOOP( "Force NTSC" )
};

static const char * LanguageText[MAX_LANGUAGE] =
{
    trNOOP( "Disc Default" ),
    trNOOP( "Console Default" ),
    trNOOP( "Japanese" ),
    trNOOP( "English" ),
    trNOOP( "German" ),
    trNOOP( "French" ),
    trNOOP( "Spanish" ),
    trNOOP( "Italian" ),
    trNOOP( "Dutch" ),
    trNOOP( "SChinese" ),
    trNOOP( "TChinese" ),
    trNOOP( "Korean" )
};

static const char * Error002Text[3] =
{
    trNOOP( "No" ),
    trNOOP( "Yes" ),
    trNOOP( "Anti" )
};

static const char * ParentalText[5] =
{
    trNOOP( "0 (Everyone)" ),
    trNOOP( "1 (Child 7+)" ),
    trNOOP( "2 (Teen 12+)" ),
    trNOOP( "3 (Mature 16+)" ),
    trNOOP( "4 (Adults Only 18+)" )
};

static const char * AlternateDOLText[] =
{
    trNOOP( "OFF" ),
    trNOOP( "Select a DOL from Game" ),
    trNOOP( "Load From SD/USB" ),
};

IndGameLoadSM::IndGameLoadSM(const char * GameID)
    : SettingsMenu(tr("Game Load"), &GuiOptions, MENU_NONE)
{
    //! Setup default settings from global settings
    snprintf(GameConfig.id, sizeof(GameConfig.id), "%s", (char *) GameID);
    SetDefaultConfig();

    GameCFG * existCFG = GameSettings.GetGameCFG(GameID);

    //! Overwrite with existing if available
	if (existCFG)
	    memcpy(&GameConfig, existCFG, sizeof(GameCFG));

    if(!btnOutline)
        btnOutline = Resources::GetImageData("button_dialogue_box.png");
    if(!trigA)
        trigA = new GuiTrigger();
    trigA->SetSimpleTrigger(-1, WPAD_BUTTON_A | WPAD_CLASSIC_BUTTON_A, PAD_BUTTON_A);

    saveBtnTxt = new GuiText(tr( "Save" ), 22, Theme.prompttext);
    saveBtnTxt->SetMaxWidth(btnOutline->GetWidth() - 30);
    saveBtnImg = new GuiImage (btnOutline);
    if (Settings.wsprompt == ON)
    {
        saveBtnTxt->SetWidescreen(Settings.widescreen);
        saveBtnImg->SetWidescreen(Settings.widescreen);
    }
    saveBtn = new GuiButton(saveBtnImg, saveBtnImg, 2, 3, 180, 400, trigA, btnSoundOver, btnSoundClick2, 1);
    saveBtn->SetLabel(saveBtnTxt);
    Append(saveBtn);

    SetOptionNames();
    SetOptionValues();
}

IndGameLoadSM::~IndGameLoadSM()
{
    HaltGui();
    //! The rest is destroyed in SettingsMenu.cpp
    Remove(saveBtn);
    delete saveBtnTxt;
    delete saveBtnImg;
    delete saveBtn;
    ResumeGui();
}

void IndGameLoadSM::SetDefaultConfig()
{
    GameConfig.video = Settings.videomode;
    GameConfig.language = Settings.language;
    GameConfig.ocarina = Settings.ocarina;
    GameConfig.vipatch = Settings.videopatch;
    GameConfig.ios = Settings.cios;
    GameConfig.parentalcontrol = 0;
    GameConfig.errorfix002 = Settings.error002;
    GameConfig.patchcountrystrings = Settings.patchcountrystrings;
    GameConfig.loadalternatedol = OFF;
    GameConfig.alternatedolstart = 0;
    GameConfig.iosreloadblock = OFF;
    strcpy(GameConfig.alternatedolname, "");
    GameConfig.returnTo = 1;
    GameConfig.Locked = 0;
}

void IndGameLoadSM::SetOptionNames()
{
    int Idx = 0;

    Options->SetName(Idx++, "%s", tr( "Video Mode" ));
    Options->SetName(Idx++, "%s", tr( "VIDTV Patch" ));
    Options->SetName(Idx++, "%s", tr( "Game Language" ));
    Options->SetName(Idx++, "%s", tr( "Patch Country Strings" ));
    Options->SetName(Idx++, "%s", tr( "Ocarina" ));
    Options->SetName(Idx++, "%s", tr( "Game IOS" ));
    Options->SetName(Idx++, "%s", tr( "Parental Control" ));
    Options->SetName(Idx++, "%s", tr( "Error 002 fix" ));
    Options->SetName(Idx++, "%s", tr( "Return To" ));
    Options->SetName(Idx++, "%s", tr( "Alternate DOL" ));
    Options->SetName(Idx++, "%s", tr( "Select DOL Offset" ));
    Options->SetName(Idx++, "%s", tr( "Block IOS Reload" ));
    Options->SetName(Idx++, "%s", tr( "Game Lock" ));
}

void IndGameLoadSM::SetOptionValues()
{
    int Idx = 0;

    //! Settings: Video Mode
    Options->SetValue(Idx++, "%s", tr(VideoModeText[GameConfig.video]));

    //! Settings: VIDTV Patch
    Options->SetValue(Idx++, "%s", tr(OnOffText[GameConfig.vipatch]));

    //! Settings: Game Language
    Options->SetValue(Idx++, "%s", tr(LanguageText[GameConfig.language]));

    //! Settings: Patch Country Strings
    Options->SetValue(Idx++, "%s", tr(OnOffText[GameConfig.patchcountrystrings]));

    //! Settings: Ocarina
    Options->SetValue(Idx++, "%s", tr(OnOffText[GameConfig.ocarina]));

    //! Settings: Game IOS
    Options->SetValue(Idx++, "%i", GameConfig.ios);

    //! Settings: Parental Control
    Options->SetValue(Idx++, "%s", tr(ParentalText[GameConfig.parentalcontrol]));

    //! Settings: Error 002 fix
    Options->SetValue(Idx++, "%s", tr(Error002Text[GameConfig.errorfix002]));

    //! Settings: Return To
    if(GameConfig.returnTo)
    {
        const char* TitleName = NULL;
        int haveTitle = NandTitles.FindU32(Settings.returnTo);
        if (haveTitle >= 0)
            TitleName = NandTitles.NameFromIndex(haveTitle);
        Options->SetValue(Idx++, "%s", TitleName ? TitleName : strlen(Settings.returnTo) > 0 ?
                                        Settings.returnTo : tr( OnOffText[0] ));
    }
    else
    {
        Options->SetValue(Idx++, "%s", tr( OnOffText[0] ));
    }

    //! Settings: Alternate DOL
    Options->SetValue(Idx++, "%s", tr( AlternateDOLText[GameConfig.loadalternatedol] ));

    //! Settings: Select DOL Offset
    if(GameConfig.loadalternatedol != 1)
        Options->SetValue(Idx++, tr("Not required"));
    else
    {
        if(strcmp(GameConfig.alternatedolname, "") != 0)
            Options->SetValue(Idx++, "%i <%s>", GameConfig.alternatedolstart, GameConfig.alternatedolname);
        else
            Options->SetValue(Idx++, "%i", GameConfig.alternatedolstart);
    }

    //! Settings: Block IOS Reload
    Options->SetValue(Idx++, "%s", tr( OnOffText[GameConfig.iosreloadblock] ));

    //! Settings: Game Lock
    Options->SetValue(Idx++, "%s", tr( OnOffText[GameConfig.Locked] ));
}

int IndGameLoadSM::GetMenuInternal()
{
    if (saveBtn->GetState() == STATE_CLICKED)
    {
        if (GameSettings.AddGame(GameConfig) && GameSettings.Save())
        {
            WindowPrompt(tr( "Successfully Saved" ), 0, tr( "OK" ));
        }
        else
            WindowPrompt(tr( "Save Failed. No device inserted?" ), 0, tr( "OK" ));

        saveBtn->ResetState();
    }

    int ret = optionBrowser->GetClickedOption();

    if (ret < 0)
        return MENU_NONE;

    int Idx = -1;

    //! Settings: Video Mode
    if (ret == ++Idx)
    {
        if (++GameConfig.video >= VIDEO_MODE_MAX) GameConfig.video = 0;
    }

    //! Settings: VIDTV Patch
    else if (ret == ++Idx)
    {
        if (++GameConfig.vipatch >= MAX_ON_OFF) GameConfig.vipatch = 0;
    }

    //! Settings: Game Language
    else if (ret == ++Idx)
    {
        if (++GameConfig.language >= MAX_LANGUAGE) GameConfig.language = 0;
    }

    //! Settings: Patch Country Strings
    else if (ret == ++Idx)
    {
        if (++GameConfig.patchcountrystrings >= MAX_ON_OFF) GameConfig.patchcountrystrings = 0;
    }

    //! Settings: Ocarina
    else if (ret == ++Idx)
    {
        if (++GameConfig.ocarina >= MAX_ON_OFF) GameConfig.ocarina = 0;
    }

    //! Settings: Game IOS
    else if (ret == ++Idx)
    {
        char entered[4];
        snprintf(entered, sizeof(entered), "%i", GameConfig.ios);
        if(OnScreenKeyboard(entered, sizeof(entered), 0))
        {
            GameConfig.ios = atoi(entered);
            if(GameConfig.ios < 200) GameConfig.ios = 200;
            else if(GameConfig.ios > 255) GameConfig.ios = 255;

            if(NandTitles.IndexOf(TITLE_ID(1, GameConfig.ios)) < 0)
            {
                WindowPrompt(tr("Warning:"), tr("This IOS was not found on the titles list. If you are sure you have it installed than ignore this warning."), tr("OK"));
            }
            else if(GameConfig.ios == 254)
            {
                WindowPrompt(tr("Warning:"), tr("This IOS is the BootMii ios. If you are sure it is not BootMii and you have something else installed there than ignore this warning."), tr("OK"));
            }
        }
    }

    //! Settings: Parental Control
    else if (ret == ++Idx)
    {
        if (++GameConfig.parentalcontrol >= 5) GameConfig.parentalcontrol = 0;
    }

    //! Settings: Error 002 fix
    else if (ret == ++Idx)
    {
        if (++GameConfig.errorfix002 >= 3) GameConfig.errorfix002 = 0;
    }

    //! Settings: Return To
    else if (ret == ++Idx)
    {
        if (++GameConfig.returnTo >= MAX_ON_OFF) GameConfig.returnTo = 0;
    }

    //! Settings: Alternate DOL
    else if (ret == ++Idx)
    {
        if (++GameConfig.loadalternatedol > 2)
            GameConfig.loadalternatedol = 0;
    }

    //! Settings: Select DOL Offset from Game
    else if (ret == ++Idx && GameConfig.loadalternatedol == 1)
    {
        char filename[10];
        snprintf(filename, 7, "%s", GameConfig.id);

        //alt dol menu for games that require more than a single alt dol
        int autodol = autoSelectDolMenu(filename, false);

        if (autodol > 0)
        {
            GameConfig.alternatedolstart = autodol;
            snprintf(GameConfig.alternatedolname, sizeof(GameConfig.alternatedolname), "%s <%i>", tr( "AUTO" ), autodol);
            SetOptionValues();
            return MENU_NONE;
        }
        else if (autodol == 0)
        {
            GameConfig.loadalternatedol = 0;
            SetOptionValues();
            return MENU_NONE;
        }

        //check to see if we already know the offset of the correct dol
        autodol = autoSelectDol(filename, false);
        //if we do know that offset ask if they want to use it
        if (autodol > 0)
        {
            int dolchoice = WindowPrompt(0, tr( "Do you want to use the alternate DOL that is known to be correct?" ),
                                            tr( "Yes" ), tr( "Pick from a list" ), tr( "Cancel" ));
            if (dolchoice == 1)
            {
                GameConfig.alternatedolstart = autodol;
                snprintf(GameConfig.alternatedolname, sizeof(GameConfig.alternatedolname), "%s <%i>", tr( "AUTO" ), autodol);
            }
            else if (dolchoice == 2) //they want to search for the correct dol themselves
            {
                int res = DiscBrowse(GameConfig.id, GameConfig.alternatedolname, sizeof(GameConfig.alternatedolname));
                if (res >= 0)
                    GameConfig.alternatedolstart = res;
            }
        }
        else
        {
            int res = DiscBrowse(GameConfig.id, GameConfig.alternatedolname, sizeof(GameConfig.alternatedolname));
            if (res >= 0)
            {
                GameConfig.alternatedolstart = res;
                char tmp[170];
                snprintf(tmp, sizeof(tmp), "%s %s - %i", tr( "It seems that you have some information that will be helpful to us. Please pass this information along to the DEV team." ), filename, GameConfig.alternatedolstart);
                WindowPrompt(0, tmp, tr( "OK" ));
            }
        }

        if(GameConfig.alternatedolstart == 0)
            GameConfig.loadalternatedol = 0;
    }

    //! Settings: Block IOS Reload
    else if (ret == ++Idx)
    {
        if (++GameConfig.iosreloadblock >= MAX_ON_OFF) GameConfig.iosreloadblock = 0;
    }

    //! Settings: Game Lock
    else if (ret == ++Idx)
    {
        if (++GameConfig.Locked >= MAX_ON_OFF) GameConfig.Locked = 0;
    }

    SetOptionValues();

    return MENU_NONE;
}