137 lines
2.5 KiB
C++
137 lines
2.5 KiB
C++
#include "STDInclude.hpp"
|
|
|
|
namespace Components
|
|
{
|
|
std::map<std::string, wink::slot<UIScript::Callback>> UIScript::UIScripts;
|
|
std::map<int, wink::slot<UIScript::CallbackRaw>> UIScript::UIOwnerDraws;
|
|
|
|
template<> int UIScript::Token::Get()
|
|
{
|
|
if (this->IsValid())
|
|
{
|
|
return atoi(this->token);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
template<> char* UIScript::Token::Get()
|
|
{
|
|
if (this->IsValid())
|
|
{
|
|
return this->token;
|
|
}
|
|
|
|
return "";
|
|
}
|
|
|
|
template<> const char* UIScript::Token::Get()
|
|
{
|
|
return this->Get<char*>();
|
|
}
|
|
|
|
template<> std::string UIScript::Token::Get()
|
|
{
|
|
return this->Get<const char*>();
|
|
}
|
|
|
|
bool UIScript::Token::IsValid()
|
|
{
|
|
return (token && token[0]);
|
|
}
|
|
|
|
void UIScript::Token::Parse(const char** args)
|
|
{
|
|
if (args)
|
|
{
|
|
this->token = Game::Com_ParseExt(args);
|
|
}
|
|
}
|
|
|
|
void UIScript::Add(std::string name, UIScript::Callback* callback)
|
|
{
|
|
UIScript::UIScripts[name] = callback;
|
|
}
|
|
|
|
void UIScript::Add(std::string name, UIScript::CallbackRaw* callback)
|
|
{
|
|
UIScript::Add(name, reinterpret_cast<UIScript::Callback*>(callback));
|
|
}
|
|
|
|
void UIScript::AddOwnerDraw(int ownerdraw, UIScript::CallbackRaw* callback)
|
|
{
|
|
UIScript::UIOwnerDraws[ownerdraw] = callback;
|
|
}
|
|
|
|
bool UIScript::RunMenuScript(const char* name, const char** args)
|
|
{
|
|
if (UIScript::UIScripts.find(name) != UIScript::UIScripts.end())
|
|
{
|
|
UIScript::UIScripts[name](UIScript::Token(args));
|
|
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 = UIScript::UIOwnerDraws.begin(); i != UIScript::UIOwnerDraws.end(); ++i)
|
|
{
|
|
if (i->first == ownerDraw)
|
|
{
|
|
i->second();
|
|
}
|
|
}
|
|
}
|
|
|
|
Utils::Hook::Call<void(int, int, float*, int)>(0x4F58A0)(ownerDraw, flags, special, key);
|
|
}
|
|
|
|
void __declspec(naked) 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 UIScript::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()
|
|
{
|
|
// Install handler
|
|
Utils::Hook::RedirectJump(0x45EC59, UIScript::RunMenuScriptStub);
|
|
|
|
// Install ownerdraw handler
|
|
Utils::Hook(0x63D233, UIScript::OwnerDrawHandleKeyStub, HOOK_CALL).Install()->Quick();
|
|
}
|
|
|
|
UIScript::~UIScript()
|
|
{
|
|
UIScript::UIScripts.clear();
|
|
UIScript::UIOwnerDraws.clear();
|
|
}
|
|
}
|