126 lines
3.4 KiB
C++
126 lines
3.4 KiB
C++
// For license of this file, see <project-root-folder>/LICENSE.md.
|
|
|
|
#include "network-web/cookiejar.h"
|
|
|
|
#include "3rd-party/boolinq/boolinq.h"
|
|
#include "definitions/definitions.h"
|
|
#include "miscellaneous/application.h"
|
|
#include "miscellaneous/iofactory.h"
|
|
#include "miscellaneous/settings.h"
|
|
|
|
#include <QDateTime>
|
|
#include <QDir>
|
|
#include <QNetworkCookie>
|
|
#include <QSettings>
|
|
|
|
CookieJar::CookieJar(QObject* parent) : QNetworkCookieJar(parent) {
|
|
loadCookies();
|
|
}
|
|
|
|
QList<QNetworkCookie> CookieJar::extractCookiesFromUrl(const QString& url) {
|
|
if (!url.contains(QSL(COOKIE_URL_IDENTIFIER))) {
|
|
return {};
|
|
}
|
|
|
|
int index = url.lastIndexOf(QSL(COOKIE_URL_IDENTIFIER), -1, Qt::CaseSensitivity::CaseInsensitive);
|
|
QString cookies_string = url.right(url.length() - index - QSL(COOKIE_URL_IDENTIFIER).size());
|
|
QStringList cookies_list = cookies_string.split(';');
|
|
QList<QNetworkCookie> cookies;
|
|
|
|
for (const QString& single_cookie : cookies_list) {
|
|
const QList<QNetworkCookie>& extracted_cookies = QNetworkCookie::parseCookies(single_cookie.toUtf8());
|
|
|
|
if (extracted_cookies.isEmpty()) {
|
|
continue;
|
|
}
|
|
|
|
QNetworkCookie cookie = extracted_cookies.at(0);
|
|
QDateTime date = QDateTime::currentDateTime();
|
|
|
|
date = date.addYears(30);
|
|
cookie.setExpirationDate(date);
|
|
cookies.append(cookie);
|
|
}
|
|
|
|
return cookies;
|
|
}
|
|
|
|
void CookieJar::loadCookies() {
|
|
Settings* sett = qApp->settings();
|
|
auto keys = sett->allKeys(Cookies::ID);
|
|
|
|
for (const QString& cookie_key : qAsConst(keys)) {
|
|
QByteArray encoded = sett->password(GROUP(Cookies), cookie_key, {}).toByteArray();
|
|
|
|
if (!encoded.isEmpty()) {
|
|
auto cookie = QNetworkCookie::parseCookies(encoded);
|
|
|
|
if (!cookie.isEmpty()) {
|
|
if (!QNetworkCookieJar::insertCookie(cookie.at(0))) {
|
|
qCriticalNN << LOGSEC_NETWORK
|
|
<< "Failed to load cookie"
|
|
<< QUOTE_W_SPACE(cookie_key)
|
|
<< "from settings.";
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void CookieJar::saveCookies() {
|
|
auto cookies = allCookies();
|
|
Settings* sett = qApp->settings();
|
|
int i = 1;
|
|
|
|
sett->beginGroup(GROUP(Cookies));
|
|
qobject_cast<QSettings*>(sett)->remove(QString());
|
|
sett->endGroup();
|
|
|
|
for (const QNetworkCookie& cookie : cookies) {
|
|
if (cookie.isSessionCookie()) {
|
|
continue;
|
|
}
|
|
|
|
sett->setPassword(GROUP(Cookies),
|
|
QSL("%1-%2").arg(QString::number(i++), QString::fromUtf8(cookie.name())),
|
|
cookie.toRawForm(QNetworkCookie::RawForm::Full));
|
|
}
|
|
}
|
|
|
|
QList<QNetworkCookie> CookieJar::cookiesForUrl(const QUrl& url) const {
|
|
return QNetworkCookieJar::cookiesForUrl(url);
|
|
}
|
|
|
|
bool CookieJar::setCookiesFromUrl(const QList<QNetworkCookie>& cookie_list, const QUrl& url) {
|
|
return QNetworkCookieJar::setCookiesFromUrl(cookie_list, url);
|
|
}
|
|
|
|
bool CookieJar::insertCookie(const QNetworkCookie& cookie) {
|
|
auto result = QNetworkCookieJar::insertCookie(cookie);
|
|
|
|
if (result) {
|
|
saveCookies();
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
bool CookieJar::updateCookie(const QNetworkCookie& cookie) {
|
|
auto result = QNetworkCookieJar::updateCookie(cookie);
|
|
|
|
if (result) {
|
|
saveCookies();
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
bool CookieJar::deleteCookie(const QNetworkCookie& cookie) {
|
|
auto result = QNetworkCookieJar::deleteCookie(cookie);
|
|
|
|
if (result) {
|
|
saveCookies();
|
|
}
|
|
|
|
return result;
|
|
}
|