From 2847cefff701dad137cd04f628c76a9282d04a83 Mon Sep 17 00:00:00 2001
From: dada513 <dada513@protonmail.com>
Date: Fri, 20 May 2022 19:56:27 +0200
Subject: [PATCH] Add cursefrog key override

---
 launcher/Application.cpp             | 11 +++++
 launcher/Application.h               |  1 +
 launcher/net/Download.cpp            |  3 +-
 launcher/ui/pages/global/APIPage.cpp |  4 ++
 launcher/ui/pages/global/APIPage.ui  | 64 ++++++++++++++++++++++------
 5 files changed, 68 insertions(+), 15 deletions(-)

diff --git a/launcher/Application.cpp b/launcher/Application.cpp
index dc8a7b0d..ce62c41a 100644
--- a/launcher/Application.cpp
+++ b/launcher/Application.cpp
@@ -679,6 +679,7 @@ Application::Application(int &argc, char **argv) : QApplication(argc, argv)
 
         // Custom MSA credentials
         m_settings->registerSetting("MSAClientIDOverride", "");
+        m_settings->registerSetting("CFKeyOverride", "");
 
         // Init page provider
         {
@@ -1508,3 +1509,13 @@ QString Application::getMSAClientID()
 
     return BuildConfig.MSA_CLIENT_ID;
 }
+
+QString Application::getCurseKey()
+{
+    QString keyOverride = m_settings->get("CFKeyOverride").toString();
+    if (!keyOverride.isEmpty()) {
+        return keyOverride;
+    }
+
+    return BuildConfig.CURSEFORGE_API_KEY;
+}
diff --git a/launcher/Application.h b/launcher/Application.h
index 172321c0..3129b4fb 100644
--- a/launcher/Application.h
+++ b/launcher/Application.h
@@ -155,6 +155,7 @@ public:
     QString getJarsPath();
 
     QString getMSAClientID();
+    QString getCurseKey();
 
     /// this is the root of the 'installation'. Used for automatic updates
     const QString &root() {
diff --git a/launcher/net/Download.cpp b/launcher/net/Download.cpp
index 65cc8f67..7a401609 100644
--- a/launcher/net/Download.cpp
+++ b/launcher/net/Download.cpp
@@ -25,6 +25,7 @@
 #include "MetaCacheSink.h"
 
 #include "BuildConfig.h"
+#include "Application.h"
 
 namespace Net {
 
@@ -96,7 +97,7 @@ void Download::startImpl()
 
     request.setHeader(QNetworkRequest::UserAgentHeader, BuildConfig.USER_AGENT);
     if (request.url().host().contains("api.curseforge.com")) {
-        request.setRawHeader("x-api-key", BuildConfig.CURSEFORGE_API_KEY.toUtf8());
+        request.setRawHeader("x-api-key", APPLICATION->getCurseKey().toUtf8());
     };
 
     QNetworkReply* rep = m_network->get(request);
diff --git a/launcher/ui/pages/global/APIPage.cpp b/launcher/ui/pages/global/APIPage.cpp
index 287eb74f..8b806bcf 100644
--- a/launcher/ui/pages/global/APIPage.cpp
+++ b/launcher/ui/pages/global/APIPage.cpp
@@ -70,6 +70,8 @@ void APIPage::loadSettings()
     ui->urlChoices->setCurrentText(pastebinURL);
     QString msaClientID = s->get("MSAClientIDOverride").toString();
     ui->msaClientID->setText(msaClientID);
+    QString curseKey = s->get("CFKeyOverride").toString();
+    ui->curseKey->setText(curseKey);
 }
 
 void APIPage::applySettings()
@@ -79,6 +81,8 @@ void APIPage::applySettings()
     s->set("PastebinURL", pastebinURL);
     QString msaClientID = ui->msaClientID->text();
     s->set("MSAClientIDOverride", msaClientID);
+    QString curseKey = ui->curseKey->text();
+    s->set("CFKeyOverride", curseKey);
 }
 
 bool APIPage::apply()
diff --git a/launcher/ui/pages/global/APIPage.ui b/launcher/ui/pages/global/APIPage.ui
index acde9aef..eaa44c88 100644
--- a/launcher/ui/pages/global/APIPage.ui
+++ b/launcher/ui/pages/global/APIPage.ui
@@ -6,8 +6,8 @@
    <rect>
     <x>0</x>
     <y>0</y>
-    <width>491</width>
-    <height>474</height>
+    <width>603</width>
+    <height>530</height>
    </rect>
   </property>
   <layout class="QVBoxLayout" name="verticalLayout">
@@ -148,17 +148,56 @@
         </widget>
        </item>
        <item>
-        <spacer name="verticalSpacer">
-         <property name="orientation">
-          <enum>Qt::Vertical</enum>
+        <widget class="QGroupBox" name="groupBox_curse">
+         <property name="enabled">
+          <bool>true</bool>
          </property>
-         <property name="sizeHint" stdset="0">
-          <size>
-           <width>20</width>
-           <height>40</height>
-          </size>
+         <property name="title">
+          <string>&amp;CurseForge Core API</string>
          </property>
-        </spacer>
+         <layout class="QVBoxLayout" name="verticalLayout_5">
+          <item>
+           <widget class="Line" name="line_3">
+            <property name="orientation">
+             <enum>Qt::Horizontal</enum>
+            </property>
+           </widget>
+          </item>
+          <item>
+           <widget class="QLabel" name="label_6">
+            <property name="text">
+             <string>Note: you probably don't need to set this if CurseForge already works.</string>
+            </property>
+           </widget>
+          </item>
+          <item>
+           <widget class="QLineEdit" name="curseKey">
+            <property name="enabled">
+             <bool>true</bool>
+            </property>
+            <property name="placeholderText">
+             <string>(Default)</string>
+            </property>
+           </widget>
+          </item>
+          <item>
+           <widget class="QLabel" name="label_5">
+            <property name="text">
+             <string>Enter a custom API Key for CurseForge here. </string>
+            </property>
+            <property name="textFormat">
+             <enum>Qt::RichText</enum>
+            </property>
+            <property name="wordWrap">
+             <bool>true</bool>
+            </property>
+            <property name="openExternalLinks">
+             <bool>true</bool>
+            </property>
+           </widget>
+          </item>
+         </layout>
+        </widget>
        </item>
       </layout>
      </widget>
@@ -166,9 +205,6 @@
    </item>
   </layout>
  </widget>
- <tabstops>
-  <tabstop>tabWidget</tabstop>
- </tabstops>
  <resources/>
  <connections/>
 </ui>