#ifndef _H
#define _H

#include "libs/libwbfs/libwbfs.h"
#include "usbloader/utils.h"
#include "usbloader/frag.h"

class Wbfs
{
    public:
        Wbfs(u32, u32, u32);

        void GetProgressValue(s32 * d, s32 * m);
        static s32 Init(u32);
        void Close();
        s32 CheckGame(u8 *);
        s32 GameSize(u8 *, f32 *);
        wbfs_t *GetHddInfo(void);
        bool Mounted();
        virtual int GetFragList(u8 *id);
        virtual int GetFragList(char *filename, _frag_append_t append_fragment, FragList *);
        virtual bool ShowFreeSpace(void);

        virtual s32 Open() = 0;
        virtual wbfs_disc_t* OpenDisc(u8 *discid) = 0;
        virtual void CloseDisc(wbfs_disc_t *disc) = 0;
        virtual s32 Format();
        virtual s32 GetCount(u32 *) = 0;
        virtual s32 GetHeaders(struct discHdr *, u32, u32) = 0;
        virtual s32 AddGame(void) = 0;
        virtual s32 RemoveGame(u8 *) = 0;
        virtual s32 DiskSpace(f32 *, f32 *) = 0;
        virtual s32 RenameGame(u8 *, const void *) = 0;
        virtual s32 ReIDGame(u8 *discid, const void *newID) = 0;
        virtual f32 EstimateGameSize(void) = 0;

        /*
         static s32 OpenPart(u32 part_fat, u32 part_idx, u32 part_lba, u32 part_size, char *partition);
         static s32 OpenNamed(char *partition);
         static s32 OpenLBA(u32 lba, u32 size);
         */
    protected:
        static u32 nb_sectors;

        /* WBFS HDD */
        wbfs_t *hdd;

        u32 device, lba, size;
    private:

        static s32 total, done;
};

#endif //_H