From 98c2b345121550bbc5183f695d4ce0af55ccdebc Mon Sep 17 00:00:00 2001
From: 20kdc <asdd2808@gmail.com>
Date: Sun, 22 May 2022 02:56:18 +0100
Subject: [PATCH] Redial Button II - Now with magic! (#8237)

* Redial Button II - Now with magic!

* 15-second timer before redial is allowed
---
 Content.Client/Entry/EntryPoint.cs            |  2 +
 Content.Client/IoC/ClientContentIoC.cs        |  2 +
 .../ExtendedDisconnectInformationManager.cs   | 45 +++++++++++++++++++
 Content.Client/Launcher/LauncherConnecting.cs | 28 ++++++++++++
 .../Launcher/LauncherConnectingGui.xaml       |  4 ++
 .../Launcher/LauncherConnectingGui.xaml.cs    | 37 +++++++++++++++
 .../en-US/launcher/launcher-connecting.ftl    |  4 +-
 7 files changed, 121 insertions(+), 1 deletion(-)
 create mode 100644 Content.Client/Launcher/ExtendedDisconnectInformationManager.cs

diff --git a/Content.Client/Entry/EntryPoint.cs b/Content.Client/Entry/EntryPoint.cs
index 46dd76003d..483f4dc733 100644
--- a/Content.Client/Entry/EntryPoint.cs
+++ b/Content.Client/Entry/EntryPoint.cs
@@ -129,6 +129,7 @@ namespace Content.Client.Entry
             IoCManager.Resolve<RulesManager>().Initialize();
             IoCManager.Resolve<ViewportManager>().Initialize();
             IoCManager.Resolve<GhostKickManager>().Initialize();
+            IoCManager.Resolve<ExtendedDisconnectInformationManager>().Initialize();
 
             IoCManager.InjectDependencies(this);
 
@@ -138,6 +139,7 @@ namespace Content.Client.Entry
             {
                 IoCManager.Resolve<IMapManager>().CreateNewMapEntity(MapId.Nullspace);
             };
+
         }
 
         /// <summary>
diff --git a/Content.Client/IoC/ClientContentIoC.cs b/Content.Client/IoC/ClientContentIoC.cs
index c7f1e95627..06618dd518 100644
--- a/Content.Client/IoC/ClientContentIoC.cs
+++ b/Content.Client/IoC/ClientContentIoC.cs
@@ -8,6 +8,7 @@ using Content.Client.GhostKick;
 using Content.Client.HUD;
 using Content.Client.Info;
 using Content.Client.Items.Managers;
+using Content.Client.Launcher;
 using Content.Client.Module;
 using Content.Client.Parallax.Managers;
 using Content.Client.Preferences;
@@ -45,6 +46,7 @@ namespace Content.Client.IoC
             IoCManager.Register<IGamePrototypeLoadManager, GamePrototypeLoadManager>();
             IoCManager.Register<NetworkResourceManager>();
             IoCManager.Register<GhostKickManager>();
+            IoCManager.Register<ExtendedDisconnectInformationManager>();
         }
     }
 }
diff --git a/Content.Client/Launcher/ExtendedDisconnectInformationManager.cs b/Content.Client/Launcher/ExtendedDisconnectInformationManager.cs
new file mode 100644
index 0000000000..0fc2eabc32
--- /dev/null
+++ b/Content.Client/Launcher/ExtendedDisconnectInformationManager.cs
@@ -0,0 +1,45 @@
+using System;
+using Robust.Client;
+using Robust.Client.UserInterface;
+using Robust.Shared.IoC;
+using Robust.Shared.Log;
+using Robust.Shared.Network;
+
+namespace Content.Client.Launcher;
+
+/// <summary>
+/// So apparently the way that disconnect information is shipped around is really indirect.
+/// But honestly, given that content might have additional flags (i.e. hide disconnect button for bans)?
+/// This is responsible for collecting any extended disconnect information.
+/// </summary>
+public sealed class ExtendedDisconnectInformationManager
+{
+    [Dependency] private readonly IClientNetManager _clientNetManager = default!;
+
+    private NetDisconnectedArgs? _lastNetDisconnectedArgs = null;
+
+    public NetDisconnectedArgs? LastNetDisconnectedArgs
+    {
+        get => _lastNetDisconnectedArgs;
+        private set
+        {
+            _lastNetDisconnectedArgs = value;
+            LastNetDisconnectedArgsChanged?.Invoke(value);
+        }
+    }
+
+    // BE CAREFUL!
+    // This may fire at an arbitrary time before or after whatever code that needs it.
+    public event Action<NetDisconnectedArgs?>? LastNetDisconnectedArgsChanged;
+
+    public void Initialize()
+    {
+        _clientNetManager.Disconnect += OnNetDisconnect;
+    }
+
+    private void OnNetDisconnect(object? sender, NetDisconnectedArgs args)
+    {
+        LastNetDisconnectedArgs = args;
+    }
+}
+
diff --git a/Content.Client/Launcher/LauncherConnecting.cs b/Content.Client/Launcher/LauncherConnecting.cs
index 511a85b1c2..f4ed4b46ae 100644
--- a/Content.Client/Launcher/LauncherConnecting.cs
+++ b/Content.Client/Launcher/LauncherConnecting.cs
@@ -2,6 +2,7 @@ using System;
 using Robust.Client;
 using Robust.Client.UserInterface;
 using Robust.Shared.IoC;
+using Robust.Shared.Log;
 using Robust.Shared.Network;
 
 namespace Content.Client.Launcher
@@ -70,6 +71,12 @@ namespace Content.Client.Launcher
 
         private void OnConnectFailed(object? _, NetConnectFailArgs args)
         {
+            if (args.RedialFlag)
+            {
+                // We've just *attempted* to connect and we've been told we need to redial, so do it.
+                // Result deliberately discarded.
+                Redial();
+            }
             ConnectFailReason = args.Reason;
             CurrentPage = Page.ConnectFailed;
         }
@@ -88,6 +95,27 @@ namespace Content.Client.Launcher
             }
         }
 
+        public bool Redial()
+        {
+            try
+            {
+                if (_gameController.LaunchState.Ss14Address != null)
+                {
+                    _gameController.Redial(_gameController.LaunchState.Ss14Address);
+                    return true;
+                }
+                else
+                {
+                    Logger.InfoS("launcher-ui", $"Redial not possible, no Ss14Address");
+                }
+            }
+            catch (Exception ex)
+            {
+                Logger.ErrorS("launcher-ui", $"Redial exception: {ex}");
+            }
+            return false;
+        }
+
         public void Exit()
         {
             _gameController.Shutdown("Exit button pressed");
diff --git a/Content.Client/Launcher/LauncherConnectingGui.xaml b/Content.Client/Launcher/LauncherConnectingGui.xaml
index 2d4aa5f1cb..787d35a754 100644
--- a/Content.Client/Launcher/LauncherConnectingGui.xaml
+++ b/Content.Client/Launcher/LauncherConnectingGui.xaml
@@ -33,6 +33,10 @@
                         <Button Name="ReconnectButton" Text="{Loc 'connecting-reconnect'}"
                                 HorizontalAlignment="Center"
                                 VerticalExpand="True" VerticalAlignment="Bottom" />
+                        <Button Name="RedialButton" Text="{Loc 'connecting-redial'}"
+                                Disabled="True"
+                                HorizontalAlignment="Center"
+                                VerticalExpand="True" VerticalAlignment="Bottom" />
                     </BoxContainer>
                 </Control>
                 <Label Name="ConnectingAddress" StyleClasses="LabelSubText" HorizontalAlignment="Center" />
diff --git a/Content.Client/Launcher/LauncherConnectingGui.xaml.cs b/Content.Client/Launcher/LauncherConnectingGui.xaml.cs
index 379ef76cdc..acf4cc91a8 100644
--- a/Content.Client/Launcher/LauncherConnectingGui.xaml.cs
+++ b/Content.Client/Launcher/LauncherConnectingGui.xaml.cs
@@ -4,6 +4,7 @@ using Robust.Client.UserInterface;
 using Robust.Client.UserInterface.Controls;
 using Robust.Client.UserInterface.XAML;
 using Robust.Shared.IoC;
+using Robust.Shared.Timing;
 using Robust.Shared.Localization;
 using Robust.Shared.Network;
 
@@ -12,7 +13,9 @@ namespace Content.Client.Launcher
     [GenerateTypedNameReferences]
     public sealed partial class LauncherConnectingGui : Control
     {
+        private const float RedialWaitTimeSeconds = 15f;
         private readonly LauncherConnecting _state;
+        private float _redialWaitTime = RedialWaitTimeSeconds;
 
         public LauncherConnectingGui(LauncherConnecting state)
         {
@@ -25,6 +28,12 @@ namespace Content.Client.Launcher
             Stylesheet = IoCManager.Resolve<IStylesheetManager>().SheetSpace;
 
             ReconnectButton.OnPressed += _ => _state.RetryConnect();
+            // Redial shouldn't fail, but if it does, try a reconnect (maybe we're being run from debug)
+            RedialButton.OnPressed += _ =>
+            {
+                if (!_state.Redial())
+                    _state.RetryConnect();
+            };
             RetryButton.OnPressed += _ => _state.RetryConnect();
             ExitButton.OnPressed += _ => _state.Exit();
 
@@ -37,6 +46,11 @@ namespace Content.Client.Launcher
             state.ConnectionStateChanged += ConnectionStateChanged;
 
             ConnectionStateChanged(state.ConnectionState);
+
+            // Redial flag setup
+            var edim = IoCManager.Resolve<ExtendedDisconnectInformationManager>();
+            edim.LastNetDisconnectedArgsChanged += LastNetDisconnectedArgsChanged;
+            LastNetDisconnectedArgsChanged(edim.LastNetDisconnectedArgs);
         }
 
         private void ConnectFailReasonChanged(string? reason)
@@ -46,6 +60,29 @@ namespace Content.Client.Launcher
                 : Loc.GetString("connecting-fail-reason", ("reason", reason));
         }
 
+        private void LastNetDisconnectedArgsChanged(NetDisconnectedArgs? args)
+        {
+            var redialFlag = args?.RedialFlag ?? false;
+            RedialButton.Visible = redialFlag;
+            ReconnectButton.Visible = !redialFlag;
+        }
+
+        protected override void FrameUpdate(FrameEventArgs args)
+        {
+            base.FrameUpdate(args);
+            _redialWaitTime -= args.DeltaSeconds;
+            if (_redialWaitTime <= 0)
+            {
+                RedialButton.Disabled = false;
+                RedialButton.Text = Loc.GetString("connecting-redial");
+            }
+            else
+            {
+                RedialButton.Disabled = true;
+                RedialButton.Text = Loc.GetString("connecting-redial-wait", ("time", _redialWaitTime));
+            }
+        }
+
         private void OnPageChanged(LauncherConnecting.Page page)
         {
             ConnectingStatus.Visible = page == LauncherConnecting.Page.Connecting;
diff --git a/Resources/Locale/en-US/launcher/launcher-connecting.ftl b/Resources/Locale/en-US/launcher/launcher-connecting.ftl
index 7f083951e2..c4fb51d266 100644
--- a/Resources/Locale/en-US/launcher/launcher-connecting.ftl
+++ b/Resources/Locale/en-US/launcher/launcher-connecting.ftl
@@ -4,6 +4,8 @@ connecting-title = Space Station 14
 connecting-exit = Exit
 connecting-retry = Retry
 connecting-reconnect = Reconnect
+connecting-redial = Relaunch
+connecting-redial-wait = Please wait: { TOSTRING($time, "G3") }
 connecting-in-progress = Connecting to server...
 connecting-disconnected = Disconnected from server:
 connecting-tip = Don't die!
@@ -14,4 +16,4 @@ connecting-state-NotConnecting = Not connecting
 connecting-state-ResolvingHost = Resolving host
 connecting-state-EstablishingConnection = Establishing connection
 connecting-state-Handshake = Handshake
-connecting-state-Connected = Connected
\ No newline at end of file
+connecting-state-Connected = Connected