#pragma once

#include <Cplt/Utils/IO/DataStream.hpp>
#include <Cplt/Utils/IO/Helper.hpp>
#include <Cplt/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& stream, tsl::array_map<char, TValue>& map)
	{
		static_assert(std::is_default_constructible_v<TValue>);
		static_assert(std::is_move_constructible_v<TValue>);

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

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

			TValue value;
			ReadHelper<TAdapter>(stream, value);

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

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

		for (auto it = map.begin(); it != map.end(); ++it) {
			stream.WriteObjectAdapted<DataStreamAdapters::StringView>(it.key_sv());
			WriteHelper<TAdapter>(stream, it.value());
		}
	}
};
} // namespace DataStreamAdapters