Add radio jammer (#14369)

2023-04-14 22:50:19 +03:00
parent 779100484f
commit 7886d27dda
12 changed files with 214 additions and 42 deletions
--- a/Content.Server/Radio/EntitySystems/JammerSystem.cs
+++ b/Content.Server/Radio/EntitySystems/JammerSystem.cs
@@ -0,0 +1,91 @@
+using Content.Server.Popups;
+using Content.Server.PowerCell;
+using Content.Server.Radio.Components;
+using Content.Shared.Examine;
+using Content.Shared.Interaction;
+using Content.Shared.PowerCell.Components;
+
+namespace Content.Server.Radio.EntitySystems;
+
+public sealed class JammerSystem : EntitySystem
+{
+    [Dependency] private readonly PowerCellSystem _powerCell = default!;
+    [Dependency] private readonly PopupSystem _popup = default!;
+    [Dependency] private readonly SharedTransformSystem _transform = default!;
+
+    public override void Initialize()
+    {
+        base.Initialize();
+
+        SubscribeLocalEvent<RadioJammerComponent, ActivateInWorldEvent>(OnActivate);
+        SubscribeLocalEvent<ActiveRadioJammerComponent, PowerCellChangedEvent>(OnPowerCellChanged);
+        SubscribeLocalEvent<RadioJammerComponent, ExaminedEvent>(OnExamine);
+        SubscribeLocalEvent<RadioSendAttemptEvent>(OnRadioSendAttempt);
+    }
+
+    public override void Update(float frameTime)
+    {
+        var query = AllEntityQuery<ActiveRadioJammerComponent, RadioJammerComponent>();
+        while (query.MoveNext(out var uid, out var _, out var jam))
+        {
+            if (_powerCell.TryGetBatteryFromSlot(uid, out var battery) &&
+                !battery.TryUseCharge(jam.Wattage * frameTime))
+            {
+                RemComp<ActiveRadioJammerComponent>(uid);
+            }
+        }
+    }
+
+    private void OnActivate(EntityUid uid, RadioJammerComponent comp, ActivateInWorldEvent args)
+    {
+        var activated = !HasComp<ActiveRadioJammerComponent>(uid) && 
+            _powerCell.TryGetBatteryFromSlot(uid, out var battery) &&
+            battery.CurrentCharge > comp.Wattage;
+        if (activated)
+        {
+            EnsureComp<ActiveRadioJammerComponent>(uid);
+        }
+        else
+        {
+            RemComp<ActiveRadioJammerComponent>(uid);
+        }
+        var state = Loc.GetString(activated ? "radio-jammer-component-on-state" : "radio-jammer-component-off-state");
+        var message = Loc.GetString("radio-jammer-component-on-use", ("state", state));
+        _popup.PopupEntity(message, args.User, args.User);
+        args.Handled = true;
+    }
+
+    private void OnPowerCellChanged(EntityUid uid, ActiveRadioJammerComponent comp, PowerCellChangedEvent args)
+    {
+        if (args.Ejected)
+            RemComp<ActiveRadioJammerComponent>(uid);
+    }
+
+    private void OnExamine(EntityUid uid, RadioJammerComponent comp, ExaminedEvent args)
+    {
+        if (args.IsInDetailsRange)
+        {
+            var msg = HasComp<ActiveRadioJammerComponent>(uid)
+                ? Loc.GetString("radio-jammer-component-examine-on-state")
+                : Loc.GetString("radio-jammer-component-examine-off-state");
+            args.PushMarkup(msg);
+            if (_powerCell.TryGetBatteryFromSlot(uid, out var battery))
+                args.PushMarkup(Loc.GetString("radio-jammer-component-charge",
+                    ("charge", (int) ((battery.CurrentCharge / battery.MaxCharge) * 100))));
+        }
+    }
+
+    private void OnRadioSendAttempt(ref RadioSendAttemptEvent args)
+    {
+        var source = Transform(args.RadioSource).Coordinates;
+        var query = AllEntityQuery<ActiveRadioJammerComponent, RadioJammerComponent, TransformComponent>();
+        while (query.MoveNext(out _, out _, out var jam, out var transform))
+        {
+            if (source.InRange(EntityManager, _transform, transform.Coordinates, jam.Range))
+            {
+                args.Cancelled = true;
+                return;
+            }
+        }
+    }
+}