Refactored RoundEndSystem (#5990)

* No RestartRound if round id changed

* Refactored RoundEndSystem

Content.Server/Communications/CommunicationsConsoleComponent.cs

@@ -17,11 +17,12 @@ using Timer = Robust.Shared.Timing.Timer;
 namespace Content.Server.Communications
 {
     [RegisterComponent]
-    public class CommunicationsConsoleComponent : SharedCommunicationsConsoleComponent
+    public class CommunicationsConsoleComponent : SharedCommunicationsConsoleComponent, IEntityEventSubscriber
     {
         [Dependency] private readonly IGameTiming _gameTiming = default!;
         [Dependency] private readonly IChatManager _chatManager = default!;
         [Dependency] private readonly IEntityManager _entities = default!;
+        [Dependency] private readonly IEntityManager _entityManager = default!;
 
         private bool Powered => !_entities.TryGetComponent(Owner, out ApcPowerReceiverComponent? receiver) || receiver.Powered;
 
@@ -42,10 +43,7 @@ namespace Content.Server.Communications
                 UserInterface.OnReceiveMessage += UserInterfaceOnOnReceiveMessage;
             }
 
-            RoundEndSystem.OnRoundEndCountdownStarted += UpdateBoundInterface;
+            _entityManager.EventBus.SubscribeEvent<RoundEndSystemChangedEvent>(EventSource.Local, this, (s) => UpdateBoundInterface());
-            RoundEndSystem.OnRoundEndCountdownCancelled += UpdateBoundInterface;
-            RoundEndSystem.OnRoundEndCountdownFinished += UpdateBoundInterface;
-            RoundEndSystem.OnCallCooldownEnded += UpdateBoundInterface;
         }
 
         protected override void Startup()
@@ -76,9 +74,7 @@ namespace Content.Server.Communications
 
         protected override void OnRemove()
         {
-            RoundEndSystem.OnRoundEndCountdownStarted -= UpdateBoundInterface;
+            _entityManager.EventBus.UnsubscribeEvent<RoundEndSystemChangedEvent>(EventSource.Local, this);
-            RoundEndSystem.OnRoundEndCountdownCancelled -= UpdateBoundInterface;
-            RoundEndSystem.OnRoundEndCountdownFinished -= UpdateBoundInterface;
             base.OnRemove();
         }
 

Content.Server/GameTicking/Commands/RestartRoundCommand.cs

@@ -1,10 +1,9 @@
-using System;
+using System;
 using Content.Server.Administration;
 using Content.Server.RoundEnd;
 using Content.Shared.Administration;
 using Robust.Shared.Console;
 using Robust.Shared.GameObjects;
-using Robust.Shared.IoC;
 
 namespace Content.Server.GameTicking.Commands
 {
@@ -13,7 +12,7 @@ namespace Content.Server.GameTicking.Commands
     {
         public string Command => "restartround";
         public string Description => "Ends the current round and starts the countdown for the next lobby.";
-        public string Help => String.Empty;
+        public string Help => string.Empty;
 
         public void Execute(IConsoleShell shell, string argStr, string[] args)
         {

Content.Server/RoundEnd/RoundEndSystem.cs

@@ -19,79 +19,60 @@ namespace Content.Server.RoundEnd
     {
         [Dependency] private readonly IGameTiming _gameTiming = default!;
         [Dependency] private readonly IChatManager _chatManager = default!;
+        [Dependency] private readonly GameTicker _gameTicker = default!;
 
         [Dependency] private readonly AdminLogSystem _adminLog = default!;
 
-        public const float RestartRoundTime = 20f;
 
-        private CancellationTokenSource _roundEndCancellationTokenSource = new();
+        private readonly TimeSpan _cooldownDuration = TimeSpan.FromSeconds(30);
-        private CancellationTokenSource _callCooldownEndedTokenSource = new();
+        private readonly TimeSpan _countdownDuration = TimeSpan.FromMinutes(4);
-        public bool IsRoundEndCountdownStarted { get; private set; }
+        private readonly TimeSpan _restartRoundDuration = TimeSpan.FromSeconds(20);
-        public TimeSpan RoundEndCountdownTime { get; set; } = TimeSpan.FromMinutes(4);
-        public TimeSpan? ExpectedCountdownEnd = null;
 
-        public TimeSpan LastCallTime { get; private set; }
+        private CancellationTokenSource? _countdownTokenSource = null;
-
+        private CancellationTokenSource? _cooldownTokenSource = null;
-        public TimeSpan CallCooldown { get; } = TimeSpan.FromSeconds(30);
+        public TimeSpan? ExpectedCountdownEnd { get; set; } = null;
-
-        // TODO: Make these regular eventbus events...
-        public delegate void RoundEndCountdownStarted();
-        public event RoundEndCountdownStarted? OnRoundEndCountdownStarted;
-
-        public delegate void RoundEndCountdownCancelled();
-        public event RoundEndCountdownCancelled? OnRoundEndCountdownCancelled;
-
-        public delegate void RoundEndCountdownFinished();
-        public event RoundEndCountdownFinished? OnRoundEndCountdownFinished;
-
-        public delegate void CallCooldownEnded();
-        public event CallCooldownEnded? OnCallCooldownEnded;
 
         public override void Initialize()
         {
             base.Initialize();
-
             SubscribeLocalEvent<RoundRestartCleanupEvent>(Reset);
         }
 
-        void Reset(RoundRestartCleanupEvent ev)
+        private void Reset(RoundRestartCleanupEvent ev)
         {
-            IsRoundEndCountdownStarted = false;
+            if (_countdownTokenSource != null)
-            _roundEndCancellationTokenSource.Cancel();
+            {
-            _roundEndCancellationTokenSource = new CancellationTokenSource();
+                _countdownTokenSource.Cancel();
-            _callCooldownEndedTokenSource.Cancel();
+                _countdownTokenSource = null;
-            _callCooldownEndedTokenSource = new CancellationTokenSource();
+            }
+
+            if (_cooldownTokenSource != null)
+            {
+                _cooldownTokenSource.Cancel();
+                _cooldownTokenSource = null;
+            }
+
             ExpectedCountdownEnd = null;
-            LastCallTime = default;
         }
 
         public bool CanCall()
         {
-            return _gameTiming.CurTime >= LastCallTime + CallCooldown;
+            return _cooldownTokenSource == null;
-        }
-
-        private void ActivateCooldown()
-        {
-            _callCooldownEndedTokenSource.Cancel();
-            _callCooldownEndedTokenSource = new CancellationTokenSource();
-            LastCallTime = _gameTiming.CurTime;
-            Timer.Spawn(CallCooldown, () => OnCallCooldownEnded?.Invoke(), _callCooldownEndedTokenSource.Token);
         }
 
         public void RequestRoundEnd(EntityUid? requester = null, bool checkCooldown = true)
         {
-            RequestRoundEnd(RoundEndCountdownTime, requester, checkCooldown);
+            RequestRoundEnd(_countdownDuration, requester, checkCooldown);
         }
 
         public void RequestRoundEnd(TimeSpan countdownTime, EntityUid? requester = null, bool checkCooldown = true)
         {
-            if (IsRoundEndCountdownStarted)
+            if (_gameTicker.RunLevel != GameRunLevel.InRound) return;
-                return;
 
-            if (checkCooldown && !CanCall())
+            if (checkCooldown && _cooldownTokenSource != null) return;
-            {
+
-                return;
+            if (_countdownTokenSource != null) return;
-            }
+            _countdownTokenSource = new();
 
             if (requester != null)
             {
@@ -102,29 +83,25 @@ namespace Content.Server.RoundEnd
                 _adminLog.Add(LogType.ShuttleCalled, LogImpact.High, $"Shuttle called");
             }
 
-            IsRoundEndCountdownStarted = true;
-
             _chatManager.DispatchStationAnnouncement(Loc.GetString("round-end-system-shuttle-called-announcement",("minutes", countdownTime.Minutes)), Loc.GetString("Station"), false);
 
             SoundSystem.Play(Filter.Broadcast(), "/Audio/Announcements/shuttlecalled.ogg");
 
             ExpectedCountdownEnd = _gameTiming.CurTime + countdownTime;
-            Timer.Spawn(countdownTime, EndRound, _roundEndCancellationTokenSource.Token);
+            Timer.Spawn(countdownTime, EndRound, _countdownTokenSource.Token);
 
             ActivateCooldown();
-
+            RaiseLocalEvent(RoundEndSystemChangedEvent.Default);
-            OnRoundEndCountdownStarted?.Invoke();
         }
 
         public void CancelRoundEndCountdown(EntityUid? requester = null, bool checkCooldown = true)
         {
-            if (!IsRoundEndCountdownStarted)
+            if (_gameTicker.RunLevel != GameRunLevel.InRound) return;
-                return;
+            if (checkCooldown && _cooldownTokenSource != null) return;
 
-            if (checkCooldown && !CanCall())
+            if (_countdownTokenSource == null) return;
-            {
+            _countdownTokenSource.Cancel();
-                return;
+            _countdownTokenSource = null;
-            }
 
             if (requester != null)
             {
@@ -135,31 +112,47 @@ namespace Content.Server.RoundEnd
                 _adminLog.Add(LogType.ShuttleRecalled, LogImpact.High, $"Shuttle recalled");
             }
 
-            IsRoundEndCountdownStarted = false;
-
             _chatManager.DispatchStationAnnouncement(Loc.GetString("round-end-system-shuttle-recalled-announcement"), Loc.GetString("Station"), false);
 
             SoundSystem.Play(Filter.Broadcast(), "/Audio/Announcements/shuttlerecalled.ogg");
 
-            _roundEndCancellationTokenSource.Cancel();
-            _roundEndCancellationTokenSource = new CancellationTokenSource();
-
             ExpectedCountdownEnd = null;
-
             ActivateCooldown();
-
+            RaiseLocalEvent(RoundEndSystemChangedEvent.Default);
-            OnRoundEndCountdownCancelled?.Invoke();
         }
 
         public void EndRound()
         {
-            OnRoundEndCountdownFinished?.Invoke();
+            if (_gameTicker.RunLevel != GameRunLevel.InRound) return;
-            var gameTicker = Get<GameTicker>();
+            RaiseLocalEvent(RoundEndSystemChangedEvent.Default);
-            gameTicker.EndRound();
+            _gameTicker.EndRound();
+            _countdownTokenSource?.Cancel();
+            _countdownTokenSource = new();
+            _chatManager.DispatchServerAnnouncement(Loc.GetString("round-end-system-round-restart-eta-announcement", ("seconds", _restartRoundDuration.Seconds)));
+            Timer.Spawn(_restartRoundDuration, AfterEndRoundRestart, _countdownTokenSource.Token);
+        }
 
-            _chatManager.DispatchServerAnnouncement(Loc.GetString("round-end-system-round-restart-eta-announcement", ("seconds", RestartRoundTime)));
+        private void AfterEndRoundRestart()
+        {
+            if (_gameTicker.RunLevel != GameRunLevel.InRound) return;
+            _gameTicker.RestartRound();
+        }
 
-            Timer.Spawn(TimeSpan.FromSeconds(RestartRoundTime), () => gameTicker.RestartRound(), CancellationToken.None);
+        private void ActivateCooldown()
+        {
+            _cooldownTokenSource?.Cancel();
+            _cooldownTokenSource = new();
+            Timer.Spawn(_cooldownDuration, () =>
+            {
+                _cooldownTokenSource.Cancel();
+                _cooldownTokenSource = null;
+                RaiseLocalEvent(RoundEndSystemChangedEvent.Default);
+            }, _cooldownTokenSource.Token);
         }
     }
+
+    public class RoundEndSystemChangedEvent : EntityEventArgs
+    {
+        public static RoundEndSystemChangedEvent Default { get; } = new();
+    }
 }