Convert StomachBehavior to a component/system + rejig body namespaces (#5249)

* Convert StomachBehavior to a component/system + rejig body namespaces * test * slightly more namespace changes * remove * Hello????? * fuck you github test runner * reviews * oobsy!
2021-11-11 16:10:57 -07:00
parent 509e1ba6e7
commit 457e8c64ee
77 changed files with 326 additions and 307 deletions
--- a/Content.Server/Body/Systems/StomachSystem.cs
+++ b/Content.Server/Body/Systems/StomachSystem.cs
@@ -0,0 +1,121 @@
+using Content.Server.Body.Components;
+using Content.Server.Chemistry.Components.SolutionManager;
+using Content.Server.Chemistry.EntitySystems;
+using Content.Shared.Chemistry.Components;
+using Robust.Shared.GameObjects;
+using Robust.Shared.IoC;
+using Robust.Shared.Utility;
+
+namespace Content.Server.Body.Systems
+{
+    public class StomachSystem : EntitySystem
+    {
+        [Dependency] private readonly SolutionContainerSystem _solutionContainerSystem = default!;
+
+        public const string DefaultSolutionName = "stomach";
+
+        public override void Initialize()
+        {
+            SubscribeLocalEvent<StomachComponent, ComponentInit>(OnComponentInit);
+        }
+
+        public override void Update(float frameTime)
+        {
+            foreach (var (stomach, mech, sol)
+                in EntityManager.EntityQuery<StomachComponent, MechanismComponent, SolutionContainerManagerComponent>(false))
+            {
+                if (mech.Body == null)
+                    continue;
+
+                stomach.AccumulatedFrameTime += frameTime;
+
+                if (stomach.AccumulatedFrameTime < stomach.UpdateInterval)
+                    continue;
+
+                stomach.AccumulatedFrameTime -= stomach.UpdateInterval;
+
+                // Get our solutions
+                if (!_solutionContainerSystem.TryGetSolution(stomach.OwnerUid, DefaultSolutionName,
+                    out var stomachSolution, sol))
+                    continue;
+
+                if (!_solutionContainerSystem.TryGetSolution(mech.Body.OwnerUid, stomach.BodySolutionName,
+                    out var bodySolution))
+                    continue;
+
+                var transferSolution = new Solution();
+
+                var queue = new RemQueue<StomachComponent.ReagentDelta>();
+                foreach (var delta in stomach.ReagentDeltas)
+                {
+                    delta.Increment(stomach.UpdateInterval);
+                    if (delta.Lifetime > stomach.DigestionDelay)
+                    {
+                        if (stomachSolution.ContainsReagent(delta.ReagentId, out var quant))
+                        {
+                            if (quant > delta.Quantity)
+                                quant = delta.Quantity;
+
+                            _solutionContainerSystem.TryRemoveReagent(stomach.OwnerUid, stomachSolution,
+                                delta.ReagentId, quant);
+                            transferSolution.AddReagent(delta.ReagentId, quant);
+                        }
+
+                        queue.Add(delta);
+                    }
+                }
+
+                foreach (var item in queue)
+                {
+                    stomach.ReagentDeltas.Remove(item);
+                }
+
+                // Transfer everything to the body solution!
+                _solutionContainerSystem.TryAddSolution(mech.Body.OwnerUid, bodySolution, transferSolution);
+            }
+        }
+
+        private void OnComponentInit(EntityUid uid, StomachComponent component, ComponentInit args)
+        {
+            var solution = _solutionContainerSystem.EnsureSolution(uid, DefaultSolutionName);
+            solution.MaxVolume = component.InitialMaxVolume;
+        }
+
+        public bool CanTransferSolution(EntityUid uid, Solution solution,
+            SolutionContainerManagerComponent? solutions=null)
+        {
+            if (!Resolve(uid, ref solutions, false))
+                return false;
+
+            if (!_solutionContainerSystem.TryGetSolution(uid, DefaultSolutionName, out var stomachSolution, solutions))
+                return false;
+
+            // TODO: For now no partial transfers. Potentially change by design
+            if (!stomachSolution.CanAddSolution(solution))
+                return false;
+
+            return true;
+        }
+
+        public bool TryTransferSolution(EntityUid uid, Solution solution,
+            StomachComponent? stomach=null,
+            SolutionContainerManagerComponent? solutions=null)
+        {
+            if (!Resolve(uid, ref stomach, ref solutions, false))
+                return false;
+
+            if (!_solutionContainerSystem.TryGetSolution(uid, DefaultSolutionName, out var stomachSolution, solutions)
+                || !CanTransferSolution(uid, solution, solutions))
+                return false;
+
+            _solutionContainerSystem.TryAddSolution(uid, stomachSolution, solution);
+            // Add each reagent to ReagentDeltas. Used to track how long each reagent has been in the stomach
+            foreach (var reagent in solution.Contents)
+            {
+                stomach.ReagentDeltas.Add(new StomachComponent.ReagentDelta(reagent.ReagentId, reagent.Quantity));
+            }
+
+            return true;
+        }
+    }
+}