2fde98a0d1
The Text element of these widgets was changed from display text to a translation key as part of adding translation support. Functions interested in the display text need to invoke GetText instead. Lots of functions have not been updated, resulting in symptoms such as measuring the font size of the translation key rather than the display text and resizing a widget to the wrong size. Update all callers to use GetText when getting or setting display text. This ensure their existing functionality that was intended to work in terms of the display text and not the translation key works as expected.
307 lines
9.2 KiB
C#
307 lines
9.2 KiB
C#
#region Copyright & License Information
|
|
/*
|
|
* Copyright (c) The OpenRA Developers and Contributors
|
|
* 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.Net.Http;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using OpenRA.FileSystem;
|
|
using OpenRA.Support;
|
|
using OpenRA.Widgets;
|
|
|
|
namespace OpenRA.Mods.Common.Widgets.Logic
|
|
{
|
|
public class DownloadPackageLogic : ChromeLogic
|
|
{
|
|
[TranslationReference("title")]
|
|
const string Downloading = "label-downloading";
|
|
|
|
[TranslationReference]
|
|
const string FetchingMirrorList = "label-fetching-mirror-list";
|
|
|
|
[TranslationReference]
|
|
const string UnknownHost = "label-unknown-host";
|
|
|
|
[TranslationReference("host", "received", "suffix")]
|
|
const string DownloadingFrom = "label-downloading-from";
|
|
|
|
[TranslationReference("host", "received", "total", "suffix", "progress")]
|
|
const string DownloadingFromProgress = "label-downloading-from-progress";
|
|
|
|
[TranslationReference]
|
|
const string VerifyingArchive = "label-verifying-archive";
|
|
|
|
[TranslationReference]
|
|
const string ArchiveValidationFailed = "label-archive-validation-failed";
|
|
|
|
[TranslationReference]
|
|
const string Extracting = "label-extracting-archive";
|
|
|
|
[TranslationReference("entry")]
|
|
const string ExtractingEntry = "label-extracting-archive-entry";
|
|
|
|
[TranslationReference]
|
|
const string ArchiveExtractionFailed = "label-archive-extraction-failed";
|
|
|
|
[TranslationReference]
|
|
const string MirrorSelectionFailed = "label-mirror-selection-failed";
|
|
|
|
static readonly string[] SizeSuffixes = { "bytes", "KB", "MB", "GB", "TB", "PB", "EB", "ZB", "YB" };
|
|
|
|
readonly ModData modData;
|
|
readonly ModContent.ModDownload download;
|
|
readonly Action onSuccess;
|
|
|
|
readonly Widget panel;
|
|
readonly ProgressBarWidget progressBar;
|
|
|
|
Func<string> getStatusText = () => "";
|
|
string downloadHost;
|
|
|
|
[ObjectCreator.UseCtor]
|
|
public DownloadPackageLogic(Widget widget, ModData modData, ModContent.ModDownload download, Action onSuccess)
|
|
{
|
|
this.modData = modData;
|
|
this.download = download;
|
|
this.onSuccess = onSuccess;
|
|
|
|
Log.AddChannel("install", "install.log");
|
|
|
|
panel = widget.Get("PACKAGE_DOWNLOAD_PANEL");
|
|
progressBar = panel.Get<ProgressBarWidget>("PROGRESS_BAR");
|
|
|
|
var statusLabel = panel.Get<LabelWidget>("STATUS_LABEL");
|
|
var statusFont = Game.Renderer.Fonts[statusLabel.Font];
|
|
var status = new CachedTransform<string, string>(s => WidgetUtils.TruncateText(s, statusLabel.Bounds.Width, statusFont));
|
|
statusLabel.GetText = () => status.Update(getStatusText());
|
|
|
|
var text = TranslationProvider.GetString(Downloading, Translation.Arguments("title", download.Title));
|
|
panel.Get<LabelWidget>("TITLE").GetText = () => text;
|
|
|
|
ShowDownloadDialog();
|
|
}
|
|
|
|
void ShowDownloadDialog()
|
|
{
|
|
getStatusText = () => TranslationProvider.GetString(FetchingMirrorList);
|
|
progressBar.Indeterminate = true;
|
|
|
|
var retryButton = panel.Get<ButtonWidget>("RETRY_BUTTON");
|
|
retryButton.IsVisible = () => false;
|
|
|
|
var cancelButton = panel.Get<ButtonWidget>("CANCEL_BUTTON");
|
|
|
|
void OnDownloadProgress(long total, long read, int progressPercentage)
|
|
{
|
|
var dataReceived = 0.0f;
|
|
var dataTotal = 0.0f;
|
|
var mag = 0;
|
|
var dataSuffix = "";
|
|
var host = downloadHost ?? TranslationProvider.GetString(UnknownHost);
|
|
|
|
if (total < 0)
|
|
{
|
|
mag = (int)Math.Log(read, 1024);
|
|
dataReceived = read / (float)(1L << (mag * 10));
|
|
dataSuffix = SizeSuffixes[mag];
|
|
|
|
getStatusText = () => TranslationProvider.GetString(DownloadingFrom,
|
|
Translation.Arguments("host", host, "received", $"{dataReceived:0.00}", "suffix", dataSuffix));
|
|
progressBar.Indeterminate = true;
|
|
}
|
|
else
|
|
{
|
|
mag = (int)Math.Log(total, 1024);
|
|
dataTotal = total / (float)(1L << (mag * 10));
|
|
dataReceived = read / (float)(1L << (mag * 10));
|
|
dataSuffix = SizeSuffixes[mag];
|
|
|
|
getStatusText = () => TranslationProvider.GetString(DownloadingFromProgress,
|
|
Translation.Arguments("host", host, "received", $"{dataReceived:0.00}", "total", $"{dataTotal:0.00}",
|
|
"suffix", dataSuffix, "progress", progressPercentage));
|
|
progressBar.Indeterminate = false;
|
|
}
|
|
|
|
progressBar.Percentage = progressPercentage;
|
|
}
|
|
|
|
void OnExtractProgress(string s) => Game.RunAfterTick(() => getStatusText = () => s);
|
|
|
|
void OnError(string s) => Game.RunAfterTick(() =>
|
|
{
|
|
var host = downloadHost ?? TranslationProvider.GetString(UnknownHost);
|
|
Log.Write("install", $"Download from {host} failed: " + s);
|
|
|
|
progressBar.Indeterminate = false;
|
|
progressBar.Percentage = 100;
|
|
getStatusText = () => $"{host}: Error: {s}";
|
|
retryButton.IsVisible = () => true;
|
|
cancelButton.OnClick = Ui.CloseWindow;
|
|
});
|
|
|
|
void DownloadUrl(string url)
|
|
{
|
|
Log.Write("install", "Downloading " + url);
|
|
|
|
var tokenSource = new CancellationTokenSource();
|
|
var token = tokenSource.Token;
|
|
downloadHost = new Uri(url).Host;
|
|
|
|
cancelButton.OnClick = () =>
|
|
{
|
|
tokenSource.Cancel();
|
|
Game.RunAfterTick(Ui.CloseWindow);
|
|
};
|
|
|
|
retryButton.OnClick = ShowDownloadDialog;
|
|
|
|
Task.Run(async () =>
|
|
{
|
|
var file = Path.Combine(Path.GetTempPath(), Path.GetRandomFileName());
|
|
|
|
try
|
|
{
|
|
var client = HttpClientFactory.Create();
|
|
|
|
var response = await client.GetAsync(url, HttpCompletionOption.ResponseHeadersRead, token);
|
|
|
|
using (var fileStream = new FileStream(file, FileMode.Create, FileAccess.Write, FileShare.ReadWrite, 8192, true))
|
|
{
|
|
await response.ReadAsStreamWithProgress(fileStream, OnDownloadProgress, token);
|
|
}
|
|
|
|
// Validate integrity
|
|
if (!string.IsNullOrEmpty(download.SHA1))
|
|
{
|
|
getStatusText = () => TranslationProvider.GetString(VerifyingArchive);
|
|
progressBar.Indeterminate = true;
|
|
|
|
var archiveValid = false;
|
|
try
|
|
{
|
|
using (var stream = File.OpenRead(file))
|
|
{
|
|
var archiveSHA1 = CryptoUtil.SHA1Hash(stream);
|
|
Log.Write("install", "Downloaded SHA1: " + archiveSHA1);
|
|
Log.Write("install", "Expected SHA1: " + download.SHA1);
|
|
|
|
archiveValid = archiveSHA1 == download.SHA1;
|
|
}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Log.Write("install", "SHA1 calculation failed: " + e.ToString());
|
|
}
|
|
|
|
if (!archiveValid)
|
|
{
|
|
OnError(TranslationProvider.GetString(ArchiveValidationFailed));
|
|
return;
|
|
}
|
|
}
|
|
|
|
// Automatically extract
|
|
getStatusText = () => TranslationProvider.GetString(Extracting);
|
|
progressBar.Indeterminate = true;
|
|
|
|
var extracted = new List<string>();
|
|
try
|
|
{
|
|
using (var stream = File.OpenRead(file))
|
|
{
|
|
var packageLoader = download.ObjectCreator.CreateObject<IPackageLoader>($"{download.Type}Loader");
|
|
|
|
if (packageLoader.TryParsePackage(stream, file, modData.ModFiles, out var package))
|
|
{
|
|
foreach (var kv in download.Extract)
|
|
{
|
|
if (!package.Contains(kv.Value))
|
|
{
|
|
Log.Write("install", $"Downloaded package does not contain {kv.Value} - skipping.");
|
|
continue;
|
|
}
|
|
|
|
OnExtractProgress(TranslationProvider.GetString(ExtractingEntry, Translation.Arguments("entry", kv.Value)));
|
|
Log.Write("install", "Extracting " + kv.Value);
|
|
var targetPath = Platform.ResolvePath(kv.Key);
|
|
Directory.CreateDirectory(Path.GetDirectoryName(targetPath));
|
|
extracted.Add(targetPath);
|
|
|
|
using (var zz = package.GetStream(kv.Value))
|
|
using (var f = File.Create(targetPath))
|
|
await zz.CopyToAsync(f);
|
|
}
|
|
|
|
package.Dispose();
|
|
}
|
|
}
|
|
|
|
Game.RunAfterTick(() =>
|
|
{
|
|
Ui.CloseWindow();
|
|
onSuccess();
|
|
});
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Log.Write("install", "Archive extraction failed: " + e.ToString());
|
|
|
|
foreach (var f in extracted)
|
|
{
|
|
Log.Write("install", "Deleting " + f);
|
|
File.Delete(f);
|
|
}
|
|
|
|
OnError(TranslationProvider.GetString(ArchiveExtractionFailed));
|
|
}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
OnError(e.ToString());
|
|
}
|
|
finally
|
|
{
|
|
File.Delete(file);
|
|
}
|
|
}, token);
|
|
}
|
|
|
|
if (download.MirrorList != null)
|
|
{
|
|
Log.Write("install", "Fetching mirrors from " + download.MirrorList);
|
|
|
|
Task.Run(async () =>
|
|
{
|
|
try
|
|
{
|
|
var client = HttpClientFactory.Create();
|
|
var httpResponseMessage = await client.GetAsync(download.MirrorList);
|
|
var result = await httpResponseMessage.Content.ReadAsStringAsync();
|
|
|
|
var mirrorList = result.Split(new[] { '\n' }, StringSplitOptions.RemoveEmptyEntries);
|
|
DownloadUrl(mirrorList.Random(new MersenneTwister()));
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Log.Write("install", "Mirror selection failed with error:");
|
|
Log.Write("install", e.ToString());
|
|
OnError(TranslationProvider.GetString(MirrorSelectionFailed));
|
|
}
|
|
});
|
|
}
|
|
else
|
|
DownloadUrl(download.URL);
|
|
}
|
|
}
|
|
}
|