using Content.Shared.Botany.Components;
using Content.Shared.Botany.Systems;
using Content.Shared.EntityEffects;
using Content.Shared.EntityEffects.Effects.Botany.PlantAttributes;
using Robust.Shared.Random;
namespace Content.Server.EntityEffects.Effects.Botany.PlantAttributes;
///
/// Entity effect that changes a plant's stat by a random amount between a minimum and maximum value.
///
///
public sealed partial class PlantChangeStatEntityEffectSystem : EntityEffectSystem
{
[Dependency] private IRobustRandom _random = default!;
[Dependency] private PlantHolderSystem _plantHolder = default!;
[Dependency] private SharedEntityEffectsSystem _entityEffects = default!;
protected override void Effect(Entity entity, ref EntityEffectEvent args)
{
if (_plantHolder.IsDead(entity.Owner))
return;
var targetDataField = args.Effect.TargetDataField;
var targetComponent = args.Effect.TargetComponent;
if (!Factory.TryGetRegistration(targetComponent, out var registration))
{
Log.Error($"{nameof(PlantChangeStat)} Error: Component '{targetComponent}' is not a valid component name.");
return;
}
if (!TryComp(entity.Owner, registration.Type, out var plantComp))
return;
var field = registration.Type.GetField(targetDataField);
if (field == null)
{
Log.Error(
$"{nameof(PlantChangeStat)} Error: Field '{targetDataField}' not found on component '{targetComponent}'. Did you misspell it?");
return;
}
var currentValue = field.GetValue(plantComp);
if (currentValue == null)
return;
float current;
switch (currentValue)
{
case float f:
current = f;
break;
case int i:
current = i;
break;
default:
Log.Error(
$"{nameof(PlantChangeStat)} Error: Field '{targetDataField}' on component '{targetComponent}' has unsupported type '{currentValue.GetType().Name}'.");
return;
}
var min = args.Effect.ApplyRange.Min;
var max = args.Effect.ApplyRange.Max;
// If the range is degenerate, only move toward that value.
if (MathHelper.CloseTo(min, max))
{
if (MathHelper.CloseTo(current, min))
return;
_entityEffects.TryApplyEffect(entity.Owner,
current < min ? args.Effect.Up : args.Effect.Down,
args.Scale,
args.User);
return;
}
bool goUp;
if (current <= min)
{
goUp = true;
}
else if (current >= max)
{
goUp = false;
}
else
{
var thermometer = (current - min) / (max - min);
thermometer = Math.Clamp(thermometer, 0f, 1f);
var probUp = 1f - thermometer;
goUp = _random.Prob(probUp);
}
_entityEffects.TryApplyEffect(entity.Owner, goUp ? args.Effect.Up : args.Effect.Down, args.Scale, args.User);
}
}