back to lobby state when server ends

src/server/Lobby.cpp

@@ -32,13 +32,19 @@ Lobby::Lobby(Server* server) : m_Server(server), m_Timer(1000, std::bind(&Lobby:
 
 }
 
+void Lobby::OpenLobby() {
+	m_Players.clear();
+	m_LobbyOpened = true;
+	m_StartTimerTime = 0;
+}
+
 void Lobby::Tick() {
-	if (m_GameStarted || m_StartTimerTime == 0)
+	if (!m_LobbyOpened || m_StartTimerTime == 0)
 		return;
 
 	if (utils::GetTime() - m_StartTimerTime >= LobbyWaitingTime) {
 		m_Server->GetGame().NotifyListeners(&game::GameListener::OnGameBegin);
-		m_GameStarted = true;
+		m_LobbyOpened = false;
 		return;
 	}
 
@@ -51,8 +57,9 @@ void Lobby::SendTimeRemaining() {
 }
 
 void Lobby::OnPlayerJoin(std::uint8_t playerID) {
-	if (m_GameStarted)
+	if (!m_LobbyOpened)
 		return;
+
 	utils::LOG("(Server) Player Joined Lobby !");
 	m_Players.push_back(playerID);
 	if (m_Players.size() == MIN_PLAYER_WAITING) { // start timer if a second player join the match
@@ -63,8 +70,9 @@ void Lobby::OnPlayerJoin(std::uint8_t playerID) {
 }
 
 void Lobby::OnPlayerLeave(std::uint8_t playerID) {
-	if (m_GameStarted)
+	if (!m_LobbyOpened)
 		return;
+
 	utils::LOG("(Server) Player Leaved Lobby !");
 
 	auto it = std::find(m_Players.begin(), m_Players.end(), playerID);

src/server/Server.cpp

@@ -52,6 +52,12 @@ void Server::StartThread() {
 	});
 }
 
+void Server::Restart() {
+	m_Game.GetWorld()->Reset();
+	m_Lobby.OpenLobby();
+	m_Game.SetGameState(game::GameState::Lobby);
+}
+
 void Server::Close() {
 	StopThread();
 }
@@ -76,6 +82,7 @@ bool Server::Start(std::uint16_t port, bool blocking) {
 	utils::LOG(utils::format("Server started at port %u !", port));
 	m_TickCounter.Reset();
 	m_ServerRunning = true;
+	m_Lobby.OpenLobby();
 	if (blocking) {
 		ServerLoop();
 	} else {
@@ -166,9 +173,9 @@ void Server::OnPlayerLeave(std::uint8_t id) {
 	protocol::PlayerLeavePacket packet(id);
 	BroadcastPacket(&packet);
 
-	if (GetPlayers().empty()) {
-		utils::LOG("All players left. Stopping server ...");
-		Stop();
+	if (GetPlayers().empty() && m_Game.GetGameState() != game::GameState::Lobby) {
+		utils::LOG("All players left. Go back to lobby ...");
+		Restart();
 	}
 }
 

src/server/game/ServerGame.cpp

@@ -122,8 +122,8 @@ void ServerGame::OnGameClose() {
 	protocol::DisconnectPacket packet("Game finished");
 	m_Server->BroadcastPacket(&packet);
 
-	// Closing server
-	m_Server->Close();
+	// Put back to lobby state
+	m_Server->Restart();
 }
 
 } // namespace game

src/td/game/World.cpp

@@ -110,6 +110,11 @@ void World::Tick(std::uint64_t delta) {
 	CleanDeadMobs();
 }
 
+void World::Reset() {
+	m_Towers.clear();
+	m_Mobs.clear();
+}
+
 void World::SpawnMobAt(MobID id, MobType type, std::uint8_t level, PlayerID sender, float x, float y, Direction dir) {
 	MobPtr mob = MobFactory::CreateMob(id, type, level, sender);
 	mob->SetCenter({ x, y });