#include <stdio.h>
#include <unistd.h>
#include <malloc.h>
#include <ogcsys.h>
#include <errno.h>

#include "usbloader/sdhc.h"
#include "usbloader/usbstorage2.h"
#include "fatmounter.h"
#include "wbfs_rw.h"

#include "wbfs_base.h"

s32 Wbfs::done = -1;
s32 Wbfs::total = -1;
u32 Wbfs::nb_sectors;

Wbfs::Wbfs(u32 device, u32 lba, u32 size) :
    hdd(NULL)
{
    this->device = device;
    this->lba = lba;
    this->size = size;
}

void Wbfs::GetProgressValue(s32 * d, s32 * m)
{
    *d = done;
    *m = total;
}

s32 Wbfs::Init(u32 device)
{
    s32 ret;

    switch (device)
    {
        case WBFS_DEVICE_USB:
            /* Initialize USB storage */
            ret = USBStorage2_Init();
            if (ret >= 0)
            {
                /* Setup callbacks */
                readCallback = __ReadUSB;
                writeCallback = __WriteUSB;
                /* Device info */
                /* Get USB capacity */
                nb_sectors = USBStorage2_GetCapacity(&sector_size);
                if (!nb_sectors) return -1;
            }
            else return ret;
            break;
        case WBFS_DEVICE_SDHC:
            /* Initialize SDHC */
            ret = SDHC_Init();

            if (ret)
            {
                /* Setup callbacks */
                readCallback = __ReadSDHC;
                writeCallback = __WriteSDHC;

                /* Device info */
                nb_sectors = 0;
                sector_size = SDHC_SECTOR_SIZE;
            }
            else return -1;
            break;
    }

    return 0;
}

void Wbfs::Close()
{
    if (hdd)
    {
        wbfs_close(hdd);
        hdd = NULL;
    }

    WBFSDevice_deInit();
}

// Default behavior: can't format
s32 Wbfs::Format()
{
    return -1;
}

s32 Wbfs::CheckGame(u8 *discid)
{
    wbfs_disc_t *disc = NULL;

    /* Try to open game disc */
    disc = OpenDisc(discid);
    if (disc)
    {
        /* Close disc */
        CloseDisc(disc);

        return 1;
    }

    return 0;
}

s32 Wbfs::GameSize(u8 *discid, f32 *size)
{
    wbfs_disc_t *disc = NULL;

    u32 sectors;

    /* Open disc */
    disc = OpenDisc(discid);
    if (!disc) return -2;

    /* Get game size in sectors */
    sectors = wbfs_sector_used(disc->p, disc->header);

    /* Copy value */
    *size = (disc->p->wbfs_sec_sz / GB_SIZE) * sectors;

    /* Close disc */
    CloseDisc(disc);

    return 0;
}

wbfs_t *Wbfs::GetHddInfo()
{
    return hdd;
}

bool Wbfs::Mounted()
{
    return hdd == NULL;
}

int Wbfs::GetFragList(u8 *id)
{
    return 0;
}

int Wbfs::GetFragList(char *filename, _frag_append_t append_fragment, FragList *)
{
    return 0;
}

bool Wbfs::ShowFreeSpace(void)
{
    return true;
}