From f4a11d22af3262cd38daf4f40ec82cfa7d5b9a3f Mon Sep 17 00:00:00 2001
From: LightCat <BenCat07@gmail.com>
Date: Sat, 15 Dec 2018 16:22:24 +0100
Subject: [PATCH] improve anti auto balance some more

---
 src/hooks/DispatchUserMessage.cpp | 24 +++++++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/src/hooks/DispatchUserMessage.cpp b/src/hooks/DispatchUserMessage.cpp
index 3525547f..a36a1eb5 100644
--- a/src/hooks/DispatchUserMessage.cpp
+++ b/src/hooks/DispatchUserMessage.cpp
@@ -91,10 +91,32 @@ DEFINE_HOOKED_METHOD(DispatchUserMessage, bool, void *this_, int type, bf_read &
     case 5:
         if (*anti_votekick && buf.GetNumBytesLeft() > 35)
         {
+            INetChannel *server = (INetChannel *)g_IEngine->GetNetChannelInfo();
+            static int anti_balance_attempts = 0;
             data = std::string(buf_data);
             logging::Info("%s", data.c_str());
             if (data.find("TeamChangeP") != data.npos && CE_GOOD(LOCAL_E))
-                g_IEngine->ClientCmd_Unrestricted("cat_disconnect read if gay;wait 100;cat_mm_join");
+            {
+                std::string server_name = server->GetName();
+                static std::string previous_name = "";
+                if (server_name != previous_name)
+                {
+                    previous_name = server_name;
+                    anti_balance_attempts = 0;
+                }
+                if (anti_balance_attempts < 2)
+                    g_IEngine->ClientCmd_Unrestricted("cat_disconnect read if gay;wait 100;cat_mm_join");
+                else
+                {
+                    std::string autobalance_msg = "tf_party_chat \"autobalanced in 3 seconds";
+                    if (ipc::peer && ipc::peer->connected)
+                        autobalance_msg += format(" IPC ID ", ipc::peer->client_id, "\"");
+                    else
+                        autobalance_msg += "\"";
+                    g_IEngine->ClientCmd_Unrestricted(autobalance_msg.c_str());
+                }
+                anti_balance_attempts++;
+            }
             buf.Seek(0);
         }
         break;