.

Lan2Play · Nov 4, 2023 · 6ed0ebc · 6ed0ebc
1 parent 3fe0543
commit 6ed0ebc
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/PugSharp/Player.cs b/PugSharp/Player.cs
@@ -11,7 +11,8 @@ namespace PugSharp;
 public class Player : IPlayer
 {
     private static readonly ILogger<Player> _Logger = LogManager.CreateLogger<Player>();
-    private readonly CCSPlayerController _PlayerController;
+    private uint _EntityIndex;
+    private CCSPlayerController _PlayerController;
 
     public Player(CCSPlayerController playerController)
     {
@@ -20,6 +21,7 @@ public Player(CCSPlayerController playerController)
             _Logger.LogError("PlayerController is invalid!");
         }
 
+        _EntityIndex = playerController.EntityIndex!.Value.Value;
         _PlayerController = playerController;
         if (_PlayerController.ActionTrackingServices != null)
         {
@@ -29,17 +31,32 @@ public Player(CCSPlayerController playerController)
 
     private T DefaultIfInvalid<T>(Func<T> loadValue) where T : struct
     {
+        if (_PlayerController == null || !_PlayerController.IsValid)
+        {
+            _PlayerController = Utilities.GetPlayerFromIndex((int)_EntityIndex);
+        }
+
         return _PlayerController != null && _PlayerController.IsValid ? loadValue() : default;
     }
 
     private T DefaultIfInvalid<T>(Func<T> loadValue, T defaultValue)
     {
+        if (_PlayerController == null || !_PlayerController.IsValid)
+        {
+            _PlayerController = Utilities.GetPlayerFromIndex((int)_EntityIndex);
+        }
+
         return _PlayerController != null && _PlayerController.IsValid ? loadValue() : defaultValue;
     }
 
 
     private T? NullIfInvalid<T>(Func<T?> loadValue)
     {
+        if (_PlayerController == null || !_PlayerController.IsValid)
+        {
+            _PlayerController = Utilities.GetPlayerFromIndex((int)_EntityIndex);
+        }
+
         return _PlayerController != null && _PlayerController.IsValid ? loadValue() : default;
     }