Save URL in NetworkRequest.

This commit is contained in:
Patrick Sean Klein 2023-05-09 22:59:40 +01:00
parent 7de7c8bcb3
commit b4249ea941
No known key found for this signature in database
GPG Key ID: B6D50F39A56F6906
2 changed files with 85 additions and 73 deletions

View File

@ -1,10 +1,10 @@
#include "NetworkRequest.h" #include "NetworkRequest.h"
#include "NetworkManager.h" #include "NetworkManager.h"
#include <QString>
#include <QtNetwork/QNetworkAccessManager> #include <QtNetwork/QNetworkAccessManager>
#include <QtNetwork/QNetworkReply> #include <QtNetwork/QNetworkReply>
#include <QtNetwork/QNetworkRequest> #include <QtNetwork/QNetworkRequest>
#include <QString>
namespace namespace
{ {
@ -17,12 +17,14 @@ namespace
} }
return url; return url;
} }
} } // namespace
void NetworkRequest::fetch(const QUrl& url) void NetworkRequest::fetch(const QUrl& url)
{ {
reset(); reset();
m_url = url;
QNetworkRequest request(url); QNetworkRequest request(url);
// Set headers // Set headers
@ -111,10 +113,7 @@ NetworkRequest::~NetworkRequest()
QUrl NetworkRequest::url() const QUrl NetworkRequest::url() const
{ {
if(m_reply) { return m_url;
return m_reply->url();
}
return {};
} }
void NetworkRequest::setMaxRedirects(int maxRedirects) void NetworkRequest::setMaxRedirects(int maxRedirects)
@ -122,9 +121,15 @@ void NetworkRequest::setMaxRedirects(int maxRedirects)
m_maxRedirects = std::max(0, maxRedirects); m_maxRedirects = std::max(0, maxRedirects);
} }
NetworkRequest::NetworkRequest(int maxRedirects, std::chrono::milliseconds timeoutDuration, NetworkRequest::NetworkRequest(int maxRedirects,
QList<QPair<QString, QString>> headers, QNetworkAccessManager* manager) std::chrono::milliseconds timeoutDuration,
: m_reply(nullptr), m_maxRedirects(maxRedirects), m_redirects(0), m_timeoutDuration(timeoutDuration), m_headers(headers) QList<QPair<QString, QString>> headers,
QNetworkAccessManager* manager)
: m_reply(nullptr)
, m_maxRedirects(maxRedirects)
, m_redirects(0)
, m_timeoutDuration(timeoutDuration)
, m_headers(headers)
{ {
m_manager = manager ? manager : getNetMgr(); m_manager = manager ? manager : getNetMgr();
connect(&m_timeout, &QTimer::timeout, this, &NetworkRequest::fetchTimeout); connect(&m_timeout, &QTimer::timeout, this, &NetworkRequest::fetchTimeout);

View File

@ -17,11 +17,11 @@
#ifndef KEEPASSXC_NETWORKREQUEST_H #ifndef KEEPASSXC_NETWORKREQUEST_H
#define KEEPASSXC_NETWORKREQUEST_H #define KEEPASSXC_NETWORKREQUEST_H
#include <QObject>
#include <QList>
#include <QUrl>
#include <QHash> #include <QHash>
#include <QList>
#include <QObject>
#include <QTimer> #include <QTimer>
#include <QUrl>
class QNetworkReply; class QNetworkReply;
class QNetworkAccessManager; class QNetworkAccessManager;
@ -32,7 +32,8 @@ class QNetworkAccessManager;
* response can be obtained by calling the url(), ContentType(), and ContentTypeParameters() * response can be obtained by calling the url(), ContentType(), and ContentTypeParameters()
* methods. * methods.
*/ */
class NetworkRequest : public QObject { class NetworkRequest : public QObject
{
Q_OBJECT Q_OBJECT
QNetworkAccessManager* m_manager; QNetworkAccessManager* m_manager;
@ -49,10 +50,14 @@ class NetworkRequest : public QObject {
int m_redirects; int m_redirects;
std::chrono::milliseconds m_timeoutDuration; std::chrono::milliseconds m_timeoutDuration;
QList<QPair<QString, QString>> m_headers; QList<QPair<QString, QString>> m_headers;
QUrl m_url;
public: public:
// TODO Disallow insecure connections by default? // TODO Disallow insecure connections by default?
explicit NetworkRequest(int maxRedirects, std::chrono::milliseconds timeoutDuration, explicit NetworkRequest(int maxRedirects,
QList<QPair<QString, QString>> headers, QNetworkAccessManager* manager = nullptr); std::chrono::milliseconds timeoutDuration,
QList<QPair<QString, QString>> headers,
QNetworkAccessManager* manager = nullptr);
~NetworkRequest() override; ~NetworkRequest() override;
void setMaxRedirects(int maxRedirects); void setMaxRedirects(int maxRedirects);
@ -71,6 +76,7 @@ class NetworkRequest : public QObject {
* @return Any parameters set in the Content-Type header. * @return Any parameters set in the Content-Type header.
*/ */
const QHash<QString, QString>& ContentTypeParameters() const; const QHash<QString, QString>& ContentTypeParameters() const;
private: private:
void reset(); void reset();
private slots: private slots:
@ -93,6 +99,7 @@ class NetworkRequest : public QObject {
*/ */
NetworkRequest createRequest(int maxRedirects = 5, NetworkRequest createRequest(int maxRedirects = 5,
std::chrono::milliseconds timeoutDuration = std::chrono::milliseconds(5000), std::chrono::milliseconds timeoutDuration = std::chrono::milliseconds(5000),
QList<QPair<QString, QString>> additionalHeaders = {}, QNetworkAccessManager* manager = nullptr); QList<QPair<QString, QString>> additionalHeaders = {},
QNetworkAccessManager* manager = nullptr);
#endif // KEEPASSXC_NETWORKREQUEST_H #endif // KEEPASSXC_NETWORKREQUEST_H