ss14-wega/Content.Server/Implants/ImplantedSystem.cs

using Content.Shared.Body.Events;
using Content.Shared.Gibbing;
using Content.Shared.Implants.Components;
using Content.Shared.Storage;
using Robust.Shared.Containers;

namespace Content.Server.Implants;

public sealed partial class ImplanterSystem
{
    public void InitializeImplanted()
    {
        SubscribeLocalEvent<ImplantedComponent, ComponentInit>(OnImplantedInit);
        SubscribeLocalEvent<ImplantedComponent, ComponentShutdown>(OnShutdown);
        SubscribeLocalEvent<ImplantedComponent, GibbedBeforeDeletionEvent>(OnGibbed);
    }

    private void OnImplantedInit(Entity<ImplantedComponent> ent, ref ComponentInit args)
    {
        ent.Comp.ImplantContainer = _container.EnsureContainer<Container>(ent.Owner, ImplanterComponent.ImplantSlotId);
        ent.Comp.ImplantContainer.OccludesLight = false;
    }

    private void OnShutdown(Entity<ImplantedComponent> ent, ref ComponentShutdown args)
    {
        //If the entity is deleted, get rid of the implants
        _container.CleanContainer(ent.Comp.ImplantContainer);
    }

    private void OnGibbed(Entity<ImplantedComponent> ent, ref GibbedBeforeDeletionEvent args)
    {
        // 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);
        }

    }
}