34 lines
1.3 KiB
C#
34 lines
1.3 KiB
C#
using Content.Shared._Offbrand.Wounds;
|
|
using Content.Shared.StatusEffectNew;
|
|
|
|
namespace Content.Shared._Offbrand.StatusEffects;
|
|
|
|
public sealed class PainSuppressionStatusEffectSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly PainSystem _pain = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<PainSuppressionStatusEffectComponent, StatusEffectAppliedEvent>(OnStatusEffectApplied);
|
|
SubscribeLocalEvent<PainSuppressionStatusEffectComponent, StatusEffectRemovedEvent>(OnStatusEffectRemoved);
|
|
SubscribeLocalEvent<PainSuppressionStatusEffectComponent, StatusEffectRelayedEvent<PainSuppressionEvent>>(OnPainSuppression);
|
|
}
|
|
|
|
private void OnPainSuppression(Entity<PainSuppressionStatusEffectComponent> ent, ref StatusEffectRelayedEvent<PainSuppressionEvent> args)
|
|
{
|
|
args.Args = args.Args with { Suppressed = true };
|
|
}
|
|
|
|
private void OnStatusEffectApplied(Entity<PainSuppressionStatusEffectComponent> ent, ref StatusEffectAppliedEvent args)
|
|
{
|
|
_pain.UpdateSuppression(args.Target);
|
|
}
|
|
|
|
private void OnStatusEffectRemoved(Entity<PainSuppressionStatusEffectComponent> ent, ref StatusEffectRemovedEvent args)
|
|
{
|
|
_pain.UpdateSuppression(args.Target);
|
|
}
|
|
}
|