#pragma once

#include "Utils/IO/DataStream.hpp"
#include "Utils/IO/StringIntegration.hpp"

#include <tsl/array_map.h>
#include <tsl/array_set.h>
#include <string>
#include <type_traits>

// TODO support custom key types

namespace DataStreamAdapters {
template <class TAdapter = void>
struct TslArrayMap
{
	template <class TValue>
	static void ReadFromDataStream(InputDataStream& s, tsl::robin_map<char, TValue>& map)
	{
		static_assert(std::is_default_constructible_v<TValue>);
		static_assert(std::is_move_constructible_v<TValue>);

		uint64_t size;
		s.Read(size);
		map.reserve(size);

		for (uint64_t i = 0; i < size; ++i) {
			std::string key;
			s.ReadObjectAdapted<DataStreamAdapters::String>(key);

			TValue value;
			if constexpr (std::is_same_v<TAdapter, void>) {
				s.ReadGeneric(value);
			} else {
				s.ReadObjectAdapted<TAdapter>(value);
			}

			map.insert(key, std::move(value));
		}
	}

	template <class TValue>
	static void WriteToDataStream(OutputDataStream& s, const tsl::robin_map<char, TValue>& map)
	{
		s.Write((uint64_t)map.size());

		for (auto it = map.begin(); it != map.end(); ++it) {
			s.WriteObjectAdapted<DataStreamAdapters::StringView>(it.key_sv());

			if constexpr (std::is_same_v<TAdapter, void>) {
				s.WriteGeneric(it.value());
			} else {
				s.WriteObjectAdapted<TAdapter>(it.value());
			}
		}
	}
};
} // namespace DataStreamAdapters