#include <std_include.hpp>
#include "loader/component_loader.hpp"

#include "profile_infos.hpp"
#include "network.hpp"
#include "party.hpp"

#include <utils/nt.hpp>
#include <utils/properties.hpp>
#include <utils/concurrency.hpp>

#include "../steam/steam.hpp"
#include <utils/io.hpp>

#include "game/utils.hpp"

namespace profile_infos
{
	namespace
	{
		using profile_map = std::unordered_map<uint64_t, profile_info>;
		utils::concurrency::container<profile_map> profile_mapping{};

		std::optional<profile_info> load_profile_info()
		{
			std::string data{};
			if (!utils::io::read_file("players/user/profile_info", &data))
			{
				return {};
			}

			profile_info info{};
			constexpr auto version_size = sizeof(info.version);

			if (data.size() < sizeof(version_size))
			{
				return {};
			}

			memcpy(&info.version, data.data(), version_size);
			info.ddl.assign(data.begin() + version_size, data.end());

			return {std::move(info)};
		}

		int get_max_client_count()
		{
			return game::get_dvar_int("com_maxclients");
		}

		void send_profile_info(const game::netadr_t& address, const std::string& buffer)
		{
			network::send(address, "profileInfo", buffer);
		}

		template <typename T>
		void distribute_profile_info(T* client_states, const std::string& buffer)
		{
			if (!client_states)
			{
				return;
			}

			for (int i = 0; i < get_max_client_count(); ++i)
			{
				if (client_states[i].client_state > 0)
				{
					send_profile_info(client_states[i].address, buffer);
				}
			}
		}

		void distribute_profile_info(const uint64_t user_id, const profile_info& info)
		{
			utils::byte_buffer buffer{};
			buffer.write(user_id);
			info.serialize(buffer);

			if (game::is_server())
			{
				distribute_profile_info(*game::svs_clients, buffer.get_buffer());
			}
			else
			{
				distribute_profile_info(*game::svs_clients_cl, buffer.get_buffer());
			}
		}
	}

	profile_info::profile_info(utils::byte_buffer& buffer)
	{
		this->version = buffer.read<int32_t>();
		this->ddl = buffer.read_string();
	}

	void profile_info::serialize(utils::byte_buffer& buffer) const
	{
		buffer.write(this->version);
		buffer.write_string(this->ddl);
	}

	void add_profile_info(const uint64_t user_id, const profile_info& info)
	{
		if (user_id == steam::SteamUser()->GetSteamID().bits)
		{
			return;
		}

		printf("Adding profile info: %llX\n", user_id);

		profile_mapping.access([&](profile_map& profiles)
		{
			profiles[user_id] = info;
		});
	}

	void distribute_profile_info_to_user(const game::netadr_t& addr, const uint64_t user_id, const profile_info& info)
	{
		utils::byte_buffer buffer{};
		buffer.write(user_id);
		info.serialize(buffer);

		send_profile_info(addr, buffer.get_buffer());
	}

	void distribute_profile_infos_to_user(const game::netadr_t& addr)
	{
		profile_mapping.access([&](const profile_map& profiles)
		{
			for (const auto& entry : profiles)
			{
				distribute_profile_info_to_user(addr, entry.first, entry.second);
			}
		});
	}

	void add_and_distribute_profile_info(const game::netadr_t& addr, const uint64_t user_id, const profile_info& info)
	{
		distribute_profile_infos_to_user(addr);

		add_profile_info(user_id, info);
		distribute_profile_info(user_id, info);
	}

	void clear_profile_infos()
	{
		profile_mapping.access([&](profile_map& profiles)
		{
			profiles = {};
		});
	}

	std::optional<profile_info> get_profile_info(const uint64_t user_id)
	{
		printf("Requesting profile info: %llX\n", user_id);

		if (user_id == steam::SteamUser()->GetSteamID().bits)
		{
			return load_profile_info();
		}

		return profile_mapping.access<std::optional<profile_info>>([user_id](const profile_map& profiles)
		{
			std::optional<profile_info> result{};

			const auto profile_entry = profiles.find(user_id);
			if (profile_entry != profiles.end())
			{
				result = profile_entry->second;
			}

			return result;
		});
	}

	void update_profile_info(const profile_info& info)
	{
		std::string data{};
		data.reserve(4 + info.ddl.size());

		data.append(reinterpret_cast<const char*>(&info.version), sizeof(info.version));
		data.append(info.ddl);

		utils::io::write_file("players/user/profile_info", data);
	}

	struct component final : client_component
	{
		void post_unpack() override
		{
			network::on("profileInfo", [](const game::netadr_t& server, const network::data_view& data)
			{
				if (!party::is_host(server))
				{
					return;
				}

				utils::byte_buffer buffer(data);
				const auto user_id = buffer.read<uint64_t>();
				const profile_info info(buffer);

				add_profile_info(user_id, info);
			});
		}
	};
}

REGISTER_COMPONENT(profile_infos::component)