121 lines
2.9 KiB
C++
121 lines
2.9 KiB
C++
#include <STDInclude.hpp>
|
|
#include <Utils/InfoString.hpp>
|
|
|
|
#include "UserInfo.hpp"
|
|
|
|
#include "GSC/Script.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 clientNum)
|
|
{
|
|
UserInfoOverrides[clientNum].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);
|
|
|
|
if (name == nullptr)
|
|
{
|
|
Game::Scr_ParamError(0, "^1SetName: Illegal parameter!\n");
|
|
return;
|
|
}
|
|
|
|
Logger::Debug("Setting name of {} to {}", ent->s.number, name);
|
|
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);
|
|
|
|
Logger::Debug("Resetting name of {}", ent->s.number);
|
|
UserInfoOverrides[ent->s.number].erase("name");
|
|
Game::ClientUserinfoChanged(ent->s.number);
|
|
});
|
|
|
|
Script::AddMethod("SetClanTag", [](Game::scr_entref_t entref) // gsc: self SetClanTag(<string>)
|
|
{
|
|
const auto* ent = Game::GetPlayerEntity(entref);
|
|
const auto* clanName = Game::Scr_GetString(0);
|
|
|
|
if (clanName == nullptr)
|
|
{
|
|
Game::Scr_ParamError(0, "^1SetClanTag: Illegal parameter!\n");
|
|
return;
|
|
}
|
|
|
|
Logger::Debug("Setting clanName of {} to {}", ent->s.number, clanName);
|
|
UserInfoOverrides[ent->s.number]["clanAbbrev"] = clanName;
|
|
Game::ClientUserinfoChanged(ent->s.number);
|
|
});
|
|
|
|
Script::AddMethod("ResetClanTag", [](Game::scr_entref_t entref) // gsc: self ResetClanTag()
|
|
{
|
|
const auto* ent = Game::GetPlayerEntity(entref);
|
|
|
|
Logger::Debug("Resetting clanName of {}", ent->s.number);
|
|
UserInfoOverrides[ent->s.number].erase("clanAbbrev");
|
|
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();
|
|
|
|
Events::OnVMShutdown([]
|
|
{
|
|
ClearAllOverrides();
|
|
});
|
|
|
|
Events::OnClientDisconnect([](const int clientNum)
|
|
{
|
|
// Clear the overrides for UserInfo
|
|
ClearClientOverrides(clientNum);
|
|
});
|
|
}
|
|
}
|