From c60303d364a9369cab716f3097b2df4687fe61d2 Mon Sep 17 00:00:00 2001
From: Lyberta <lyberta@lyberta.net>
Date: Tue, 13 Jun 2017 06:43:19 +0300
Subject: [PATCH] Updated to new balance API.

---
 .../mutators/mutator/gamemode_survival.qc     | 34 +++++++++++++------
 1 file changed, 24 insertions(+), 10 deletions(-)

diff --git a/qcsrc/server/mutators/mutator/gamemode_survival.qc b/qcsrc/server/mutators/mutator/gamemode_survival.qc
index 3f701bc24..18dd43b52 100644
--- a/qcsrc/server/mutators/mutator/gamemode_survival.qc
+++ b/qcsrc/server/mutators/mutator/gamemode_survival.qc
@@ -501,9 +501,10 @@ bool Surv_AddPlayerToTeam(entity player, int teamnum)
 			});
 			if (lowestplayer != NULL)
 			{
+				bool savedautobalance = surv_autobalance;
 				surv_autobalance = false;
 				SetPlayerTeamSimple(lowestplayer, surv_defenderteam);
-				surv_autobalance = true;
+				surv_autobalance = savedautobalance;
 				return true;
 			}
 			FOREACH_CLIENT(true,
@@ -521,9 +522,10 @@ bool Surv_AddPlayerToTeam(entity player, int teamnum)
 			});
 			if (lowestplayer != NULL)
 			{
+				bool savedautobalance = surv_autobalance;
 				surv_autobalance = false;
 				SetPlayerTeamSimple(lowestplayer, surv_defenderteam);
-				surv_autobalance = true;
+				surv_autobalance = savedautobalance;
 			}
 			return true;
 		}
@@ -592,10 +594,11 @@ bool Surv_AddPlayerToTeam(entity player, int teamnum)
 				{
 					player.surv_savedplayerstate = Surv_SavePlayerState(bot);
 				}
+				bool savedautobalance = surv_autobalance;
 				surv_autobalance = false;
 				surv_announcefrags = false;
 				SetPlayerTeamSimple(bot, surv_attackerteam);
-				surv_autobalance = true;
+				surv_autobalance = savedautobalance;
 				surv_announcefrags = true;
 				LOG_TRACE("Removed bot");
 			}
@@ -624,9 +627,10 @@ bool Surv_AddPlayerToTeam(entity player, int teamnum)
 			});
 			if (lowestplayer != NULL)
 			{
+				bool savedautobalance = surv_autobalance;
 				surv_autobalance = false;
 				SetPlayerTeamSimple(lowestplayer, surv_attackerteam);
-				surv_autobalance = true;
+				surv_autobalance = savedautobalance;
 				return true;
 			}
 			FOREACH_CLIENT(true,
@@ -643,9 +647,10 @@ bool Surv_AddPlayerToTeam(entity player, int teamnum)
 			});
 			if (lowestplayer != NULL)
 			{
+				bool savedautobalance = surv_autobalance;
 				surv_autobalance = false;
 				SetPlayerTeamSimple(lowestplayer, surv_attackerteam);
-				surv_autobalance = true;
+				surv_autobalance = savedautobalance;
 			}
 			return true;
 		}
@@ -728,9 +733,10 @@ void Surv_RemovePlayerFromTeam(entity player, int teamnum)
 			});
 			if (lowestplayer != NULL)
 			{
+				bool savedautobalance = surv_autobalance;
 				surv_autobalance = false;
 				SetPlayerTeamSimple(lowestplayer, surv_attackerteam);
-				surv_autobalance = true;
+				surv_autobalance = savedautobalance;
 				return;
 			}
 			FOREACH_CLIENT(true,
@@ -749,9 +755,10 @@ void Surv_RemovePlayerFromTeam(entity player, int teamnum)
 			{
 				return;
 			}
+			bool savedautobalance = surv_autobalance;
 			surv_autobalance = false;
 			SetPlayerTeamSimple(lowestplayer, surv_attackerteam);
-			surv_autobalance = true;
+			surv_autobalance = savedautobalance;
 			return;
 		}
 		case surv_defenderteam:
@@ -793,10 +800,11 @@ void Surv_RemovePlayerFromTeam(entity player, int teamnum)
 					{
 						it.surv_savedplayerstate = Surv_SavePlayerState(player);
 					}
+					bool savedautobalance = surv_autobalance;
 					surv_autobalance = false;
 					surv_announcefrags = false;
 					SetPlayerTeamSimple(it, surv_defenderteam);
-					surv_autobalance = true;
+					surv_autobalance = savedautobalance;
 					surv_announcefrags = true;					
 					return;
 				}
@@ -817,10 +825,11 @@ void Surv_RemovePlayerFromTeam(entity player, int teamnum)
 			});
 			if (lowestplayer != NULL)
 			{
+				bool savedautobalance = surv_autobalance;
 				surv_autobalance = false;
 				surv_announcefrags = false;
 				SetPlayerTeamSimple(lowestplayer, surv_defenderteam);
-				surv_autobalance = true;
+				surv_autobalance = savedautobalance;
 				surv_announcefrags = true;
 				return;
 			}
@@ -840,10 +849,11 @@ void Surv_RemovePlayerFromTeam(entity player, int teamnum)
 			{
 				return;
 			}
+			bool savedautobalance = surv_autobalance;
 			surv_autobalance = false;
 			surv_announcefrags = false;
 			SetPlayerTeamSimple(lowestplayer, surv_defenderteam);
-			surv_autobalance = true;
+			surv_autobalance = savedautobalance;
 			surv_announcefrags = true;
 			return;
 		}
@@ -1606,6 +1616,10 @@ MUTATOR_HOOKFUNCTION(surv, CheckAllowedTeams, CBC_ORDER_EXCLUSIVE)
 {
 	entity player = M_ARGV(2, entity);
 	LOG_TRACE("Survival: CheckAllowedTeams, player = ", player.netname);
+	if (player == NULL)
+	{
+		return SURVIVAL_TEAM_BITS;
+	}
 	if (IS_BOT_CLIENT(player))
 	{
 		int teambits = surv_attackerteambit;
-- 
2.39.5