59 lines
2.1 KiB
C#
59 lines
2.1 KiB
C#
using System.Linq;
|
|
using Content.Server.Body.Components;
|
|
using Content.Shared.Implants.Components;
|
|
using Robust.Shared.Containers;
|
|
|
|
namespace Content.Server.Implants;
|
|
|
|
public sealed partial class ImplanterSystem
|
|
{
|
|
public const string BaseStorageId = "storagebase";
|
|
|
|
public void InitializeImplanted()
|
|
{
|
|
SubscribeLocalEvent<ImplantedComponent, ComponentInit>(OnImplantedInit);
|
|
SubscribeLocalEvent<ImplantedComponent, ComponentShutdown>(OnShutdown);
|
|
SubscribeLocalEvent<ImplantedComponent, BeingGibbedEvent>(OnGibbed);
|
|
}
|
|
|
|
private void OnImplantedInit(EntityUid uid, ImplantedComponent component, ComponentInit args)
|
|
{
|
|
component.ImplantContainer = _container.EnsureContainer<Container>(uid, ImplanterComponent.ImplantSlotId);
|
|
component.ImplantContainer.OccludesLight = false;
|
|
}
|
|
|
|
private void OnShutdown(EntityUid uid, ImplantedComponent component, ComponentShutdown args)
|
|
{
|
|
//If the entity is deleted, get rid of the implants
|
|
_container.CleanContainer(component.ImplantContainer);
|
|
}
|
|
|
|
private void OnGibbed(Entity<ImplantedComponent> ent, ref BeingGibbedEvent args)
|
|
{
|
|
foreach (var implant in ent.Comp.ImplantContainer.ContainedEntities)
|
|
{
|
|
if (!TryComp<SubdermalImplantComponent>(implant, out var subdermalImplant))
|
|
continue;
|
|
|
|
if (!subdermalImplant.DropContainerItemsIfGib)
|
|
continue;
|
|
|
|
if (!_container.TryGetContainer(implant, BaseStorageId, out var storageImplant))
|
|
continue;
|
|
|
|
var entCoords = Transform(ent.Owner).Coordinates;
|
|
|
|
var containedEntites = storageImplant.ContainedEntities.ToArray();
|
|
|
|
foreach (var entity in containedEntites)
|
|
{
|
|
if (Terminating(entity))
|
|
continue;
|
|
|
|
_container.RemoveEntity(storageImplant.Owner, entity, force: true, destination: entCoords);
|
|
}
|
|
}
|
|
//If the entity is gibbed, get rid of the implants
|
|
_container.CleanContainer(ent.Comp.ImplantContainer);
|
|
}
|
|
}
|