67 lines
2.2 KiB
C#
67 lines
2.2 KiB
C#
|
using Content.Shared.StatusEffectNew;
|
||
|
|
using JetBrains.Annotations;
|
||
|
|
using Robust.Shared.Prototypes;
|
||
|
|
|
||
|
|
namespace Content.Shared.EntityEffects.Effects.StatusEffects;
|
||
|
|
|
||
|
|
/// <summary>
|
||
|
|
/// Changes status effects on entities: Adds, removes or sets time.
|
||
|
|
/// </summary>
|
||
|
|
[UsedImplicitly]
|
||
|
|
public sealed partial class ModifyStatusEffect : EntityEffect
|
||
|
|
{
|
||
|
|
[DataField(required: true)]
|
||
|
|
public EntProtoId EffectProto;
|
||
|
|
|
||
|
|
/// <summary>
|
||
|
|
/// Time for which status effect should be applied. Behaviour changes according to <see cref="Refresh" />.
|
||
|
|
/// </summary>
|
||
|
|
[DataField]
|
||
|
|
public float Time = 2.0f;
|
||
|
|
|
||
|
|
/// <remarks>
|
||
|
|
/// true - refresh status effect time, false - accumulate status effect time.
|
||
|
|
/// </remarks>
|
||
|
|
[DataField]
|
||
|
|
public bool Refresh = true;
|
||
|
|
|
||
|
|
/// <summary>
|
||
|
|
/// Should this effect add the status effect, remove time from it, or set its cooldown?
|
||
|
|
/// </summary>
|
||
|
|
[DataField]
|
||
|
|
public StatusEffectMetabolismType Type = StatusEffectMetabolismType.Add;
|
||
|
|
|
||
|
|
/// <inheritdoc />
|
||
|
|
public override void Effect(EntityEffectBaseArgs args)
|
||
|
|
{
|
||
|
|
var statusSys = args.EntityManager.EntitySysManager.GetEntitySystem<SharedStatusEffectsSystem>();
|
||
|
|
|
||
|
|
var time = Time;
|
||
|
|
if (args is EntityEffectReagentArgs reagentArgs)
|
||
|
|
time *= reagentArgs.Scale.Float();
|
||
|
|
|
||
|
|
switch (Type)
|
||
|
|
{
|
||
|
|
case StatusEffectMetabolismType.Add:
|
||
|
|
statusSys.TryAddStatusEffect(args.TargetEntity, EffectProto, TimeSpan.FromSeconds(time), Refresh);
|
||
|
|
break;
|
||
|
|
case StatusEffectMetabolismType.Remove:
|
||
|
|
statusSys.TryAddTime(args.TargetEntity, EffectProto, -TimeSpan.FromSeconds(time));
|
||
|
|
break;
|
||
|
|
case StatusEffectMetabolismType.Set:
|
||
|
|
statusSys.TrySetTime(args.TargetEntity, EffectProto, TimeSpan.FromSeconds(time));
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/// <inheritdoc />
|
||
|
|
protected override string? ReagentEffectGuidebookText(IPrototypeManager prototype, IEntitySystemManager entSys)
|
||
|
|
=> Loc.GetString(
|
||
|
|
"reagent-effect-guidebook-status-effect",
|
||
|
|
("chance", Probability),
|
||
|
|
("type", Type),
|
||
|
|
("time", Time),
|
||
|
|
("key", prototype.Index(EffectProto).Name)
|
||
|
|
);
|
||
|
|
}
|