Replaced hardcoded source actions by user defined ISourceAction.

2022-11-06 19:23:49 +01:00
parent fcc8f53b59
commit 7188f88ba1
33 changed files with 546 additions and 385 deletions
--- a/OpenRA.Mods.Common/Installer/SourceActions/ExtractBlastSourceAction.cs
+++ b/OpenRA.Mods.Common/Installer/SourceActions/ExtractBlastSourceAction.cs
@@ -0,0 +1,77 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2022 The OpenRA Developers (see AUTHORS)
+ * This file is part of OpenRA, which is free software. It is made
+ * available to you 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. For more
+ * information, see COPYING.
+ */
+#endregion
+
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using OpenRA.Mods.Common.FileFormats;
+using OpenRA.Mods.Common.Widgets.Logic;
+using FS = OpenRA.FileSystem.FileSystem;
+
+namespace OpenRA.Mods.Common.Installer
+{
+	public class ExtractBlastSourceAction : ISourceAction
+	{
+		public void RunActionOnSource(MiniYaml actionYaml, string path, ModData modData, List<string> extracted, Action<string> updateMessage)
+		{
+			// Yaml path may be specified relative to a named directory (e.g. ^SupportDir) or the detected source path
+			var sourcePath = actionYaml.Value.StartsWith("^") ? Platform.ResolvePath(actionYaml.Value) : FS.ResolveCaseInsensitivePath(Path.Combine(path, actionYaml.Value));
+
+			using (var source = File.OpenRead(sourcePath))
+			{
+				foreach (var node in actionYaml.Nodes)
+				{
+					var targetPath = Platform.ResolvePath(node.Key);
+
+					if (File.Exists(targetPath))
+					{
+						Log.Write("install", "Skipping installed file " + targetPath);
+						continue;
+					}
+
+					var offsetNode = node.Value.Nodes.FirstOrDefault(n => n.Key == "Offset");
+					if (offsetNode == null)
+					{
+						Log.Write("install", "Skipping entry with missing Offset definition " + targetPath);
+						continue;
+					}
+
+					var lengthNode = node.Value.Nodes.FirstOrDefault(n => n.Key == "Length");
+					if (lengthNode == null)
+					{
+						Log.Write("install", "Skipping entry with missing Length definition " + targetPath);
+						continue;
+					}
+
+					var length = FieldLoader.GetValue<int>("Length", lengthNode.Value.Value);
+					source.Position = FieldLoader.GetValue<int>("Offset", offsetNode.Value.Value);
+
+					extracted.Add(targetPath);
+					Directory.CreateDirectory(Path.GetDirectoryName(targetPath));
+					var displayFilename = Path.GetFileName(Path.GetFileName(targetPath));
+
+					Action<long> onProgress = null;
+					if (length < InstallFromDiscLogic.ShowPercentageThreshold)
+						updateMessage(modData.Translation.GetString(InstallFromDiscLogic.Extracing, Translation.Arguments("filename", displayFilename)));
+					else
+						onProgress = b => updateMessage(modData.Translation.GetString(InstallFromDiscLogic.ExtracingProgress, Translation.Arguments("filename", displayFilename, "progress", 100 * b / length)));
+
+					using (var target = File.OpenWrite(targetPath))
+					{
+						Log.Write("install", $"Extracting {sourcePath} -> {targetPath}");
+						Blast.Decompress(source, target, (read, _) => onProgress?.Invoke(read));
+					}
+				}
+			}
+		}
+	}
+}