CatStation/Content.Shared/Nutrition/EntitySystems/IngestionSystem.Blockers.cs

using System.Linq;
using Content.Shared.Chemistry.Components;
using Content.Shared.Clothing;
using Content.Shared.Containers.ItemSlots;
using Content.Shared.Fluids.Components;
using Content.Shared.Foldable;
using Content.Shared.Interaction.Components;
using Content.Shared.Inventory;
using Content.Shared.Nutrition.Components;
using Content.Shared.Storage;

namespace Content.Shared.Nutrition.EntitySystems;

public sealed partial class IngestionSystem
{
    [Dependency] private OpenableSystem _openable = default!;

    public void InitializeBlockers()
    {
        SubscribeLocalEvent<UnremoveableComponent, IngestibleEvent>(OnUnremovableIngestion);
        SubscribeLocalEvent<IngestionBlockerComponent, ItemMaskToggledEvent>(OnBlockerMaskToggled);
        SubscribeLocalEvent<IngestionBlockerComponent, FoldedEvent>(OnBlockerFolded);
        SubscribeLocalEvent<IngestionBlockerComponent, IngestionAttemptEvent>(OnIngestionBlockerAttempt);
        SubscribeLocalEvent<IngestionBlockerComponent, InventoryRelayedEvent<IngestionAttemptEvent>>(OnIngestionBlockerAttempt);

        // Edible Event
        SubscribeLocalEvent<EdibleComponent, EdibleEvent>(OnEdible);
        SubscribeLocalEvent<StorageComponent, EdibleEvent>(OnStorageEdible);
        SubscribeLocalEvent<ItemSlotsComponent, EdibleEvent>(OnItemSlotsEdible);
        SubscribeLocalEvent<OpenableComponent, EdibleEvent>(OnOpenableEdible);

        // Digestion Events
        SubscribeLocalEvent<EdibleComponent, IsDigestibleEvent>(OnEdibleIsDigestible);
        SubscribeLocalEvent<DrainableSolutionComponent, IsDigestibleEvent>(OnDrainableIsDigestible);
        SubscribeLocalEvent<PuddleComponent, IsDigestibleEvent>(OnPuddleIsDigestible);

        SubscribeLocalEvent<PillComponent, BeforeIngestedEvent>(OnPillBeforeEaten);
    }

    private void OnUnremovableIngestion(Entity<UnremoveableComponent> entity, ref IngestibleEvent args)
    {
        // If we can't remove it we probably shouldn't be able to eat it.
        // TODO: Separate glue and Unremovable component.
        args.Cancelled = true;
    }

    private void OnBlockerMaskToggled(Entity<IngestionBlockerComponent> entity, ref ItemMaskToggledEvent args)
    {
        entity.Comp.Enabled = !args.Mask.Comp.IsToggled;
        Dirty(entity);
    }

    private void OnBlockerFolded(Entity<IngestionBlockerComponent> entity, ref FoldedEvent args)
    {
        entity.Comp.Enabled = !args.IsFolded;
        Dirty(entity);
    }

    private void OnIngestionBlockerAttempt(Entity<IngestionBlockerComponent> entity, ref IngestionAttemptEvent args)
    {
        if (!args.Cancelled && entity.Comp.Enabled)
            args.Cancelled = true;
    }

    /// <summary>
    ///     Block ingestion attempts based on the equipped mask or head-wear
    /// </summary>
    private void OnIngestionBlockerAttempt(Entity<IngestionBlockerComponent> entity, ref InventoryRelayedEvent<IngestionAttemptEvent> args)
    {
        if (args.Args.Cancelled || !entity.Comp.Enabled)
            return;

        args.Args.Cancelled = true;
        args.Args.Blocker = entity;
    }

    private void OnEdible(Entity<EdibleComponent> entity, ref EdibleEvent args)
    {
        if (args.Cancelled || args.Solution != null)
            return;

        if (entity.Comp.UtensilRequired && !HasRequiredUtensils(args.User, entity.Comp.Utensil))
        {
            args.Cancelled = true;
            return;
        }

        // Check this last
        if (!_solutionContainer.TryGetSolution(entity.Owner, entity.Comp.Solution, out args.Solution) || IsEmpty(entity) && !entity.Comp.DestroyOnEmpty)
        {
            args.Cancelled = true;

            _popup.PopupEntity(Loc.GetString("ingestion-try-use-is-empty", ("entity", entity)), entity, args.User);
            return;
        }

        // Time is additive because I said so.
        args.Time += entity.Comp.Delay;
    }

    private void OnStorageEdible(Entity<StorageComponent> ent, ref EdibleEvent args)
    {
        if (args.Cancelled)
            return;

        if (!ent.Comp.Container.ContainedEntities.Any())
            return;

        args.Cancelled = true;

        _popup.PopupEntity(Loc.GetString("edible-has-used-storage", ("food", ent), ("verb", GetEdibleVerb(ent.Owner))), args.User, args.User);
    }

    private void OnItemSlotsEdible(Entity<ItemSlotsComponent> ent, ref EdibleEvent args)
    {
        if (args.Cancelled)
            return;

        if (!ent.Comp.Slots.Any(slot => slot.Value.HasItem))
            return;

        args.Cancelled = true;

        _popup.PopupEntity(Loc.GetString("edible-has-used-storage", ("food", ent), ("verb", GetEdibleVerb(ent.Owner))), args.User, args.User);
    }

    private void OnOpenableEdible(Entity<OpenableComponent> ent, ref EdibleEvent args)
    {
        if (args.Cancelled)
            return;

        if (_openable.IsClosed(ent, args.User, ent.Comp, predicted: true))
            args.Cancelled = true;
    }

    private void OnEdibleIsDigestible(Entity<EdibleComponent> ent, ref IsDigestibleEvent args)
    {
        if (ent.Comp.RequireDead && _mobState.IsAlive(ent))
            return;

        args.AddDigestible(ent.Comp.RequiresSpecialDigestion);
    }

    /// <remarks>
    /// Both of these assume that having this component means there's nothing stopping you from slurping up
    /// pure reagent juice with absolutely nothing to stop you.
    /// </remarks>
    private void OnDrainableIsDigestible(Entity<DrainableSolutionComponent> ent, ref IsDigestibleEvent args)
    {
        args.UniversalDigestion();
    }

    private void OnPuddleIsDigestible(Entity<PuddleComponent> ent, ref IsDigestibleEvent args)
    {
        args.UniversalDigestion();
    }

    /// <remarks>
    /// I mean you have to eat the *whole* pill no?
    /// </remarks>
    private void OnPillBeforeEaten(Entity<PillComponent> ent, ref BeforeIngestedEvent args)
    {
        if (args.Cancelled || args.Solution is not { } sol)
            return;

        if (args.TryNewMinimum(sol.Volume))
            return;

        args.Cancelled = true;
    }
}