MpqArchive.h

//
// Created by Lectem on 11/11/2016.
//

#pragma once

#include <vector>
#include "Archive.h"
#include "Stream.h"

namespace WorldStone
{

class MpqArchive : public Archive
{
    using HANDLE = void*; // Do not expose stormlib
public:
    MpqArchive() { setstate(badbit); }
    MpqArchive(const char* MpqFileName, const char* listFilePath = nullptr);
    MpqArchive(MpqArchive&& toMove);
    MpqArchive& operator=(MpqArchive&& toMove);
    ~MpqArchive() override;

    bool exists(const Path& filePath) override;
    StreamPtr open(const Path& filePath) override;

    HANDLE getInternalHandle() { return mpqHandle; }

    void addListFile(const char* listFilePAth);
    std::vector<Path> findFiles(const Path& searchMask = "*");

private:
    bool load() override;
    bool is_loaded() override;
    bool unload() override;

    Path   mpqFileName;
    HANDLE mpqHandle = nullptr;
};

class MpqFileStream : public IStream
{
    using HANDLE = void*;

    HANDLE file = nullptr;

protected:
    MpqFileStream() = default; // Needed to make tests easier
public:
    MpqFileStream(MpqArchive& archive, const Path& filename);
    bool open(MpqArchive& archive, const Path& filename);
    bool is_open() const { return file != nullptr; }
    bool close();

    size_t read(void* buffer, size_t size) override;

    long tell() override;
    long size() override;

    bool seek(long offset, seekdir origin) override;

    ~MpqFileStream() override;
};
}