using System; using System.Linq; using System.Threading.Tasks; using Alphaleonis.Win32.Filesystem; using MongoDB.Driver; using MongoDB.Driver.Linq; using Wabbajack.BuildServer.Model.Models; using Wabbajack.BuildServer.Models.JobQueue; using Wabbajack.Common; using Wabbajack.Lib; using Wabbajack.Lib.Downloaders; using Wabbajack.Lib.ModListRegistry; using Wabbajack.Lib.NexusApi; using Wabbajack.Lib.Validation; using File = Alphaleonis.Win32.Filesystem.File; namespace Wabbajack.BuildServer.Models.Jobs { public class UpdateModLists : AJobPayload, IFrontEndJob { public override string Description => "Validate curated modlists"; public override async Task Execute(DBContext db, SqlService sql, AppSettings settings) { Utils.Log("Starting Modlist Validation"); var modlists = await ModlistMetadata.LoadFromGithub(); using (var queue = new WorkQueue()) { var whitelists = new ValidateModlist(queue); await whitelists.LoadListsFromGithub(); foreach (var list in modlists) { try { await ValidateList(db, list, queue, whitelists); } catch (Exception ex) { Utils.Log(ex.ToString()); } } } return JobResult.Success(); } private async Task ValidateList(DBContext db, ModlistMetadata list, WorkQueue queue, ValidateModlist whitelists) { var modlist_path = Path.Combine(Consts.ModListDownloadFolder, list.Links.MachineURL + Consts.ModListExtension); if (list.NeedsDownload(modlist_path)) { if (File.Exists(modlist_path)) File.Delete(modlist_path); var state = DownloadDispatcher.ResolveArchive(list.Links.Download); Utils.Log($"Downloading {list.Links.MachineURL} - {list.Title}"); await state.Download(modlist_path); } else { Utils.Log($"No changes detected from downloaded modlist"); } Utils.Log($"Loading {modlist_path}"); var installer = AInstaller.LoadFromFile(modlist_path); Utils.Log($"{installer.Archives.Count} archives to validate"); DownloadDispatcher.PrepareAll(installer.Archives.Select(a => a.State)); var validated = (await installer.Archives .PMap(queue, async archive => { var isValid = await IsValid(db, whitelists, archive); return new DetailedStatusItem {IsFailing = !isValid, Archive = archive}; })) .ToList(); var status = new DetailedStatus { Name = list.Title, Archives = validated.OrderBy(v => v.Archive.Name).ToList(), DownloadMetaData = list.DownloadMetadata, HasFailures = validated.Any(v => v.IsFailing) }; var dto = new ModListStatus { Id = list.Links.MachineURL, Summary = new ModlistSummary { Name = status.Name, MachineURL = list.Links?.MachineURL ?? status.Name, Checked = status.Checked, Failed = status.Archives.Count(a => a.IsFailing), Passed = status.Archives.Count(a => !a.IsFailing), }, DetailedStatus = status, Metadata = list }; Utils.Log( $"Writing Update for {dto.Summary.Name} - {dto.Summary.Failed} failed - {dto.Summary.Passed} passed"); await ModListStatus.Update(db, dto); Utils.Log( $"Done updating {dto.Summary.Name}"); } private async Task IsValid(DBContext db, ValidateModlist whitelists, Archive archive) { try { if (!archive.State.IsWhitelisted(whitelists.ServerWhitelist)) return false; try { if (archive.State is NexusDownloader.State state) { if (await ValidateNexusFast(db, state)) return true; } else if (archive.State is GoogleDriveDownloader.State) { // Disabled for now return true; } else if (archive.State is HTTPDownloader.State hstate && hstate.Url.StartsWith("https://wabbajack")) { return true; } else { if (await archive.State.Verify(archive)) return true; } } catch (Exception) { // ignored } Utils.Log($"{archive.State.PrimaryKeyString} is broken, looking for upgrade: {archive.Name}"); var result = await ClientAPI.GetModUpgrade(archive.Hash); if (result != null) { Utils.Log($"{archive.State.PrimaryKeyString} is broken, upgraded to {result.State.PrimaryKeyString} {result.Name}"); return true; } Utils.Log($"{archive.State.PrimaryKeyString} is broken, no alternative found"); return false; } catch (Exception ex) { Utils.Log(ex.ToString()); return false; } return false; } private async Task ValidateNexusFast(DBContext db, NexusDownloader.State state) { try { var gameMeta = GameRegistry.GetByFuzzyName(state.GameName); if (gameMeta == null) return false; var game = gameMeta.Game; if (!int.TryParse(state.ModID, out var modID)) return false; var modFiles = (await db.NexusModFiles.AsQueryable().Where(g => g.Game == gameMeta.NexusName && g.ModId == state.ModID).FirstOrDefaultAsync())?.Data; if (modFiles == null) { Utils.Log($"No Cache for {state.PrimaryKeyString} falling back to HTTP"); var nexusApi = await NexusApiClient.Get(); modFiles = await nexusApi.GetModFiles(game, modID); } if (!ulong.TryParse(state.FileID, out var fileID)) return false; var found = modFiles.files .FirstOrDefault(file => file.file_id == fileID && file.category_name != null); return found != null; } catch (Exception ex) { return false; } } } }