78 lines
2.7 KiB
C#
78 lines
2.7 KiB
C#
using Content.Server.Cuffs;
|
|
using Content.Shared.Cuffs.Components;
|
|
using Content.Shared.Implants;
|
|
using Content.Shared.Implants.Components;
|
|
using Content.Shared.Interaction.Events;
|
|
using Content.Shared.Mobs;
|
|
using Robust.Shared.Containers;
|
|
|
|
namespace Content.Server.Implants;
|
|
|
|
public sealed class SubdermalImplantSystem : SharedSubdermalImplantSystem
|
|
{
|
|
[Dependency] private readonly CuffableSystem _cuffable = default!;
|
|
[Dependency] private readonly SharedContainerSystem _container = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<SubdermalImplantComponent, UseFreedomImplantEvent>(OnFreedomImplant);
|
|
|
|
SubscribeLocalEvent<ImplantedComponent, MobStateChangedEvent>(RelayToImplantEvent);
|
|
SubscribeLocalEvent<ImplantedComponent, SuicideEvent>(RelayToImplantEvent);
|
|
}
|
|
|
|
private void OnFreedomImplant(EntityUid uid, SubdermalImplantComponent component, UseFreedomImplantEvent args)
|
|
{
|
|
if (!TryComp<CuffableComponent>(component.ImplantedEntity, out var cuffs) || cuffs.Container.ContainedEntities.Count < 1)
|
|
return;
|
|
|
|
_cuffable.Uncuff(component.ImplantedEntity.Value, cuffs.LastAddedCuffs, cuffs.LastAddedCuffs);
|
|
args.Handled = true;
|
|
}
|
|
|
|
#region Relays
|
|
|
|
//Relays from the implanted to the implant
|
|
private void RelayToImplantEvent<T>(EntityUid uid, ImplantedComponent component, T args) where T: notnull
|
|
{
|
|
if (!_container.TryGetContainer(uid, ImplanterComponent.ImplantSlotId, out var implantContainer))
|
|
return;
|
|
foreach (var implant in implantContainer.ContainedEntities)
|
|
{
|
|
RaiseLocalEvent(implant, args);
|
|
}
|
|
}
|
|
|
|
//Relays from the implanted to the implant
|
|
private void RelayToImplantEventByRef<T>(EntityUid uid, ImplantedComponent component, ref T args) where T: notnull
|
|
{
|
|
if (!_container.TryGetContainer(uid, ImplanterComponent.ImplantSlotId, out var implantContainer))
|
|
return;
|
|
foreach (var implant in implantContainer.ContainedEntities)
|
|
{
|
|
RaiseLocalEvent(implant,ref args);
|
|
}
|
|
}
|
|
|
|
//Relays from the implant to the implanted
|
|
private void RelayToImplantedEvent<T>(EntityUid uid, SubdermalImplantComponent component, T args) where T : EntityEventArgs
|
|
{
|
|
if (component.ImplantedEntity != null)
|
|
{
|
|
RaiseLocalEvent(component.ImplantedEntity.Value, args);
|
|
}
|
|
}
|
|
|
|
private void RelayToImplantedEventByRef<T>(EntityUid uid, SubdermalImplantComponent component, ref T args) where T : EntityEventArgs
|
|
{
|
|
if (component.ImplantedEntity != null)
|
|
{
|
|
RaiseLocalEvent(component.ImplantedEntity.Value, ref args);
|
|
}
|
|
}
|
|
|
|
#endregion
|
|
}
|