/****************************************************************************
 * DiscBrowser
 * USB Loader GX 2009
 *
 * DiscBrowser.h
 ***************************************************************************/
#include "language/gettext.h"
#include "libwiigui/gui.h"
#include "libwiigui/gui_customoptionbrowser.h"
#include "prompts/PromptWindows.h"
#include "filelist.h"
#include "menu.h"
#include "usbloader/disc.h"
#include "usbloader/fstfile.h"
#include "usbloader/wdvd.h"
#include "main.h"
#include "sys.h"
#include "settings/cfg.h"
#include "memory.h"
#include "../gecko.h"
#include "../patches/dvd_broadway.h"

/*** Extern functions ***/
extern void ResumeGui();
extern void HaltGui();

/*** Extern variables ***/
extern GuiWindow * mainWindow;
extern u8 shutdown;
extern u8 reset;
extern u8 mountMethod;

/********************************************************************************
*Disk Browser
*********************************************************************************/
int DiscBrowse( struct discHdr * header )
{
    gprintf( "\nDiscBrowser() started" );
    bool exit = false;
    int ret, choice;
    u64 offset;

    HaltGui();
    if ( !mountMethod )
    {
        ret = Disc_SetUSB( header->id );
        if ( ret < 0 )
        {
            ResumeGui();
            WindowPrompt( tr( "ERROR:" ), tr( "Could not set USB." ), tr( "OK" ) );
            return ret;
        }
    }

    ret = Disc_Open();
    if ( ret < 0 )
    {
        ResumeGui();
        WindowPrompt( tr( "ERROR:" ), tr( "Could not open Disc" ), tr( "OK" ) );
        return ret;
    }

    ret = __Disc_FindPartition( &offset );
    if ( ret < 0 )
    {
        ResumeGui();
        WindowPrompt( tr( "ERROR:" ), tr( "Could not find a WBFS partition." ), tr( "OK" ) );
        return ret;
    }

    ret = WDVD_OpenPartition( offset );
    if ( ret < 0 )
    {
        ResumeGui();
        WindowPrompt( tr( "ERROR:" ), tr( "Could not open WBFS partition" ), tr( "OK" ) );
        return ret;
    }

    int *buffer = ( int* )allocate_memory( 0x20 );

    if ( buffer == NULL )
    {
        ResumeGui();
        WindowPrompt( tr( "ERROR:" ), tr( "Not enough free memory." ), tr( "OK" ) );
        return -1;
    }

    ret = WDVD_Read( buffer, 0x20, 0x420 );
    if ( ret < 0 )
    {
        ResumeGui();
        WindowPrompt( tr( "ERROR:" ), tr( "Could not read the disc." ), tr( "OK" ) );
        return ret;
    }

    void *fstbuffer = allocate_memory( buffer[2] * 4 );
    FST_ENTRY *fst = ( FST_ENTRY * )fstbuffer;

    if ( fst == NULL )
    {
        ResumeGui();
        WindowPrompt( tr( "ERROR:" ), tr( "Not enough free memory." ), tr( "OK" ) );
        free( buffer );
        return -1;
    }

    ret = WDVD_Read( fstbuffer, buffer[2] * 4, buffer[1] * 4 );

    if ( ret < 0 )
    {
        ResumeGui();
        WindowPrompt( tr( "ERROR:" ), tr( "Could not read the disc." ), tr( "OK" ) );
        free( buffer );
        free( fstbuffer );
        return ret;
    }
    ResumeGui();
    free( buffer );

    WDVD_Reset();
    //Disc_SetUSB(NULL);
    WDVD_ClosePartition();

    u32 discfilecount = fst[0].filelen;
    u32 dolfilecount = 0;
    //int offsetselect[20];

    customOptionList options3( discfilecount );

    for ( u32 i = 0; i < discfilecount; i++ )
    {

        //don't add files that aren't .dol to the list
        int len = ( strlen( fstfiles( fst, i ) ) );
        if ( fstfiles( fst, i )[len-4] == '.' &&
                fstfiles( fst, i )[len-3] == 'd' &&
                fstfiles( fst, i )[len-2] == 'o' &&
                fstfiles( fst, i )[len-1] == 'l' )
        {
            options3.SetName( i, "%i", i );
            options3.SetValue( i, fstfiles( fst, i ) );
            //options3.SetName(i, fstfiles(fst, i));

            dolfilecount++;
        }
    }
    gprintf( "\n%i alt dols found", dolfilecount );
    if ( dolfilecount <= 0 )
    {
        WindowPrompt( tr( "ERROR" ), tr( "No DOL file found on disc." ), tr( "OK" ) );
        free( fstbuffer );
        return -1;
    }

    GuiSound btnSoundOver( button_over_pcm, button_over_pcm_size, Settings.sfxvolume );
    // because destroy GuiSound must wait while sound playing is finished, we use a global sound
    if ( !btnClick2 ) btnClick2 = new GuiSound( button_click2_pcm, button_click2_pcm_size, Settings.sfxvolume );
    //  GuiSound btnClick(button_click2_pcm, button_click2_pcm_size, Settings.sfxvolume);

    char imgPath[100];

    snprintf( imgPath, sizeof( imgPath ), "%sbutton_dialogue_box.png", CFG.theme_path );
    GuiImageData btnOutline( imgPath, button_dialogue_box_png );
    snprintf( imgPath, sizeof( imgPath ), "%sgamesettings_background.png", CFG.theme_path );
    GuiImageData settingsbg( imgPath, settings_background_png );

    GuiTrigger trigA;
    trigA.SetSimpleTrigger( -1, WPAD_BUTTON_A | WPAD_CLASSIC_BUTTON_A, PAD_BUTTON_A );
    GuiTrigger trigHome;
    trigHome.SetButtonOnlyTrigger( -1, WPAD_BUTTON_HOME | WPAD_CLASSIC_BUTTON_HOME, 0 );
    GuiTrigger trigB;
    trigB.SetButtonOnlyTrigger( -1, WPAD_BUTTON_B | WPAD_CLASSIC_BUTTON_B, PAD_BUTTON_B );

    GuiText titleTxt( get_title( header ), 28, ( GXColor ) {0, 0, 0, 255} );
    titleTxt.SetAlignment( ALIGN_CENTRE, ALIGN_TOP );
    titleTxt.SetPosition( 12, 40 );
    titleTxt.SetMaxWidth( 356, SCROLL_HORIZONTAL );

    GuiImage settingsbackground( &settingsbg );
    GuiButton settingsbackgroundbtn( settingsbackground.GetWidth(), settingsbackground.GetHeight() );
    settingsbackgroundbtn.SetAlignment( ALIGN_LEFT, ALIGN_TOP );
    settingsbackgroundbtn.SetPosition( 0, 0 );
    settingsbackgroundbtn.SetImage( &settingsbackground );

    GuiText cancelBtnTxt( tr( "Back" ), 22, THEME.prompttext );
    cancelBtnTxt.SetMaxWidth( btnOutline.GetWidth() - 30 );
    GuiImage cancelBtnImg( &btnOutline );
    if ( Settings.wsprompt == yes )
    {
        cancelBtnTxt.SetWidescreen( CFG.widescreen );
        cancelBtnImg.SetWidescreen( CFG.widescreen );
    }
    GuiButton cancelBtn( &cancelBtnImg, &cancelBtnImg, 2, 3, 180, 400, &trigA, &btnSoundOver, btnClick2, 1 );
    cancelBtn.SetScale( 0.9 );
    cancelBtn.SetLabel( &cancelBtnTxt );
    cancelBtn.SetTrigger( &trigB );

    u8 scrollbaron = 0;
    if ( dolfilecount > 9 )
        scrollbaron = 1;

    GuiCustomOptionBrowser optionBrowser3( 396, 280, &options3, CFG.theme_path, "bg_options_gamesettings.png", bg_options_settings_png, dolfilecount > 9 ? 1 : 0, 200 );
    optionBrowser3.SetPosition( 0, 90 );
    optionBrowser3.SetAlignment( ALIGN_CENTRE, ALIGN_TOP );

    HaltGui();
    GuiWindow w( screenwidth, screenheight );
    w.Append( &settingsbackgroundbtn );
    w.Append( &titleTxt );
    w.Append( &cancelBtn );
    w.Append( &optionBrowser3 );

    mainWindow->Append( &w );

    ResumeGui();
    while ( !exit )
    {
        VIDEO_WaitVSync();

        if ( shutdown == 1 )
            Sys_Shutdown();
        if ( reset == 1 )
            Sys_Reboot();

        ret = optionBrowser3.GetClickedOption();

        if ( ret > 0 )
        {
            char temp[100];
            strlcpy( temp, fstfiles( fst, ret ), sizeof( temp ) );
            choice = WindowPrompt( temp, tr( "Load this DOL as alternate DOL?" ), tr( "OK" ), tr( "Cancel" ) );
            if ( choice )
            {
                //ret = offsetselect[ret];
                strlcpy( alternatedname, temp, sizeof( alternatedname ) );
                exit = true;
            }
        }

        if ( cancelBtn.GetState() == STATE_CLICKED )
        {
            ret = 696969;
            exit = true;
        }
    }

    HaltGui();
    mainWindow->Remove( &w );
    ResumeGui();

    //free not needed list buffer anymore
    free( fstbuffer );

    return ret;
}


int autoSelectDol( const char *id, bool force )
{
    gprintf( "\nautoSelectDol() started" );

    char id4[10];
    sprintf( id4, "%c%c%c%c", id[0], id[1], id[2], id[3] );

////// games that can be forced (always need alt dol)

    //Boogie
    if ( strcmp( id, "RBOP69" ) == 0 ) return 675;//previous value was 657
    if ( strcmp( id, "RBOE69" ) == 0 ) return 675;//starstremr

    //Fifa 08
    if ( strcmp( id, "RF8E69" ) == 0 ) return 439;//from isostar
    if ( strcmp( id, "RF8P69" ) == 0 ) return 463;//from isostar
    if ( strcmp( id, "RF8X69" ) == 0 ) return 464;//from isostar

    //Madden NFL07
    if ( strcmp( id, "RMDP69" ) == 0 ) return 39;//from isostar

    //Madden NFL08
    if ( strcmp( id, "RNFP69" ) == 0 ) return 1079;//from isostar

    //Medal of Honor: Heroes 2
    if ( strcmp( id, "RM2X69" ) == 0 )return 601;//dj_skual
    if ( strcmp( id, "RM2P69" ) == 0 )return 517;//MZottel
    if ( strcmp( id, "RM2E69" ) == 0 ) return 492;//Old8oy

    //Mortal Kombat
    if ( strcmp( id, "RKMP5D" ) == 0 ) return 290;//from isostar
    if ( strcmp( id, "RKME5D" ) == 0 ) return 290;//starstremr

    //NBA 08
    if ( strcmp( id, "RNBX69" ) == 0 ) return 964;//from isostar

    //Pangya! Golf with Style
    if ( strcmp( id, "RPYP9B" ) == 0 ) return 12490;//from isostar

    //Redsteel
    if ( strcmp( id, "REDP41" ) == 0 ) return 1957;//from isostar
    if ( strcmp( id, "REDE41" ) == 0 ) return 1957;//starstremr

    //SSX
    if ( strcmp( id, "RSXP69" ) == 0 ) return 377;//previous value was 337
    if ( strcmp( id, "RSXE69" ) == 0 ) return 377;//previous value was 337

    //Wii Sports Resort, needs alt dol one time only, to show the Motion Plus video
    //if (strcmp(id,"RZTP01") == 0 && CheckForSave(id4)==0) return 952;//from isostar
    //if (strcmp(id,"RZTE01") == 0 && CheckForSave(id4)==0) return 674;//from starstremr
    //as well as Grand Slam Tennis, Tiger Woods 10, Virtual Tennis 2009

///// games that can't be forced (alt dol is not always needed)
    if ( !force )
    {

        //Grand Slam Tennis
        if ( strcmp( id, "R5TP69" ) == 0 ) return 1493;//from isostar
        if ( strcmp( id, "R5TE69" ) == 0 ) return 1493;//starstremr

        //Medal of Honor Heroes
        if ( strcmp( id, "RMZX69" ) == 0 ) return 492;//from isostar
        if ( strcmp( id, "RMZP69" ) == 0 ) return 492;//from isostar
        if ( strcmp( id, "RMZE69" ) == 0 ) return 492;//starstremr

        //Tiger Woods 10
        if ( strcmp( id, "R9OP69" ) == 0 ) return 1991;//from isostar
        if ( strcmp( id, "R9OE69" ) == 0 ) return 1973;//starstremr

        //Virtual Tennis 2009
        if ( strcmp( id, "RVUP8P" ) == 0 ) return 16426;//from isostar
        if ( strcmp( id, "RVUE8P" ) == 0 ) return 16405;//from isostar

        //Wii Sports Resort
        if ( strcmp( id, "RZTP01" ) == 0 ) return 952;//from isostar
        if ( strcmp( id, "RZTE01" ) == 0 ) return 674;//from starstremr
    }

    return -1;
}

int autoSelectDolMenu( const char *id, bool force )
{

    /*
    char id4[10];
    sprintf(id4,"%c%c%c%c",id[0],id[1],id[2],id[3]);

    switch (CheckForSave(id4)) {
        case 0:
            WindowPrompt(tr("NO save"),0,tr("OK"));
            break;
        case 1:
            WindowPrompt(tr("save"),0,tr("OK"));
            break;
        default:
            char test[10];
            sprintf(test,"%d",CheckForSave(id4));
            WindowPrompt(test,0,tr("OK"));
            break;
    }
    return -1;
    */

    //Indiana Jones and the Staff of Kings (Fate of Atlantis)
    if ( strcmp( id, "RJ8E64" ) == 0 )
    {
        int choice = WindowPrompt( tr( "Select a DOL" ), 0, "Fate of Atlantis", tr( "Default" ) );
        switch ( choice )
        {
            case 1:
                choice = 8; //from starstremr
                break;
            default: // no alt dol
                choice = 0;
                break;
        }
        return choice;
    }
    if ( strcmp( id, "RJ8P64" ) == 0 )
    {
        int choice = WindowPrompt( tr( "Select a DOL" ), 0, "Fate of Atlantis", tr( "Default" ) );
        switch ( choice )
        {
            case 1:
                choice = 8; //from isostar
                break;
            default: // no alt dol
                choice = 0;
                break;
        }
        return choice;
    }

    //Metal Slug Anthology (Metal Slug 6)
    if ( strcmp( id, "RMLEH4" ) == 0 )
    {
        int choice = WindowPrompt( tr( "Select a DOL" ), 0, "Metal Slug 6", tr( "Default" ) );
        switch ( choice )
        {
            case 1:
                choice = 54; //from lustar
                break;
            default: // no alt dol
                choice = 0;
                break;
        }
        return choice;
    }
    if ( strcmp( id, "RMLP7U" ) == 0 )
    {
        int choice = WindowPrompt( tr( "Select a DOL" ), 0, "Metal Slug 6", tr( "Default" ) );
        switch ( choice )
        {
            case 1:
                choice = 56; //from isostar
                break;
            default: // no alt dol
                choice = 0;
                break;
        }
        return choice;
    }

    //Metroid Prime Trilogy
    if ( strcmp( id, "R3ME01" ) == 0 )
    {
        //do not use any alt dol if there is no save game in the nand
        /*
        if (CheckForSave(id4)==0 && force) {
            WindowPrompt(0,tr("You need to start this game one time to create a save file, then exit and start it again."),tr("OK"));
            return -1;
        }
        */
        int choice = WindowPrompt( tr( "Select a DOL" ), 0, "Metroid Prime", "Metroid Prime 2", "Metroid Prime 3", tr( "Default" ) );
        switch ( choice )
        {
            case 1:
                choice = 780;
                break;
            case 2:
                choice = 781;
                break;
            case 3:
                choice = 782;
                break;
            default: // no alt dol
                choice = 0;
                break;
        }
        return choice;
    }
    if ( strcmp( id, "R3MP01" ) == 0 )
    {
        /*
        if (CheckForSave(id4)==0 && force) {
            WindowPrompt(0,tr("You need to start this game one time to create a save file, then exit and start it again."),tr("OK"));
            return -1;
        }
        */
        int choice = WindowPrompt( tr( "Select a DOL" ), 0, "Metroid Prime", "Metroid Prime 2", "Metroid Prime 3", tr( "Default" ) );
        switch ( choice )
        {
            case 1:
                choice = 782;
                break;
            case 2:
                choice = 783;
                break;
            case 3:
                choice = 784;
                break;
            default: // no alt dol
                choice = 0;
                break;
        }
        return choice;
    }

    //Rampage: Total Destruction (M1.dol=Rampage, jarvos.dol=Rampage World Tour)
    if ( strcmp( id, "RPGP5D" ) == 0 )
    {
        int choice = WindowPrompt( tr( "Select a DOL" ), 0, "Rampage", "World Tour", tr( "Default" ) );
        switch ( choice )
        {
            case 1:
                choice = 369; //from Ramzee
                break;
            case 2:
                choice = 368; //from Ramzee
                break;
            default: // no alt dol
                choice = 0;
                break;
        }
        return choice;
    }

    //The House Of The Dead 2 & 3 Return (only to play 2)
    if ( strcmp( id, "RHDE8P" ) == 0 )
    {
        int choice = WindowPrompt( tr( "Select a DOL" ), 0, "HotD 2", tr( "Default" ) );
        switch ( choice )
        {
            case 1:
                choice = 149; //from starstremr
                break;
            default: // no alt dol
                choice = 0;
                break;
        }
        return choice;
    }
    if ( strcmp( id, "RHDP8P" ) == 0 )
    {
        int choice = WindowPrompt( tr( "Select a DOL" ), 0, "HotD 2", tr( "Default" ) );
        switch ( choice )
        {
            case 1:
                choice = 149; //from isostar
                break;
            default: // no alt dol
                choice = 0;
                break;
        }
        return choice;
    }


    return -1;
}


/********************************************************************************
* Mount a DVD, get the type and ID.
*********************************************************************************/
static vu32 dvddone = 0;
static dvddiskid *g_diskID = ( dvddiskid* )0x80000000; // If you change this address, the read functions will FAIL!
void __dvd_readidcb( s32 result )
{
    dvddone = result;
}

u8 DiscMount( discHdr *header )
{
    gprintf( "\nDiscMount() " );
    int ret;
    HaltGui();

    u8 *tmpBuff = ( u8 * ) malloc( 0x60 );
    memcpy( tmpBuff, g_diskID, 0x60 ); // Make a backup of the first 96 bytes at 0x80000000

    ret = bwDVD_LowInit();
    dvddone = 0;
    ret = bwDVD_LowReset( __dvd_readidcb );
    while ( ret >= 0 && dvddone == 0 );

    dvddone = 0;
    ret = bwDVD_LowReadID( g_diskID, __dvd_readidcb ); // Leave this one here, or you'll get an IOCTL error
    while ( ret >= 0 && dvddone == 0 );

    dvddone = 0;
    ret = bwDVD_LowUnencryptedRead( g_diskID, 0x60, 0x00, __dvd_readidcb ); // Overwrite the g_diskID thing
    while ( ret >= 0 && dvddone == 0 );

    memcpy( header, g_diskID, 0x60 );
    memcpy( g_diskID, tmpBuff, 0x60 ); // Put the backup back, or games won't load
    free( tmpBuff );


    ResumeGui();
    if ( dvddone != 1 )
    {
        return 0;
    }
    return ( header->magic == 0x5D1C9EA3 ) ? 1 : 2; // Don't check gamecube magic (0xC2339F3D)
}