ECS AME (#16779)

Content.Server/Ame/EntitySystems/AmeShieldingSystem.cs

@@ -0,0 +1,41 @@
+using Content.Server.Ame.Components;
+using Content.Shared.Ame;
+using Robust.Server.GameObjects;
+
+namespace Content.Server.Ame.EntitySystems;
+
+public sealed class AmeShieldingSystem : EntitySystem
+{
+    [Dependency] private readonly AppearanceSystem _appearanceSystem = default!;
+    [Dependency] private readonly PointLightSystem _pointLightSystem = default!;
+
+    public void SetCore(EntityUid uid, bool value, AmeShieldComponent? shield = null)
+    {
+        if (!Resolve(uid, ref shield))
+            return;
+        if (value == shield.IsCore)
+            return;
+
+        shield.IsCore = value;
+        _appearanceSystem.SetData(uid, AmeShieldVisuals.Core, value);
+        if (!value)
+            UpdateCoreVisuals(uid, 0, false, shield);
+    }
+
+    public void UpdateCoreVisuals(EntityUid uid, int injectionStrength, bool injecting, AmeShieldComponent? shield = null)
+    {
+        if (!Resolve(uid, ref shield))
+            return;
+
+        if (!injecting)
+        {
+            _appearanceSystem.SetData(uid, AmeShieldVisuals.CoreState, AmeCoreState.Off);
+            _pointLightSystem.SetEnabled(uid, false);
+            return;
+        }
+
+        _pointLightSystem.SetRadius(uid, Math.Clamp(injectionStrength, 1, 12));
+        _pointLightSystem.SetEnabled(uid, true);
+        _appearanceSystem.SetData(uid, AmeShieldVisuals.CoreState, injectionStrength > 2 ? AmeCoreState.Strong : AmeCoreState.Weak);
+    }
+}