Blast door/shutter, timer and or gate device linking fixes (#16347)

Content.Server/DeviceLinking/Systems/SignalTimerSystem.cs

@@ -0,0 +1,162 @@
+using Content.Server.DeviceLinking.Components;
+using Content.Server.Interaction;
+using Content.Server.UserInterface;
+using Content.Shared.Access.Systems;
+using Content.Shared.MachineLinking;
+using Content.Shared.TextScreen;
+using Robust.Server.GameObjects;
+using Robust.Shared.Timing;
+
+namespace Content.Server.DeviceLinking.Systems;
+
+public sealed class SignalTimerSystem : EntitySystem
+{
+    [Dependency] private readonly SharedAudioSystem _audio = default!;
+    [Dependency] private readonly IGameTiming _gameTiming = default!;
+    [Dependency] private readonly DeviceLinkSystem _signalSystem = default!;
+    [Dependency] private readonly SharedAppearanceSystem _appearanceSystem = default!;
+    [Dependency] private readonly UserInterfaceSystem _ui = default!;
+    [Dependency] private readonly AccessReaderSystem _accessReader = default!;
+    [Dependency] private readonly InteractionSystem _interaction = default!;
+
+    public override void Initialize()
+    {
+        base.Initialize();
+
+        SubscribeLocalEvent<SignalTimerComponent, ComponentInit>(OnInit);
+        SubscribeLocalEvent<SignalTimerComponent, AfterActivatableUIOpenEvent>(OnAfterActivatableUIOpen);
+
+        SubscribeLocalEvent<SignalTimerComponent, SignalTimerTextChangedMessage>(OnTextChangedMessage);
+        SubscribeLocalEvent<SignalTimerComponent, SignalTimerDelayChangedMessage>(OnDelayChangedMessage);
+        SubscribeLocalEvent<SignalTimerComponent, SignalTimerStartMessage>(OnTimerStartMessage);
+    }
+
+    private void OnInit(EntityUid uid, SignalTimerComponent component, ComponentInit args)
+    {
+        _appearanceSystem.SetData(uid, TextScreenVisuals.ScreenText, component.Label);
+    }
+
+    private void OnAfterActivatableUIOpen(EntityUid uid, SignalTimerComponent component, AfterActivatableUIOpenEvent args)
+    {
+        var time = TryComp<ActiveSignalTimerComponent>(uid, out var active) ? active.TriggerTime : TimeSpan.Zero;
+
+        if (_ui.TryGetUi(uid, SignalTimerUiKey.Key, out var bui))
+        {
+            _ui.SetUiState(bui, new SignalTimerBoundUserInterfaceState(component.Label,
+                TimeSpan.FromSeconds(component.Delay).Minutes.ToString("D2"),
+                TimeSpan.FromSeconds(component.Delay).Seconds.ToString("D2"),
+                component.CanEditLabel,
+                time,
+                active != null,
+                _accessReader.IsAllowed(args.User, uid)));
+        }
+    }
+
+    public void Trigger(EntityUid uid, SignalTimerComponent signalTimer)
+    {
+        RemComp<ActiveSignalTimerComponent>(uid);
+
+        _signalSystem.InvokePort(uid, signalTimer.TriggerPort);
+
+        _appearanceSystem.SetData(uid, TextScreenVisuals.Mode, TextScreenMode.Text);
+
+        if (_ui.TryGetUi(uid, SignalTimerUiKey.Key, out var bui))
+        {
+            _ui.SetUiState(bui, new SignalTimerBoundUserInterfaceState(signalTimer.Label,
+                TimeSpan.FromSeconds(signalTimer.Delay).Minutes.ToString("D2"),
+                TimeSpan.FromSeconds(signalTimer.Delay).Seconds.ToString("D2"),
+                signalTimer.CanEditLabel,
+                TimeSpan.Zero,
+                false,
+                true));
+        }
+    }
+
+    public override void Update(float frameTime)
+    {
+        base.Update(frameTime);
+        UpdateTimer();
+    }
+
+    private void UpdateTimer()
+    {
+        var query = EntityQueryEnumerator<ActiveSignalTimerComponent, SignalTimerComponent>();
+        while (query.MoveNext(out var uid, out var active, out var timer))
+        {
+            if (active.TriggerTime > _gameTiming.CurTime)
+                continue;
+
+            Trigger(uid, timer);
+
+            if (timer.DoneSound == null)
+                continue;
+            _audio.PlayPvs(timer.DoneSound, uid);
+        }
+    }
+
+    /// <summary>
+    ///     Checks if a UI <paramref name="message"/> is allowed to be sent by the user.
+    /// </summary>
+    /// <param name="uid">The entity that is interacted with.</param>
+    /// <param name="message"></param>
+    private bool IsMessageValid(EntityUid uid, BoundUserInterfaceMessage message)
+    {
+        if (message.Session.AttachedEntity is not { Valid: true } mob)
+            return false;
+
+        if (!_accessReader.IsAllowed(mob, uid))
+            return false;
+
+        return true;
+    }
+
+    private void OnTextChangedMessage(EntityUid uid, SignalTimerComponent component, SignalTimerTextChangedMessage args)
+    {
+        if (!IsMessageValid(uid, args))
+            return;
+
+        component.Label = args.Text[..Math.Min(5,args.Text.Length)];
+        _appearanceSystem.SetData(uid, TextScreenVisuals.ScreenText, component.Label);
+    }
+
+    private void OnDelayChangedMessage(EntityUid uid, SignalTimerComponent component, SignalTimerDelayChangedMessage args)
+    {
+        if (!IsMessageValid(uid, args))
+            return;
+
+        component.Delay = args.Delay.TotalSeconds;
+    }
+
+    private void OnTimerStartMessage(EntityUid uid, SignalTimerComponent component, SignalTimerStartMessage args)
+    {
+        if (!IsMessageValid(uid, args))
+            return;
+
+        TryComp<AppearanceComponent>(uid, out var appearance);
+
+        if (!HasComp<ActiveSignalTimerComponent>(uid))
+        {
+            var activeTimer = EnsureComp<ActiveSignalTimerComponent>(uid);
+            activeTimer.TriggerTime = _gameTiming.CurTime + TimeSpan.FromSeconds(component.Delay);
+
+            if (appearance != null)
+            {
+                _appearanceSystem.SetData(uid, TextScreenVisuals.Mode, TextScreenMode.Timer, appearance);
+                _appearanceSystem.SetData(uid, TextScreenVisuals.TargetTime, activeTimer.TriggerTime, appearance);
+                _appearanceSystem.SetData(uid, TextScreenVisuals.ScreenText, component.Label, appearance);
+            }
+
+            _signalSystem.InvokePort(uid, component.StartPort);
+        }
+        else
+        {
+            RemComp<ActiveSignalTimerComponent>(uid);
+
+            if (appearance != null)
+            {
+                _appearanceSystem.SetData(uid, TextScreenVisuals.Mode, TextScreenMode.Text, appearance);
+                _appearanceSystem.SetData(uid, TextScreenVisuals.ScreenText, component.Label, appearance);
+            }
+        }
+    }
+}