b9fa941ca6
* Oh the possibilities * Merge fixes * Forgot to remote LavaSystem oops * Changed EntityEffectArgs to EntityEffectBaseArgs and EntityEffectReagentArgs * Throw exception for unimplemented effectargs * Remove Json and overrideable datafields * Fix test issues * Actually fix the compiling issue * Fix comments and remove EntityEffectArgs (no longer used, replaced with EntityEffectBaseArgs)
78 lines
2.5 KiB
C#
78 lines
2.5 KiB
C#
using Content.Shared.Chemistry.Reagent;
|
|
using Content.Shared.EntityEffects;
|
|
using Content.Shared.StatusEffect;
|
|
using JetBrains.Annotations;
|
|
using Robust.Shared.Prototypes;
|
|
|
|
namespace Content.Server.EntityEffects.Effects.StatusEffects;
|
|
|
|
/// <summary>
|
|
/// Adds a generic status effect to the entity,
|
|
/// not worrying about things like how to affect the time it lasts for
|
|
/// or component fields or anything. Just adds a component to an entity
|
|
/// for a given time. Easy.
|
|
/// </summary>
|
|
/// <remarks>
|
|
/// Can be used for things like adding accents or something. I don't know. Go wild.
|
|
/// </remarks>
|
|
[UsedImplicitly]
|
|
public sealed partial class GenericStatusEffect : EntityEffect
|
|
{
|
|
[DataField(required: true)]
|
|
public string Key = default!;
|
|
|
|
[DataField]
|
|
public string Component = String.Empty;
|
|
|
|
[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;
|
|
|
|
public override void Effect(EntityEffectBaseArgs args)
|
|
{
|
|
var statusSys = args.EntityManager.EntitySysManager.GetEntitySystem<StatusEffectsSystem>();
|
|
|
|
var time = Time;
|
|
if (args is EntityEffectReagentArgs reagentArgs)
|
|
time *= reagentArgs.Scale.Float();
|
|
|
|
if (Type == StatusEffectMetabolismType.Add && Component != String.Empty)
|
|
{
|
|
statusSys.TryAddStatusEffect(args.TargetEntity, Key, TimeSpan.FromSeconds(time), Refresh, Component);
|
|
}
|
|
else if (Type == StatusEffectMetabolismType.Remove)
|
|
{
|
|
statusSys.TryRemoveTime(args.TargetEntity, Key, TimeSpan.FromSeconds(time));
|
|
}
|
|
else if (Type == StatusEffectMetabolismType.Set)
|
|
{
|
|
statusSys.TrySetTime(args.TargetEntity, Key, TimeSpan.FromSeconds(time));
|
|
}
|
|
}
|
|
|
|
protected override string? ReagentEffectGuidebookText(IPrototypeManager prototype, IEntitySystemManager entSys) => Loc.GetString(
|
|
"reagent-effect-guidebook-status-effect",
|
|
("chance", Probability),
|
|
("type", Type),
|
|
("time", Time),
|
|
("key", $"reagent-effect-status-effect-{Key}"));
|
|
}
|
|
|
|
public enum StatusEffectMetabolismType
|
|
{
|
|
Add,
|
|
Remove,
|
|
Set
|
|
}
|