Merge pull request #749 from isanae/manual-nexus-api-key

New dialog to manually enter Nexus API key

1 files changed, 150 insertions, 14 deletions

diff --git a/src/settingsdialog.cpp b/src/settingsdialog.cpp
index 7f53a9bb..b3e8c8a7 100644
--- a/src/settingsdialog.cpp
+++ b/src/settingsdialog.cpp
@@ -20,6 +20,7 @@ along with Mod Organizer.  If not, see <http://www.gnu.org/licenses/>.
 #include "settingsdialog.h"
 
 #include "ui_settingsdialog.h"
+#include "ui_nexusmanualkey.h"
 #include "categoriesdialog.h"
 #include "helper.h"
 #include "noeditdelegate.h"
@@ -27,6 +28,7 @@ along with Mod Organizer.  If not, see <http://www.gnu.org/licenses/>.
 #include "settings.h"
 #include "instancemanager.h"
 #include "nexusinterface.h"
+#include "nxmaccessmanager.h"
 #include "plugincontainer.h"
 
 #include <boost/uuid/uuid_generators.hpp>
@@ -47,9 +49,59 @@ along with Mod Organizer.  If not, see <http://www.gnu.org/licenses/>.
 
 using namespace MOBase;
 
-SettingsDialog::SettingsDialog(PluginContainer *pluginContainer, QWidget *parent)
+
+class NexusManualKeyDialog : public QDialog
+{
+public:
+  NexusManualKeyDialog(QWidget* parent)
+    : QDialog(parent), ui(new Ui::NexusManualKeyDialog)
+  {
+    ui->setupUi(this);
+
+    connect(ui->openBrowser, &QPushButton::clicked, [&]{ openBrowser(); });
+    connect(ui->paste, &QPushButton::clicked, [&]{ paste(); });
+    connect(ui->clear, &QPushButton::clicked, [&]{ clear(); });
+  }
+
+  void accept() override
+  {
+    m_key = ui->key->toPlainText();
+    QDialog::accept();
+  }
+
+  const QString& key() const
+  {
+    return m_key;
+  }
+
+  void openBrowser()
+  {
+    shell::OpenLink(QUrl("https://www.nexusmods.com/users/myaccount?tab=api"));
+  }
+
+  void paste()
+  {
+    const auto text = QApplication::clipboard()->text();
+    if (!text.isEmpty()) {
+      ui->key->setPlainText(text);
+    }
+  }
+
+  void clear()
+  {
+    ui->key->clear();
+  }
+
+private:
+  std::unique_ptr<Ui::NexusManualKeyDialog> ui;
+  QString m_key;
+};
+
+
+SettingsDialog::SettingsDialog(PluginContainer *pluginContainer, Settings* settings, QWidget *parent)
   : TutorableDialog("SettingsDialog", parent)
   , ui(new Ui::SettingsDialog)
+  , m_settings(settings)
   , m_PluginContainer(pluginContainer)
   , m_nexusLogin(new QWebSocket)
   , m_KeyReceived(false)
@@ -66,8 +118,9 @@ SettingsDialog::SettingsDialog(PluginContainer *pluginContainer, QWidget *parent
   connect(m_nexusLogin, SIGNAL(error(QAbstractSocket::SocketError)), this, SLOT(authError(QAbstractSocket::SocketError)));
   connect(m_nexusLogin, SIGNAL(textMessageReceived(const QString &)), this, SLOT(receiveApiKey(const QString &)));
   connect(m_nexusLogin, SIGNAL(disconnected()), this, SLOT(completeApiConnection()));
-  connect(this, SIGNAL(retryApiConnection()), this, SLOT(on_nexusConnect_clicked()));
   m_loginTimer.callOnTimeout(this, &SettingsDialog::loginPing);
+
+  updateNexusButtons();
 }
 
 SettingsDialog::~SettingsDialog()
@@ -338,10 +391,33 @@ void SettingsDialog::on_resetDialogsButton_clicked()
 
 void SettingsDialog::on_nexusConnect_clicked()
 {
-  ui->nexusConnect->setText("Connecting the API. Please login within the browser and accept the request. This will time out after 30 minutes.");
-  ui->nexusConnect->setDisabled(true);
+  fetchNexusApiKey();
+}
+
+void SettingsDialog::on_nexusManualKey_clicked()
+{
+  NexusManualKeyDialog dialog(this);
+
+  if (dialog.exec() != QDialog::Accepted) {
+    return;
+  }
+
+  const auto key = dialog.key();
+
+  if (key.isEmpty()) {
+    clearKey();
+  } else {
+    if (setKey(key)) {
+      testApiKey();
+    }
+  }
+}
+
+void SettingsDialog::fetchNexusApiKey()
+{
   QUrl url = QUrl("wss://sso.nexusmods.com");
   m_nexusLogin->open(url);
+  updateNexusButtons();
 }
 
 void SettingsDialog::dispatchLogin()
@@ -391,12 +467,17 @@ void SettingsDialog::receiveApiKey(const QString &response)
     if (data.contains("connection_token")) {
       m_AuthToken = data["connection_token"].toString();
     } else {
-      m_KeyReceived = true;
-      emit processApiKey(data["api_key"].toString());
+      const auto key = data["api_key"].toString();
+
       m_nexusLogin->close();
-      ui->nexusConnect->setText("Nexus API Key Stored");
       m_loginTimer.stop();
       m_totalPings = 0;
+
+      if (key.isEmpty()) {
+        clearKey();
+      } else {
+        setKey(key);
+      }
     }
   } else {
     QString error("There was a problem with SSO initialization: %1");
@@ -407,10 +488,66 @@ void SettingsDialog::receiveApiKey(const QString &response)
 
 void SettingsDialog::completeApiConnection()
 {
-  if (m_KeyReceived == true || !m_loginTimer.isActive())
-    emit closeApiConnection(ui->nexusConnect);
-  else
-    emit retryApiConnection();
+  if (!m_KeyReceived && !m_loginTimer.isActive()) {
+    QMessageBox::warning(qApp->activeWindow(), tr("Error"),
+      tr("Failed to retrieve a Nexus API key! Please try again. "
+        "A browser window should open asking you to authorize."));
+
+    // try again
+    fetchNexusApiKey();
+  }
+}
+
+bool SettingsDialog::setKey(const QString& key)
+{
+  m_KeyReceived = true;
+  const bool ret = m_settings->setNexusApiKey(key);
+  updateNexusButtons();
+  return ret;
+}
+
+bool SettingsDialog::clearKey()
+{
+  m_KeyCleared = true;
+  const auto ret = m_settings->clearNexusApiKey();
+  updateNexusButtons();
+  return ret;
+}
+
+void SettingsDialog::testApiKey()
+{
+  QString key;
+  if (!m_settings->getNexusApiKey(key)) {
+    qWarning().nospace() << "can't test API key, nothing stored";
+    return;
+  }
+
+  auto* am = NexusInterface::instance(m_PluginContainer)->getAccessManager();
+  am->apiCheck(key, true);
+}
+
+void SettingsDialog::updateNexusButtons()
+{
+  if (m_nexusLogin->state() != QAbstractSocket::UnconnectedState) {
+    // api key is in the process of being retrieved
+    ui->nexusConnect->setText("Connecting the API. Please login within the browser and accept the request. This will time out after 30 minutes.");
+    ui->nexusConnect->setEnabled(false);
+    ui->nexusDisconnect->setEnabled(false);
+    ui->nexusManualKey->setEnabled(false);
+  }
+  else if (m_settings->hasNexusApiKey()) {
+    // api key is present
+    ui->nexusConnect->setText("Nexus API Key Stored");
+    ui->nexusConnect->setEnabled(false);
+    ui->nexusDisconnect->setEnabled(true);
+    ui->nexusManualKey->setEnabled(false);
+  } else {
+    // api key not present
+    ui->nexusConnect->setText("Connect to Nexus");
+    ui->nexusConnect->setEnabled(true);
+    ui->nexusDisconnect->setEnabled(false);
+    ui->nexusManualKey->setEnabled(true);
+  }
 }
 
 void SettingsDialog::storeSettings(QListWidgetItem *pluginItem)
@@ -480,10 +617,9 @@ void SettingsDialog::on_clearCacheButton_clicked()
   NexusInterface::instance(m_PluginContainer)->clearCache();
 }
 
-void SettingsDialog::on_revokeNexusAuthButton_clicked()
+void SettingsDialog::on_nexusDisconnect_clicked()
 {
-  emit revokeApiKey(ui->nexusConnect);
-  m_KeyCleared = true;
+  clearKey();
 }
 
 void SettingsDialog::normalizePath(QLineEdit *lineEdit)