#include <STDInclude.hpp>

namespace Components
{
	std::unordered_map<int, UserInfo::userInfoMap> UserInfo::UserInfoOverrides;

	void UserInfo::SV_GetUserInfo_Stub(int index, char* buffer, int bufferSize)
	{
		Utils::Hook::Call<void(int, char*, int)>(0x49A160)(index, buffer, bufferSize);

		Utils::InfoString map(buffer);

		if (!UserInfoOverrides.contains(index))
		{
			UserInfoOverrides[index] = {};
		}

		for (const auto& [key, val] : UserInfoOverrides[index])
		{
			if (val.empty())
			{
				map.remove(key);
			}
			else
			{
				map.set(key, val);
			}
		}

		const auto userInfo = map.build();
		strncpy_s(buffer, bufferSize, userInfo.data(), _TRUNCATE);
	}

	void UserInfo::ClearClientOverrides(const int client)
	{
		UserInfoOverrides[client].clear();
	}

	void UserInfo::ClearAllOverrides()
	{
		UserInfoOverrides.clear();
	}

	void UserInfo::AddScriptMethods()
	{
		Script::AddMethod("SetName", [](Game::scr_entref_t entref)  // gsc: self SetName(<string>)
		{
			const auto* ent = Game::GetPlayerEntity(entref);
			const auto* name = Game::Scr_GetString(0);

			UserInfoOverrides[ent->s.number]["name"] = name;
			Game::ClientUserinfoChanged(ent->s.number);
		});

		Script::AddMethod("ResetName", [](Game::scr_entref_t entref)  // gsc: self ResetName()
		{
			const auto* ent = Game::GetPlayerEntity(entref);

			UserInfoOverrides[ent->s.number].erase("name");
			Game::ClientUserinfoChanged(ent->s.number);
		});
	}

	UserInfo::UserInfo()
	{
		Utils::Hook(0x445268, SV_GetUserInfo_Stub, HOOK_CALL).install()->quick();
		Utils::Hook(0x478B04, SV_GetUserInfo_Stub, HOOK_CALL).install()->quick();

		AddScriptMethods();

		Script::OnVMShutdown([]
		{
			ClearAllOverrides();
		});
	}
}