PrismLauncher/logic/screenshots/ImgurUpload.cpp

107 lines
3.1 KiB
C++
Raw Normal View History

2014-02-24 10:30:27 +00:00
#include "ImgurUpload.h"
2014-02-23 21:14:24 +00:00
#include <QNetworkRequest>
#include <QHttpMultiPart>
2014-02-23 21:14:24 +00:00
#include <QJsonDocument>
#include <QJsonObject>
#include <QHttpPart>
#include <QFile>
#include <QUrl>
2014-02-24 23:51:24 +00:00
#include "logic/screenshots/ScreenshotList.h"
#include "logic/net/URLConstants.h"
2014-02-23 21:14:24 +00:00
#include "MultiMC.h"
#include "logger/QsLog.h"
2014-02-23 21:14:24 +00:00
2014-02-24 23:51:24 +00:00
ImgurUpload::ImgurUpload(ScreenshotPtr shot) : NetAction(), m_shot(shot)
2014-02-23 21:14:24 +00:00
{
2014-02-24 10:30:27 +00:00
m_url = URLConstants::IMGUR_BASE_URL + "upload.json";
2014-02-23 21:14:24 +00:00
m_status = Job_NotStarted;
}
2014-02-24 10:30:27 +00:00
void ImgurUpload::start()
2014-02-23 21:14:24 +00:00
{
m_status = Job_InProgress;
QNetworkRequest request(m_url);
2014-02-23 21:14:24 +00:00
request.setHeader(QNetworkRequest::UserAgentHeader, "MultiMC/5.0 (Uncached)");
request.setRawHeader("Authorization", "Client-ID 5b97b0713fba4a3");
request.setRawHeader("Accept", "application/json");
2014-02-23 21:14:24 +00:00
QFile f(m_shot->file);
if (!f.open(QFile::ReadOnly))
{
emit failed(m_index_within_job);
return;
}
QHttpMultiPart *multipart = new QHttpMultiPart(QHttpMultiPart::FormDataType);
QHttpPart filePart;
filePart.setBody(f.readAll().toBase64());
filePart.setHeader(QNetworkRequest::ContentTypeHeader, "image/png");
filePart.setHeader(QNetworkRequest::ContentDispositionHeader, "form-data; name=\"image\"");
multipart->append(filePart);
QHttpPart typePart;
typePart.setHeader(QNetworkRequest::ContentDispositionHeader, "form-data; name=\"type\"");
typePart.setBody("base64");
multipart->append(typePart);
QHttpPart namePart;
namePart.setHeader(QNetworkRequest::ContentDispositionHeader, "form-data; name=\"name\"");
2014-02-24 09:34:51 +00:00
namePart.setBody(m_shot->timestamp.toString(Qt::ISODate).toUtf8());
multipart->append(namePart);
2014-02-23 21:14:24 +00:00
auto worker = MMC->qnam();
QNetworkReply *rep = worker->post(request, multipart);
2014-02-23 21:14:24 +00:00
m_reply = std::shared_ptr<QNetworkReply>(rep);
2014-02-24 10:30:27 +00:00
connect(rep, &QNetworkReply::uploadProgress, this, &ImgurUpload::downloadProgress);
connect(rep, &QNetworkReply::finished, this, &ImgurUpload::downloadFinished);
2014-02-23 21:14:24 +00:00
connect(rep, SIGNAL(error(QNetworkReply::NetworkError)),
SLOT(downloadError(QNetworkReply::NetworkError)));
}
2014-02-24 10:30:27 +00:00
void ImgurUpload::downloadError(QNetworkReply::NetworkError error)
2014-02-23 21:14:24 +00:00
{
QLOG_DEBUG() << m_reply->errorString();
2014-02-23 21:14:24 +00:00
m_status = Job_Failed;
}
2014-02-24 10:30:27 +00:00
void ImgurUpload::downloadFinished()
2014-02-23 21:14:24 +00:00
{
if (m_status != Job_Failed)
{
QByteArray data = m_reply->readAll();
m_reply.reset();
QJsonParseError jsonError;
QJsonDocument doc = QJsonDocument::fromJson(data, &jsonError);
if (jsonError.error != QJsonParseError::NoError)
{
QLOG_DEBUG() << jsonError.errorString();
2014-02-23 21:14:24 +00:00
emit failed(m_index_within_job);
return;
}
auto object = doc.object();
if (!object.value("success").toBool())
{
QLOG_DEBUG() << doc.toJson();
2014-02-23 21:14:24 +00:00
emit failed(m_index_within_job);
return;
}
2014-02-24 10:30:27 +00:00
m_shot->imgurId = object.value("data").toObject().value("id").toString();
2014-02-24 09:34:51 +00:00
m_shot->url = object.value("data").toObject().value("link").toString();
2014-02-23 21:14:24 +00:00
m_status = Job_Finished;
emit succeeded(m_index_within_job);
return;
}
else
{
QLOG_DEBUG() << m_reply->readAll();
2014-02-23 21:14:24 +00:00
m_reply.reset();
emit failed(m_index_within_job);
return;
}
}
2014-02-24 10:30:27 +00:00
void ImgurUpload::downloadProgress(qint64 bytesReceived, qint64 bytesTotal)
2014-02-23 21:14:24 +00:00
{
m_total_progress = bytesTotal;
m_progress = bytesReceived;
emit progress(m_index_within_job, bytesReceived, bytesTotal);
}