fix: clear zone.Data on round start

fix: cache gameRules
oscar-wos · Aug 7, 2024 · 968f806 · 968f806
1 parent a0d0406
commit 968f806
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 4 deletions.
diff --git a/src/AntiRush.cs b/src/AntiRush.cs
@@ -31,6 +31,8 @@ public override void Load(bool isReload)
 
         Server.NextFrame(() =>
         {
+            _gameRules = Utilities.FindAllEntitiesByDesignerName<CCSGameRulesProxy>("cs_gamerules").First().GameRules!;
+
             foreach (var controller in Utilities.GetPlayers())
                 _playerData[controller] = new PlayerData();
 

diff --git a/src/Events.cs b/src/Events.cs
@@ -11,14 +11,17 @@ private HookResult OnRoundStart(EventRoundStart @event, GameEventInfo info)
         _roundStart = Server.CurrentTime;
         _bombPlanted = false;
 
-        var gameRules = Utilities.FindAllEntitiesByDesignerName<CCSGameRulesProxy>("cs_gamerules").First().GameRules!;
-        _warmup = gameRules.WarmupPeriod;
+        _gameRules ??= Utilities.FindAllEntitiesByDesignerName<CCSGameRulesProxy>("cs_gamerules").First().GameRules!;
+        _warmup = _gameRules.WarmupPeriod;
 
         if (!Config.DrawZones)
             return HookResult.Continue;
 
         foreach (var zone in _zones)
+        {
+            zone.Data = [];
             zone.Draw();
+        }
 
         return HookResult.Continue;
     }

diff --git a/src/Globals.cs b/src/Globals.cs
@@ -8,7 +8,7 @@ namespace AntiRush;
 public partial class AntiRush
 {
     public override string ModuleName => "AntiRush";
-    public override string ModuleVersion => "1.0.5";
+    public override string ModuleVersion => "1.0.6";
     public override string ModuleAuthor => "https://github.com/oscar-wos/AntiRush";
     public AntiRushConfig Config { get; set; } = new();
     public Menu.Menu Menu { get; } = new();
@@ -20,4 +20,5 @@ public partial class AntiRush
     private bool _bombPlanted;
     private bool _warmup;
     private float[] _countdown = [];
+    private CCSGameRules? _gameRules;
 }
diff --git a/src/Zone.cs b/src/Zone.cs
@@ -16,7 +16,7 @@ public class Zone(string name, ZoneType type, float delay, int damage, CsTeam[]
     public CsTeam[] Teams { get; init; } = teams;
     public Vector MinPoint { get; init; } = minPoint;
     public Vector MaxPoint { get; init; } = maxPoint;
-    public Dictionary<CCSPlayerController, ZoneData> Data { get; } = [];
+    public Dictionary<CCSPlayerController, ZoneData> Data { get; set; } = [];
     public CBeam[] Beams { get; } = [];
 
     public bool IsInZone(Vector point)