Fix Engineer able to build multiple dispensers (#167)

redsunservers · Oct 25, 2024 · 412dc72 · 412dc72
1 parent 563d185
commit 412dc72
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 1 deletion.
diff --git a/addons/sourcemod/scripting/superzombiefortress.sp b/addons/sourcemod/scripting/superzombiefortress.sp
@@ -19,7 +19,7 @@
 
 #include "include/superzombiefortress.inc"
 
-#define PLUGIN_VERSION				"4.6.6"
+#define PLUGIN_VERSION				"4.6.7"
 #define PLUGIN_VERSION_REVISION		"manual"
 
 #define MAX_CONTROL_POINTS	8

diff --git a/addons/sourcemod/scripting/szf/console.sp b/addons/sourcemod/scripting/szf/console.sp
@@ -310,5 +310,9 @@ public Action Console_Build(int iClient, const char[] sCommand, int iArgs)
 	if (nObjectType != TFObject_Dispenser && nObjectType != TFObject_Sentry)
 		return Plugin_Handled;
 
+	// Dispenser have m_bCarried set to 1 to disable healing, but also allowed to build multiples, which we want to prevent that
+	if (TF2_GetBuilding(iClient, nObjectType) != INVALID_ENT_REFERENCE)
+		return Plugin_Handled;
+
 	return Plugin_Continue;
 }
diff --git a/addons/sourcemod/scripting/szf/stocks.sp b/addons/sourcemod/scripting/szf/stocks.sp
@@ -812,6 +812,22 @@ stock void CheckClientWeapons(int iClient)
 	}
 }
 
+stock int TF2_GetBuilding(int iClient, TFObjectType nType, TFObjectMode nMode = TFObjectMode_None)
+{
+	int iBuilding = MaxClients+1;
+	while ((iBuilding = FindEntityByClassname(iBuilding, "obj_*")) > MaxClients)
+	{
+		if (GetEntPropEnt(iBuilding, Prop_Send, "m_hBuilder") == iClient
+			&& view_as<TFObjectType>(GetEntProp(iBuilding, Prop_Send, "m_iObjectType")) == nType
+			&& view_as<TFObjectMode>(GetEntProp(iBuilding, Prop_Send, "m_iObjectMode")) == nMode)
+		{
+			return iBuilding;
+		}
+	}
+
+	return INVALID_ENT_REFERENCE;
+}
+
 ////////////////
 // Cookie
 ////////////////