tommy/tbd-station-14
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ContainerComp fix (#31434)

Browse Source 
Forgot to push these ones on the branch.
This commit is contained in:
metalgearsloth
2024-08-26 01:22:01 +10:00
committed by GitHub
parent ff412a65ba
commit ae4c7ad0bc
1 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
Content.Shared/Containers/ContainerCompSystem.cs
View File

@@ -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))