#ifndef ARCHIVE_HPP
#define ARCHIVE_HPP

#include "result.hpp"

#include <cstdlib>
#include <memory>
#include <optional>
#include <string>
#include <vector>
#include <zip.h>

namespace bookmouse {

enum class ArchiveError
{
    BadIndex,
    ZipError,
    BadData,
    Unimplemented
};

using ArchiveResult = Result<std::reference_wrapper<const std::vector<char>>, ArchiveError>;

class Archive {
  public:
    virtual ~Archive() = default;
    virtual ArchiveResult getPage(size_t page) = 0;
    virtual size_t numPages() const = 0;
};

class ZipArchive : public Archive {
  public:
    explicit ZipArchive(const std::string& filename);
    virtual ~ZipArchive() override;
    ZipArchive(const ZipArchive& rhs) = delete;
    ZipArchive(ZipArchive&& rhs);
    ZipArchive& operator=(const ZipArchive& rhs) = delete;
    ZipArchive& operator=(ZipArchive&& rhs);

    virtual ArchiveResult getPage(size_t page) override;
    virtual size_t numPages() const override;

  private:
    void populate();

    zip_t* m_archive{nullptr};

    std::vector<size_t> m_sortedIndices{};
    std::vector<std::string> m_filenames{};
    std::vector<size_t> m_fileSizes{};

    std::vector<std::optional<std::vector<char>>> m_pages{};
};

} // namespace bookmouse

#endif