138 lines
2.5 KiB
C++
138 lines
2.5 KiB
C++
#include <STDInclude.hpp>
|
|
|
|
namespace Components
|
|
{
|
|
std::unordered_map<std::string, UIScript::UIScriptHandler> UIScript::UIScripts;
|
|
std::unordered_map<int, std::function<void()>> UIScript::UIOwnerDraws;
|
|
|
|
template<> int UIScript::Token::get() const
|
|
{
|
|
if (this->isValid())
|
|
{
|
|
return std::strtol(this->token, nullptr, 0);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
template<> const char* UIScript::Token::get() const
|
|
{
|
|
if (this->isValid())
|
|
{
|
|
return this->token;
|
|
}
|
|
|
|
return "";
|
|
}
|
|
|
|
template<> std::string UIScript::Token::get() const
|
|
{
|
|
return {this->get<const char*>()};
|
|
}
|
|
|
|
bool UIScript::Token::isValid() const
|
|
{
|
|
return (this->token && this->token[0]);
|
|
}
|
|
|
|
void UIScript::Token::parse(const char** args)
|
|
{
|
|
if (args)
|
|
{
|
|
this->token = Game::Com_Parse(args);
|
|
}
|
|
}
|
|
|
|
Game::uiInfo_s* UIScript::UI_GetClientInfo(int localClientNum)
|
|
{
|
|
AssertIn(localClientNum, Game::STATIC_MAX_LOCAL_CLIENTS);
|
|
return &Game::uiInfoArray[localClientNum];
|
|
}
|
|
|
|
void UIScript::Add(const std::string& name, const UIScriptHandler& callback)
|
|
{
|
|
UIScripts[name] = callback;
|
|
}
|
|
|
|
void UIScript::AddOwnerDraw(int ownerdraw, const std::function<void()>& callback)
|
|
{
|
|
UIOwnerDraws[ownerdraw] = callback;
|
|
}
|
|
|
|
bool UIScript::RunMenuScript(const char* name, const char** args)
|
|
{
|
|
if (const auto itr = UIScripts.find(name); itr != UIScripts.end())
|
|
{
|
|
const auto* info = UI_GetClientInfo(0);
|
|
itr->second(Token(args), info);
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void UIScript::OwnerDrawHandleKeyStub(int ownerDraw, int flags, float *special, int key)
|
|
{
|
|
if (key == 200 || key == 201) // mouse buttons
|
|
{
|
|
for (auto i = UIOwnerDraws.begin(); i != UIOwnerDraws.end(); ++i)
|
|
{
|
|
if (i->first == ownerDraw)
|
|
{
|
|
i->second();
|
|
}
|
|
}
|
|
}
|
|
|
|
Utils::Hook::Call<void(int, int, float*, int)>(0x4F58A0)(ownerDraw, flags, special, key);
|
|
}
|
|
|
|
__declspec(naked) void UIScript::RunMenuScriptStub()
|
|
{
|
|
__asm
|
|
{
|
|
mov eax, esp
|
|
add eax, 8h
|
|
mov edx, eax // UIScript name
|
|
mov eax, [esp + 0C10h] // UIScript args
|
|
|
|
push eax
|
|
push edx
|
|
call RunMenuScript
|
|
add esp, 8h
|
|
|
|
test al, al
|
|
jz continue
|
|
|
|
// if returned
|
|
pop edi
|
|
pop esi
|
|
add esp, 0C00h
|
|
retn
|
|
|
|
continue:
|
|
mov eax, 45ED00h
|
|
jmp eax
|
|
}
|
|
}
|
|
|
|
UIScript::UIScript()
|
|
{
|
|
AssertSize(Game::uiInfo_s, 0x22FC);
|
|
|
|
if (Dedicated::IsEnabled()) return;
|
|
|
|
// Install handler
|
|
Utils::Hook::RedirectJump(0x45EC59, RunMenuScriptStub);
|
|
|
|
// Install ownerdraw handler
|
|
Utils::Hook(0x63D233, OwnerDrawHandleKeyStub, HOOK_CALL).install()->quick();
|
|
}
|
|
|
|
UIScript::~UIScript()
|
|
{
|
|
UIScripts.clear();
|
|
UIOwnerDraws.clear();
|
|
}
|
|
}
|