Re-organize all projects (#4166)

2021-06-09 22:19:39 +02:00
parent 9f50e4061b
commit ff1a2d97ea
1773 changed files with 5258 additions and 5508 deletions
--- a/Content.Server/Gravity/EntitySystems/WeightlessSystem.cs
+++ b/Content.Server/Gravity/EntitySystems/WeightlessSystem.cs
@@ -0,0 +1,130 @@
+using System.Collections.Generic;
+using Content.Server.Alert;
+using Content.Shared.Alert;
+using Content.Shared.GameTicking;
+using Content.Shared.Gravity;
+using JetBrains.Annotations;
+using Robust.Shared.GameObjects;
+using Robust.Shared.IoC;
+using Robust.Shared.Map;
+using Robust.Shared.Utility;
+
+namespace Content.Server.Gravity.EntitySystems
+{
+    [UsedImplicitly]
+    public class WeightlessSystem : EntitySystem, IResettingEntitySystem
+    {
+        [Dependency] private readonly IMapManager _mapManager = default!;
+
+        private readonly Dictionary<GridId, List<ServerAlertsComponent>> _alerts = new();
+
+        public override void Initialize()
+        {
+            base.Initialize();
+
+            SubscribeLocalEvent<GravityChangedMessage>(GravityChanged);
+            SubscribeLocalEvent<EntParentChangedMessage>(EntParentChanged);
+        }
+
+        public override void Shutdown()
+        {
+            base.Shutdown();
+
+            UnsubscribeLocalEvent<GravityChangedMessage>();
+            UnsubscribeLocalEvent<EntParentChangedMessage>();
+        }
+
+        public void Reset()
+        {
+            _alerts.Clear();
+        }
+
+        public void AddAlert(ServerAlertsComponent status)
+        {
+            var gridId = status.Owner.Transform.GridID;
+            var alerts = _alerts.GetOrNew(gridId);
+
+            alerts.Add(status);
+
+            if (_mapManager.TryGetGrid(status.Owner.Transform.GridID, out var grid))
+            {
+                if (grid.HasGravity)
+                {
+                    RemoveWeightless(status);
+                }
+                else
+                {
+                    AddWeightless(status);
+                }
+            }
+        }
+
+        public void RemoveAlert(ServerAlertsComponent status)
+        {
+            var grid = status.Owner.Transform.GridID;
+            if (!_alerts.TryGetValue(grid, out var statuses))
+            {
+                return;
+            }
+
+            statuses.Remove(status);
+        }
+
+        private void GravityChanged(GravityChangedMessage ev)
+        {
+            if (!_alerts.TryGetValue(ev.Grid.Index, out var statuses))
+            {
+                return;
+            }
+
+            if (ev.HasGravity)
+            {
+                foreach (var status in statuses)
+                {
+                    RemoveWeightless(status);
+                }
+            }
+            else
+            {
+                foreach (var status in statuses)
+                {
+                    AddWeightless(status);
+                }
+            }
+        }
+
+        private void AddWeightless(ServerAlertsComponent status)
+        {
+            status.ShowAlert(AlertType.Weightless);
+        }
+
+        private void RemoveWeightless(ServerAlertsComponent status)
+        {
+            status.ClearAlert(AlertType.Weightless);
+        }
+
+        private void EntParentChanged(EntParentChangedMessage ev)
+        {
+            if (!ev.Entity.TryGetComponent(out ServerAlertsComponent? status))
+            {
+                return;
+            }
+
+            if (ev.OldParent != null &&
+                ev.OldParent.TryGetComponent(out IMapGridComponent? mapGrid))
+            {
+                var oldGrid = mapGrid.GridIndex;
+
+                if (_alerts.TryGetValue(oldGrid, out var oldStatuses))
+                {
+                    oldStatuses.Remove(status);
+                }
+            }
+
+            var newGrid = ev.Entity.Transform.GridID;
+            var newStatuses = _alerts.GetOrNew(newGrid);
+
+            newStatuses.Add(status);
+        }
+    }
+}