Add new API connection to retrieve game info

2 files changed, 96 insertions, 0 deletions

diff --git a/src/nexusinterface.cpp b/src/nexusinterface.cpp
index 2c536b14..00ecdae3 100644
--- a/src/nexusinterface.cpp
+++ b/src/nexusinterface.cpp
@@ -91,6 +91,11 @@ void NexusBridge::requestToggleTracking(QString gameName, int modID, bool track,
                                                          userData, m_SubModule));
 }
 
+void NexusBridge::requestGameInfo(QString gameName, QVariant userData)
+{
+  m_RequestIDs.insert(m_Interface->requestGameInfo(gameName, this, userData, m_SubModule));
+}
+
 void NexusBridge::nxmDescriptionAvailable(QString gameName, int modID,
                                           QVariant userData, QVariant resultData,
                                           int requestID)
@@ -194,6 +199,15 @@ void NexusBridge::nxmTrackingToggled(QString gameName, int modID, QVariant userD
   }
 }
 
+void NexusBridge::nxmGameInfoAvailable(QString gameName, QVariant userData, QVariant resultData, int requestID)
+{
+  std::set<int>::iterator iter = m_RequestIDs.find(requestID);
+  if (iter != m_RequestIDs.end()) {
+    m_RequestIDs.erase(iter);
+    emit gameInfoAvailable(gameName, userData, resultData);
+  }
+}
+
 void NexusBridge::nxmRequestFailed(QString gameName, int modID, int fileID,
                                    QVariant userData, int requestID, int errorCode,
                                    const QString& errorMessage)
@@ -735,6 +749,26 @@ int NexusInterface::requestToggleTracking(QString gameName, int modID, bool trac
   return requestInfo.m_ID;
 }
 
+int NexusInterface::requestGameInfo(QString gameName, QObject* receiver, QVariant userData, const QString& subModule, MOBase::IPluginGame const* game)
+{
+  if (m_User.shouldThrottle()) {
+    throttledWarning(m_User);
+    return -1;
+  }
+
+  NXMRequestInfo requestInfo(NXMRequestInfo::TYPE_GAMEINFO, userData, subModule, game);
+  m_RequestQueue.enqueue(requestInfo);
+
+  connect(this, SIGNAL(nxmGameInfoAvailable(QString, QVariant, QVariant, int)),
+    receiver, SLOT(nxmGameInfoAvailable(QString, QVariant, QVariant, int)), Qt::UniqueConnection);
+
+  connect(this, SIGNAL(nxmRequestFailed(QString, int, int, QVariant, int, QNetworkReply::NetworkError, QString)),
+    receiver, SLOT(nxmRequestFailed(QString, int, int, QVariant, int, QNetworkReply::NetworkError, QString)), Qt::UniqueConnection);
+
+  nextRequest();
+  return requestInfo.m_ID;
+}
+
 int NexusInterface::requestInfoFromMd5(QString gameName, QByteArray& hash,
                                        QObject* receiver, QVariant userData,
                                        const QString& subModule,
@@ -1201,6 +1235,30 @@ NexusInterface::NXMRequestInfo::NXMRequestInfo(
       m_Endorse(false), m_Track(false), m_Hash(QByteArray())
 {}
 
+NexusInterface::NXMRequestInfo::NXMRequestInfo(Type type
+  , QVariant userData
+  , const QString & subModule
+  , MOBase::IPluginGame const *game
+)
+  : m_ModID(0)
+  , m_ModVersion("0")
+  , m_FileID(0)
+  , m_Reply(nullptr)
+  , m_Type(type)
+  , m_UpdatePeriod(UpdatePeriod::NONE)
+  , m_UserData(userData)
+  , m_Timeout(nullptr)
+  , m_Reroute(false)
+  , m_ID(s_NextID.fetchAndAddAcquire(1))
+  , m_URL(get_management_url())
+  , m_SubModule(subModule)
+  , m_NexusGameID(game->nexusGameID())
+  , m_GameName(game->gameNexusName())
+  , m_Endorse(false)
+  , m_Track(false)
+  , m_Hash(QByteArray())
+{}
+
 NexusInterface::NXMRequestInfo::NXMRequestInfo(
     int modID, int fileID, NexusInterface::NXMRequestInfo::Type type, QVariant userData,
     const QString& subModule, MOBase::IPluginGame const* game)
diff --git a/src/nexusinterface.h b/src/nexusinterface.h
index 331ce55d..95e46168 100644
--- a/src/nexusinterface.h
+++ b/src/nexusinterface.h
@@ -113,6 +113,12 @@ public:
   virtual void requestToggleTracking(QString gameName, int modID, bool track,
                                      QVariant userData);
 
+  /**
+   * @brief requestGameInfo
+   * @param userData user data to be returned with the result
+   */
+  virtual void requestGameInfo(QString gameName, QVariant userData);
+
 public slots:
 
   void nxmDescriptionAvailable(QString gameName, int modID, QVariant userData,
@@ -129,6 +135,7 @@ public slots:
   void nxmTrackedModsAvailable(QVariant userData, QVariant resultData, int requestID);
   void nxmTrackingToggled(QString gameName, int modID, QVariant userData, bool tracked,
                           int requestID);
+  void nxmGameInfoAvailable(QString gameName, QVariant userData, QVariant resultData, int requestID);
   void nxmRequestFailed(QString gameName, int modID, int fileID, QVariant userData,
                         int requestID, int errorCode, const QString& errorMessage);
 
@@ -445,6 +452,34 @@ public:
                             MOBase::IPluginGame const* game);
 
   /**
+  * @param gameName the game short name to support multiple game sources
+  * @brief toggle tracking state of the mod
+  * @param modID id of the mod
+  * @param track true if the mod should be tracked, false for not tracked
+  * @param receiver the object to receive the result asynchronously via a signal (nxmFilesAvailable)
+  * @param userData user data to be returned with the result
+  * @param game the game with which the mods are associated
+  * @return int an id to identify the request
+  */
+  int requestGameInfo(QString gameName, QObject* receiver, QVariant userData, const QString& subModule)
+  {
+    return requestGameInfo(gameName, receiver, userData, subModule, getGame(gameName));
+  }
+
+  /**
+  * @param gameName the game short name to support multiple game sources
+  * @brief toggle tracking state of the mod
+  * @param modID id of the mod
+  * @param track true if the mod should be tracked, false for not tracked
+  * @param receiver the object to receive the result asynchronously via a signal (nxmFilesAvailable)
+  * @param userData user data to be returned with the result
+  * @param game the game with which the mods are associated
+  * @return int an id to identify the request
+  */
+  int requestGameInfo(QString gameName, QObject* receiver, QVariant userData, const QString& subModule,
+    MOBase::IPluginGame const* game);
+
+  /**
    *
    */
   int requestInfoFromMd5(QString gameName, QByteArray& hash, QObject* receiver,
@@ -552,6 +587,7 @@ signals:
   void nxmTrackedModsAvailable(QVariant userData, QVariant resultData, int requestID);
   void nxmTrackingToggled(QString gameName, int modID, QVariant userData, bool tracked,
                           int requestID);
+  void nxmGameInfoAvailable(QString gameName, QVariant userData, QVariant resultData, int requestID);
   void nxmRequestFailed(QString gameName, int modID, int fileID, QVariant userData,
                         int requestID, int errorCode, const QString& errorString);
   void requestsChanged(const APIStats& stats, const APIUserAccount& user);
@@ -592,6 +628,7 @@ private:
       TYPE_TOGGLETRACKING,
       TYPE_TRACKEDMODS,
       TYPE_FILEINFO_MD5,
+      TYPE_GAMEINFO,
     } m_Type;
     UpdatePeriod m_UpdatePeriod;
     QVariant m_UserData;
@@ -614,6 +651,7 @@ private:
                    const QString& subModule, MOBase::IPluginGame const* game);
     NXMRequestInfo(int modID, int fileID, Type type, QVariant userData,
                    const QString& subModule, MOBase::IPluginGame const* game);
+    NXMRequestInfo(Type type, QVariant userData, const QString &subModule, MOBase::IPluginGame const *game);
     NXMRequestInfo(Type type, QVariant userData, const QString& subModule);
     NXMRequestInfo(UpdatePeriod period, Type type, QVariant userData,
                    const QString& subModule, MOBase::IPluginGame const* game);