summaryrefslogtreecommitdiffstats
path: root/logic/net/ScreenshotUploader.cpp
blob: 490c74b3fd7fc91f7f5e73588774a3f721fd57ae (plain)
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
#include "ScreenshotUploader.h"

#include <QNetworkRequest>
#include <QHttpMultiPart>
#include <QJsonDocument>
#include <QJsonObject>
#include <QHttpPart>
#include <QFile>
#include <QUrl>

#include "logic/lists/ScreenshotList.h"
#include "URLConstants.h"
#include "MultiMC.h"
#include "logger/QsLog.h"

ScreenShotUpload::ScreenShotUpload(ScreenShot *shot) : NetAction(), m_shot(shot)
{
	m_url = URLConstants::IMGUR_UPLOAD_URL;
	m_status = Job_NotStarted;
}

void ScreenShotUpload::start()
{
	m_status = Job_InProgress;
	QNetworkRequest request(m_url);
	request.setHeader(QNetworkRequest::UserAgentHeader, "MultiMC/5.0 (Uncached)");
	request.setRawHeader("Authorization", "Client-ID 5b97b0713fba4a3");
	request.setRawHeader("Accept", "application/json");

	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\"");
	namePart.setBody(m_shot->timestamp.toString(Qt::ISODate).toUtf8());
	multipart->append(namePart);

	auto worker = MMC->qnam();
	QNetworkReply *rep = worker->post(request, multipart);

	m_reply = std::shared_ptr<QNetworkReply>(rep);
	connect(rep, &QNetworkReply::uploadProgress, this, &ScreenShotUpload::downloadProgress);
	connect(rep, &QNetworkReply::finished, this, &ScreenShotUpload::downloadFinished);
	connect(rep, SIGNAL(error(QNetworkReply::NetworkError)),
			SLOT(downloadError(QNetworkReply::NetworkError)));
}
void ScreenShotUpload::downloadError(QNetworkReply::NetworkError error)
{
	QLOG_DEBUG() << m_reply->errorString();
	m_status = Job_Failed;
}
void ScreenShotUpload::downloadFinished()
{
	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();
			emit failed(m_index_within_job);
			return;
		}
		auto object = doc.object();
		if (!object.value("success").toBool())
		{
			QLOG_DEBUG() << doc.toJson();
			emit failed(m_index_within_job);
			return;
		}
		m_shot->imgurIndex = object.value("data").toObject().value("id").toString();
		m_shot->url = object.value("data").toObject().value("link").toString();
		m_status = Job_Finished;
		emit succeeded(m_index_within_job);
		return;
	}
	else
	{
		QLOG_DEBUG() << m_reply->readAll();
		m_reply.reset();
		emit failed(m_index_within_job);
		return;
	}
}
void ScreenShotUpload::downloadProgress(qint64 bytesReceived, qint64 bytesTotal)
{
	m_total_progress = bytesTotal;
	m_progress = bytesReceived;
	emit progress(m_index_within_job, bytesReceived, bytesTotal);
}