fix: add support for custom headers in tag type api

Bionus · Dec 14, 2022 · 7e058b5 · 7e058b5
1 parent 430a09f
commit 7e058b5
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 7 deletions.
diff --git a/src/lib/src/tags/tag-api-base.cpp b/src/lib/src/tags/tag-api-base.cpp
@@ -16,9 +16,10 @@ TagApiBase::~TagApiBase()
 	}
 }
 
-void TagApiBase::setUrl(QUrl url)
+void TagApiBase::setUrl(QUrl url, QMap<QString, QString> headers)
 {
 	m_url = std::move(url);
+	m_headers = std::move(headers);
 }
 
 void TagApiBase::load(bool rateLimit)
@@ -34,7 +35,7 @@ void TagApiBase::load(bool rateLimit)
 	}
 
 	Site::QueryType type = rateLimit ? Site::QueryType::Retry : Site::QueryType::List;
-	m_reply = m_site->get(m_url, type);
+	m_reply = m_site->get(m_url, type, {}, "", nullptr, m_headers);
 	connect(m_reply, &NetworkReply::finished, this, &TagApiBase::parseInternal);
 }
 

diff --git a/src/lib/src/tags/tag-api-base.h b/src/lib/src/tags/tag-api-base.h
@@ -1,6 +1,7 @@
 #ifndef TAG_API_BASE_H
 #define TAG_API_BASE_H
 
+#include <QMap>
 #include <QObject>
 #include <QUrl>
 
@@ -27,7 +28,7 @@ class TagApiBase : public QObject
 		void load(bool rateLimit = false);
 
 	protected:
-		void setUrl(QUrl url);
+		void setUrl(QUrl url, QMap<QString, QString> headers = {});
 		virtual void parse(const QString &source, int statusCode, Site *site) = 0;
 
 	public slots:
@@ -46,6 +47,7 @@ class TagApiBase : public QObject
 
 	private:
 		QUrl m_url;
+		QMap<QString, QString> m_headers;
 		NetworkReply *m_reply;
 };
 

diff --git a/src/lib/src/tags/tag-api.cpp b/src/lib/src/tags/tag-api.cpp
@@ -7,8 +7,8 @@
 TagApi::TagApi(Profile *profile, Site *site, Api *api, int page, int limit, const QString &order, QObject *parent)
 	: TagApiBase(profile, site, api, parent)
 {
-	const QString url = api->tagsUrl(page, limit, order, site).url;
-	setUrl(site->fixUrl(url));
+	const auto ret = api->tagsUrl(page, limit, order, site);
+	setUrl(site->fixUrl(ret.url), ret.headers);
 }
 
 void TagApi::parse(const QString &source, int statusCode, Site *site)

diff --git a/src/lib/src/tags/tag-type-api.cpp b/src/lib/src/tags/tag-type-api.cpp
@@ -7,8 +7,8 @@
 TagTypeApi::TagTypeApi(Profile *profile, Site *site, Api *api, QObject *parent)
 	: TagApiBase(profile, site, api, parent)
 {
-	const QString url = api->tagTypesUrl(site).url;
-	setUrl(site->fixUrl(url));
+	const auto ret = api->tagTypesUrl(site);
+	setUrl(site->fixUrl(ret.url), ret.headers);
 }
 
 void TagTypeApi::parse(const QString &source, int statusCode, Site *site)