#include <unistd.h>
#include "StartUpProcess.h"
#include "libwiigui/gui.h"
#include "video.h"
#include "audio.h"
#include "input.h"
#include "themes/CTheme.h"
#include "gecko.h"
#include "Controls/DeviceHandler.hpp"
#include "wad/nandtitle.h"
#include "system/IosLoader.h"
#include "utils/timer.h"
#include "settings/CSettings.h"
#include "settings/CGameSettings.h"
#include "settings/CGameStatistics.h"
#include "usbloader/usbstorage2.h"
#include "sys.h"

StartUpProcess::StartUpProcess()
{
    //! Load default font for the next text outputs
    Theme::LoadFont("");

    background = new GuiImage(screenwidth, screenheight, (GXColor) {0, 0, 0, 255});

    GXImageData = Resources::GetImageData("gxlogo.png");
    GXImage = new GuiImage(GXImageData);
    GXImage->SetAlignment(ALIGN_CENTER, ALIGN_MIDDLE);
    GXImage->SetPosition(screenwidth/2, screenheight/2-50);

    titleTxt = new GuiText("Loading...", 24, (GXColor) {255, 255, 255, 255});
    titleTxt->SetAlignment(ALIGN_CENTER, ALIGN_MIDDLE);
    titleTxt->SetPosition(screenwidth/2, screenheight/2+30);

    messageTxt = new GuiText(" ", 22, (GXColor) {255, 255, 255, 255});
    messageTxt->SetAlignment(ALIGN_CENTER, ALIGN_MIDDLE);
    messageTxt->SetPosition(screenwidth/2, screenheight/2+60);
}

StartUpProcess::~StartUpProcess()
{
    delete background;
    delete GXImageData;
    delete GXImage;
    delete titleTxt;
    delete messageTxt;
}

void StartUpProcess::TextFade(int direction)
{
    if(direction > 0)
    {
        for(int i = 0; i < 255; i += direction)
        {
            messageTxt->SetAlpha(i);
            Draw();
        }
        messageTxt->SetAlpha(255);
        Draw();
    }
    else if(direction < 0)
    {
        for(int i = 255; i > 0; i += direction)
        {
            messageTxt->SetAlpha(i);
            Draw();
        }
        messageTxt->SetAlpha(0);
        Draw();
    }
}

void StartUpProcess::SetTextf(const char * format, ...)
{
	char * tmp = NULL;
	va_list va;
	va_start(va, format);
	if((vasprintf(&tmp, format, va) >= 0) && tmp)
    {
        TextFade(-40);
        gprintf(tmp);
        messageTxt->SetText(tmp);
        TextFade(40);
	}
	va_end(va);

	if(tmp)
        free(tmp);
}

bool StartUpProcess::USBSpinUp()
{
    bool started = false;
    const DISC_INTERFACE * handle = DeviceHandler::GetUSBInterface();
    Timer countDown;
    // wait 10 sec for the USB to spin up...stupid slow ass HDD
    do
    {
        started = (handle->startup() && handle->isInserted());
        if(started)
            break;

        messageTxt->SetTextf("Waiting for HDD: %i sec left\n", 20-(int)countDown.elapsed());
        Draw();
        usleep(50000);
    }
    while(countDown.elapsed() < 20.f);

    return started;
}

bool StartUpProcess::Run()
{
    StartUpProcess Process;

    return Process.Execute();
}

bool StartUpProcess::Execute()
{
    SetTextf("Start up\n");

    if(IosLoader::LoadAppCios() < 0)
    {
        SetTextf("Failed loading any cIOS. USB Loader GX requires at least cIOS 222 or 245-250. Exiting...\n");
        sleep(5);
        Sys_BackToLoader();
    }

    SetTextf("Initialize sd card\n");
    DeviceHandler::Instance()->MountSD();

    SetTextf("Initialize usb device\n");
    USBSpinUp();
    DeviceHandler::Instance()->MountAllUSB(false);

    SetTextf("Loading config files");
    gprintf("\tLoading config...%s\n", Settings.Load() ? "done" : "failed");
    gprintf("\tLoading language...%s\n", Settings.LoadLanguage(Settings.language_path, CONSOLE_DEFAULT) ? "done" : "failed");
    gprintf("\tLoading game settings...%s\n", GameSettings.Load(Settings.ConfigPath) ? "done" : "failed");
    gprintf("\tLoading game statistics...%s\n", GameStatistics.Load(Settings.ConfigPath) ? "done" : "failed");

    if(Settings.cios != IOS_GetVersion())
    {
        SetTextf("Loading cIOS %i\n", Settings.cios);

        DeviceHandler::DestroyInstance();

        // Loading now the cios setup in the settings
        IosLoader::LoadAppCios();

        SetTextf("Loaded cIOS %i R%i\n", IOS_GetVersion(), IOS_GetRevision());

        DeviceHandler::Instance()->MountSD();
        USBSpinUp();
        DeviceHandler::Instance()->MountAllUSB(false);
    }

    if(!IosLoader::IsHermesIOS())
    {
        Settings.USBPort = 0;
    }
    else if(Settings.USBPort == 1)
    {
        SetTextf("Changing USB Port to %i\n", Settings.USBPort);
        DeviceHandler::Instance()->UnMountAllUSB();
        DeviceHandler::SetUSBPort(Settings.USBPort);
        DeviceHandler::Instance()->MountAllUSB();
    }

    gprintf("\tLoading font...%s\n", Theme::LoadFont(Settings.theme_path) ? "done" : "failed (using default)");
    gprintf("\tLoading theme...%s\n", Theme::Load(Settings.theme) ? "done" : "failed (using default)");

    //! Init the rest of the System
    Sys_Init();
    SetupPads();
    InitAudio();
    setlocale(LC_CTYPE, "C-UTF-8");
    setlocale(LC_MESSAGES, "C-UTF-8");

    return true;
}

void StartUpProcess::Draw()
{
    background->Draw();
    GXImage->Draw();
    titleTxt->Draw();
    messageTxt->Draw();
    Menu_Render();
}