using System; using System.Collections.Generic; using System.Linq; using System.Threading.Tasks; using Alphaleonis.Win32.Filesystem; using Wabbajack.Common; using Wabbajack.Lib; using Wabbajack.Lib.Downloaders; using Wabbajack.Lib.NexusApi; using Wabbajack.VirtualFileSystem; using Xunit; using Xunit.Abstractions; namespace Wabbajack.Test { public class EndToEndTests : XunitContextBase, IDisposable { private AbsolutePath _downloadFolder = "downloads".RelativeTo(AbsolutePath.EntryPoint); private TestUtils utils = new TestUtils(); private IDisposable _unsub; public WorkQueue Queue { get; set; } public EndToEndTests(ITestOutputHelper helper) : base(helper) { Queue = new WorkQueue(); Consts.TestMode = true; utils = new TestUtils(); utils.Game = Game.SkyrimSpecialEdition; _unsub = Utils.LogMessages.Subscribe(f => XunitContext.WriteLine($"{DateTime.Now} - {f}")); _downloadFolder.CreateDirectory(); } public override void Dispose() { Queue.Dispose(); _unsub.Dispose(); base.Dispose(); } [Fact] public async Task CreateModlist() { var profile = utils.AddProfile("Default"); var mod = utils.AddMod(); await DownloadAndInstall( "https://github.com/ModOrganizer2/modorganizer/releases/download/v2.2.1/Mod.Organizer.2.2.1.7z", "Mod.Organizer.2.2.1.7z"); await utils.DownloadsFolder.Combine("Mod.Organizer.2.2.1.7z.meta").WriteAllLinesAsync( "[General]", "directURL=https://github.com/ModOrganizer2/modorganizer/releases/download/v2.2.1/Mod.Organizer.2.2.1.7z" ); var modfiles = await Task.WhenAll( DownloadAndInstall(Game.SkyrimSpecialEdition, 12604, "SkyUI"), DownloadAndInstall(Game.Fallout4, 11925, "Anti-Tank Rifle"), DownloadAndInstall(Game.SkyrimSpecialEdition, 4783, "Frost Armor UNP"), DownloadAndInstall(Game.SkyrimSpecialEdition, 32359, "Frost Armor HDT")); // We're going to fully patch this mod from another source. modfiles[3].Download.Delete(); await utils.Configure(); await modfiles[3].ModFolder.Combine("meta.ini").WriteAllLinesAsync( "[General]", $"matchAll= {modfiles[2].Download.FileName}" ); await utils.MO2Folder.Combine("startup.bat").WriteAllLinesAsync( "ModOrganizer2.exe SKSE" ); await CompileAndInstall(profile); utils.VerifyAllFiles(); await utils.InstallFolder.Combine(Consts.LOOTFolderFilesDir).DeleteDirectory(); var compiler = new MO2Compiler( mo2Folder: utils.InstallFolder, mo2Profile: profile, outputFile: profile.RelativeTo(AbsolutePath.EntryPoint).WithExtension(Consts.ModListExtension)); compiler.MO2DownloadsFolder = utils.DownloadsFolder; Assert.True(await compiler.Begin()); } private async Task DownloadAndInstall(string url, string filename, string modName = null) { var src = _downloadFolder.Combine(filename); if (!src.Exists) { var state = DownloadDispatcher.ResolveArchive(url); await state.Download(new Archive { Name = "Unknown"}, src); } utils.DownloadsFolder.CreateDirectory(); await src.CopyToAsync(utils.DownloadsFolder.Combine(filename)); await FileExtractor.ExtractAll(Queue, src, modName == null ? utils.MO2Folder : utils.ModsFolder.Combine(modName)); } private async Task<(AbsolutePath Download, AbsolutePath ModFolder)> DownloadAndInstall(Game game, int modId, string modName) { utils.AddMod(modName); var client = await NexusApiClient.Get(); var resp = await client.GetModFiles(game, modId); var file = resp.files.FirstOrDefault(f => f.is_primary) ?? resp.files.FirstOrDefault(f => !string.IsNullOrEmpty(f.category_name)); var src = _downloadFolder.Combine(file.file_name); var ini = string.Join("\n", new List { "[General]", $"gameName={game.MetaData().MO2ArchiveName}", $"modID={modId}", $"fileID={file.file_id}" }); if (!src.Exists) { var state = (AbstractDownloadState)await DownloadDispatcher.ResolveArchive(ini.LoadIniString()); await state.Download(src); } utils.DownloadsFolder.CreateDirectory(); var dest = utils.DownloadsFolder.Combine(file.file_name); await src.CopyToAsync(dest); var modFolder = utils.ModsFolder.Combine(modName); await FileExtractor.ExtractAll(Queue, src, modFolder); await dest.WithExtension(Consts.MetaFileExtension).WriteAllTextAsync(ini); return (dest, modFolder); } private async Task CompileAndInstall(string profile) { var compiler = await ConfigureAndRunCompiler(profile); await Install(compiler); return compiler.ModList; } private async Task Install(MO2Compiler compiler) { var modlist = AInstaller.LoadFromFile(compiler.ModListOutputFile); var installer = new MO2Installer( archive: compiler.ModListOutputFile, modList: modlist, outputFolder: utils.InstallFolder, downloadFolder: utils.DownloadsFolder, parameters: ACompilerTest.CreateDummySystemParameters()); installer.GameFolder = utils.GameFolder; await installer.Begin(); } private async Task ConfigureAndRunCompiler(string profile) { var compiler = new MO2Compiler( mo2Folder: utils.MO2Folder, mo2Profile: profile, outputFile: profile.RelativeTo(AbsolutePath.EntryPoint).WithExtension(Consts.ModListExtension)); Assert.True(await compiler.Begin()); return compiler; } } }