From 94d7e7d78a5d93448c7af8d181e8ef0b98b19e18 Mon Sep 17 00:00:00 2001
From: Techjar <tecknojar@gmail.com>
Date: Thu, 15 Nov 2018 00:58:07 -0500
Subject: [PATCH] Qt/NetPlayDialog: Add external IP to interface combo box

The interface address isn't particularly useful in most circumstances
(playing over internet), and we have a way to get the external IP now,
so displaying it in the dialog is useful.
---
 .../Core/DolphinQt/NetPlay/NetPlayDialog.cpp  | 59 ++++++++++++++-----
 Source/Core/DolphinQt/NetPlay/NetPlayDialog.h |  4 +-
 2 files changed, 46 insertions(+), 17 deletions(-)

diff --git a/Source/Core/DolphinQt/NetPlay/NetPlayDialog.cpp b/Source/Core/DolphinQt/NetPlay/NetPlayDialog.cpp
index 63d0ccab73b7..111c1f18e276 100644
--- a/Source/Core/DolphinQt/NetPlay/NetPlayDialog.cpp
+++ b/Source/Core/DolphinQt/NetPlay/NetPlayDialog.cpp
@@ -69,6 +69,7 @@ NetPlayDialog::NetPlayDialog(QWidget* parent)
   m_pad_mapping = new PadMappingDialog(this);
   m_md5_dialog = new MD5Dialog(this);
 
+  ResetExternalIP();
   CreateChatLayout();
   CreatePlayersLayout();
   CreateMainLayout();
@@ -497,6 +498,7 @@ void NetPlayDialog::show(std::string nickname, bool use_traversal)
   {
     if (use_traversal)
       m_room_box->addItem(tr("Room ID"));
+    m_room_box->addItem(tr("External"));
 
     for (const auto& iface : Settings::Instance().GetNetPlayServer()->GetInterfaceSet())
     {
@@ -528,6 +530,21 @@ void NetPlayDialog::show(std::string nickname, bool use_traversal)
   UpdateGUI();
 }
 
+void NetPlayDialog::ResetExternalIP()
+{
+  m_external_ip_address = Common::Lazy<std::string>([]() -> std::string {
+    Common::HttpRequest request;
+    // ENet does not support IPv6, so IPv4 has to be used
+    request.UseIPv4();
+    Common::HttpRequest::Response response =
+        request.Get("https://ip.dolphin-emu.org/", {{"X-Is-Dolphin", "1"}});
+
+    if (response.has_value())
+      return std::string(response->begin(), response->end());
+    return "";
+  });
+}
+
 void NetPlayDialog::UpdateDiscordPresence()
 {
 #ifdef USE_DISCORD_PRESENCE
@@ -555,23 +572,13 @@ void NetPlayDialog::UpdateDiscordPresence()
     }
     else
     {
-      if (m_external_ip_address.empty())
-      {
-        Common::HttpRequest request;
-        // ENet does not support IPv6, so IPv4 has to be used
-        request.UseIPv4();
-        Common::HttpRequest::Response response =
-            request.Get("https://ip.dolphin-emu.org/", {{"X-Is-Dolphin", "1"}});
-
-        if (!response.has_value())
-          return use_default();
-        m_external_ip_address = std::string(response->begin(), response->end());
-      }
+      if (m_external_ip_address->empty())
+        return use_default();
       const int port = Settings::Instance().GetNetPlayServer()->GetPort();
 
       Discord::UpdateDiscordPresence(
           m_player_count, Discord::SecretType::IPAddress,
-          Discord::CreateSecretFromIPAddress(m_external_ip_address, port), m_current_game);
+          Discord::CreateSecretFromIPAddress(*m_external_ip_address, port), m_current_game);
     }
   }
   else
@@ -683,10 +690,30 @@ void NetPlayDialog::UpdateGUI()
   }
   else if (server)
   {
-    m_hostcode_label->setText(QString::fromStdString(
-        server->GetInterfaceHost(m_room_box->currentData().toString().toStdString())));
+    if (m_room_box->currentIndex() == (m_use_traversal ? 1 : 0))
+    {
+      if (!m_external_ip_address->empty())
+      {
+        const int port = Settings::Instance().GetNetPlayServer()->GetPort();
+        m_hostcode_label->setText(QStringLiteral("%1:%2").arg(
+            QString::fromStdString(*m_external_ip_address), QString::number(port)));
+        m_hostcode_action_button->setEnabled(true);
+      }
+      else
+      {
+        m_hostcode_label->setText(tr("Unknown"));
+        m_hostcode_action_button->setEnabled(false);
+      }
+    }
+    else
+    {
+      m_hostcode_label->setText(QString::fromStdString(
+          server->GetInterfaceHost(m_room_box->currentData().toString().toStdString())));
+      m_hostcode_action_button->setEnabled(true);
+    }
+
     m_hostcode_action_button->setText(tr("Copy"));
-    m_hostcode_action_button->setEnabled(true);
+    m_is_copy_button_retry = false;
   }
 
   if (m_old_player_count != m_player_count)
diff --git a/Source/Core/DolphinQt/NetPlay/NetPlayDialog.h b/Source/Core/DolphinQt/NetPlay/NetPlayDialog.h
index c696eedaa53a..c2fbc0df7261 100644
--- a/Source/Core/DolphinQt/NetPlay/NetPlayDialog.h
+++ b/Source/Core/DolphinQt/NetPlay/NetPlayDialog.h
@@ -6,6 +6,7 @@
 
 #include <QDialog>
 
+#include "Common/Lazy.h"
 #include "Core/NetPlayClient.h"
 #include "VideoCommon/OnScreenDisplay.h"
 
@@ -79,6 +80,7 @@ class NetPlayDialog : public QDialog, public NetPlay::NetPlayUI
   void OnStart();
   void DisplayMessage(const QString& msg, const std::string& color,
                       int duration = OSD::Duration::NORMAL);
+  void ResetExternalIP();
   void UpdateDiscordPresence();
   void UpdateGUI();
   void GameStatusChanged(bool running);
@@ -122,7 +124,7 @@ class NetPlayDialog : public QDialog, public NetPlay::NetPlayUI
   MD5Dialog* m_md5_dialog;
   PadMappingDialog* m_pad_mapping;
   std::string m_current_game;
-  std::string m_external_ip_address;
+  Common::Lazy<std::string> m_external_ip_address;
   std::string m_nickname;
   GameListModel* m_game_list_model = nullptr;
   bool m_use_traversal = false;