/* SPDX-FileCopyrightText: 2026 Project Tick
 * SPDX-FileContributor: Project Tick
 * SPDX-License-Identifier: GPL-3.0-or-later
 *
 *   MeshMC - A Custom Launcher for Minecraft
 *   Copyright (C) 2026 Project Tick
 *
 *   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 <https://www.gnu.org/licenses/>.
 */

#include "ImgurUpload.h"
#include "BuildConfig.h"

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

ImgurUpload::ImgurUpload(ScreenShot::Ptr shot) : NetAction(), m_shot(shot)
{
	m_url = BuildConfig.IMGUR_BASE_URL + "upload.json";
	m_status = Job_NotStarted;
}

void ImgurUpload::startImpl()
{
	finished = false;
	m_status = Job_InProgress;
	QNetworkRequest request(m_url);
	request.setHeader(QNetworkRequest::UserAgentHeader,
					  BuildConfig.USER_AGENT_UNCACHED);
	request.setRawHeader("Authorization", QString("Client-ID %1")
											  .arg(BuildConfig.IMGUR_CLIENT_ID)
											  .toStdString()
											  .c_str());
	request.setRawHeader("Accept", "application/json");

	QFile f(m_shot->m_file.absoluteFilePath());
	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->m_file.baseName().toUtf8());
	multipart->append(namePart);

	QNetworkReply* rep = m_network->post(request, multipart);

	m_reply.reset(rep);
	connect(rep, &QNetworkReply::uploadProgress, this,
			&ImgurUpload::downloadProgress);
	connect(rep, &QNetworkReply::finished, this,
			&ImgurUpload::downloadFinished);
	connect(rep, SIGNAL(errorOccurred(QNetworkReply::NetworkError)),
			SLOT(downloadError(QNetworkReply::NetworkError)));
}
void ImgurUpload::downloadError(QNetworkReply::NetworkError error)
{
	qCritical() << "ImgurUpload failed with error" << m_reply->errorString()
				<< "Server reply:\n"
				<< m_reply->readAll();
	if (finished) {
		qCritical() << "Double finished ImgurUpload!";
		return;
	}
	m_status = Job_Failed;
	finished = true;
	m_reply.reset();
	emit failed(m_index_within_job);
}
void ImgurUpload::downloadFinished()
{
	if (finished) {
		qCritical() << "Double finished ImgurUpload!";
		return;
	}
	QByteArray data = m_reply->readAll();
	m_reply.reset();
	QJsonParseError jsonError;
	QJsonDocument doc = QJsonDocument::fromJson(data, &jsonError);
	if (jsonError.error != QJsonParseError::NoError) {
		qDebug() << "imgur server did not reply with JSON"
				 << jsonError.errorString();
		finished = true;
		m_reply.reset();
		emit failed(m_index_within_job);
		return;
	}
	auto object = doc.object();
	if (!object.value("success").toBool()) {
		qDebug() << "Screenshot upload not successful:" << doc.toJson();
		finished = true;
		m_reply.reset();
		emit failed(m_index_within_job);
		return;
	}
	m_shot->m_imgurId = object.value("data").toObject().value("id").toString();
	m_shot->m_url = object.value("data").toObject().value("link").toString();
	m_shot->m_imgurDeleteHash =
		object.value("data").toObject().value("deletehash").toString();
	m_status = Job_Finished;
	finished = true;
	emit succeeded(m_index_within_job);
	return;
}
void ImgurUpload::downloadProgress(qint64 bytesReceived, qint64 bytesTotal)
{
	m_total_progress = bytesTotal;
	m_progress = bytesReceived;
	emit netActionProgress(m_index_within_job, bytesReceived, bytesTotal);
}