EuropaTools/include/europa/io/PakFile.hpp

//
// EuropaTools
//
// (C) 2021-2022 modeco80 <lily.modeco80@protonmail.ch>
//
// SPDX-License-Identifier: GPL-3.0-or-later
//

#ifndef EUROPA_IO_PAKFILE_H
#define EUROPA_IO_PAKFILE_H

#include <cstdint>
#include <europa/structs/Pak.hpp>
#include <vector>

namespace europa::io {

	struct PakReader;
	struct PakWriter;

	struct PakFile {
		using DataType = std::vector<std::uint8_t>;

		template<class T>
		void InitAs(const T& value) {
			toc = value;
		}

		void InitAs(structs::PakVersion version) {
			switch(version) {
				case structs::PakVersion::Ver3:
					toc = structs::PakHeader_V3::TocEntry{};
					break;
				case structs::PakVersion::Ver4:
					toc = structs::PakHeader_V4::TocEntry{};
					break;
				case structs::PakVersion::Ver5:
					toc = structs::PakHeader_V5::TocEntry{};
					break;
			}
		}

		/**
		 * Get the file data.
		 */
		[[nodiscard]] const DataType& GetData() const {
			return data;
		}

		/**
		 * Get the TOC entry responsible.
		 */
		template<class T>
		[[nodiscard]] const T& GetTOCEntry() const {
			return std::get<T>(toc);
		}

		void SetData(DataType&& data) {
			this->data = std::move(data);
		}

		std::uint32_t GetOffset() const {
			std::uint32_t size{};

			std::visit([&](auto& entry) {
				size = entry.offset;
			}, toc);

			return size;
		}

		std::uint32_t GetSize() const {
			std::uint32_t size{};

			std::visit([&](auto& entry) {
				size = entry.size;
			}, toc);

			return size;
		}

		void FillTOCEntry() {
			std::visit([&](auto& entry) {
				entry.size = static_cast<std::uint32_t>(data.size());
			}, toc);
		}

		template<class Cb>
		void Visit(const Cb& cb) {
			std::visit(cb, toc);
		}

	   private:
		friend PakReader;
		friend PakWriter;

		std::vector<std::uint8_t> data;
		structs::PakTocEntryVariant toc;
	};

} // namespace europa::io

#endif // EUROPA_IO_PAKFILE_H