balance team on player join

include/server/game/ServerGame.h

@@ -33,6 +33,8 @@ public:
 	virtual void OnGameBegin() override;
 	virtual void OnGameEnd() override;
 	virtual void OnGameClose() override;
+
+	virtual void OnPlayerJoin(game::PlayerID id) override;
 private:
 	void BalanceTeams();
 	void InitPlayerStats();

src/server/ServerConnexion.cpp

@@ -115,6 +115,8 @@ void ServerConnexion::HandlePacket(const protocol::PlayerLoginPacket* packet) {
 
 		protocol::PlayerListPacket listPacket(playerNames);
 		SendPacket(&listPacket);
+
+		m_Server->GetGame().NotifyListeners(&game::GameListener::OnPlayerJoin, m_ID);
 	}
 }
 

src/server/game/ServerGame.cpp

@@ -134,5 +134,10 @@ void ServerGame::OnGameClose() {
 	m_Server->Restart();
 }
 
+void ServerGame::OnPlayerJoin(game::PlayerID id){
+	if (m_GameState == game::GameState::Game)
+		BalanceTeams();
+}
+
 } // namespace game
 } // namespace td