QtUtils/HttpRequest.cpp at master · dcourtois/QtUtils · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
#include "./HttpRequest.h"

#include <QEventLoop>
#include <QNetworkAccessManager>
#include <QNetworkRequest>
#include <QNetworkReply>
#include <QRegularExpression>
#include <QUrl>


QT_UTILS_NAMESPACE_BEGIN

	//! The network manager
	QNetworkAccessManager s_NetworkManager;

	//!
	//! Synchronously request a url.
	//!
	//! @param url
	//!		The url to request.
	//!
	//! @param networkManager
	//!		Optional network manager to use. If not specified or nullptr, a global one defined in the
	//!		QtUtils library is used.
	//!
	//! @returns
	//!		The reply or an empty array.
	//!
	QByteArray RequestUrl(const QString & url, QNetworkAccessManager * networkManager)
	{
		// check the manager
		if (networkManager == nullptr)
		{
			networkManager = &s_NetworkManager;
		}

		// to handle http redirections
		QString address = url;
		for (int i = 0; i < 2; ++i)
		{
			// prepare the request
			QNetworkRequest request;
			request.setUrl(QUrl(address));

			// send
			QNetworkReply * reply = networkManager->get(request);

			// wait for the reply
			QEventLoop loop;
			while (reply->isFinished() == false)
			{
				loop.processEvents();
			}

			if (reply->error() != QNetworkReply::NoError)
			{
				reply->deleteLater();
				return QByteArray();
			}
			else
			{
				QVariant statusCode = reply->attribute(QNetworkRequest::Attribute::HttpStatusCodeAttribute);
				if (statusCode.isValid())
				{
					// redirect
					if (statusCode.toInt() == 301)
					{
						address = reply->header(QNetworkRequest::KnownHeaders::LocationHeader).toString();
						reply->deleteLater();
						continue;
					}
				}

				QByteArray result = reply->readAll();
				reply->deleteLater();
				return result;
			}
		}

		return QByteArray();
	}

	//!
	//! Asynchronously request a url.
	//!
	//! @param networkManager
	//!		The network manager to use for this request
	//!
	//! @param url
	//!		The url to request.
	//!
	//! @param success
	//!		A function that will be called when the request successfully completed.
	//!
	//! @param failure
	//!		A function that will be called if the request failed.
	//!
	//! @param networkManager
	//!		Optional network manager to use. If not specified or nullptr, a global one defined in the
	//!		QtUtils library is used.
	//!
	void RequestUrl(const QString & url, const SuccessCallback & success, const FailureCallback & failure, QNetworkAccessManager * networkManager)
	{
		// check the manager
		if (networkManager == nullptr)
		{
			networkManager = &s_NetworkManager;
		}

		// prepare the request
		QNetworkRequest request;
		request.setUrl(QUrl(url));

		// send
		QNetworkReply * reply = networkManager->get(request);
		QObject::connect(reply, &QNetworkReply::finished, [=] (void) {
			QNetworkReply::NetworkError error = reply->error();
			QVariant statusCode = reply->attribute(QNetworkRequest::Attribute::HttpStatusCodeAttribute);
			if (statusCode.isValid())
			{
				// redirect
				if (statusCode.toInt() == 301)
				{
					QString redirection = reply->header(QNetworkRequest::KnownHeaders::LocationHeader).toString();
					reply->deleteLater();
					RequestUrl(redirection, success, failure);
					return;
				}
			}
			if (error != QNetworkReply::NoError)
			{
				QString errorString = reply->errorString();
				reply->deleteLater();
				failure(error, errorString);
			}
			else
			{
				QByteArray result = reply->readAll();
				reply->deleteLater();
				success(result);
			}
		});
	}

QT_UTILS_NAMESPACE_END