/****************************************************************************
 * 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 <sys/statvfs.h>
#include "LoaderSettings.hpp"
#include "Controls/DeviceHandler.hpp"
#include "usbloader/usbstorage2.h"
#include "settings/CSettings.h"
#include "prompts/ProgressWindow.h"
#include "prompts/PromptWindows.h"
#include "language/gettext.h"
#include "wad/nandtitle.h"
#include "prompts/TitleBrowser.h"
#include "usbloader/GameList.h"
#include "usbloader/wbfs.h"
#include "usbloader/utils.h"
#include "system/IosLoader.h"
#include "settings/GameTitles.h"
#include "xml/xml.h"
#include "menu.h"

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

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

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

static const char * InstallToText[INSTALL_TO_MAX] =
{
    trNOOP( "None" ),
    trNOOP( "GAMEID_Gamename" ),
    trNOOP( "Gamename [GAMEID]" )
};

static const char * SplitSizeText[INSTALL_TO_MAX] =
{
    trNOOP( "No Splitting" ),
    trNOOP( "Split each 2GB" ),
    trNOOP( "Split each 4GB" ),
};

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

static inline bool IsValidPartition(int fs_type, int cios)
{
    if (IosLoader::IsWaninkokoIOS() && NandTitles.VersionOf(TITLE_ID(1, cios)) < 18)
    {
        return fs_type == PART_FS_WBFS;
    }
    else
    {
        return fs_type == PART_FS_WBFS || fs_type == PART_FS_FAT || fs_type == PART_FS_NTFS || fs_type == PART_FS_EXT;
    }
}

LoaderSettings::LoaderSettings()
    : SettingsMenu(tr("Loader Settings"), &GuiOptions, MENU_NONE)
{
    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( "Boot/Standard" ));
    Options->SetName(Idx++, "%s", tr( "Game/Install Partition" ));
    Options->SetName(Idx++, "%s", tr( "Multiple Partitions" ));
    Options->SetName(Idx++, "%s", tr( "USB Port" ));
    Options->SetName(Idx++, "%s", tr( "Install directories" ));
    Options->SetName(Idx++, "%s", tr( "Game Split Size" ));
    Options->SetName(Idx++, "%s", tr( "Quick Boot" ));
    Options->SetName(Idx++, "%s", tr( "Error 002 fix" ));
    Options->SetName(Idx++, "%s", tr( "Install partitions" ));
    Options->SetName(Idx++, "%s", tr( "Return To" ));
    Options->SetName(Idx++, "%s", tr( "Messageboard Update" ));
    Options->SetName(Idx++, "%s", tr( "Sync FAT32 FS Info" ));

    SetOptionValues();

    OldSettingsPartition = Settings.partition;
    OldSettingsMultiplePartitions = Settings.MultiplePartitions;
    OldSettingsUSBPort = Settings.USBPort;
}

LoaderSettings::~LoaderSettings()
{
    //! if partition has changed, Reinitialize it
    if (Settings.partition != OldSettingsPartition ||
        Settings.MultiplePartitions != OldSettingsMultiplePartitions ||
        Settings.USBPort != OldSettingsUSBPort)
    {
        int tempPort = Settings.USBPort;
        Settings.USBPort = OldSettingsUSBPort;
        WBFS_CloseAll();
        DeviceHandler::Instance()->UnMountAllUSB();
        Settings.USBPort = tempPort;
        DeviceHandler::SetUSBPort(Settings.USBPort);
        if(Settings.USBPort == 2) DeviceHandler::Instance()->MountAllUSB();

        if(Settings.MultiplePartitions)
            WBFS_OpenAll();
        else
            WBFS_OpenPart(Settings.partition);

        //! Reload the new game titles
        gameList.ReadGameList();
        GameTitles.LoadTitlesFromWiiTDB(Settings.titlestxt_path);
    }
}

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

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

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

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

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

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

    //! Settings: Boot/Standard
    if (Settings.godmode)
        Options->SetValue(Idx++, "IOS %i", Settings.cios);
    else
        Options->SetValue(Idx++, "********");

    //! Settings: Game/Install Partition
    PartitionHandle * usbHandle = DeviceHandler::Instance()->GetUSBHandle();
    // Get the partition name and it's size in GB's
    Options->SetValue(Idx++, "%s (%.2fGB)", usbHandle->GetFSName(Settings.partition), usbHandle->GetSize(Settings.partition)/GB_SIZE);

    //! Settings: Multiple Partitions
    Options->SetValue(Idx++, "%s", tr( OnOffText[Settings.MultiplePartitions] ));

    //! Settings: USB Port
    if(Settings.USBPort == 2)
        Options->SetValue(Idx++, tr("Both Ports"));
    else
        Options->SetValue(Idx++, "%i", Settings.USBPort);

    //! Settings: Install directories
    Options->SetValue(Idx++, "%s", tr( InstallToText[Settings.InstallToDir] ));

    //! Settings: Game Split Size
    Options->SetValue(Idx++, "%s", tr( SplitSizeText[Settings.GameSplit] ));

    //! Settings: Quick Boot
    Options->SetValue(Idx++, "%s", tr( OnOffText[Settings.quickboot] ));

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

    //! Settings: Install partitions
    if(Settings.InstallPartitions == ONLY_GAME_PARTITION)
        Options->SetValue(Idx++, "%s", tr("Only Game Partition"));
    else if(Settings.InstallPartitions == ALL_PARTITIONS)
        Options->SetValue(Idx++, "%s", tr("All Partitions"));
    else if(Settings.InstallPartitions == REMOVE_UPDATE_PARTITION)
        Options->SetValue(Idx++, "%s", tr("Remove update"));

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

    //! Settings: Messageboard Update
    Options->SetValue(Idx++, "%s", tr( OnOffText[Settings.PlaylogUpdate] ));

    //! Settings: Sync FAT32 FS Info
    Options->SetValue(Idx++, " ");
}

int LoaderSettings::GetMenuInternal()
{
    int ret = optionBrowser->GetClickedOption();

    if (ret < 0)
        return MENU_NONE;

    int Idx = -1;

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

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

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

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

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

    //! Settings: Boot/Standard
    else if (ret == ++Idx)
    {
        if(!Settings.godmode)
            return MENU_NONE;

        char entered[4];
        snprintf(entered, sizeof(entered), "%i", Settings.cios);
        if(OnScreenKeyboard(entered, sizeof(entered), 0))
        {
            Settings.cios = atoi(entered);
            if(Settings.cios < 200) Settings.cios = 200;
            else if(Settings.cios > 255) Settings.cios = 255;

            if(NandTitles.IndexOf(TITLE_ID(1, Settings.cios)) < 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(Settings.cios == 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: Game/Install Partition
    else if (ret == ++Idx)
    {
        // Select the next valid partition, even if that's the same one
        int fs_type = 0;
        int ios = IOS_GetVersion();
        int retries = 20;
        do
        {
            Settings.partition = (Settings.partition + 1) % DeviceHandler::Instance()->GetUSBHandle()->GetPartitionTotalCount();
			fs_type = DeviceHandler::GetUSBFilesystemType(Settings.partition);
        }
        while (!IsValidPartition(fs_type, ios) && --retries > 0);

        if(fs_type == PART_FS_FAT && Settings.GameSplit == GAMESPLIT_NONE)
            Settings.GameSplit = GAMESPLIT_4GB;
    }

    //! Settings: Multiple Partitions
    else if (ret == ++Idx)
    {
        if (++Settings.MultiplePartitions >= MAX_ON_OFF) Settings.MultiplePartitions = 0;
    }

    //! Settings: USB Port
    else if (ret == ++Idx)
    {
        if(!IosLoader::IsHermesIOS())
        {
            WindowPrompt(tr("ERROR:"), tr("USB Port changing is only supported on Hermes cIOS."), tr("OK"));
            Settings.USBPort = 0;
        }

        else if (++Settings.USBPort >= 3) // 2 = both ports
            Settings.USBPort = 0;
    }

    //! Settings: Install directories
    else if (ret == ++Idx)
    {
        if (++Settings.InstallToDir >= INSTALL_TO_MAX) Settings.InstallToDir = 0;
    }

    //! Settings: Game Split Size
    else if (ret == ++Idx)
    {
        if (++Settings.GameSplit >= GAMESPLIT_MAX)
        {
            if(DeviceHandler::GetUSBFilesystemType(Settings.partition) == PART_FS_FAT)
                Settings.GameSplit = GAMESPLIT_2GB;
            else
                Settings.GameSplit = GAMESPLIT_NONE;
        }
    }

    //! Settings: Quick Boot
    else if (ret == ++Idx)
    {
        if (++Settings.quickboot >= MAX_ON_OFF) Settings.quickboot = 0;
    }

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

    //! Settings: Install partitions
    else if (ret == ++Idx)
    {
        switch(Settings.InstallPartitions)
        {
            case ONLY_GAME_PARTITION:
                Settings.InstallPartitions = ALL_PARTITIONS;
                break;
            case ALL_PARTITIONS:
                Settings.InstallPartitions = REMOVE_UPDATE_PARTITION;
                break;
            default:
            case REMOVE_UPDATE_PARTITION:
                Settings.InstallPartitions = ONLY_GAME_PARTITION;
                break;
        }
    }

    //! Settings: Return To
    else if (ret == ++Idx)
    {
        char tidChar[10];
        bool getChannel = TitleSelector(tidChar);
        if (getChannel)
            snprintf(Settings.returnTo, sizeof(Settings.returnTo), "%s", tidChar);
    }

    //! Settings: Messageboard Update
    else if (ret == ++Idx )
    {
        if (++Settings.PlaylogUpdate >= MAX_ON_OFF) Settings.PlaylogUpdate = 0;
    }

    //! Settings: Sync FAT32 FS Info
    else if (ret == ++Idx )
    {
        int choice = WindowPrompt(0, tr("Do you want to sync free space info sector on all FAT32 partitions?"), tr("Yes"), tr("Cancel"));
        if(choice)
        {
            StartProgress(tr("Synchronizing..."), tr("Please wait..."), 0, false, false);
            PartitionHandle * usb = DeviceHandler::Instance()->GetUSBHandle();
            for(int i = 0; i < usb->GetPartitionCount(); ++i)
            {
                ShowProgress(i, usb->GetPartitionCount());
                if(strncmp(usb->GetFSName(i), "FAT", 3) == 0)
                {
                    struct statvfs stats;
                    char drive[20];
                    snprintf(drive, sizeof(drive), "%s:/", usb->MountName(i));
                    memset(&stats, 0, sizeof(stats));
                    memcpy(&stats.f_flag, "SCAN", 4);
                    statvfs(drive, &stats);
                }
            }
            ProgressStop();
        }
    }

    SetOptionValues();

    return MENU_NONE;
}