CatStation/Content.Shared/Botany/Systems/PlantWeedPestSystem.cs

using JetBrains.Annotations;
using Content.Shared.Botany.Components;
using Content.Shared.Botany.Events;
using Robust.Shared.Random;

namespace Content.Shared.Botany.Systems;

/// <summary>
/// Manages weed growth and pest damage per growth tick, and handles tray-level
/// weed spawning.
/// </summary>
public sealed partial class PlantWeedPestSystem : EntitySystem
{
    [Dependency] private BotanySystem _botany = default!;
    [Dependency] private IRobustRandom _random = default!;
    [Dependency] private PlantMutationSystem _mutation = default!;
    [Dependency] private PlantHolderSystem _plantHolder = default!;
    [Dependency] private PlantTraySystem _plantTray = default!;

    [Dependency] private EntityQuery<PlantTrayComponent> _trayQuery = default!;

    [SubscribeLocalEvent]
    private void OnCrossPollinate(Entity<PlantWeedPestComponent> ent, ref PlantCrossPollinateEvent args)
    {
        if (!_botany.TryGetPlantComponent<PlantWeedPestComponent>(args.PollenData, args.PollenProtoId, out var pollenData))
            return;

        _mutation.CrossFloat(ref ent.Comp.WeedTolerance, pollenData.WeedTolerance);
        _mutation.CrossFloat(ref ent.Comp.PestTolerance, pollenData.PestTolerance);
        Dirty(ent);
    }

    [SubscribeLocalEvent]
    private void OnPlantGrow(Entity<PlantWeedPestComponent> ent, ref PlantGrowEvent args)
    {
        var trayUid = GetEntity(args.Tray);
        if (!_trayQuery.TryComp(trayUid, out var tray))
            return;

        if (_random.Prob(ent.Comp.PestGrowthChance))
            _plantTray.AdjustPest((trayUid, tray), ent.Comp.PestGrowthAmount);

        if (tray.PestLevel > ent.Comp.PestTolerance)
            _plantHolder.AdjustsHealth(ent.Owner, -ent.Comp.PestDamageAmount);
    }

    /// <summary>
    /// Adjusts maximum weed level the plant can tolerate before taking damage.
    /// </summary>
    [PublicAPI]
    public void AdjustWeedTolerance(Entity<PlantWeedPestComponent?> ent, float amount)
    {
        if (!Resolve(ent.Owner, ref ent.Comp, false))
            return;

        ent.Comp.WeedTolerance = MathF.Max(0f, ent.Comp.WeedTolerance + amount);
        DirtyField(ent, nameof(ent.Comp.WeedTolerance));
    }

    /// <summary>
    /// Adjusts maximum pest level the plant can tolerate before taking damage.
    /// </summary>
    [PublicAPI]
    public void AdjustPestTolerance(Entity<PlantWeedPestComponent?> ent, float amount)
    {
        if (!Resolve(ent.Owner, ref ent.Comp, false))
            return;

        ent.Comp.PestTolerance = MathF.Max(0f, ent.Comp.PestTolerance + amount);
        DirtyField(ent, nameof(ent.Comp.PestTolerance));
    }
}