#include <gctypes.h>
#include <ogc/system.h>
#include <wiiuse/wpad.h>

#include "mload/mload.h"
#include "Controls/DeviceHandler.hpp"
#include "FileOperations/fileops.h"
#include "homebrewboot/BootHomebrew.h"
#include "settings/CSettings.h"
#include "settings/GameTitles.h"
#include "settings/newtitles.h"
#include "language/gettext.h"
#include "network/networkops.h"
#include "utils/ResourceManager.h"
#include "usbloader/playlog.h"
#include "usbloader/wbfs.h"
#include "themes/CTheme.h"
#include "SoundOperations/SoundHandler.hpp"
#include "utils/ThreadedTask.hpp"
#include "audio.h"
#include "lstub.h"
#include "menu.h"
#include "video.h"
#include "gecko.h"
#include "xml/xml.h"
#include "wad/nandtitle.h"
#include "buildtype.h"

extern "C"
{
    extern s32 MagicPatches(s32);
}

extern char game_partition[6];
extern u8 load_from_fs;

//Wiilight stuff
static vu32 *_wiilight_reg = (u32*) 0xCD0000C0;
void wiilight(int enable) // Toggle wiilight (thanks Bool for wiilight source)
{
    u32 val = (*_wiilight_reg & ~0x20);
    if (enable && Settings.wiilight) val |= 0x20;
    *_wiilight_reg = val;
}

/* Variables */
u8 shutdown = 0;
u8 reset = 0;

void __Sys_ResetCallback(void)
{
    /* Reboot console */
    reset = 1;
}

void __Sys_PowerCallback(void)
{
    /* Poweroff console */
    shutdown = 1;
}

void Sys_Init(void)
{
    /* Initialize video subsytem */
    //VIDEO_Init();

    /* Set RESET/POWER button callback */
    SYS_SetResetCallback(__Sys_ResetCallback);
    SYS_SetPowerCallback(__Sys_PowerCallback);
}

void AppCleanUp(void)
{
    static bool app_clean = false;
    if(app_clean)
        return;

    app_clean = true;
	extern u8 mountMethod;
    gprintf("Exiting main GUI.  mountMethod = %d\n", mountMethod);

    Settings.Save();

    ExitGUIThreads();
    StopGX();

    delete btnSoundClick;
    delete btnSoundOver;
    delete btnSoundClick2;
    delete bgMusic;
    delete background;
    delete bgImg;
    delete mainWindow;
    for (int i = 0; i < 4; i++)
        delete pointer[i];

    gettextCleanUp();
    Theme::CleanUp();
    NewTitles::DestroyInstance();
    ThreadedTask::DestroyInstance();
    SoundHandler::DestroyInstance();
    DeinitNetwork();
    GameTitles.SetDefault();

    ShutdownAudio();

    ResourceManager::DestroyInstance();

    WPAD_Shutdown();
}

void ExitApp(void)
{
    AppCleanUp();
    WBFS_CloseAll();
    DeviceHandler::DestroyInstance();
    USB_Deinitialize();
    if(Settings.PlaylogUpdate)
        Playlog_Delete(); // Don't show USB Loader GX in the Wii message board

    MagicPatches(0);
}

void Sys_Reboot(void)
{
    /* Restart console */
    ExitApp();
    STM_RebootSystem();
}

#define ShutdownToDefault   0
#define ShutdownToIdle      1
#define ShutdownToStandby   2

static void _Sys_Shutdown(int SHUTDOWN_MODE)
{
    ExitApp();

    /* Poweroff console */
    if ((CONF_GetShutdownMode() == CONF_SHUTDOWN_IDLE && SHUTDOWN_MODE != ShutdownToStandby) || SHUTDOWN_MODE
            == ShutdownToIdle)
    {
        s32 ret;

        /* Set LED mode */
        ret = CONF_GetIdleLedMode();
        if (ret >= 0 && ret <= 2) STM_SetLedMode(ret);

        /* Shutdown to idle */
        STM_ShutdownToIdle();
    }
    else
    {
        /* Shutdown to standby */
        STM_ShutdownToStandby();
    }
}

void Sys_Shutdown(void)
{
    _Sys_Shutdown(ShutdownToDefault);
}

void Sys_ShutdownToIdle(void)
{
    _Sys_Shutdown(ShutdownToIdle);
}
void Sys_ShutdownToStandby(void)
{
    _Sys_Shutdown(ShutdownToStandby);
}

void Sys_LoadMenu(void)
{
    ExitApp();
    /* Return to the Wii system menu */
    SYS_ResetSystem(SYS_RETURNTOMENU, 0, 0);
}

void Sys_BackToLoader(void)
{
    ExitApp();

    if (hbcStubAvailable())
        exit(0);
    // Channel Version
    SYS_ResetSystem(SYS_RETURNTOMENU, 0, 0);
}

#define HBC_HAXX    0x0001000148415858LL
#define HBC_JODI    0x000100014A4F4449LL
#define HBC_1_0_7   0x00010001AF1BF516LL

void Sys_LoadHBC(void)
{
    ExitApp();

    WII_Initialize();

    int ret = WII_LaunchTitle(HBC_1_0_7);
    if(ret < 0)
        WII_LaunchTitle(HBC_JODI);
    if(ret < 0)
        WII_LaunchTitle(HBC_HAXX);

    //Back to system menu if all fails
    SYS_ResetSystem(SYS_RETURNTOMENU, 0, 0);
}

bool RebootApp(void)
{
#ifdef FULLCHANNEL
	ExitApp();
	WII_Initialize();
	return !(WII_LaunchTitle(TITLE_ID(0x00010001, 0x554c4e52)) < 0);
#else
	char filepath[255];
	snprintf(filepath, sizeof(filepath), "%s/boot.dol", Settings.update_path);
	return !(BootHomebrew(filepath) < 0);
#endif
}

void ScreenShot()
{
    time_t rawtime;
    struct tm * timeinfo;
    char buffer[150];
    char buffer2[300];

    time(&rawtime);
    timeinfo = localtime(&rawtime);
    //USBLoader_GX_ScreenShot-Month_Day_Hour_Minute_Second_Year.png
    strftime(buffer, 80, "USBLoader_GX_ScreenShot-%b%d%H%M%S%y.png", timeinfo);
    sprintf(buffer2, "%s%s", Settings.ConfigPath, buffer);

    if(!CreateSubfolder(Settings.ConfigPath))
    {
        gprintf("Can't create screenshot folder\n");
        return;
    }

    TakeScreenshot(buffer2);
}