#pragma once

#include "Utils/IO/DataStream.hpp"

#include <type_traits>
#include <vector>

namespace DataStreamAdapters {
template <class TAdapter = void>
struct Vector
{
	template <class TElement>
	static void ReadFromDataStream(InputDataStream& s, std::vector<TElement>& vec)
	{
		static_assert(std::is_default_constructible_v<TElement>);
		static_assert(std::is_move_constructible_v<TElement>);

		uint64_t size;
		s.Read(size);

		vec.clear();
		vec.reserve(size);

		for (uint64_t i = 0; i < size; ++i) {
			TElement element;
			if constexpr (std::is_same_v<TAdapter, void>) {
				s.ReadGeneric(element);
			} else {
				s.ReadObjectAdapted<TAdapter>(element);
			}

			vec.push_back(std::move(element));
		}
	}

	template <class TElement>
	static void WriteToDataStream(OutputDataStream& s, const std::vector<TElement>& vec)
	{
		s.Write((uint64_t)vec.size());
		for (auto& element : vec) {
			if constexpr (std::is_same_v<TAdapter, void>) {
				s.WriteGeneric(element);
			} else {
				s.WriteObjectAdapted<TAdapter>(element);
			}
		}
	}
};
} // namespace DataStreamAdapters