gta5view/src/DatabaseThread.cpp

224 lines
8.7 KiB
C++
Raw Normal View History

2017-10-09 08:35:48 +02:00
/*****************************************************************************
* gta5view Grand Theft Auto V Profile Viewer
* Copyright (C) 2016-2021 Syping
2017-10-09 08:35:48 +02:00
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*****************************************************************************/
#include "DatabaseThread.h"
#include "CrewDatabase.h"
#include "AppEnv.h"
#include <QNetworkAccessManager>
#include <QNetworkRequest>
#include <QNetworkReply>
#include <QJsonDocument>
#include <QJsonObject>
#include <QStringList>
#include <QVariantMap>
#include <QEventLoop>
#include <QTimer>
#include <QDebug>
#include <QUrl>
#define crewMaxPages 83
#define maxLoadFails 3
2017-10-09 08:35:48 +02:00
DatabaseThread::DatabaseThread(CrewDatabase *crewDB, QObject *parent) : QThread(parent), crewDB(crewDB)
{
continueLastCrew = true;
2017-10-09 08:35:48 +02:00
threadRunning = true;
}
void DatabaseThread::run()
{
QEventLoop threadLoop;
QObject::connect(this, SIGNAL(threadTerminated()), &threadLoop, SLOT(quit()));
2017-10-09 08:35:48 +02:00
while (threadRunning) {
2020-09-13 02:10:15 +02:00
QTimer::singleShot(300000, &threadLoop, SLOT(quit()));
threadLoop.exec();
2017-10-09 08:35:48 +02:00
}
}
void DatabaseThread::scanCrewReference(const QStringList &crewList, const int &requestDelay)
{
for (const QString &crewID : crewList) {
if (threadRunning && crewID != QLatin1String("0")) {
2017-10-09 08:35:48 +02:00
QNetworkAccessManager *netManager = new QNetworkAccessManager();
QNetworkRequest netRequest(AppEnv::getCrewFetchingUrl(crewID));
#if QT_VERSION >= 0x050600
2020-09-28 05:33:04 +02:00
#if QT_VERSION < 0x060000
2017-10-09 08:35:48 +02:00
netRequest.setAttribute(QNetworkRequest::FollowRedirectsAttribute, true);
2020-09-28 05:33:04 +02:00
#endif
2017-10-09 08:35:48 +02:00
#endif
netRequest.setRawHeader("User-Agent", AppEnv::getUserAgent());
2018-06-29 02:50:00 +02:00
netRequest.setRawHeader("Accept", "text/html");
netRequest.setRawHeader("Accept-Charset", "utf-8");
netRequest.setRawHeader("Accept-Language", "en-US,en;q=0.9");
2017-10-09 08:35:48 +02:00
netRequest.setRawHeader("Connection", "keep-alive");
QNetworkReply *netReply = netManager->get(netRequest);
QEventLoop *downloadLoop = new QEventLoop();
QObject::connect(netReply, SIGNAL(finished()), downloadLoop, SLOT(quit()));
if (!continueLastCrew)
QObject::connect(this, SIGNAL(threadTerminated()), downloadLoop, SLOT(quit()));
2017-10-09 08:35:48 +02:00
QTimer::singleShot(30000, downloadLoop, SLOT(quit()));
downloadLoop->exec();
downloadLoop->disconnect();
2017-10-09 08:35:48 +02:00
delete downloadLoop;
if (netReply->isFinished()) {
2017-10-09 08:35:48 +02:00
QString crewName;
QByteArray crewHtml = netReply->readAll();
QStringList crewHtmlSplit1 = QString::fromUtf8(crewHtml).split("<title>Rockstar Games Social Club - Crew : ");
if (crewHtmlSplit1.length() >= 2) {
2017-10-09 08:35:48 +02:00
QStringList crewHtmlSplit2 = QString(crewHtmlSplit1.at(1)).split("</title>");
if (crewHtmlSplit2.length() >= 1) {
2017-10-09 08:35:48 +02:00
crewName = crewHtmlSplit2.at(0);
}
}
if (!crewName.isEmpty()) {
2017-10-09 08:35:48 +02:00
emit crewNameFound(crewID.toInt(), crewName);
}
}
else {
2018-06-17 09:27:10 +02:00
netReply->abort();
}
2017-10-09 08:35:48 +02:00
if (threadRunning) {
QEventLoop *waitingLoop = new QEventLoop();
QTimer::singleShot(requestDelay, waitingLoop, SLOT(quit()));
if (!continueLastCrew)
QObject::connect(this, SIGNAL(threadTerminated()), waitingLoop, SLOT(quit()));
waitingLoop->exec();
waitingLoop->disconnect();
delete waitingLoop;
}
2017-10-09 08:35:48 +02:00
delete netReply;
delete netManager;
}
}
}
void DatabaseThread::scanCrewMembersList(const QStringList &crewList, const int &maxPages, const int &requestDelay)
{
for (const QString &crewID : crewList) {
if (threadRunning && crewID != QLatin1String("0")) {
int currentFail = 0;
2017-10-09 08:35:48 +02:00
int currentPage = 0;
int foundPlayers = 0;
int totalPlayers = 1000;
while(foundPlayers < totalPlayers && currentPage < maxPages && (continueLastCrew ? true : threadRunning)) {
2017-10-09 08:35:48 +02:00
QNetworkAccessManager *netManager = new QNetworkAccessManager();
QNetworkRequest netRequest(AppEnv::getPlayerFetchingUrl(crewID, currentPage));
#if QT_VERSION >= 0x050600
2020-09-28 05:33:04 +02:00
#if QT_VERSION < 0x060000
2017-10-09 08:35:48 +02:00
netRequest.setAttribute(QNetworkRequest::FollowRedirectsAttribute, true);
2020-09-28 05:33:04 +02:00
#endif
2017-10-09 08:35:48 +02:00
#endif
netRequest.setRawHeader("User-Agent", AppEnv::getUserAgent());
2018-06-29 02:50:00 +02:00
netRequest.setRawHeader("Accept", "application/json");
netRequest.setRawHeader("Accept-Charset", "utf-8");
netRequest.setRawHeader("Accept-Language", "en-US,en;q=0.9");
2017-10-09 08:35:48 +02:00
netRequest.setRawHeader("Connection", "keep-alive");
QNetworkReply *netReply = netManager->get(netRequest);
QEventLoop *downloadLoop = new QEventLoop();
QObject::connect(netReply, SIGNAL(finished()), downloadLoop, SLOT(quit()));
if (!continueLastCrew)
QObject::connect(this, SIGNAL(threadTerminated()), downloadLoop, SLOT(quit()));
2017-10-09 08:35:48 +02:00
QTimer::singleShot(30000, downloadLoop, SLOT(quit()));
downloadLoop->exec();
downloadLoop->disconnect();
2017-10-09 08:35:48 +02:00
delete downloadLoop;
if (netReply->isFinished()) {
2017-10-09 08:35:48 +02:00
QByteArray crewJson = netReply->readAll();
QJsonDocument crewDocument = QJsonDocument::fromJson(crewJson);
QJsonObject crewObject = crewDocument.object();
QVariantMap crewMap = crewObject.toVariantMap();
if (crewMap.contains("Total")) { totalPlayers = crewMap["Total"].toInt(); }
if (crewMap.contains("Members")) {
2017-10-21 05:59:10 +02:00
const QList<QVariant> memberList = crewMap["Members"].toList();
for (const QVariant &memberVariant : memberList) {
2017-10-09 08:35:48 +02:00
QMap<QString, QVariant> memberMap = memberVariant.toMap();
if (memberMap.contains("RockstarId") && memberMap.contains("Name")) {
2017-10-09 08:35:48 +02:00
int RockstarId = memberMap["RockstarId"].toInt();
QString memberName = memberMap["Name"].toString();
if (!memberName.isEmpty() && RockstarId != 0) {
foundPlayers++;
2017-10-09 08:35:48 +02:00
emit playerNameFound(RockstarId, memberName);
}
}
}
}
currentPage++;
}
else {
currentFail++;
if (currentFail == maxLoadFails) {
currentFail = 0;
currentPage++;
}
}
2017-10-09 08:35:48 +02:00
delete netReply;
delete netManager;
if (foundPlayers < totalPlayers && currentPage < maxPages && (continueLastCrew ? true : threadRunning)) {
QEventLoop *waitingLoop = new QEventLoop();
QTimer::singleShot(requestDelay, waitingLoop, SLOT(quit()));
if (!continueLastCrew) { QObject::connect(this, SIGNAL(threadTerminated()), waitingLoop, SLOT(quit())); }
waitingLoop->exec();
waitingLoop->disconnect();
delete waitingLoop;
}
2017-10-09 08:35:48 +02:00
}
}
}
}
2017-10-21 05:59:10 +02:00
void DatabaseThread::deleteCompatibleCrews(QStringList *crewList)
{
for (const QString &crewNID : *crewList) {
if (crewDB->isCompatibleCrew(crewNID)) {
2017-10-21 05:59:10 +02:00
crewList->removeAll(crewNID);
}
}
}
2017-10-09 08:35:48 +02:00
QStringList DatabaseThread::deleteCompatibleCrews(const QStringList &crewList)
{
QStringList crewListR = crewList;
for (const QString &crewNID : crewListR) {
if (crewDB->isCompatibleCrew(crewNID)) {
2017-10-09 08:35:48 +02:00
crewListR.removeAll(crewNID);
}
}
return crewListR;
}
void DatabaseThread::terminateThread()
2017-10-09 08:35:48 +02:00
{
threadRunning = false;
emit threadTerminated();
2017-10-09 08:35:48 +02:00
}