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/BodySystem.cs
+++ b/Content.Server/Body/Systems/BodySystem.cs
@@ -0,0 +1,78 @@
+using System.Collections.Generic;
+using System.Diagnostics.CodeAnalysis;
+using System.Linq;
+using Content.Server.Body.Components;
+using Content.Server.GameTicking;
+using Content.Server.Mind.Components;
+using Content.Shared.Body.Components;
+using Content.Shared.MobState.Components;
+using Content.Shared.Movement.EntitySystems;
+using Robust.Shared.GameObjects;
+using Robust.Shared.IoC;
+using Robust.Shared.Timing;
+
+namespace Content.Server.Body.Systems
+{
+    public sealed class BodySystem : EntitySystem
+    {
+        [Dependency] private readonly GameTicker _ticker = default!;
+        [Dependency] private readonly IGameTiming _gameTiming = default!;
+
+        public override void Initialize()
+        {
+            base.Initialize();
+            SubscribeLocalEvent<BodyComponent, RelayMoveInputEvent>(OnRelayMoveInput);
+        }
+
+        private void OnRelayMoveInput(EntityUid uid, BodyComponent component, RelayMoveInputEvent args)
+        {
+            if (EntityManager.TryGetComponent<MobStateComponent>(uid, out var mobState) &&
+                mobState.IsDead() &&
+                EntityManager.TryGetComponent<MindComponent>(uid, out var mind) &&
+                mind.HasMind)
+            {
+                if (!mind.Mind!.TimeOfDeath.HasValue)
+                {
+                    mind.Mind.TimeOfDeath = _gameTiming.RealTime;
+                }
+
+                _ticker.OnGhostAttempt(mind.Mind!, true);
+            }
+        }
+
+        public IEnumerable<T> GetComponentsOnMechanisms<T>(EntityUid uid,
+            SharedBodyComponent? body) where T : Component
+        {
+            if (!Resolve(uid, ref body))
+                yield break;
+
+            foreach (var (part, _) in body.Parts)
+            foreach (var mechanism in part.Mechanisms)
+            {
+                if (EntityManager.TryGetComponent<T>(mechanism.OwnerUid, out var comp))
+                    yield return comp;
+            }
+        }
+
+        public bool TryGetComponentsOnMechanisms<T>(EntityUid uid,
+            [NotNullWhen(true)] out IEnumerable<T>? comps,
+            SharedBodyComponent? body) where T: Component
+        {
+            if (!Resolve(uid, ref body))
+            {
+                comps = null;
+                return false;
+            }
+
+            comps = GetComponentsOnMechanisms<T>(uid, body).ToArray();
+
+            if (!comps.Any())
+            {
+                comps = null;
+                return false;
+            }
+
+            return true;
+        }
+    }
+}