From fb3e689be9d23f4bfd24502bdf4d5b56ac8c23b3 Mon Sep 17 00:00:00 2001
From: z411 <z411@omaera.org>
Date: Tue, 15 Aug 2023 14:43:10 -0400
Subject: [PATCH] Add queue logic when dealing with bots

---
 qcsrc/server/teamplay.qc | 20 ++++++++++++++------
 qcsrc/server/teamplay.qh |  2 +-
 2 files changed, 15 insertions(+), 7 deletions(-)

diff --git a/qcsrc/server/teamplay.qc b/qcsrc/server/teamplay.qc
index e303eb565..f7df74baa 100644
--- a/qcsrc/server/teamplay.qc
+++ b/qcsrc/server/teamplay.qc
@@ -236,7 +236,7 @@ bool Player_SetTeamIndex(entity player, int index)
 
 bool IsQueueNeeded(entity ignore)
 {
-	return (teamplay && autocvar_g_balance_teams_queue && !autocvar_g_campaign && AVAILABLE_TEAMS == 2 && TeamBalance_AreEqual(ignore));
+	return (teamplay && autocvar_g_balance_teams_queue && !autocvar_g_campaign && AVAILABLE_TEAMS == 2 && TeamBalance_AreEqual(ignore, true));
 }
 
 entity SpectatorWantsJoin(entity this)
@@ -695,26 +695,34 @@ int TeamBalance_GetAllowedTeams(entity balance)
 	return result;
 }
 
-bool TeamBalance_AreEqual(entity ignore)
+bool TeamBalance_AreEqual(entity ignore, bool would_leave)
 {
 	entity balance = TeamBalance_CheckAllowedTeams(ignore);
 	TeamBalance_GetTeamCounts(balance, ignore);
 
+	bool equality = true;
 	int total;
 	int prev_total = 0;
+	int bots = 0;
 
 	for(int i = 1; i <= AVAILABLE_TEAMS; ++i)
 	{
 		total = TeamBalance_GetTeamFromIndex(balance, i).m_num_players;
+		bots += TeamBalance_GetTeamFromIndex(balance, i).m_num_bots;
 		if(i > 1 && total != prev_total)
 		{
-			TeamBalance_Destroy(balance);
-			return false;
+			equality = false;
+			break;
 		}
 		prev_total = total;
 	}
 	TeamBalance_Destroy(balance);
-	return true;
+
+	// Ignore if there are "ghost" bots that would leave if anyone joined
+	if (would_leave && bots > autocvar_bot_number)
+		return false;
+
+	return equality;
 }
 
 entity remove_countdown;
@@ -723,7 +731,7 @@ int remove_timeleft;
 
 void Remove_Countdown(entity this)
 {
-	if(remove_timeleft <= 0 || TeamBalance_AreEqual(NULL))
+	if(remove_timeleft <= 0 || TeamBalance_AreEqual(NULL, false))
 	{
 		if(remove_timeleft <= 0)
 		{
diff --git a/qcsrc/server/teamplay.qh b/qcsrc/server/teamplay.qh
index 7996379e7..f677f45d7 100644
--- a/qcsrc/server/teamplay.qh
+++ b/qcsrc/server/teamplay.qh
@@ -189,7 +189,7 @@ void TeamBalance_Destroy(entity balance);
 /// \return Bitmask of allowed teams.
 int TeamBalance_GetAllowedTeams(entity balance);
 
-bool TeamBalance_AreEqual(entity ignore);
+bool TeamBalance_AreEqual(entity ignore, bool would_leave);
 void TeamBalance_RemoveExcessPlayers(entity ignore);
 
 /// \brief Returns whether the team change to the specified team is allowed.
-- 
2.39.2