using Content.Server.Body.Components;
using Content.Shared.Implants.Components;
+using Content.Shared.Storage;
using Robust.Shared.Containers;
namespace Content.Server.Implants;
SubscribeLocalEvent<ImplantedComponent, BeingGibbedEvent>(OnGibbed);
}
- private void OnImplantedInit(EntityUid uid, ImplantedComponent component, ComponentInit args)
+ private void OnImplantedInit(Entity<ImplantedComponent> ent, ref ComponentInit args)
{
- component.ImplantContainer = _container.EnsureContainer<Container>(uid, ImplanterComponent.ImplantSlotId);
- component.ImplantContainer.OccludesLight = false;
+ ent.Comp.ImplantContainer = _container.EnsureContainer<Container>(ent.Owner, ImplanterComponent.ImplantSlotId);
+ ent.Comp.ImplantContainer.OccludesLight = false;
}
- private void OnShutdown(EntityUid uid, ImplantedComponent component, ComponentShutdown args)
+ private void OnShutdown(Entity<ImplantedComponent> ent, ref ComponentShutdown args)
{
//If the entity is deleted, get rid of the implants
- _container.CleanContainer(component.ImplantContainer);
+ _container.CleanContainer(ent.Comp.ImplantContainer);
}
private void OnGibbed(Entity<ImplantedComponent> ent, ref BeingGibbedEvent args)
{
- //If the entity is gibbed, get rid of the implants
- _container.CleanContainer(ent.Comp.ImplantContainer);
+ // Drop the storage implant contents before the implants are deleted by the body being gibbed
+ foreach (var implant in ent.Comp.ImplantContainer.ContainedEntities)
+ {
+ if (TryComp<StorageComponent>(implant, out var storage))
+ _container.EmptyContainer(storage.Container, destination: Transform(ent).Coordinates);
+ }
+
}
}