76 lines
2.7 KiB
C#
76 lines
2.7 KiB
C#
using System.Linq;
|
|
using Content.Shared._DV.Polymorph;
|
|
using Content.Shared.Body.Organ;
|
|
using Content.Shared.Intellicard;
|
|
using Content.Shared.Mind.Components;
|
|
using Robust.Shared.Containers;
|
|
|
|
namespace Content.Shared._DV.Containers;
|
|
|
|
public sealed class ContentContainerSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly SharedContainerSystem _container = default!;
|
|
[Dependency] private readonly SharedTransformSystem _transform = default!;
|
|
|
|
private List<EntityUid> _found = new();
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<ContainerManagerComponent, BeforePolymorphedEvent>(OnBeforePolymorphed);
|
|
}
|
|
|
|
private void OnBeforePolymorphed(Entity<ContainerManagerComponent> ent, ref BeforePolymorphedEvent args)
|
|
{
|
|
// Recursively drop all entities with MindContainerComponent in all containers on the entity
|
|
// This prevents players from being sent to null-space when carried by a polymorphing entity
|
|
var stack = new Stack<EntityUid>();
|
|
_found.Clear();
|
|
stack.Push(ent);
|
|
|
|
while (stack.Count > 0)
|
|
{
|
|
var currentUid = stack.Pop();
|
|
|
|
if (!HasComp<ContainerManagerComponent>(currentUid))
|
|
continue;
|
|
|
|
foreach (var container in _container.GetAllContainers(currentUid))
|
|
{
|
|
foreach (var entity in container.ContainedEntities)
|
|
{
|
|
// We will end up recursing over each of our body parts, so
|
|
// we need to make sure that we don't end up dropping our
|
|
// own brain, since that can gain a MindContainer if it was
|
|
// operated on.
|
|
if (TryComp<OrganComponent>(entity, out var organ)
|
|
&& organ.Body.HasValue) // Yeah so that's KINDA in use.
|
|
continue;
|
|
|
|
if (HasComp<MindContainerComponent>(entity))
|
|
{
|
|
_found.Add(entity);
|
|
continue;
|
|
}
|
|
|
|
if (HasComp<IntellicardComponent>(entity))
|
|
{
|
|
// Dump out the WHOLE intellicard rather than just the entities contained inside.
|
|
_found.Add(entity);
|
|
continue;
|
|
}
|
|
|
|
if (stack.Count < 1000) // Unlikely to have over 1000 nested entities unless there is an infinite loop.
|
|
stack.Push(entity); // Process this entity's containers next
|
|
}
|
|
}
|
|
}
|
|
|
|
foreach (var entity in _found)
|
|
{
|
|
_transform.AttachToGridOrMap(entity);
|
|
}
|
|
}
|
|
}
|