diff --git a/include/client/render/gui/GameMenu.h b/include/client/render/gui/GameMenu.h index 9c08ca3..dc3c49f 100644 --- a/include/client/render/gui/GameMenu.h +++ b/include/client/render/gui/GameMenu.h @@ -1,16 +1,19 @@ #pragma once #include "SummonMenu.h" +#include "td/game/BaseGame.h" namespace td { namespace gui { -class GameMenu : public GuiWidget { +class GameMenu : public GuiWidget, public game::GameListener { private: std::unique_ptr m_SummonMenu; public: GameMenu(client::Client* client); + virtual void OnGameBegin(); + virtual void Render(); private: void ShowTPS(); diff --git a/src/client/render/gui/GameMenu.cpp b/src/client/render/gui/GameMenu.cpp index 0d13c86..495c232 100644 --- a/src/client/render/gui/GameMenu.cpp +++ b/src/client/render/gui/GameMenu.cpp @@ -11,7 +11,7 @@ namespace td { namespace gui { GameMenu::GameMenu(client::Client* client) : GuiWidget(client), m_SummonMenu(std::make_unique(client)) { - + m_Client->GetGame().BindListener(this); } void GameMenu::Render() { @@ -105,5 +105,9 @@ void GameMenu::DisconnectButton() { } } +void GameMenu::OnGameBegin() { + m_SummonMenu->SetCooldown(10); +} + } // namespace gui } // namespace td