1 files changed, 129 insertions, 0 deletions
diff --git a/meshmc/launcher/updater/DownloadTask.h b/meshmc/launcher/updater/DownloadTask.h
new file mode 100644
index 0000000000..e26d40dae9
--- /dev/null
+++ b/meshmc/launcher/updater/DownloadTask.h
@@ -0,0 +1,129 @@
+/* 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