send team update after restart

include/server/game/ServerGame.h

@@ -23,6 +23,8 @@ public:
 	ServerWorld* GetServerWorld() { return &m_ServerWorld; }
 
 	virtual void Tick(std::uint64_t delta);
+	virtual void Reset() override;
+
 	void StartGame();
 
 	// GameListener

include/td/game/BaseGame.h

@@ -42,7 +42,7 @@ public:
 
 	virtual void Tick(std::uint64_t delta);
 
-	void Reset();
+	virtual void Reset();
 
 	Team& GetRedTeam() { return m_Teams[static_cast<std::uint8_t>(TeamColor::Red)]; }
 	const Team& GetRedTeam() const { return m_Teams[static_cast<std::uint8_t>(TeamColor::Red)]; }

src/server/game/ServerGame.cpp

@@ -100,6 +100,17 @@ void ServerGame::BalanceTeams() {
 	}
 }
 
+void ServerGame::Reset() {
+	Game::Reset();
+
+	// update players teams to none
+	for (auto& playerInfo : m_Players) {
+		game::Player& player = playerInfo.second;
+		protocol::UpdatePlayerTeamPacket packet(player.GetID(), player.GetTeamColor());
+		m_Server->BroadcastPacket(&packet);
+	}
+}
+
 void ServerGame::OnGameStateUpdate(game::GameState newState) {
 	SetGameState(newState);
 	protocol::UpdateGameStatePacket packet(newState);