Puddles & spreader refactor (#15191)

2023-04-10 15:37:03 +10:00
parent 3178ab83f6
commit 317a4013eb
141 changed files with 3046 additions and 3201 deletions
--- a/Content.Server/Fluids/EntitySystems/PuddleSystem.Transfers.cs
+++ b/Content.Server/Fluids/EntitySystems/PuddleSystem.Transfers.cs
@@ -0,0 +1,69 @@
+using Content.Shared.Chemistry.Components;
+using Content.Shared.DragDrop;
+using Content.Shared.FixedPoint;
+using Content.Shared.Fluids;
+using Content.Shared.Fluids.Components;
+
+namespace Content.Server.Fluids.EntitySystems;
+
+public sealed partial class PuddleSystem
+{
+    private void InitializeTransfers()
+    {
+        SubscribeLocalEvent<RefillableSolutionComponent, DragDropDraggedEvent>(OnRefillableDragged);
+    }
+
+    private void OnRefillableDragged(EntityUid uid, RefillableSolutionComponent component, ref DragDropDraggedEvent args)
+    {
+        _solutionContainerSystem.TryGetSolution(uid, component.Solution, out var solution);
+
+        if (solution?.Volume == FixedPoint2.Zero)
+        {
+            _popups.PopupEntity(Loc.GetString("mopping-system-empty", ("used", uid)), uid, args.User);
+            return;
+        }
+
+        TryComp<DrainableSolutionComponent>(args.Target, out var drainable);
+
+        _solutionContainerSystem.TryGetDrainableSolution(args.Target, out var drainableSolution, drainable);
+
+        // Dump reagents into drain
+        if (TryComp<DrainComponent>(args.Target, out var drain) && drainable != null)
+        {
+            if (drainableSolution == null || solution == null)
+                return;
+
+            var split = _solutionContainerSystem.SplitSolution(uid, solution, drainableSolution.AvailableVolume);
+
+            // TODO: Drane refactor
+            if (_solutionContainerSystem.TryAddSolution(args.Target, drainableSolution, split))
+            {
+                _audio.PlayPvs(AbsorbentComponent.DefaultTransferSound, args.Target);
+            }
+            else
+            {
+                _popups.PopupEntity(Loc.GetString("mopping-system-full", ("used", args.Target)), args.Target, args.User);
+            }
+
+            return;
+        }
+
+        // Take reagents from target
+        if (drainable != null)
+        {
+            if (drainableSolution == null || solution == null)
+                return;
+
+            var split = _solutionContainerSystem.SplitSolution(args.Target, drainableSolution, solution.AvailableVolume);
+
+            if (_solutionContainerSystem.TryAddSolution(uid, solution, split))
+            {
+                _audio.PlayPvs(AbsorbentComponent.DefaultTransferSound, uid);
+            }
+            else
+            {
+                _popups.PopupEntity(Loc.GetString("mopping-system-full", ("used", uid)), uid, args.User);
+            }
+        }
+    }
+}