From 0279d1205504f1187d1a1a4cdc2b3ef6356e491e Mon Sep 17 00:00:00 2001
From: spycrab <spycrab@users.noreply.github.com>
Date: Sat, 30 Mar 2019 14:50:41 +0100
Subject: [PATCH] Core/NetPlayServer: Integrate NetPlayIndex

---
 Source/Core/Core/NetPlayServer.cpp | 56 ++++++++++++++++++++++++++++--
 Source/Core/Core/NetPlayServer.h   |  3 ++
 2 files changed, 56 insertions(+), 3 deletions(-)

diff --git a/Source/Core/Core/NetPlayServer.cpp b/Source/Core/Core/NetPlayServer.cpp
index 166ce881bf..b6fc2b8e78 100644
--- a/Source/Core/Core/NetPlayServer.cpp
+++ b/Source/Core/Core/NetPlayServer.cpp
@@ -24,6 +24,7 @@
 #include "Common/ENetUtil.h"
 #include "Common/File.h"
 #include "Common/FileUtil.h"
+#include "Common/HttpRequest.h"
 #include "Common/Logging/Log.h"
 #include "Common/MsgHandler.h"
 #include "Common/SFMLHelper.h"
@@ -132,6 +133,8 @@ NetPlayServer::NetPlayServer(const u16 port, const bool forward_port,
     m_server = enet_host_create(&serverAddr, 10, CHANNEL_COUNT, 0, 0);
     if (m_server != nullptr)
       m_server->intercept = ENetUtil::InterceptCallback;
+
+    SetupIndex();
   }
   if (m_server != nullptr)
   {
@@ -162,6 +165,45 @@ static void ClearPeerPlayerId(ENetPeer* peer)
   }
 }
 
+void NetPlayServer::SetupIndex()
+{
+  if (!Config::Get(Config::NETPLAY_USE_INDEX))
+    return;
+
+  NetPlaySession session;
+
+  session.name = Config::Get(Config::NETPLAY_INDEX_NAME);
+  session.region = Config::Get(Config::NETPLAY_INDEX_REGION);
+  session.has_password = !Config::Get(Config::NETPLAY_INDEX_PASSWORD).empty();
+  session.method = m_traversal_client ? "traversal" : "direct";
+  session.game_id = m_selected_game.empty() ? "UNKNOWN" : m_selected_game;
+  session.player_count = static_cast<int>(m_players.size());
+  session.in_game = m_is_running;
+  session.port = GetPort();
+
+  if (m_traversal_client)
+  {
+    session.server_id = std::string(g_TraversalClient->GetHostID().data(), 8);
+  }
+  else
+  {
+    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;
+
+    session.server_id = std::string(response->begin(), response->end());
+  }
+
+  session.EncryptID(Config::Get(Config::NETPLAY_INDEX_PASSWORD));
+
+  m_index.Add(session);
+}
+
 // called from ---NETPLAY--- thread
 void NetPlayServer::ThreadFunc()
 {
@@ -178,6 +220,11 @@ void NetPlayServer::ThreadFunc()
 
       m_ping_timer.Start();
       SendToClients(spac);
+
+      m_index.SetPlayerCount(static_cast<int>(m_players.size()));
+      m_index.SetGame(m_selected_game);
+      m_index.SetInGame(m_is_running);
+
       m_update_pings = false;
     }
 
@@ -283,7 +330,7 @@ void NetPlayServer::ThreadFunc()
     ClearPeerPlayerId(player_entry.second.socket);
     enet_peer_disconnect(player_entry.second.socket, 0);
   }
-}
+}  // namespace NetPlay
 
 // called from ---NETPLAY--- thread
 unsigned int NetPlayServer::OnConnect(ENetPeer* socket, sf::Packet& rpac)
@@ -1017,11 +1064,14 @@ unsigned int NetPlayServer::OnData(sf::Packet& packet, Client& player)
 
 void NetPlayServer::OnTraversalStateChanged()
 {
+  const TraversalClient::State state = m_traversal_client->GetState();
+
+  if (g_TraversalClient->GetHostID()[0] != '\0')
+    SetupIndex();
+
   if (!m_dialog)
     return;
 
-  const TraversalClient::State state = m_traversal_client->GetState();
-
   if (state == TraversalClient::Failure)
     m_dialog->OnTraversalError(m_traversal_client->GetFailureReason());
 
diff --git a/Source/Core/Core/NetPlayServer.h b/Source/Core/Core/NetPlayServer.h
index 70c354a41d..ecb6ddc063 100644
--- a/Source/Core/Core/NetPlayServer.h
+++ b/Source/Core/Core/NetPlayServer.h
@@ -21,6 +21,7 @@
 #include "Common/TraversalClient.h"
 #include "Core/NetPlayProto.h"
 #include "InputCommon/GCPadStatus.h"
+#include "UICommon/NetPlayIndex.h"
 
 namespace NetPlay
 {
@@ -137,6 +138,7 @@ private:
   std::vector<std::pair<std::string, std::string>> GetInterfaceListInternal() const;
   void ChunkedDataThreadFunc();
   void ChunkedDataSend(sf::Packet&& packet, PlayerId pid, const TargetMode target_mode);
+  void SetupIndex();
 
   NetSettings m_settings;
 
@@ -187,5 +189,6 @@ private:
   ENetHost* m_server = nullptr;
   TraversalClient* m_traversal_client = nullptr;
   NetPlayUI* m_dialog = nullptr;
+  NetPlayIndex m_index;
 };
 }  // namespace NetPlay