From 377316999ea9ef6adbc75c52a858eb8e526616e7 Mon Sep 17 00:00:00 2001
From: Alexia <alexia@robotbrain.info>
Date: Thu, 12 May 2016 16:51:25 -0400
Subject: GH-767 Basic skin upload

---
 api/logic/CMakeLists.txt           |  4 ++-
 api/logic/minecraft/SkinUpload.cpp | 69 ++++++++++++++++++++++++++++++++++++++
 api/logic/minecraft/SkinUpload.h   | 40 ++++++++++++++++++++++
 3 files changed, 112 insertions(+), 1 deletion(-)
 create mode 100644 api/logic/minecraft/SkinUpload.cpp
 create mode 100644 api/logic/minecraft/SkinUpload.h

(limited to 'api/logic')

diff --git a/api/logic/CMakeLists.txt b/api/logic/CMakeLists.txt
index 32f5fd12..da465992 100644
--- a/api/logic/CMakeLists.txt
+++ b/api/logic/CMakeLists.txt
@@ -276,7 +276,9 @@ set(MINECRAFT_SOURCES
 	minecraft/liteloader/LiteLoaderInstaller.cpp
 	minecraft/liteloader/LiteLoaderVersionList.h
 	minecraft/liteloader/LiteLoaderVersionList.cpp
-)
+	minecraft/SkinUpload.cpp
+	minecraft/SkinUpload.h
+	)
 
 add_unit_test(GradleSpecifier
 	SOURCES minecraft/GradleSpecifier_test.cpp
diff --git a/api/logic/minecraft/SkinUpload.cpp b/api/logic/minecraft/SkinUpload.cpp
new file mode 100644
index 00000000..2c67c7aa
--- /dev/null
+++ b/api/logic/minecraft/SkinUpload.cpp
@@ -0,0 +1,69 @@
+#include "SkinUpload.h"
+#include <QNetworkRequest>
+#include <QHttpMultiPart>
+#include <Env.h>
+
+QByteArray getModelString(SkinUpload::Model model) {
+	switch (model) {
+		case SkinUpload::STEVE:
+			return "steve";
+		case SkinUpload::ALEX:
+			return "alex";
+		default:
+			qDebug() << "Unknown skin type!";
+			return "";
+	}
+}
+
+SkinUpload::SkinUpload(QObject *parent, AuthSessionPtr session, QByteArray skin, SkinUpload::Model model)
+	: Task(parent), m_model(model), m_skin(skin), m_session(session)
+{
+}
+
+void SkinUpload::executeTask()
+{
+	QNetworkRequest request(QUrl(QString("https://api.mojang.com/user/profile/%1/skin").arg(m_session->uuid)));
+	request.setRawHeader("Authorization", QString("Bearer: %1").arg(m_session->access_token).toLocal8Bit());
+
+	QHttpMultiPart *multiPart = new QHttpMultiPart(QHttpMultiPart::FormDataType);
+
+	QHttpPart model;
+	model.setHeader(QNetworkRequest::ContentDispositionHeader, QVariant("form-data; name=\"model\""));
+	model.setBody(getModelString(m_model));
+
+	QHttpPart skin;
+	skin.setHeader(QNetworkRequest::ContentTypeHeader, QVariant("image/png"));
+	skin.setHeader(QNetworkRequest::ContentDispositionHeader,
+				QVariant("form-data; name=\"file\"; filename=\"skin.png\""));
+	skin.setBody(m_skin);
+
+	multiPart->append(model);
+	multiPart->append(skin);
+
+	QNetworkReply *rep = ENV.qnam()->put(request, multiPart);
+	m_reply = std::shared_ptr<QNetworkReply>(rep);
+
+	setStatus(tr("Uploading skin"));
+	connect(rep, &QNetworkReply::uploadProgress, this, &Task::setProgress);
+	connect(rep, SIGNAL(error(QNetworkReply::NetworkError)), this, SLOT(downloadError(QNetworkReply::NetworkError)));
+	connect(rep, SIGNAL(finished()), this, SLOT(downloadFinished()));
+}
+
+void SkinUpload::downloadError(QNetworkReply::NetworkError error)
+{
+	// error happened during download.
+	qCritical() << "Network error: " << error;
+	emitFailed(m_reply->errorString());
+}
+
+void SkinUpload::downloadFinished()
+{
+	// if the download failed
+	if (m_reply->error() != QNetworkReply::NetworkError::NoError)
+	{
+		emitFailed(QString("Network error: %1").arg(m_reply->errorString()));
+		m_reply.reset();
+		return;
+	}
+	emitSucceeded();
+}
diff --git a/api/logic/minecraft/SkinUpload.h b/api/logic/minecraft/SkinUpload.h
new file mode 100644
index 00000000..86944b82
--- /dev/null
+++ b/api/logic/minecraft/SkinUpload.h
@@ -0,0 +1,40 @@
+#pragma once
+
+#include <QFile>
+#include <QtNetwork/QtNetwork>
+#include <memory>
+#include <minecraft/auth/AuthSession.h>
+#include "tasks/Task.h"
+#include "multimc_logic_export.h"
+
+typedef std::shared_ptr<class SkinUpload> SkinUploadPtr;
+
+class MULTIMC_LOGIC_EXPORT SkinUpload : public Task\
+{
+Q_OBJECT
+public:
+	enum Model
+	{
+		STEVE,
+		ALEX
+	};
+
+	// Note this class takes ownership of the file.
+	SkinUpload(QObject *parent, AuthSessionPtr session, QByteArray skin, Model model = STEVE);
+
+	virtual ~SkinUpload() {}
+
+private:
+	Model m_model;
+	QByteArray m_skin;
+	AuthSessionPtr m_session;
+	std::shared_ptr<QNetworkReply> m_reply;
+protected:
+	virtual void executeTask();
+
+public slots:
+
+	void downloadError(QNetworkReply::NetworkError);
+
+	void downloadFinished();
+};
-- 
cgit v1.2.3