FUCKING ENTITYTEST

Author: MilonPL

Content.Server/_EE/FootPrint/FootPrintsSystem.cs

@@ -163,17 +163,35 @@ private void OnFootPrintInit(Entity<FootPrintComponent> ent, ref ComponentInit a
     /// </summary>
     private void OnFootPrintRemove(Entity<FootPrintComponent> ent, ref ComponentRemove args)
     {
-        if (!TryGetTilePos(ent.Owner, out var tilePos))
-            return;
+        // Try to get tile pos if entity is still valid
+        Vector2i? tilePos = null;
+        if (!Deleted(ent) && !EntityManager.IsQueuedForDeletion(ent) && Transform(ent).ParentUid.IsValid())
+        {
+            TryGetTilePos(ent.Owner, out var pos);
+            tilePos = pos;
+        }
+
+        // If we couldn't get tile pos, search all tiles to find and remove this footprint
+        if (tilePos == null)
+        {
+            foreach (var (pos, prints) in _footprintsPerTile)
+            {
+                if (!prints.Contains(ent.Owner))
+                    continue;
+
+                tilePos = pos;
+                break;
+            }
+        }
 
-        if (_footprintsPerTile.TryGetValue(tilePos, out var footprints))
+        // Clean up the footprint tracking if we found it
+        if (tilePos.HasValue && _footprintsPerTile.TryGetValue(tilePos.Value, out var footprints))
         {
-            // Create a new queue without the removed footprint
             var newQueue = new Queue<EntityUid>(footprints.Where(x => x != ent.Owner));
             if (newQueue.Count > 0)
-                _footprintsPerTile[tilePos] = newQueue;
+                _footprintsPerTile[tilePos.Value] = newQueue;
             else
-                _footprintsPerTile.Remove(tilePos);
+                _footprintsPerTile.Remove(tilePos.Value);
         }
     }