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

#include "mload/mload.h"
#include "settings/CSettings.h"
#include "audio.h"
#include "fatmounter.h"
#include "lstub.h"
#include "menu.h"
#include "video.h"

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);
}

static void _ExitApp()
{
    ExitGUIThreads();
    StopGX();
    ShutdownAudio();

    UnmountNTFS();
    SDCard_deInit();
    USBDevice_deInit();
    mload_set_ES_ioctlv_vector(NULL);
    mload_close();
}

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();
    WPAD_Flush(0);
    WPAD_Disconnect(0);
    WPAD_Shutdown();

    /* 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_ShutdownToIdel(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)
{

    if (hbcStubAvailable())
    {
        _ExitApp();
        exit(0);
    }
    // Channel Version
    Sys_LoadMenu();
}

bool Sys_IsHermes()
{
    return IOS_GetVersion() == 222 || IOS_GetVersion() == 223;
}

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

    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/config/%s", bootDevice, buffer);

    TakeScreenshot(buffer2);
}