using Content.Server.Administration.Logs; using Content.Server.Chemistry.Containers.EntitySystems; using Content.Shared.Chemistry; using Content.Shared.Chemistry.Components; using Content.Shared.Database; using Content.Shared.FixedPoint; using Content.Shared.Interaction; using Content.Shared.Popups; using Content.Shared.Verbs; using JetBrains.Annotations; using Robust.Server.GameObjects; using Robust.Shared.Player; namespace Content.Server.Chemistry.EntitySystems { [UsedImplicitly] public sealed class SolutionTransferSystem : EntitySystem { [Dependency] private readonly SharedPopupSystem _popupSystem = default!; [Dependency] private readonly SolutionContainerSystem _solutionContainerSystem = default!; [Dependency] private readonly UserInterfaceSystem _userInterfaceSystem = default!; [Dependency] private readonly IAdminLogManager _adminLogger = default!; /// /// Default transfer amounts for the set-transfer verb. /// public static readonly List DefaultTransferAmounts = new() { 1, 5, 10, 25, 50, 100, 250, 500, 1000 }; public override void Initialize() { base.Initialize(); SubscribeLocalEvent>(AddSetTransferVerbs); SubscribeLocalEvent(OnAfterInteract); SubscribeLocalEvent(OnTransferAmountSetValueMessage); } private void OnTransferAmountSetValueMessage(Entity entity, ref TransferAmountSetValueMessage message) { var newTransferAmount = FixedPoint2.Clamp(message.Value, entity.Comp.MinimumTransferAmount, entity.Comp.MaximumTransferAmount); entity.Comp.TransferAmount = newTransferAmount; if (message.Session.AttachedEntity is { Valid: true } user) _popupSystem.PopupEntity(Loc.GetString("comp-solution-transfer-set-amount", ("amount", newTransferAmount)), entity.Owner, user); } private void AddSetTransferVerbs(Entity entity, ref GetVerbsEvent args) { var (uid, component) = entity; if (!args.CanAccess || !args.CanInteract || !component.CanChangeTransferAmount || args.Hands == null) return; if (!EntityManager.TryGetComponent(args.User, out ActorComponent? actor)) return; // Custom transfer verb AlternativeVerb custom = new(); custom.Text = Loc.GetString("comp-solution-transfer-verb-custom-amount"); custom.Category = VerbCategory.SetTransferAmount; custom.Act = () => _userInterfaceSystem.TryOpen(uid, TransferAmountUiKey.Key, actor.PlayerSession); custom.Priority = 1; args.Verbs.Add(custom); // Add specific transfer verbs according to the container's size var priority = 0; var user = args.User; foreach (var amount in DefaultTransferAmounts) { if (amount < component.MinimumTransferAmount.Int() || amount > component.MaximumTransferAmount.Int()) continue; AlternativeVerb verb = new(); verb.Text = Loc.GetString("comp-solution-transfer-verb-amount", ("amount", amount)); verb.Category = VerbCategory.SetTransferAmount; verb.Act = () => { component.TransferAmount = FixedPoint2.New(amount); _popupSystem.PopupEntity(Loc.GetString("comp-solution-transfer-set-amount", ("amount", amount)), uid, user); }; // we want to sort by size, not alphabetically by the verb text. verb.Priority = priority; priority--; args.Verbs.Add(verb); } } private void OnAfterInteract(Entity entity, ref AfterInteractEvent args) { if (!args.CanReach || args.Target == null) return; var target = args.Target!.Value; var (uid, component) = entity; //Special case for reagent tanks, because normally clicking another container will give solution, not take it. if (component.CanReceive && !EntityManager.HasComponent(target) // target must not be refillable (e.g. Reagent Tanks) && _solutionContainerSystem.TryGetDrainableSolution(target, out var targetSoln, out _) // target must be drainable && EntityManager.TryGetComponent(uid, out RefillableSolutionComponent? refillComp) && _solutionContainerSystem.TryGetRefillableSolution((uid, refillComp, null), out var ownerSoln, out var ownerRefill)) { var transferAmount = component.TransferAmount; // This is the player-configurable transfer amount of "uid," not the target reagent tank. if (EntityManager.TryGetComponent(uid, out RefillableSolutionComponent? refill) && refill.MaxRefill != null) // uid is the entity receiving solution from target. { transferAmount = FixedPoint2.Min(transferAmount, (FixedPoint2) refill.MaxRefill); // if the receiver has a smaller transfer limit, use that instead } var transferred = Transfer(args.User, target, targetSoln.Value, uid, ownerSoln.Value, transferAmount); if (transferred > 0) { var toTheBrim = ownerRefill.AvailableVolume == 0; var msg = toTheBrim ? "comp-solution-transfer-fill-fully" : "comp-solution-transfer-fill-normal"; _popupSystem.PopupEntity(Loc.GetString(msg, ("owner", args.Target), ("amount", transferred), ("target", uid)), uid, args.User); args.Handled = true; return; } } // if target is refillable, and owner is drainable if (component.CanSend && _solutionContainerSystem.TryGetRefillableSolution(target, out targetSoln, out var targetRefill) && _solutionContainerSystem.TryGetDrainableSolution(uid, out ownerSoln, out var ownerDrain)) { var transferAmount = component.TransferAmount; if (EntityManager.TryGetComponent(target, out RefillableSolutionComponent? refill) && refill.MaxRefill != null) { transferAmount = FixedPoint2.Min(transferAmount, (FixedPoint2) refill.MaxRefill); } var transferred = Transfer(args.User, uid, ownerSoln.Value, target, targetSoln.Value, transferAmount); if (transferred > 0) { var message = Loc.GetString("comp-solution-transfer-transfer-solution", ("amount", transferred), ("target", target)); _popupSystem.PopupEntity(message, uid, args.User); args.Handled = true; } } } /// /// Transfer from a solution to another. /// /// The actual amount transferred. public FixedPoint2 Transfer(EntityUid user, EntityUid sourceEntity, Entity source, EntityUid targetEntity, Entity target, FixedPoint2 amount) { var transferAttempt = new SolutionTransferAttemptEvent(sourceEntity, targetEntity); // Check if the source is cancelling the transfer RaiseLocalEvent(sourceEntity, transferAttempt, broadcast: true); if (transferAttempt.Cancelled) { _popupSystem.PopupEntity(transferAttempt.CancelReason!, sourceEntity, user); return FixedPoint2.Zero; } var sourceSolution = source.Comp.Solution; if (sourceSolution.Volume == 0) { _popupSystem.PopupEntity(Loc.GetString("comp-solution-transfer-is-empty", ("target", sourceEntity)), sourceEntity, user); return FixedPoint2.Zero; } // Check if the target is cancelling the transfer RaiseLocalEvent(targetEntity, transferAttempt, broadcast: true); if (transferAttempt.Cancelled) { _popupSystem.PopupEntity(transferAttempt.CancelReason!, sourceEntity, user); return FixedPoint2.Zero; } var targetSolution = target.Comp.Solution; if (targetSolution.AvailableVolume == 0) { _popupSystem.PopupEntity(Loc.GetString("comp-solution-transfer-is-full", ("target", targetEntity)), targetEntity, user); return FixedPoint2.Zero; } var actualAmount = FixedPoint2.Min(amount, FixedPoint2.Min(sourceSolution.Volume, targetSolution.AvailableVolume)); var solution = _solutionContainerSystem.Drain(sourceEntity, source, actualAmount); _solutionContainerSystem.Refill(targetEntity, target, solution); _adminLogger.Add(LogType.Action, LogImpact.Medium, $"{EntityManager.ToPrettyString(user):player} transferred {string.Join(", ", solution.Contents)} to {EntityManager.ToPrettyString(targetEntity):entity}, which now contains {SolutionContainerSystem.ToPrettyString(targetSolution)}"); return actualAmount; } } /// /// Raised when attempting to transfer from one solution to another. /// public sealed class SolutionTransferAttemptEvent : CancellableEntityEventArgs { public SolutionTransferAttemptEvent(EntityUid from, EntityUid to) { From = from; To = to; } public EntityUid From { get; } public EntityUid To { get; } /// /// Why the transfer has been cancelled. /// public string? CancelReason { get; private set; } /// /// Cancels the transfer. /// public void Cancel(string reason) { base.Cancel(); CancelReason = reason; } } }