/****************************************************************************
 * 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 "usbloader/wbfs.h"
#include "libwbfs/libwbfs.h"
#include "libwbfs/wiidisc.h"
#include "main.h"
#include "sys.h"
#include "settings/cfg.h"
#include "themes/CTheme.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 = 0, choice;

    HaltGui();

    Disc_SetUSB(NULL);
    wbfs_disc_t *disc = WBFS_OpenDisc((u8 *) header->id);
    if (!disc)
    {
        ResumeGui();
        WindowPrompt(tr( "ERROR:" ), tr( "Could not open Disc" ), tr( "OK" ));
        return ret;
    }
    wiidisc_t *wdisc = wd_open_disc((int(*)(void *, u32, u32, void *)) wbfs_disc_read, disc);
    if (!wdisc)
    {
        ResumeGui();
        WindowPrompt(tr( "ERROR:" ), tr( "Could not open Disc" ), tr( "OK" ));
        return ret;
    }

    FST_ENTRY * fstbuffer = (FST_ENTRY *) wd_get_fst(wdisc, ONLY_GAME_PARTITION);
    if (!fstbuffer)
    {
        ResumeGui();
        WindowPrompt(tr( "ERROR:" ), tr( "Not enough free memory." ), tr( "OK" ));
        return -1;
    }

    wd_close_disc(wdisc);
    WBFS_CloseDisc(disc);

    u32 discfilecount = fstbuffer[0].filelen;
    u32 dolfilecount = 0;

    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(fstbuffer, i)));
        if (fstfiles(fstbuffer, i)[len - 4] == '.' && fstfiles(fstbuffer, i)[len - 3] == 'd' && fstfiles(fstbuffer, i)[len - 2] == 'o'
                && fstfiles(fstbuffer, i)[len - 1] == 'l')
        {
            options3.SetName(i, "%i", i);
            options3.SetValue(i, fstfiles(fstbuffer, 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);

    GuiImageData btnOutline(Resources::GetFile("button_dialogue_box.png"), Resources::GetFileSize("button_dialogue_box.png"));
    GuiImageData settingsbg(Resources::GetFile("settings_background.png"), Resources::GetFileSize("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(Settings.widescreen);
        cancelBtnImg.SetWidescreen(Settings.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, "bg_options_gamesettings.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(fstbuffer, 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)
}