remove towers on player leave

include/server/game/ServerGame.h

@@ -35,6 +35,7 @@ public:
 	virtual void OnGameClose() override;
 
 	virtual void OnPlayerJoin(game::PlayerID id) override;
+	virtual void OnPlayerLeave(game::PlayerID id) override;
 private:
 	void BalanceTeams();
 	void InitPlayerStats();

include/td/game/World.h

@@ -219,7 +219,9 @@ public:
 
 	const TeamList& GetTeams() const;
 
+	TowerList& GetTowers() { return m_Towers; }
 	const TowerList& GetTowers() const { return m_Towers; }
+
 	TowerPtr GetTowerById(TowerID tower);
 
 	const Player* GetPlayerById(PlayerID id) const;

src/server/game/ServerGame.cpp

@@ -2,6 +2,7 @@
 #include "server/Server.h"
 
 #include "td/protocol/packets/DisconnectPacket.h"
+#include "td/protocol/packets/RemoveTowerPacket.h"
 #include "td/protocol/packets/UpdatePlayerTeamPacket.h"
 #include "td/protocol/packets/UpdateGameStatePacket.h"
 #include "td/protocol/packets/UpdateMoneyPacket.h"
@@ -139,5 +140,20 @@ void ServerGame::OnPlayerJoin(game::PlayerID id){
 		BalanceTeams();
 }
 
+void ServerGame::OnPlayerLeave(game::PlayerID playerId){
+	// temporary fix
+	auto& towerList = GetWorld()->GetTowers();
+	for(std::size_t i = 0; i < towerList.size(); i++) {
+		auto tower = towerList.at(i);
+
+		if(tower->GetBuilder() == playerId) {
+			protocol::RemoveTowerPacket packet(tower->GetID());
+			m_Server->BroadcastPacket(&packet);
+
+			towerList.erase(towerList.begin() + i);
+		}
+	}
+}
+
 } // namespace game
 } // namespace td