CatStation/Content.Server/EntityEffects/Effects/Botany/PlantAttributes/PlantChangeStatEntityEffectSystem.cs

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;

/// <summary>
/// Entity effect that changes a plant's stat by a random amount between a minimum and maximum value.
/// </summary>
/// <inheritdoc cref="EntityEffectSystem{T,TEffect}"/>
public sealed partial class PlantChangeStatEntityEffectSystem : EntityEffectSystem<PlantComponent, PlantChangeStat>
{
    [Dependency] private IComponentFactory _componentFactory = default!;
    [Dependency] private IRobustRandom _random = default!;
    [Dependency] private PlantHolderSystem _plantHolder = default!;
    [Dependency] private SharedEntityEffectsSystem _entityEffects = default!;

    protected override void Effect(Entity<PlantComponent> entity, ref EntityEffectEvent<PlantChangeStat> args)
    {
        if (_plantHolder.IsDead(entity.Owner))
            return;

        var targetDataField = args.Effect.TargetDataField;
        var targetComponent = args.Effect.TargetComponent;

        if (!_componentFactory.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);
    }
}