#ifndef ARCHIVE_HPP
#define ARCHIVE_HPP

#include "result.hpp"

#include <QtWidgets>
#include <cstdlib>
#include <string>
#include <vector>
#include <zip.h>

namespace getsuyomi {

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

using ArchiveResult = Result<std::reference_wrapper<const QImage>, 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 QString& filename);
    virtual ~ZipArchive() override;
    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<QImage>> m_pages{};
};

} // namespace getsuyomi

#endif