tbd-station-14/Content.Server/Chemistry/Components/SolutionTransferComponent.cs

using System;
using System.Threading.Tasks;
using Content.Server.UserInterface;
using Content.Shared.ActionBlocker;
using Content.Shared.Chemistry;
using Content.Shared.Chemistry.Components;
using Content.Shared.Chemistry.Components.SolutionManager;
using Content.Shared.Chemistry.EntitySystems;
using Content.Shared.Chemistry.Reagent;
using Content.Shared.Interaction;
using Content.Shared.Interaction.Helpers;
using Content.Shared.Notification.Managers;
using Content.Shared.Verbs;
using Microsoft.EntityFrameworkCore.Metadata.Builders;
using Robust.Server.GameObjects;
using Robust.Shared.GameObjects;
using Robust.Shared.Localization;
using Robust.Shared.Serialization.Manager.Attributes;
using Robust.Shared.ViewVariables;

namespace Content.Server.Chemistry.Components
{
    /// <summary>
    ///     Gives click behavior for transferring to/from other reagent containers.
    /// </summary>
    [RegisterComponent]
    public sealed class SolutionTransferComponent : Component, IAfterInteract
    {
        // Behavior is as such:
        // If it's a reagent tank, TAKE reagent.
        // If it's anything else, GIVE reagent.
        // Of course, only if possible.

        public override string Name => "SolutionTransfer";

        /// <summary>
        ///     The amount of solution to be transferred from this solution when clicking on other solutions with it.
        /// </summary>
        [DataField("transferAmount")]
        [ViewVariables(VVAccess.ReadWrite)]
        public ReagentUnit TransferAmount { get; set; } = ReagentUnit.New(5);

        /// <summary>
        ///     The minimum amount of solution that can be transferred at once from this solution.
        /// </summary>
        [DataField("minTransferAmount")]
        [ViewVariables(VVAccess.ReadWrite)]
        public ReagentUnit MinimumTransferAmount { get; set; } = ReagentUnit.New(5);

        /// <summary>
        ///     The maximum amount of solution that can be transferred at once from this solution.
        /// </summary>
        [DataField("maxTransferAmount")]
        [ViewVariables(VVAccess.ReadWrite)]
        public ReagentUnit MaximumTransferAmount { get; set; } = ReagentUnit.New(50);

        /// <summary>
        ///     Subjectively, which transfer amount would be best for most activities given the maximum
        ///     transfer amount.
        /// </summary>
        public ReagentUnit SubjectiveBestTransferAmount() =>
            MaximumTransferAmount.Int() switch
            {
                <= 5 => ReagentUnit.New(1),
                (> 5) and (<= 25) => ReagentUnit.New(5),
                (> 25) and (<= 50) => ReagentUnit.New(10),
                (> 50) and (<= 100) => ReagentUnit.New(20),
                (> 100) and (<= 500) => ReagentUnit.New(50),
                (> 500) => ReagentUnit.New(100)
            };

        /// <summary>
        ///     Can this entity take reagent from reagent tanks?
        /// </summary>
        [DataField("canReceive")]
        [ViewVariables(VVAccess.ReadWrite)]
        public bool CanReceive { get; set; } = true;

        /// <summary>
        ///     Can this entity give reagent to other reagent containers?
        /// </summary>
        [DataField("canSend")]
        [ViewVariables(VVAccess.ReadWrite)]
        public bool CanSend { get; set; } = true;

        /// <summary>
        /// Whether you're allowed to change the transfer amount.
        /// </summary>
        [DataField("canChangeTransferAmount")]
        [ViewVariables(VVAccess.ReadWrite)]
        public bool CanChangeTransferAmount { get; set; } = false;

        [ViewVariables] private BoundUserInterface? UserInterface => Owner.GetUIOrNull(TransferAmountUiKey.Key);

        protected override void Initialize()
        {
            base.Initialize();

            if (UserInterface != null)
            {
                UserInterface.OnReceiveMessage += UserInterfaceOnReceiveMessage;
            }
        }

        public void UserInterfaceOnReceiveMessage(ServerBoundUserInterfaceMessage serverMsg)
        {
            switch (serverMsg.Message)
            {
                case TransferAmountSetValueMessage svm:
                    var sval = svm.Value.Float();
                    var amount = Math.Clamp(sval, MinimumTransferAmount.Float(),
                        MaximumTransferAmount.Float());

                    serverMsg.Session.AttachedEntity?.PopupMessage(Loc.GetString("comp-solution-transfer-set-amount",
                        ("amount", amount)));
                    SetTransferAmount(ReagentUnit.New(amount));
                    break;
            }
        }

        public void SetTransferAmount(ReagentUnit amount)
        {
            amount = ReagentUnit.New(Math.Clamp(amount.Int(), MinimumTransferAmount.Int(),
                MaximumTransferAmount.Int()));
            TransferAmount = amount;
        }

        async Task<bool> IAfterInteract.AfterInteract(AfterInteractEventArgs eventArgs)
        {
            var solutionsSys = EntitySystem.Get<SolutionContainerSystem>();

            if (!eventArgs.InRangeUnobstructed() || eventArgs.Target == null)
                return false;

            if (!Owner.HasComponent<SolutionContainerManagerComponent>())
                return false;

            var target = eventArgs.Target!;
            if (!target.HasComponent<SolutionContainerManagerComponent>())
            {
                return false;
            }


            if (CanReceive && target.TryGetComponent(out ReagentTankComponent? tank)
                           && solutionsSys.TryGetRefillableSolution(Owner.Uid, out var ownerRefill)
                           && solutionsSys.TryGetDrainableSolution(eventArgs.Target.Uid, out var targetDrain))
            {
                var transferred = DoTransfer(eventArgs.User, eventArgs.Target, targetDrain, Owner, ownerRefill, tank.TransferAmount);
                if (transferred > 0)
                {
                    var toTheBrim = ownerRefill.AvailableVolume == 0;
                    var msg = toTheBrim
                        ? "comp-solution-transfer-fill-fully"
                        : "comp-solution-transfer-fill-normal";

                    target.PopupMessage(eventArgs.User,
                        Loc.GetString(msg, ("owner", eventArgs.Target), ("amount", transferred), ("target", Owner)));
                    return true;
                }
            }

            if (CanSend && solutionsSys.TryGetRefillableSolution(eventArgs.Target.Uid, out var targetRefill)
                        && solutionsSys.TryGetDrainableSolution(Owner.Uid, out var ownerDrain))
            {
                var transferred = DoTransfer(eventArgs.User, Owner, ownerDrain, target, targetRefill, TransferAmount);

                if (transferred > 0)
                {
                    Owner.PopupMessage(eventArgs.User,
                        Loc.GetString("comp-solution-transfer-transfer-solution",
                            ("amount", transferred),
                            ("target", target)));

                    return true;
                }
            }

            return true;
        }

        /// <returns>The actual amount transferred.</returns>
        private static ReagentUnit DoTransfer(IEntity user,
            IEntity sourceEntity,
            Solution source,
            IEntity targetEntity,
            Solution target,
            ReagentUnit amount)
        {

            if (source.DrainAvailable == 0)
            {
                sourceEntity.PopupMessage(user,
                    Loc.GetString("comp-solution-transfer-is-empty", ("target", sourceEntity)));
                return ReagentUnit.Zero;
            }

            if (target.AvailableVolume == 0)
            {
                targetEntity.PopupMessage(user,
                    Loc.GetString("comp-solution-transfer-is-full", ("target", targetEntity)));
                return ReagentUnit.Zero;
            }

            var actualAmount =
                ReagentUnit.Min(amount, ReagentUnit.Min(source.DrainAvailable, target.AvailableVolume));

            var solution = EntitySystem.Get<SolutionContainerSystem>().Drain(sourceEntity.Uid, source, actualAmount);
            EntitySystem.Get<SolutionContainerSystem>().Refill(targetEntity.Uid, target, solution);

            return actualAmount;
        }

        // TODO refactor when dynamic verbs are a thing

        [Verb]
        public sealed class MinimumTransferVerb : Verb<SolutionTransferComponent>
        {
            protected override void GetData(IEntity user, SolutionTransferComponent component, VerbData data)
            {
                if (!EntitySystem.Get<ActionBlockerSystem>().CanInteract(user) || !component.CanChangeTransferAmount)
                {
                    data.Visibility = VerbVisibility.Invisible;
                    return;
                }

                data.Visibility = VerbVisibility.Visible;
                data.Text = Loc.GetString("comp-solution-transfer-verb-transfer-amount-min",
                    ("amount", component.MinimumTransferAmount.Int()));
                data.CategoryData = VerbCategories.SetTransferAmount;
            }

            protected override void Activate(IEntity user, SolutionTransferComponent component)
            {
                component.TransferAmount = component.MinimumTransferAmount;
                user.PopupMessage(Loc.GetString("comp-solution-transfer-set-amount",
                    ("amount", component.TransferAmount.Int())));
            }
        }

        [Verb]
        public sealed class DefaultTransferVerb : Verb<SolutionTransferComponent>
        {
            protected override void GetData(IEntity user, SolutionTransferComponent component, VerbData data)
            {
                if (!EntitySystem.Get<ActionBlockerSystem>().CanInteract(user) || !component.CanChangeTransferAmount)
                {
                    data.Visibility = VerbVisibility.Invisible;
                    return;
                }

                var amt = component.SubjectiveBestTransferAmount();
                if (amt > component.MinimumTransferAmount && amt < component.MaximumTransferAmount)
                {
                    data.Visibility = VerbVisibility.Visible;
                    data.Text = Loc.GetString("comp-solution-transfer-verb-transfer-amount-ideal",
                        ("amount", amt.Int()));
                    data.CategoryData = VerbCategories.SetTransferAmount;
                }
                else
                {
                    data.Visibility = VerbVisibility.Invisible;
                }
            }

            protected override void Activate(IEntity user, SolutionTransferComponent component)
            {
                component.TransferAmount = component.SubjectiveBestTransferAmount();
                user.PopupMessage(Loc.GetString("comp-solution-transfer-set-amount",
                    ("amount", component.TransferAmount.Int())));
            }
        }

        [Verb]
        public sealed class MaximumTransferVerb : Verb<SolutionTransferComponent>
        {
            protected override void GetData(IEntity user, SolutionTransferComponent component, VerbData data)
            {
                if (!EntitySystem.Get<ActionBlockerSystem>().CanInteract(user) || !component.CanChangeTransferAmount)
                {
                    data.Visibility = VerbVisibility.Invisible;
                    return;
                }

                data.Visibility = VerbVisibility.Visible;
                data.Text = Loc.GetString("comp-solution-transfer-verb-transfer-amount-max",
                    ("amount", component.MaximumTransferAmount));
                data.CategoryData = VerbCategories.SetTransferAmount;
            }

            protected override void Activate(IEntity user, SolutionTransferComponent component)
            {
                component.TransferAmount = component.MaximumTransferAmount;
                user.PopupMessage(Loc.GetString("comp-solution-transfer-set-amount",
                    ("amount", component.TransferAmount.Int())));
            }
        }

        [Verb]
        public sealed class CustomTransferVerb : Verb<SolutionTransferComponent>
        {
            public override bool AlternativeInteraction => true;

            protected override void GetData(IEntity user, SolutionTransferComponent component, VerbData data)
            {
                if (!EntitySystem.Get<ActionBlockerSystem>().CanInteract(user) || !component.CanChangeTransferAmount)
                {
                    data.Visibility = VerbVisibility.Invisible;
                    return;
                }

                data.Visibility = VerbVisibility.Visible;
                data.Text = Loc.GetString("comp-solution-transfer-verb-transfer-amount-custom");
                data.CategoryData = VerbCategories.SetTransferAmount;
            }

            protected override void Activate(IEntity user, SolutionTransferComponent component)
            {
                if (!user.TryGetComponent<ActorComponent>(out var actor))
                {
                    return;
                }

                component.UserInterface?.Open(actor.PlayerSession);
            }
        }
    }
}