wabbajack/Wabbajack.CLI/Verbs/ValidateLists.cs

691 lines
28 KiB
C#
Raw Normal View History

2021-09-27 12:42:46 +00:00
using System;
using System.Collections.Generic;
using System.CommandLine;
using System.CommandLine.NamingConventionBinder;
2021-09-27 12:42:46 +00:00
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Net.Http;
2021-09-27 12:42:46 +00:00
using System.Threading;
using System.Threading.Tasks;
using Microsoft.Extensions.Logging;
using SixLabors.ImageSharp;
2022-05-15 12:47:51 +00:00
using SixLabors.ImageSharp.Formats.Webp;
using SixLabors.ImageSharp.Processing;
2021-09-27 12:42:46 +00:00
using Wabbajack.CLI.Services;
using Wabbajack.Common;
using Wabbajack.Compression.Zip;
2021-09-27 12:42:46 +00:00
using Wabbajack.Downloaders;
using Wabbajack.Downloaders.Interfaces;
using Wabbajack.DTOs;
2022-01-03 04:44:16 +00:00
using Wabbajack.DTOs.Configs;
using Wabbajack.DTOs.Directives;
2021-09-27 12:42:46 +00:00
using Wabbajack.DTOs.DownloadStates;
using Wabbajack.DTOs.JsonConverters;
using Wabbajack.DTOs.ModListValidation;
using Wabbajack.DTOs.ServerResponses;
using Wabbajack.Hashing.xxHash64;
using Wabbajack.Installer;
using Wabbajack.Networking.Discord;
2021-10-23 16:51:17 +00:00
using Wabbajack.Networking.GitHub;
2021-09-27 12:42:46 +00:00
using Wabbajack.Paths;
using Wabbajack.Paths.IO;
using Wabbajack.RateLimiter;
using Wabbajack.Server.Lib.TokenProviders;
2021-10-23 16:51:17 +00:00
namespace Wabbajack.CLI.Verbs;
public class ValidateLists : IVerb
2021-09-27 12:42:46 +00:00
{
2021-10-23 16:51:17 +00:00
private static readonly Uri MirrorPrefix = new("https://mirror.wabbajack.org");
private readonly WriteOnlyClient _discord;
private readonly DownloadDispatcher _dispatcher;
private readonly DTOSerializer _dtos;
private readonly IResource<IFtpSiteCredentials> _ftpRateLimiter;
private readonly IFtpSiteCredentials _ftpSiteCredentials;
private readonly Client _gitHubClient;
private readonly ILogger<ValidateLists> _logger;
private readonly ParallelOptions _parallelOptions;
private readonly Random _random;
private readonly TemporaryFileManager _temporaryFileManager;
private readonly Networking.WabbajackClientApi.Client _wjClient;
private readonly HttpClient _httpClient;
private readonly IResource<HttpClient> _httpLimiter;
2021-10-23 16:51:17 +00:00
public ValidateLists(ILogger<ValidateLists> logger, Networking.WabbajackClientApi.Client wjClient,
Client gitHubClient, TemporaryFileManager temporaryFileManager,
DownloadDispatcher dispatcher, DTOSerializer dtos, ParallelOptions parallelOptions,
IFtpSiteCredentials ftpSiteCredentials, IResource<IFtpSiteCredentials> ftpRateLimiter,
WriteOnlyClient discordClient, HttpClient httpClient, IResource<HttpClient> httpLimiter)
2021-09-27 12:42:46 +00:00
{
2021-10-23 16:51:17 +00:00
_logger = logger;
_wjClient = wjClient;
_gitHubClient = gitHubClient;
_temporaryFileManager = temporaryFileManager;
_dispatcher = dispatcher;
_dtos = dtos;
_parallelOptions = parallelOptions;
_ftpSiteCredentials = ftpSiteCredentials;
_ftpRateLimiter = ftpRateLimiter;
_discord = discordClient;
_random = new Random();
_httpClient = httpClient;
_httpLimiter = httpLimiter;
2021-10-23 16:51:17 +00:00
}
2021-09-27 12:42:46 +00:00
2021-10-23 16:51:17 +00:00
public Command MakeCommand()
{
var command = new Command("validate-lists");
command.Add(new Option<AbsolutePath>(new[] {"-r", "--reports"}, "Location to store validation report outputs"));
command.Add(new Option<AbsolutePath>(new[] {"--other-archives"},
"Look for files here before downloading (stored by hex hash name)")
{IsRequired = false});
command.Description = "Gets a list of modlists, validates them and exports a result list";
command.Handler = CommandHandler.Create(Run);
return command;
}
2021-09-27 12:42:46 +00:00
2022-04-02 15:36:54 +00:00
public async Task<int> Run(AbsolutePath reports, AbsolutePath otherArchives)
2021-10-23 16:51:17 +00:00
{
_logger.LogInformation("Cleaning {Reports}", reports);
if (reports.DirectoryExists())
reports.DeleteDirectory();
2021-10-23 16:51:17 +00:00
reports.CreateDirectory();
var token = CancellationToken.None;
2021-09-27 12:42:46 +00:00
2021-10-23 16:51:17 +00:00
_logger.LogInformation("Scanning for existing patches/mirrors");
2021-12-18 16:14:39 +00:00
var mirroredFiles = (await _wjClient.GetAllMirroredFileDefinitions(token)).Select(m => m.Hash).ToHashSet();
_logger.LogInformation("Found {Count} mirrored files", mirroredFiles.Count);
var patchFiles = await _wjClient.GetAllPatches(token);
_logger.LogInformation("Found {Count} patches", patchFiles.Length);
2021-09-27 12:42:46 +00:00
2022-01-03 04:44:16 +00:00
var forcedRemovals = (await _wjClient.GetForcedRemovals(token)).ToLookup(f => f.Hash);
_logger.LogInformation("Found {Count} forced removals", forcedRemovals.Count);
2021-10-23 16:51:17 +00:00
var validationCache = new LazyCache<string, Archive, (ArchiveStatus Status, Archive archive)>
(x => x.State.PrimaryKeyString + x.Hash,
2022-01-03 04:44:16 +00:00
archive => DownloadAndValidate(archive, forcedRemovals, token));
2021-12-18 16:14:39 +00:00
2021-10-23 16:51:17 +00:00
var stopWatch = Stopwatch.StartNew();
2022-04-02 15:36:54 +00:00
var listData = await _wjClient.LoadLists();
2021-10-23 16:51:17 +00:00
var validatedLists = await listData.PMapAll(async modList =>
{
var validatedList = new ValidatedModList
2021-09-27 12:42:46 +00:00
{
2021-10-23 16:51:17 +00:00
Name = modList.Title,
ModListHash = modList.DownloadMetadata?.Hash ?? default,
2022-03-30 03:39:48 +00:00
MachineURL = modList.NamespacedName,
2021-10-23 16:51:17 +00:00
Version = modList.Version
};
2021-09-27 12:42:46 +00:00
2021-10-23 16:51:17 +00:00
if (modList.ForceDown)
{
_logger.LogWarning("List is ForceDown, skipping");
validatedList.Status = ListStatus.ForcedDown;
return validatedList;
}
2021-09-27 12:42:46 +00:00
2022-03-30 03:39:48 +00:00
using var scope = _logger.BeginScope("MachineURL: {MachineUrl}", modList.NamespacedName);
_logger.LogInformation("Verifying {MachineUrl} - {Title}", modList.NamespacedName, modList.Title);
//await DownloadModList(modList, archiveManager, CancellationToken.None);
2021-09-27 12:42:46 +00:00
2021-10-23 16:51:17 +00:00
ModList modListData;
try
{
_logger.LogInformation("Loading Modlist");
modListData =
await StandardInstaller.Load(_dtos, _dispatcher, modList, token);
2021-10-23 16:51:17 +00:00
}
catch (Exception ex)
2021-10-23 16:51:17 +00:00
{
_logger.LogError(ex, "Forcing down {Modlist} due to error while loading: ", modList.NamespacedName);
2021-10-23 16:51:17 +00:00
validatedList.Status = ListStatus.ForcedDown;
return validatedList;
}
2021-09-27 12:42:46 +00:00
2021-12-18 16:14:39 +00:00
_logger.LogInformation("Verifying {Count} archives", modListData.Archives.Length);
2021-09-27 12:42:46 +00:00
2021-10-23 16:51:17 +00:00
var archives = await modListData.Archives.PMapAll(async archive =>
{
var result = await validationCache.Get(archive);
2021-12-18 17:23:38 +00:00
if (result.Status == ArchiveStatus.Valid)
{
return new ValidatedArchive
{
Status = ArchiveStatus.Valid,
Original = archive
};
}
2021-09-27 12:42:46 +00:00
2021-10-23 16:51:17 +00:00
if (result.Status == ArchiveStatus.InValid)
{
2021-12-18 16:14:39 +00:00
if (mirroredFiles.Contains(archive.Hash))
{
return new ValidatedArchive
{
Status = ArchiveStatus.Mirrored,
Original = archive,
PatchedFrom = new Archive
{
State = new WabbajackCDN
{
Url = _wjClient.GetMirrorUrl(archive.Hash)!
},
Size = archive.Size,
Name = archive.Name,
Hash = archive.Hash
}
};
}
2021-10-23 16:51:17 +00:00
}
2021-09-27 12:42:46 +00:00
2021-10-23 16:51:17 +00:00
if (result.Status == ArchiveStatus.InValid)
{
2021-12-18 16:14:39 +00:00
_logger.LogInformation("Looking for patch for {Hash}", archive.Hash);
foreach (var file in patchFiles.Where(p => p.Original.Hash == archive.Hash && p.Status == ArchiveStatus.Updated))
2021-09-27 12:42:46 +00:00
{
2021-12-18 16:14:39 +00:00
if (await _dispatcher.Verify(file.PatchedFrom!, token))
{
return new ValidatedArchive()
{
Original = archive,
Status = ArchiveStatus.Updated,
PatchUrl = file.PatchUrl,
PatchedFrom = file.PatchedFrom
};
}
}
2021-10-23 16:51:17 +00:00
}
2021-12-18 16:14:39 +00:00
return new ValidatedArchive()
2021-10-23 16:51:17 +00:00
{
2021-12-18 16:14:39 +00:00
Status = ArchiveStatus.InValid,
Original = archive
2021-10-23 16:51:17 +00:00
};
2021-09-27 12:42:46 +00:00
}).ToArray();
2021-10-23 16:51:17 +00:00
validatedList.Archives = archives;
validatedList.Status = archives.Any(a => a.Status == ArchiveStatus.InValid)
? ListStatus.Failed
: ListStatus.Available;
2022-06-22 20:30:43 +00:00
try
{
var (smallImage, largeImage) = await ProcessModlistImage(reports, modList, token);
validatedList.SmallImage =
new Uri(
$"https://raw.githubusercontent.com/wabbajack-tools/mod-lists/master/reports/{smallImage.ToString().Replace("\\", "/")}");
validatedList.LargeImage =
new Uri(
$"https://raw.githubusercontent.com/wabbajack-tools/mod-lists/master/reports/{largeImage.ToString().Replace("\\", "/")}");
}
catch (Exception ex)
{
_logger.LogError(ex, "While processing modlist images for {MachineURL}", modList.NamespacedName);
}
2021-10-23 16:51:17 +00:00
return validatedList;
}).ToArray();
var allArchives = validatedLists.SelectMany(l => l.Archives).ToList();
2021-12-17 17:19:50 +00:00
_logger.LogInformation("Validated {Count} lists in {Elapsed}", validatedLists.Length, stopWatch.Elapsed);
_logger.LogInformation(" - {Count} Valid", allArchives.Count(a => a.Status is ArchiveStatus.Valid));
_logger.LogInformation(" - {Count} Invalid", allArchives.Count(a => a.Status is ArchiveStatus.InValid));
_logger.LogInformation(" - {Count} Mirrored", allArchives.Count(a => a.Status is ArchiveStatus.Mirrored));
_logger.LogInformation(" - {Count} Updated", allArchives.Count(a => a.Status is ArchiveStatus.Updated));
2021-10-23 16:51:17 +00:00
foreach (var invalid in allArchives.Where(a => a.Status is ArchiveStatus.InValid)
.DistinctBy(a => a.Original.Hash))
{
_logger.LogInformation("-- Invalid {Hash}: {PrimaryKeyString}", invalid.Original.Hash.ToHex(),
invalid.Original.State.PrimaryKeyString);
}
2021-09-27 12:42:46 +00:00
2021-10-23 16:51:17 +00:00
await ExportReports(reports, validatedLists, token);
2021-12-19 22:20:24 +00:00
var usedMirroredFiles = validatedLists.SelectMany(a => a.Archives)
.Where(m => m.Status == ArchiveStatus.Mirrored)
.Select(m => m.Original.Hash)
.ToHashSet();
await DeleteOldMirrors(mirroredFiles, usedMirroredFiles);
2021-09-27 12:42:46 +00:00
2021-10-23 16:51:17 +00:00
return 0;
}
2021-09-27 12:42:46 +00:00
2022-05-15 04:33:17 +00:00
private async Task<(RelativePath SmallImage, RelativePath LargeImage)> ProcessModlistImage(AbsolutePath reports, ModlistMetadata validatedList,
CancellationToken token)
{
_logger.LogInformation("Processing Modlist Image for {MachineUrl}", validatedList.NamespacedName);
var baseFolder = reports.Combine(validatedList.NamespacedName);
baseFolder.CreateDirectory();
await using var imageStream = await _httpClient.GetStreamAsync(validatedList.Links.ImageUri, token);
var ms = new MemoryStream();
var hash = await imageStream.HashingCopy(ms, token);
2022-05-15 04:33:17 +00:00
RelativePath smallImage, largeImage;
2022-05-15 04:40:40 +00:00
ms.Position = 0;
2022-05-15 04:33:17 +00:00
// Large Image
{
var standardWidth = 1152;
using var image = await Image.LoadAsync(ms, token);
var height = standardWidth * image.Height / image.Width;
image.Mutate(x => x
.Resize(standardWidth, height));
largeImage = validatedList.RepositoryName.ToRelativePath().Combine(hash.ToHex()+"_large").WithExtension(Ext.Webp);
2022-05-15 12:47:51 +00:00
await image.SaveAsync(largeImage.RelativeTo(reports).ToString(), new WebpEncoder {Quality = 85}, cancellationToken: token);
2022-05-15 04:33:17 +00:00
}
2022-05-15 04:40:40 +00:00
ms.Position = 0;
2022-05-15 04:33:17 +00:00
// Small Image
{
var standardWidth = 466;
using var image = await Image.LoadAsync(ms, token);
var height = standardWidth * image.Height / image.Width;
image.Mutate(x => x
.Resize(standardWidth, height));
smallImage = validatedList.RepositoryName.ToRelativePath().Combine(hash.ToHex()+"_small").WithExtension(Ext.Webp);
2022-05-15 12:47:51 +00:00
await image.SaveAsync(smallImage.RelativeTo(reports).ToString(), new WebpEncoder {Quality = 75}, cancellationToken: token);
2022-05-15 04:33:17 +00:00
}
return (smallImage, largeImage);
}
private async Task SendDefinitionToLoadOrderLibrary(ModlistMetadata metadata, ModList modListData, CancellationToken token)
{
var lolGame = modListData.GameType switch
{
Game.Morrowind => 1,
Game.Oblivion => 2,
Game.Skyrim => 3,
Game.SkyrimSpecialEdition => 4,
Game.SkyrimVR => 5,
Game.Fallout3 => 6,
Game.FalloutNewVegas => 7,
Game.Fallout4 => 8,
Game.Fallout4VR => 9,
_ => 0
};
if (lolGame == 0) return;
var files = (await GetFiles(modListData, metadata, token))
.Where(f => f.Key.Depth == 3)
.Where(f => f.Key.Parent.Parent == "profiles".ToRelativePath())
.GroupBy(f => f.Key.Parent.FileName.ToString())
.ToArray();
foreach (var profile in files)
{
var formData = new MultipartFormDataContent();
if (files.Length > 1)
{
formData.Add(new StringContent(modListData.Name + $"({metadata.NamespacedName})"), "name");
}
else
{
formData.Add(new StringContent(modListData.Name + $" - Profile: {profile.Key} ({metadata.NamespacedName})"), "name");
}
formData.Add(new StringContent(lolGame.ToString()), "game_id");
formData.Add(new StringContent(metadata.Description), "description");
formData.Add(new StringContent((metadata.Version ?? Version.Parse("0.0.0.0")).ToString()), "version");
formData.Add(new StringContent("0"), "is_private");
2022-04-18 17:28:33 +00:00
formData.Add(new StringContent("perm"), "expires_at");
if (modListData.Website != null)
{
formData.Add(new StringContent(modListData.Website!.ToString()), "website");
}
if (metadata.Links.DiscordURL != null)
{
formData.Add(new StringContent(metadata.Links.DiscordURL), "discord");
}
if (metadata.Links.Readme != null)
{
formData.Add(new StringContent(metadata.Links.Readme), "readme");
}
foreach (var file in profile)
{
formData.Add(new ByteArrayContent(file.Value), "files[]", file.Key.FileName.ToString());
}
using var job = await _httpLimiter.Begin("Posting to load order library", 0, token);
var msg = new HttpRequestMessage(HttpMethod.Post, "https://api.loadorderlibrary.com/v1/lists");
msg.Content = formData;
using var result = await _httpClient.SendAsync(msg, token);
if (result.IsSuccessStatusCode)
return;
//var data = await result.Content.ReadFromJsonAsync<string>(token);
}
}
private static HashSet<RelativePath> LoadOrderFiles = new HashSet<string>()
{
"enblocal.ini",
"enbseries.ini",
"fallout.ini",
"falloutprefs.ini",
"fallout4.ini",
"fallout4custom.ini",
"fallout4prefs.ini",
"falloutcustom.ini",
"geckcustom.ini",
"geckprefs.ini",
"loadorder.txt",
"mge.ini",
"modlist.txt",
"morrowind.ini",
"mwse-version.ini",
"oblivion.ini",
"oblivionprefs.ini",
"plugins.txt",
"settings.ini",
"skyrim.ini",
"skyrimcustom.ini",
"skyrimprefs.ini",
"skyrimvr.ini",
}.Select(f => f.ToRelativePath()).ToHashSet();
private async Task<Dictionary<RelativePath, byte[]>> GetFiles(ModList modlist, ModlistMetadata metadata, CancellationToken token)
{
var archive = new Archive
{
State = _dispatcher.Parse(new Uri(metadata.Links.Download))!,
Size = metadata.DownloadMetadata!.Size,
Hash = metadata.DownloadMetadata.Hash
};
var stream = await _dispatcher.ChunkedSeekableStream(archive, token);
await using var reader = new ZipReader(stream);
var files = await reader.GetFiles();
var indexed = files.ToDictionary(f => f.FileName.ToRelativePath());
var entriesToGet = modlist.Directives.OfType<InlineFile>()
.Where(f => LoadOrderFiles.Contains(f.To.FileName))
.Select(f => (f, indexed[f.SourceDataID]))
.ToArray();
var fileData = new Dictionary<RelativePath, byte[]>();
foreach (var entry in entriesToGet)
{
var ms = new MemoryStream();
await reader.Extract(entry.Item2, ms, token);
fileData.Add(entry.f.To, ms.ToArray());
}
return fileData;
}
2021-09-27 12:42:46 +00:00
2021-10-23 16:51:17 +00:00
private async Task ExportReports(AbsolutePath reports, ValidatedModList[] validatedLists, CancellationToken token)
{
foreach (var validatedList in validatedLists)
2021-09-27 12:42:46 +00:00
{
2021-10-23 16:51:17 +00:00
var baseFolder = reports.Combine(validatedList.MachineURL);
baseFolder.CreateDirectory();
await using var jsonFile = baseFolder.Combine("status").WithExtension(Ext.Json)
.Open(FileMode.Create, FileAccess.Write, FileShare.None);
await _dtos.Serialize(validatedList, jsonFile, true);
await using var mdFile = baseFolder.Combine("status").WithExtension(Ext.Md)
.Open(FileMode.Create, FileAccess.Write, FileShare.None);
await using var sw = new StreamWriter(mdFile);
await sw.WriteLineAsync($"## Validation Report - {validatedList.Name} ({validatedList.MachineURL})");
await sw.WriteAsync("\n\n");
async Task WriteSection(TextWriter w, ArchiveStatus status, string sectionName)
2021-09-27 12:42:46 +00:00
{
2021-10-23 16:51:17 +00:00
var archives = validatedList.Archives.Where(a => a.Status == status).ToArray();
await w.WriteLineAsync($"### {sectionName} ({archives.Length})");
2021-09-27 12:42:46 +00:00
2021-10-23 16:51:17 +00:00
foreach (var archive in archives.OrderBy(a => a.Original.Name))
{
if (_dispatcher.TryGetDownloader(archive.Original, out var downloader) &&
downloader is IUrlDownloader u)
2021-09-27 12:42:46 +00:00
{
2021-10-23 16:51:17 +00:00
await w.WriteLineAsync(
$"* [{archive.Original.Name}]({u.UnParse(archive.Original.State)})");
}
else
{
await w.WriteLineAsync(
$"* {archive.Original.Name}");
2021-09-27 12:42:46 +00:00
}
}
2021-10-23 16:51:17 +00:00
}
2021-09-27 12:42:46 +00:00
2021-10-23 16:51:17 +00:00
await WriteSection(sw, ArchiveStatus.InValid, "Invalid");
await WriteSection(sw, ArchiveStatus.Updated, "Updated");
await WriteSection(sw, ArchiveStatus.Mirrored, "Mirrored");
await WriteSection(sw, ArchiveStatus.Valid, "Valid");
2021-09-27 12:42:46 +00:00
2021-10-23 16:51:17 +00:00
try
{
var oldSummary = await _wjClient.GetDetailedStatus(validatedList.MachineURL);
if (oldSummary.ModListHash != validatedList.ModListHash)
2021-09-27 12:42:46 +00:00
{
try
{
await SendDefinitionToLoadOrderLibrary(validatedList, token);
}
catch (Exception ex)
{
_logger.LogError("While uploading to load order library", ex);
}
2021-10-23 16:51:17 +00:00
await _discord.SendAsync(Channel.Ham,
$"Finished processing {validatedList.Name} ({validatedList.MachineURL}) v{validatedList.Version} ({oldSummary.ModListHash} -> {validatedList.ModListHash})",
token);
}
2021-09-27 12:42:46 +00:00
2021-10-23 16:51:17 +00:00
if (oldSummary.Failures != validatedList.Failures)
{
if (validatedList.Failures == 0)
2021-09-27 12:42:46 +00:00
{
await _discord.SendAsync(Channel.Ham,
2021-10-23 16:51:17 +00:00
new DiscordMessage
{
Embeds = new[]
2021-09-27 12:42:46 +00:00
{
2021-10-23 16:51:17 +00:00
new DiscordEmbed
2021-09-27 12:42:46 +00:00
{
2021-10-23 16:51:17 +00:00
Title =
$"{validatedList.Name} (`{validatedList.MachineURL}`) is now passing.",
Url = new Uri(
$"https://github.com/wabbajack-tools/mod-lists/blob/master/reports/{validatedList.MachineURL}/status.md")
2021-09-27 12:42:46 +00:00
}
2021-10-23 16:51:17 +00:00
}
}, token);
}
else
{
await _discord.SendAsync(Channel.Ham,
new DiscordMessage
{
Embeds = new[]
2021-09-27 12:42:46 +00:00
{
2021-10-23 16:51:17 +00:00
new DiscordEmbed
2021-09-27 12:42:46 +00:00
{
2021-10-23 16:51:17 +00:00
Title =
$"Number of failures in {validatedList.Name} (`{validatedList.MachineURL}`) was {oldSummary.Failures} is now {validatedList.Failures}",
Url = new Uri(
$"https://github.com/wabbajack-tools/mod-lists/blob/master/reports/{validatedList.MachineURL}/status.md")
2021-09-27 12:42:46 +00:00
}
2021-10-23 16:51:17 +00:00
}
}, token);
2021-09-27 12:42:46 +00:00
}
}
}
2021-10-23 16:51:17 +00:00
catch (Exception ex)
2021-09-27 12:42:46 +00:00
{
2021-10-23 16:51:17 +00:00
_logger.LogCritical(ex, "While sending discord message for {MachineURl}", validatedList.MachineURL);
}
2021-09-27 12:42:46 +00:00
}
2021-10-23 16:51:17 +00:00
var summaries = validatedLists.Select(l => new ModListSummary
{
Failed = l.Archives.Count(f => f.Status == ArchiveStatus.InValid),
Mirrored = l.Archives.Count(f => f.Status == ArchiveStatus.Mirrored),
Passed = l.Archives.Count(f => f.Status == ArchiveStatus.Valid),
MachineURL = l.MachineURL,
Name = l.Name,
2022-05-15 03:05:55 +00:00
Updating = 0,
2022-05-15 04:33:17 +00:00
SmallImage = l.SmallImage,
LargeImage = l.LargeImage
2021-10-23 16:51:17 +00:00
}).ToArray();
await using var summaryFile = reports.Combine("modListSummary.json")
.Open(FileMode.Create, FileAccess.Write, FileShare.None);
await _dtos.Serialize(summaries, summaryFile, true);
var upgradedMetas = validatedLists.SelectMany(v => v.Archives)
.Where(a => a.Status is ArchiveStatus.Mirrored or ArchiveStatus.Updated)
.DistinctBy(a => a.Original.Hash)
.OrderBy(a => a.Original.Hash)
.ToArray();
await using var upgradedMetasFile = reports.Combine("upgraded.json")
.Open(FileMode.Create, FileAccess.Write, FileShare.None);
await _dtos.Serialize(upgradedMetas, upgradedMetasFile, true);
2021-12-19 22:20:24 +00:00
}
private async Task SendDefinitionToLoadOrderLibrary(ValidatedModList validatedModList, CancellationToken token)
{
var modlistMetadata = (await _wjClient.LoadLists())
.First(l => l.NamespacedName == validatedModList.MachineURL);
var modList = await StandardInstaller.Load(_dtos, _dispatcher, modlistMetadata, token);
await SendDefinitionToLoadOrderLibrary(modlistMetadata, modList, token);
}
2021-12-19 22:20:24 +00:00
private async Task DeleteOldMirrors(IEnumerable<Hash> mirroredFiles, IReadOnlySet<Hash> usedMirroredFiles)
{
foreach (var file in mirroredFiles.Where(file => !usedMirroredFiles.Contains(file)))
{
await _wjClient.DeleteMirror(file);
}
2021-10-23 16:51:17 +00:00
}
2021-09-27 12:42:46 +00:00
2022-01-03 04:44:16 +00:00
private async Task<(ArchiveStatus, Archive)> DownloadAndValidate(Archive archive,
ILookup<Hash, ForcedRemoval> forcedRemovals, CancellationToken token)
2021-10-23 16:51:17 +00:00
{
2022-01-03 04:44:16 +00:00
if (forcedRemovals.Contains(archive.Hash))
return (ArchiveStatus.InValid, archive);
2021-10-23 16:51:17 +00:00
switch (archive.State)
{
case GameFileSource:
return (ArchiveStatus.Valid, archive);
case Manual:
return (ArchiveStatus.Valid, archive);
case TESAlliance:
return (ArchiveStatus.Valid, archive);
2021-11-14 00:30:05 +00:00
case Mega:
return (ArchiveStatus.Valid, archive);
2021-11-28 02:20:34 +00:00
case Nexus:
return (ArchiveStatus.Valid, archive);
2021-12-18 17:59:47 +00:00
case VectorPlexus:
2022-06-22 20:30:43 +00:00
return (ArchiveStatus.InValid, archive);
2021-10-23 16:51:17 +00:00
}
if (archive.State is Http http && http.Url.Host.EndsWith("github.com"))
return (ArchiveStatus.Valid, archive);
2021-12-18 17:18:27 +00:00
2021-10-23 16:51:17 +00:00
try
2021-09-27 12:42:46 +00:00
{
2021-10-23 16:51:17 +00:00
for (var attempts = 0; attempts < 3; attempts++)
2021-09-27 12:42:46 +00:00
{
2021-10-23 16:51:17 +00:00
var valid = await _dispatcher.Verify(archive, token);
if (valid)
return (ArchiveStatus.Valid, archive);
var delay = _random.Next(200, 1200);
_logger.LogWarning(
"Archive {primaryKeyString} is invalid retrying in {Delay} ms ({Attempt} of {MaxAttempts})",
archive.State.PrimaryKeyString, delay, attempts, 3);
await Task.Delay(delay, token);
2021-09-27 12:42:46 +00:00
}
2021-10-23 16:51:17 +00:00
_logger.LogWarning("Archive {primaryKeyString} is invalid", archive.State.PrimaryKeyString);
return (ArchiveStatus.InValid, archive);
}
catch (Exception ex)
{
_logger.LogCritical(ex, "While verifying {primaryKeyString}", archive.State.PrimaryKeyString);
return (ArchiveStatus.InValid, archive);
2021-09-27 12:42:46 +00:00
}
2021-10-23 16:51:17 +00:00
}
2021-09-27 12:42:46 +00:00
2021-10-23 16:51:17 +00:00
private async Task<Hash> DownloadModList(ModlistMetadata modList, ArchiveManager archiveManager,
CancellationToken token)
{
if (archiveManager.HaveArchive(modList.DownloadMetadata!.Hash))
2021-09-27 12:42:46 +00:00
{
2022-03-30 03:39:48 +00:00
_logger.LogInformation("Previously downloaded {hash} not re-downloading", modList.NamespacedName);
2021-10-23 16:51:17 +00:00
return modList.DownloadMetadata!.Hash;
2021-09-27 12:42:46 +00:00
}
2021-10-23 16:51:17 +00:00
else
2021-09-27 12:42:46 +00:00
{
2022-03-30 03:39:48 +00:00
_logger.LogInformation("Downloading {hash}", modList.NamespacedName);
2021-10-23 16:51:17 +00:00
await _discord.SendAsync(Channel.Ham,
2022-03-30 03:39:48 +00:00
$"Downloading and ingesting {modList.Title} ({modList.NamespacedName}) v{modList.Version}", token);
2021-10-23 16:51:17 +00:00
return await DownloadWabbajackFile(modList, archiveManager, token);
2021-09-27 12:42:46 +00:00
}
2021-10-23 16:51:17 +00:00
}
private async Task<Hash> DownloadWabbajackFile(ModlistMetadata modList, ArchiveManager archiveManager,
CancellationToken token)
{
var state = _dispatcher.Parse(new Uri(modList.Links.Download));
if (state == null)
_logger.LogCritical("Can't download {url}", modList.Links.Download);
2021-09-27 12:42:46 +00:00
2021-10-23 16:51:17 +00:00
var archive = new Archive
2021-09-27 12:42:46 +00:00
{
2021-10-23 16:51:17 +00:00
State = state!,
Size = modList.DownloadMetadata!.Size,
Hash = modList.DownloadMetadata.Hash
};
await using var tempFile = _temporaryFileManager.CreateFile(Ext.Wabbajack);
_logger.LogInformation("Downloading {primaryKeyString}", state.PrimaryKeyString);
var hash = await _dispatcher.Download(archive, tempFile.Path, token);
if (hash != modList.DownloadMetadata.Hash)
2021-09-27 12:42:46 +00:00
{
2021-10-23 16:51:17 +00:00
_logger.LogCritical("Downloaded modlist was {actual} expected {expected}", hash,
modList.DownloadMetadata.Hash);
throw new Exception();
2021-09-27 12:42:46 +00:00
}
2021-10-23 16:51:17 +00:00
_logger.LogInformation("Archiving {hash}", hash);
await archiveManager.Ingest(tempFile.Path, token);
return hash;
2021-09-27 12:42:46 +00:00
}
2022-04-18 17:28:33 +00:00
}