ContainerComp fix (#31434)

Forgot to push these ones on the branch.

diff --git a/Content.Shared/Containers/ContainerCompSystem.cs b/Content.Shared/Containers/ContainerCompSystem.cs
index 1e1983a331b9f5805a0076bfbc10084f8c4ee3e0..75908efea3aca9c192cfe36795d7b97d023a32de 100644 (file)
--- a/Content.Shared/Containers/ContainerCompSystem.cs
+++ b/Content.Shared/Containers/ContainerCompSystem.cs
@@ -1,5 +1,6 @@
 using Robust.Shared.Containers;
 using Robust.Shared.Prototypes;
+using Robust.Shared.Timing;
 
 namespace Content.Shared.Containers;
 
@@ -8,6 +9,7 @@ namespace Content.Shared.Containers;
 /// </summary>
 public sealed class ContainerCompSystem : EntitySystem
 {
+    [Dependency] private readonly IGameTiming _timing = default!;
     [Dependency] private readonly IPrototypeManager _proto = default!;
 
     public override void Initialize()
@@ -19,21 +21,18 @@ public sealed class ContainerCompSystem : EntitySystem
 
     private void OnConRemove(Entity<ContainerCompComponent> ent, ref EntRemovedFromContainerMessage args)
     {
-        if (args.Container.ID != ent.Comp.Container)
+        if (args.Container.ID != ent.Comp.Container || _timing.ApplyingState)
             return;
 
-        if (_proto.TryIndex(ent.Comp.Container, out var entProto))
+        if (_proto.TryIndex(ent.Comp.Proto, out var entProto))
         {
-            foreach (var entry in entProto.Components.Values)
-            {
-                RemComp(args.Entity, entry.Component);
-            }
+            EntityManager.RemoveComponents(args.Entity, entProto.Components);
         }
     }
 
     private void OnConInsert(Entity<ContainerCompComponent> ent, ref EntInsertedIntoContainerMessage args)
     {
-        if (args.Container.ID != ent.Comp.Container)
+        if (args.Container.ID != ent.Comp.Container || _timing.ApplyingState)
             return;
 
         if (_proto.TryIndex(ent.Comp.Proto, out var entProto))