Add reconnect button to disconnection notice.

src/aoapplication.cpp

@@ -1,7 +1,6 @@
 #include "aoapplication.h"
 
 #include "courtroom.h"
-#include "debug_functions.h"
 #include "lobby.h"
 #include "networkmanager.h"
 #include "options.h"
@@ -20,6 +19,7 @@ AOApplication::AOApplication(QObject *parent)
     : QObject(parent)
 {
   net_manager = new NetworkManager(this);
+
   discord = new AttorneyOnline::Discord();
 
   asset_lookup_cache.reserve(2048);
@@ -138,16 +138,23 @@ QString AOApplication::find_image(QStringList p_list)
 
 void AOApplication::server_disconnected()
 {
+  bool try_reconnect = false;
   if (is_courtroom_constructed())
   {
-    if (w_courtroom->isVisible())
-    {
-      call_notice(tr("Disconnected from server."));
-    }
+    try_reconnect = w_courtroom->isVisible();
     construct_lobby();
     destruct_courtroom();
   }
   Options::getInstance().setServerSubTheme(QString());
+
+  if (try_reconnect && QMessageBox::question(nullptr,
+                                             tr("Server Disconnected"),
+                                             tr("Connection to the server has been lost. "
+                                                "Do you want to reconnect?"),
+                                             QMessageBox::Yes | QMessageBox::No) == QMessageBox::Yes)
+  {
+    net_manager->reconnect_to_last_server();
+  }
 }
 
 void AOApplication::loading_cancelled()
@@ -158,6 +165,7 @@ void AOApplication::loading_cancelled()
 void AOApplication::call_settings_menu()
 {
   AOOptionsDialog *l_dialog = new AOOptionsDialog(this);
+
   if (is_courtroom_constructed())
   {
     connect(l_dialog, &AOOptionsDialog::reloadThemeRequest, w_courtroom, &Courtroom::on_reload_theme_clicked);

src/networkmanager.cpp

@@ -152,6 +152,8 @@ void NetworkManager::connect_to_server(ServerInfo server)
 {
   disconnect_from_server();
 
+  m_last_server = server;
+
   qInfo().noquote() << QObject::tr("Connecting to %1").arg(server.toString());
   m_connection = new WebSocketConnection(ao_app, this);
 
@@ -173,6 +175,13 @@ void NetworkManager::disconnect_from_server()
   }
 }
 
+void NetworkManager::reconnect_to_last_server()
+{
+  connect(this, &NetworkManager::server_connected, this, &NetworkManager::join_to_server);
+
+  connect_to_server(m_last_server);
+}
+
 void NetworkManager::ship_server_packet(AOPacket packet)
 {
   if (!m_connection)
@@ -194,6 +203,7 @@ void NetworkManager::ship_server_packet(AOPacket packet)
 
 void NetworkManager::join_to_server()
 {
+  disconnect(this, &NetworkManager::server_connected, this, &NetworkManager::join_to_server);
   ship_server_packet(AOPacket("askchaa"));
 }
 

src/networkmanager.h

@@ -28,6 +28,7 @@ class NetworkManager : public QObject
 
   void connect_to_server(ServerInfo p_server);
   void disconnect_from_server();
+  void reconnect_to_last_server();
 
   QString get_user_agent() const;
 
@@ -60,4 +61,6 @@ private Q_SLOTS:
   const int heartbeat_interval = 60 * 5 * 1000;
 
   unsigned int s_decryptor = 5;
+
+  ServerInfo m_last_server;
 };