#ifndef __URDE_CMEMORYCARDDRIVER_HPP__
#define __URDE_CMEMORYCARDDRIVER_HPP__

#include "CMemoryCardSys.hpp"
#include "CGameState.hpp"

namespace urde
{
namespace MP1
{

class CMemoryCardDriver
{
    friend class CSaveGameScreen;
public:
    enum class EState
    {
        Initial = 0,
        Ready = 1,
        NoCard = 2,
        DriverClosed = 3,
        CardFormatted = 4,
        CardProbeDone = 5,
        CardMountDone = 6,
        CardCheckDone = 7,
        FileCreateDone = 8,
        FileCreateTransactionalDone = 9,
        FileWriteTransactionalDone = 10,
        FileDeleteAltTransactionalDone = 11,
        CardProbeFailed = 12,
        CardMountFailed = 13,
        CardCheckFailed = 14,
        FileDeleteBadFailed = 15,
        FileDeleteAltFailed = 16,
        FileBad = 17,
        FileCreateFailed = 18,
        FileWriteFailed = 19,
        FileCreateTransactionalFailed = 20,
        FileWriteTransactionalFailed = 21,
        FileDeleteAltTransactionalFailed = 22,
        FileRenameBtoAFailed = 23,
        CardFormatFailed = 24,
        CardProbe = 25,
        CardMount = 26,
        CardCheck = 27,
        FileDeleteBad = 28,
        FileRead = 29,
        FileDeleteAlt = 30,
        FileCreate = 31,
        FileWrite = 32,
        FileCreateTransactional = 33,
        FileWriteTransactional = 34,
        FileDeleteAltTransactional = 35,
        FileRenameBtoA = 36,
        CardFormat = 37
    };

    enum class EError
    {
        OK,
        CardBroken,
        CardWrongCharacterSet,
        CardIOError,
        CardWrongDevice,
        CardFull,
        CardStillFull, /* After attempting alt-delete (if needed) */
        CardNon8KSectors,
        FileMissing,
        FileCorrupted
    };

private:

    struct SFileInfo
    {
        CMemoryCardSys::CardFileHandle x0_fileInfo;

        std::string x14_name;
        std::vector<u8> x24_saveFileData;
        std::vector<u8> x34_saveData;
        SFileInfo(kabufuda::ECardSlot cardPort, const std::string& name);
        kabufuda::ECardResult Open();
        kabufuda::ECardResult Close();
        kabufuda::ECardSlot GetFileCardPort() const { return x0_fileInfo.slot; }
        int GetFileNo() const { return x0_fileInfo.getFileNo(); }
        kabufuda::ECardResult StartRead();
        kabufuda::ECardResult TryFileRead();
        kabufuda::ECardResult FileRead();
        kabufuda::ECardResult GetSaveDataOffset(u32& offOut);
    };

    struct SSaveHeader
    {
        u32 x0_version = 0;
        bool x4_savePresent[3];
        void DoPut(CMemoryOutStream& out) const
        {
            out.writeUint32Big(x0_version);
            for (int i=0 ; i<3 ; ++i)
                out.writeBool(x4_savePresent[i]);
        }
    };

    struct SGameFileSlot
    {
        u8 x0_saveBuffer[940] = {};
        CGameState::GameFileStateInfo x944_fileInfo;
        SGameFileSlot();
        SGameFileSlot(CMemoryInStream& in);
        void InitializeFromGameState();
        void LoadGameState(u32 idx);
        void DoPut(CMemoryOutStream& w) const
        {
            w.writeBytes(x0_saveBuffer, 940);
        }
    };

    enum class EFileState
    {
        Unknown,
        NoFile,
        File,
        BadFile
    };

    kabufuda::ECardSlot x0_cardPort;
    ResId x4_saveBanner;
    ResId x8_saveIcon0;
    ResId xc_saveIcon1;
    EState x10_state = EState::Initial;
    EError x14_error = EError::OK;
    s32 x18_cardFreeBytes = 0;
    s32 x1c_cardFreeFiles = 0;
    u32 x20_fileTime = 0;
    u64 x28_cardSerial = 0;
    u8 x30_systemData[174] = {};
    std::unique_ptr<SGameFileSlot> xe4_fileSlots[3];
    std::vector<std::pair<EFileState, SFileInfo>> x100_mcFileInfos;
    u32 x194_fileIdx = -1;
    std::unique_ptr<CMemoryCardSys::CCardFileInfo> x198_fileInfo;
    bool x19c_ = false;
    bool x19d_importPersistent;

public:
    CMemoryCardDriver(kabufuda::ECardSlot cardPort, ResId saveBanner,
                      ResId saveIcon0, ResId saveIcon1, bool importPersistent);

    void NoCardFound();
    const CGameState::GameFileStateInfo* GetGameFileStateInfo(int idx);
    static SSaveHeader LoadSaveHeader(CMemoryInStream& in);
    static std::unique_ptr<SGameFileSlot> LoadSaveFile(CMemoryInStream& in);
    void ReadFinished();
    void ImportPersistentOptions();
    void ExportPersistentOptions();
    void CheckCardCapacity();
    void IndexFiles();

    void StartCardProbe(); // 25
    void StartMountCard(); // 26
    void StartCardCheck(); // 27
    void StartFileDeleteBad(); // 28
    void StartFileRead(); // 29
    void StartFileDeleteAlt(); // 30
    void StartFileCreate(); // 31
    void StartFileWrite(); // 32
    void StartFileCreateTransactional(); // 33
    void StartFileWriteTransactional(); // 34
    void StartFileDeleteAltTransactional(); // 35
    void StartFileRenameBtoA(); // 36
    void StartCardFormat(); // 37

    void UpdateCardProbe(); // 25
    void UpdateMountCard(kabufuda::ECardResult result); // 26
    void UpdateCardCheck(kabufuda::ECardResult result); // 27
    void UpdateFileDeleteBad(kabufuda::ECardResult result); // 28
    void UpdateFileRead(kabufuda::ECardResult result); // 29
    void UpdateFileDeleteAlt(kabufuda::ECardResult result); // 30
    void UpdateFileCreate(kabufuda::ECardResult result); // 31
    void UpdateFileWrite(kabufuda::ECardResult result); // 32
    void UpdateFileCreateTransactional(kabufuda::ECardResult result); // 33
    void UpdateFileWriteTransactional(kabufuda::ECardResult result); // 34
    void UpdateFileDeleteAltTransactional(kabufuda::ECardResult result); // 35
    void UpdateFileRenameBtoA(kabufuda::ECardResult result); // 36
    void UpdateCardFormat(kabufuda::ECardResult result); // 37

    void ClearFileInfo() { x198_fileInfo.reset(); }
    void BuildNewFileSlot(u32 saveIdx);
    void EraseFileSlot(u32 saveIdx);
    void BuildExistingFileSlot(u32 saveIdx);
    void InitializeFileInfo();
    void WriteBackupBuf();
    bool GetCardFreeBytes();
    void HandleCardError(kabufuda::ECardResult result, EState state);
    void Update();
    void ClearError() { x14_error = EError::OK; }

    static bool IsCardBusy(EState v)
    {
        return v >= EState::CardMount && v <= EState::CardFormat;
    }

    static bool IsCardWriting(EState v)
    {
        if (v < EState::CardProbe)
            return false;
        if (v == EState::CardCheck)
            return false;
        if (v == EState::FileRead)
            return false;
        return true;
    }
};

}
}

#endif // __URDE_CMEMORYCARDDRIVER_HPP__