/* 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/>.
 *
 *  This file incorporates work covered by the following copyright and
 *  permission notice:
 *
 * Copyright 2013-2021 MultiMC Contributors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#pragma once

#include "tasks/Task.h"
#include "net/NetJob.h"
#include "GoUpdate.h"

namespace GoUpdate
{
	/*!
	 * The DownloadTask is a task that takes a given version ID and repository
	 * URL, downloads that version's files from the repository, and prepares to
	 * install them.
	 */
	class DownloadTask : public Task
	{
		Q_OBJECT

	  public:
		/**
		 * Create a download task
		 *
		 * target is a template - XXXXXX at the end will be replaced with a
		 * random generated string, ensuring uniqueness
		 */
		explicit DownloadTask(shared_qobject_ptr<QNetworkAccessManager> network,
							  Status status, QString target,
							  QObject* parent = 0);
		virtual ~DownloadTask() {};

		/// Get the directory that will contain the update files.
		QString updateFilesDir();

		/// Get the list of operations that should be done
		OperationList operations();

		/// set updater download behavior
		void setUseLocalUpdater(bool useLocal);

	  protected:
		//! Entry point for tasks.
		virtual void executeTask() override;

		/*!
		 * Downloads the version info files from the repository.
		 * The files for both the current build, and the build that we're
		 * updating to need to be downloaded. If the current version's info file
		 * can't be found, MeshMC will not delete files that were removed
		 * between versions. It will still replace files that have changed,
		 * however. Note that although the repository URL for the current
		 * version is not given to the update task, the task will attempt to
		 * look it up in the UpdateChecker's channel list. If an error occurs
		 * here, the function will call emitFailed and return false.
		 */
		void loadVersionInfo();

		NetJob::Ptr m_vinfoNetJob;
		QByteArray currentVersionFileListData;
		QByteArray newVersionFileListData;
		Net::Download::Ptr m_currentVersionFileListDownload;
		Net::Download::Ptr m_newVersionFileListDownload;

		NetJob::Ptr m_filesNetJob;

		Status m_status;

		OperationList m_operations;

		/*!
		 * Temporary directory to store update files in.
		 * This will be set to not auto delete. Task will fail if this fails to
		 * be created.
		 */
		QTemporaryDir m_updateFilesDir;

	  protected slots:
		/*!
		 * This function is called when version information is finished
		 * downloading and at least the new file list download succeeded
		 */
		void processDownloadedVersionInfo();
		void vinfoDownloadFailed();

		void fileDownloadFinished();
		void fileDownloadFailed(QString reason);
		void fileDownloadProgressChanged(qint64 current, qint64 total);

	  private:
		shared_qobject_ptr<QNetworkAccessManager> m_network;
	};

} // namespace GoUpdate