Mirrored_Repos/wabbajack
1
0
Fork 0
mirror of https://github.com/wabbajack-tools/wabbajack.git synced 2024-08-30 18:42:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Whole app now useses Async IO exclusively

Browse Source
This commit is contained in:
Timothy Baldridge 2020-05-25 11:34:25 -06:00
parent 191f321dc2
commit b2a112bd37
37 changed files with 218 additions and 213 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
Compression.BSA.Test/BSATests.cs
View File

@ -96,16 +96,16 @@ namespace Compression.BSA.Test
var tempFile = ((RelativePath)"tmp.bsa").RelativeToEntryPoint();
var size = bsa.Size;
await using var a = BSADispatch.OpenRead(bsa);
await a.Files.PMap(Queue, file =>
await using var a = await BSADispatch.OpenRead(bsa);
await a.Files.PMap(Queue, async file =>
{
var absName = _tempDir.Combine(file.Path);
ViaJson(file.State);
absName.Parent.CreateDirectory();
using (var fs = absName.Create())
await using (var fs = await absName.Create())
{
file.CopyDataTo(fs);
await file.CopyDataTo(fs);
}
Assert.Equal(file.Size, absName.Size);
@ -123,7 +123,7 @@ namespace Compression.BSA.Test
var streams = await a.Files.PMap(Queue, async file =>
{
var absPath = _tempDir.Combine(file.Path);
var str = absPath.OpenRead();
var str = await absPath.OpenRead();
await w.AddFile(ViaJson(file.State), str);
return str;
});
@ -132,7 +132,7 @@ namespace Compression.BSA.Test
}
TestContext.WriteLine($"Verifying {bsa}");
await using var b = BSADispatch.OpenRead(tempFile);
await using var b = await BSADispatch.OpenRead(tempFile);
TestContext.WriteLine($"Performing A/B tests on {bsa}");
Assert.Equal(a.State.ToJson(), b.State.ToJson());

8
Wabbajack.App.Test/SettingsTests.cs
View File

@ -8,16 +8,16 @@ namespace Wabbajack.Test
[Fact]
public async Task CanSaveAndLoadSettings()
{
MainSettings.TryLoadTypicalSettings(out var settings);
var (settings, loadedSettings) = await MainSettings.TryLoadTypicalSettings();
if (settings == null)
if (settings == null || !loadedSettings)
{
settings = new MainSettings();
}
MainSettings.SaveSettings(settings);
await MainSettings.SaveSettings(settings);
Assert.True(MainSettings.TryLoadTypicalSettings(out settings));
Assert.True((await MainSettings.TryLoadTypicalSettings()).loaded);
}
}

4
Wabbajack.CLI/Verbs/BSADump.cs
View File

@ -14,8 +14,8 @@ namespace Wabbajack.CLI.Verbs
protected override async Task<ExitCode> Run()
{
await using var bsa = BSADispatch.OpenRead(Input.RelativeTo(AbsolutePath.GetCurrentDirectory()));
bsa.Dump(line => Console.WriteLine(line));
await using var bsa = await BSADispatch.OpenRead(Input.RelativeTo(AbsolutePath.GetCurrentDirectory()));
bsa.Dump(Console.WriteLine);
return ExitCode.Ok;
}
}

2
Wabbajack.CLI/Verbs/Encrypt.cs
View File

@ -18,7 +18,7 @@ namespace Wabbajack.CLI.Verbs
protected override async Task<ExitCode> Run()
{
File.ReadAllBytes(Input).ToEcryptedData(Name);
await File.ReadAllBytes(Input).ToEcryptedData(Name);
return 0;
}
}

28
Wabbajack.Lib/ACompiler.cs
View File

@ -81,10 +81,10 @@ namespace Wabbajack.Lib
return id;
}
internal FileStream IncludeFile(out RelativePath id)
internal AbsolutePath IncludeFile(out RelativePath id)
{
id = IncludeId();
return ModListOutputFolder.Combine(id).Create();
return ModListOutputFolder.Combine(id);
}
internal async Task<RelativePath> IncludeFile(string data)
@ -144,34 +144,32 @@ namespace Wabbajack.Lib
ModList.Image = (RelativePath)"modlist-image.png";
}
using (var of = ModListOutputFolder.Combine("modlist").Create())
using (var of = await ModListOutputFolder.Combine("modlist").Create())
ModList.ToJson(of);
ModListOutputFile.Delete();
using (var fs = ModListOutputFile.Create())
using (var fs = await ModListOutputFile.Create())
{
using (var za = new ZipArchive(fs, ZipArchiveMode.Create))
{
ModListOutputFolder.EnumerateFiles()
await ModListOutputFolder.EnumerateFiles()
.DoProgress("Compressing ModList",
f =>
async f =>
{
var ze = za.CreateEntry((string)f.FileName);
using var os = ze.Open();
using var ins = f.OpenRead();
ins.CopyTo(os);
await using var os = ze.Open();
await using var ins = await f.OpenRead();
await ins.CopyToAsync(os);
});
// Copy in modimage
if (ModListImage.Exists)
{
var ze = za.CreateEntry((string)ModList.Image);
using (var os = ze.Open())
using (var ins = ModListImage.OpenRead())
{
ins.CopyTo(os);
}
await using var os = ze.Open();
await using var ins = await ModListImage.OpenRead();
await ins.CopyToAsync(os);
}
}
}
@ -180,7 +178,7 @@ namespace Wabbajack.Lib
var metadata = new DownloadMetadata
{
Size = ModListOutputFile.Size,
Hash = ModListOutputFile.FileHash(),
Hash = await ModListOutputFile.FileHashAsync(),
NumberOfArchives = ModList.Archives.Count,
SizeOfArchives = ModList.Archives.Sum(a => a.Size),
NumberOfInstalledFiles = ModList.Directives.Count,

4
Wabbajack.Lib/AInstaller.cs
View File

@ -71,7 +71,7 @@ namespace Wabbajack.Lib
public async Task<byte[]> LoadBytesFromPath(RelativePath path)
{
await using var e = ExtractedModListFiles![path].OpenRead();
await using var e = await ExtractedModListFiles![path].OpenRead();
return await e.ReadAllAsync();
}
@ -213,7 +213,7 @@ namespace Wabbajack.Lib
toFile.Delete();
// Patch it
await using (var outStream = toFile.Create())
await using (var outStream = await toFile.Create())
{
Utils.ApplyPatch(oldData, () => new MemoryStream(patchData), outStream);
}

4
Wabbajack.Lib/CompilationSteps/DeconstructBSAs.cs
View File

@ -62,7 +62,7 @@ namespace Wabbajack.Lib.CompilationSteps
if (source.AbsolutePath.Size >= (long) 2 << 31)
{
await using var bsa = BSADispatch.OpenRead(source.AbsolutePath);
await using var bsa = await BSADispatch.OpenRead(source.AbsolutePath);
if (bsa.State is BSAStateObject)
{
Utils.Error(
@ -95,7 +95,7 @@ namespace Wabbajack.Lib.CompilationSteps
}
CreateBSA directive;
await using (var bsa = BSADispatch.OpenRead(source.AbsolutePath))
await using (var bsa = await BSADispatch.OpenRead(source.AbsolutePath))
{
directive = new CreateBSA(
state: bsa.State,

2
Wabbajack.Lib/CompilationSteps/IncludeAll.cs
View File

@ -14,7 +14,7 @@ namespace Wabbajack.Lib.CompilationSteps
public override async ValueTask<Directive?> Run(RawSourceFile source)
{
var inline = source.EvolveTo<InlineFile>();
await using var file = source.File.StagedFile.OpenRead();
await using var file = await source.File.StagedFile.OpenRead();
inline.SourceDataID = await _compiler.IncludeFile(await file.ReadAllAsync());
return inline;
}

2
Wabbajack.Lib/Downloaders/AbstractIPS4Downloader.cs
View File

@ -124,7 +124,7 @@ namespace Wabbajack.Lib.Downloaders
using var stream = await ResolveDownloadStream(a);
if (stream == null) return false;
await using var fromStream = await stream.Content.ReadAsStreamAsync();
await using var toStream = destination.Create();
await using var toStream = await destination.Create();
await fromStream.CopyToAsync(toStream);
return true;
}

2
Wabbajack.Lib/Downloaders/AbstractNeedsLoginDownloader.cs
View File

@ -81,7 +81,7 @@ namespace Wabbajack.Lib.Downloaders
await Task.Delay(500, cancel);
}
cookies.ToEcryptedJson(_encryptedKeyName);
await cookies.ToEcryptedJson(_encryptedKeyName);
return cookies;
}

6
Wabbajack.Lib/Downloaders/BethesdaNetDownloader.cs
View File

@ -110,7 +110,7 @@ namespace Wabbajack.Lib.Downloaders
try
{
var result = last_line.FromJsonString<BethesdaNetData>();
result.ToEcryptedJson(DataName);
await result.ToEcryptedJson(DataName);
return result;
}
catch (Exception ex)
@ -180,7 +180,7 @@ namespace Wabbajack.Lib.Downloaders
private const uint CKM_Magic = 0x52415442; // BTAR
private async Task ConvertCKMToZip(AbsolutePath src, AbsolutePath dest)
{
using var reader = new BinaryReader(src.OpenRead());
using var reader = new BinaryReader(await src.OpenRead());
var magic = reader.ReadUInt32();
if (magic != CKM_Magic)
throw new InvalidDataException("Invalid magic format in CKM parsing");
@ -193,7 +193,7 @@ namespace Wabbajack.Lib.Downloaders
if (minorVersion < 2 || minorVersion > 4)
throw new InvalidDataException("Archive minor version is unknown. Should be 2, 3, or 4.");
await using var fos = dest.Create();
await using var fos = await dest.Create();
using var archive = new ZipArchive(fos, ZipArchiveMode.Create);
while (reader.PeekChar() != -1)
{

2
Wabbajack.Lib/Downloaders/DownloadDispatcher.cs
View File

@ -129,7 +129,7 @@ namespace Wabbajack.Lib.Downloaders
Utils.Log($"Applying patch to {archive.Name}");
await using(var src = await result.NewFile.Path.OpenShared())
await using (var final = destination.Create())
await using (var final = await destination.Create())
{
Utils.ApplyPatch(src, () => tempFile.Path.OpenShared().Result, final);
}

4
Wabbajack.Lib/Downloaders/GameFileSourceDownloader.cs
View File

@ -71,8 +71,8 @@ namespace Wabbajack.Lib.Downloaders
public override async Task<bool> Download(Archive a, AbsolutePath destination)
{
await using var src = SourcePath.OpenRead();
await using var dest = destination.Create();
await using var src = await SourcePath.OpenRead();
await using var dest = await destination.Create();
var size = SourcePath.Size;
await src.CopyToWithStatusAsync(size, dest, "Copying from Game folder");

2
Wabbajack.Lib/Downloaders/HTTPDownloader.cs
View File

@ -81,7 +81,7 @@ namespace Wabbajack.Lib.Downloaders
destination.Parent.CreateDirectory();
}
using (var fs = download ? destination.Create() : null)
using (var fs = download ? await destination.Create() : null)
{
var client = Client ?? new Common.Http.Client();
client.Headers.Add(("User-Agent", Consts.UserAgent));

2
Wabbajack.Lib/Downloaders/WabbajackCDNDownloader.cs
View File

@ -57,7 +57,7 @@ namespace Wabbajack.Lib.Downloaders
{
destination.Parent.CreateDirectory();
var definition = await GetDefinition();
using var fs = destination.Create();
await using var fs = await destination.Create();
using var mmfile = MemoryMappedFile.CreateFromFile(fs, null, definition.Size, MemoryMappedFileAccess.ReadWrite, HandleInheritability.None, false);
var client = new Common.Http.Client();
using var queue = new WorkQueue();

6
Wabbajack.Lib/Downloaders/YouTubeDownloader.cs
View File

@ -114,7 +114,7 @@ namespace Wabbajack.Lib.Downloaders
var trackFolder = folder.Dir.Combine("tracks");
await using (var fs = initialDownload.Create())
await using (var fs = await initialDownload.Create())
{
await client.Videos.Streams.CopyToAsync(stream, fs, new Progress($"Downloading {a.Name}"),
CancellationToken.None);
@ -128,7 +128,7 @@ namespace Wabbajack.Lib.Downloaders
await ExtractTrack(initialDownload, trackFolder, track);
});
await using var dest = destination.Create();
await using var dest = await destination.Create();
using var ar = new ZipArchive(dest, ZipArchiveMode.Create);
foreach (var track in trackFolder.EnumerateFiles().OrderBy(e => e))
{
@ -136,7 +136,7 @@ namespace Wabbajack.Lib.Downloaders
var entry = ar.CreateEntry(Path.Combine("Data", "tracks", (string)track.RelativeTo(trackFolder)), CompressionLevel.NoCompression);
entry.LastWriteTime = meta.UploadDate;
await using var es = entry.Open();
await using var ins = track.OpenRead();
await using var ins = await track.OpenRead();
await ins.CopyToAsync(es);
}

16
Wabbajack.Lib/Extensions/ReactiveUIExt.cs
View File

@ -4,6 +4,8 @@ using System.Linq.Expressions;
using System.Reactive;
using System.Reactive.Disposables;
using System.Reactive.Linq;
using System.Reactive.Subjects;
using System.Threading.Tasks;
using DynamicData;
using DynamicData.Kernel;
using ReactiveUI;
@ -38,6 +40,20 @@ namespace Wabbajack
return source.ObserveOn(RxApp.MainThreadScheduler);
}
/// <summary>
/// Like IObservable.Select but supports async map functions
/// </summary>
/// <param name="source"></param>
/// <param name="f"></param>
/// <typeparam name="T"></typeparam>
/// <returns></returns>
public static IObservable<TOut> SelectAsync<TIn, TOut>(this IObservable<TIn> source, Func<TIn, Task<TOut>> f)
{
return source.Select(itm => Observable.FromAsync(async () => await f(itm))).Merge(10);
}
public static IObservable<Unit> StartingExecution(this IReactiveCommand cmd)
{
return cmd.IsExecuting

10
Wabbajack.Lib/MO2Compiler.cs
View File

@ -484,8 +484,8 @@ namespace Wabbajack.Lib
Info($"Patching {entry.To}");
Status($"Patching {entry.To}");
var srcFile = byPath[string.Join("|", entry.ArchiveHashPath.Paths)];
await using var srcStream = srcFile.OpenRead();
await using var outputStream = IncludeFile(out var id);
await using var srcStream = await srcFile.OpenRead();
await using var outputStream = await IncludeFile(out var id).Create();
entry.PatchID = id;
await using var destStream = await LoadDataForTo(entry.To, absolutePaths);
await Utils.CreatePatch(srcStream, srcFile.Hash, destStream, entry.Hash, outputStream);
@ -496,18 +496,18 @@ namespace Wabbajack.Lib
private async Task<FileStream> LoadDataForTo(RelativePath to, Dictionary<RelativePath, AbsolutePath> absolutePaths)
{
if (absolutePaths.TryGetValue(to, out var absolute))
return absolute.OpenRead();
return await absolute.OpenRead();
if (to.StartsWith(Consts.BSACreationDir))
{
var bsaId = (RelativePath)((string)to).Split('\\')[1];
var bsa = InstallDirectives.OfType<CreateBSA>().First(b => b.TempID == bsaId);
await using var a = BSADispatch.OpenRead(MO2Folder.Combine(bsa.To));
await using var a = await BSADispatch.OpenRead(MO2Folder.Combine(bsa.To));
var find = (RelativePath)Path.Combine(((string)to).Split('\\').Skip(2).ToArray());
var file = a.Files.First(e => e.Path == find);
var returnStream = new TempStream();
file.CopyDataTo(returnStream);
await file.CopyDataTo(returnStream);
returnStream.Position = 0;
return returnStream;
}

12
Wabbajack.Lib/MO2Installer.cs
View File

@ -84,7 +84,7 @@ namespace Wabbajack.Lib
if (cancel.IsCancellationRequested) return false;
UpdateTracker.NextStep("Validating Game ESMs");
ValidateGameESMs();
await ValidateGameESMs();
if (cancel.IsCancellationRequested) return false;
UpdateTracker.NextStep("Validating Modlist");
@ -238,14 +238,14 @@ namespace Wabbajack.Lib
});
}
private void ValidateGameESMs()
private async ValueTask ValidateGameESMs()
{
foreach (var esm in ModList.Directives.OfType<CleanedESM>().ToList())
{
var filename = esm.To.FileName;
var gameFile = GameFolder!.Value.Combine((RelativePath)"Data", filename);
Utils.Log($"Validating {filename}");
var hash = gameFile.FileHash();
var hash = await gameFile.FileHashAsync();
if (hash != esm.SourceESMHash)
{
Utils.ErrorThrow(new InvalidGameESMError(esm, hash, gameFile));
@ -269,7 +269,7 @@ namespace Wabbajack.Lib
var streams = await bsa.FileStates.PMap(Queue, async state =>
{
Status($"Adding {state.Path} to BSA");
var fs = sourceDir.Combine(state.Path).OpenRead();
var fs = await sourceDir.Combine(state.Path).OpenRead();
await a.AddFile(state, fs);
return fs;
});
@ -328,8 +328,8 @@ namespace Wabbajack.Lib
var patchData = await LoadBytesFromPath(directive.SourceDataID);
var toFile = OutputFolder.Combine(directive.To);
Status($"Patching {filename}");
using var output = toFile.Create();
using var input = gameFile.OpenRead();
await using var output = await toFile.Create();
await using var input = await gameFile.OpenRead();
Utils.ApplyPatch(input, () => new MemoryStream(patchData), output);
}

4
Wabbajack.Lib/ModListRegistry/ModListMetadata.cs
View File

@ -84,14 +84,14 @@ namespace Wabbajack.Lib.ModListRegistry
return metadata.OrderBy(m => (m.ValidationSummary?.HasFailures ?? false ? 1 : 0, m.Title)).ToList();
}
public bool NeedsDownload(AbsolutePath modlistPath)
public async ValueTask<bool> NeedsDownload(AbsolutePath modlistPath)
{
if (!modlistPath.Exists) return true;
if (DownloadMetadata?.Hash == null)
{
return true;
}
return DownloadMetadata.Hash != modlistPath.FileHashCached(true);
return DownloadMetadata.Hash != await modlistPath.FileHashCachedAsync(true);
}
}

2
Wabbajack.Lib/NexusApi/NexusApi.cs
View File

@ -77,7 +77,7 @@ namespace Wabbajack.Lib.NexusApi
public static async Task<string> RequestAndCacheAPIKey()
{
var result = await Utils.Log(new RequestNexusAuthorization()).Task;
result.ToEcryptedJson("nexusapikey");
await result.ToEcryptedJson("nexusapikey");
return result;
}

2
Wabbajack.Lib/zEditIntegration.cs
View File

@ -270,7 +270,7 @@ namespace Wabbajack.Lib
var patchData = await installer.LoadBytesFromPath(m.PatchID);
await using var fs = installer.OutputFolder.Combine(m.To).Create();
await using var fs = await installer.OutputFolder.Combine(m.To).Create();
Utils.ApplyPatch(new MemoryStream(srcData), () => new MemoryStream(patchData), fs);
});
}

23
Wabbajack.Server.Test/ABuildServerSystemTest.cs
View File

@ -36,18 +36,19 @@ namespace Wabbajack.BuildServer.Test
public AbsolutePath ServerUpdatesFolder => "updates".RelativeTo(AbsolutePath.EntryPoint);
public BuildServerFixture()
public static async Task Start()
{
ServerArchivesFolder.DeleteDirectory().Wait();
ServerArchivesFolder.CreateDirectory();
var fixture = new BuildServerFixture();
fixture.ServerArchivesFolder.DeleteDirectory().Wait();
fixture.ServerArchivesFolder.CreateDirectory();
var builder = Program.CreateHostBuilder(
new[]
{
$"WabbajackSettings:DownloadDir={"tmp".RelativeTo(AbsolutePath.EntryPoint)}",
$"WabbajackSettings:ArchiveDir={"archives".RelativeTo(AbsolutePath.EntryPoint)}",
$"WabbajackSettings:TempFolder={ServerTempFolder}",
$"WabbajackSettings:SQLConnection={PublicConnStr}",
$"WabbajackSettings:TempFolder={fixture.ServerTempFolder}",
$"WabbajackSettings:SQLConnection={fixture.PublicConnStr}",
$"WabbajackSettings:BunnyCDN_User=TEST",
$"WabbajackSettings:BunnyCDN_Password=TEST",
"WabbajackSettings:JobScheduler=false",
@ -56,12 +57,12 @@ namespace Wabbajack.BuildServer.Test
"WabbajackSettings:RunFrontEndJobs=false",
"WabbajackSettinss:DisableNexusForwarding=true"
}, true);
_host = builder.Build();
_token = new CancellationTokenSource();
_task = _host.RunAsync(_token.Token);
fixture._host = builder.Build();
fixture._token = new CancellationTokenSource();
fixture._task = fixture._host.RunAsync(fixture._token.Token);
Consts.WabbajackBuildServerUri = new Uri("http://localhost:8080");
"ServerWhitelist.yaml".RelativeTo(ServerPublicFolder).WriteAllText(
await "ServerWhitelist.yaml".RelativeTo(fixture.ServerPublicFolder).WriteAllTextAsync(
"GoogleIDs:\nAllowedPrefixes:\n - http://localhost");
}
@ -208,7 +209,7 @@ namespace Wabbajack.BuildServer.Test
var modListPath = "test_modlist.wabbajack".RelativeTo(Fixture.ServerPublicFolder);
await using (var fs = modListPath.Create())
await using (var fs = await modListPath.Create())
{
using var za = new ZipArchive(fs, ZipArchiveMode.Create);
var entry = za.CreateEntry("modlist");
@ -254,7 +255,7 @@ namespace Wabbajack.BuildServer.Test
var metadataPath = "test_mod_list_metadata.json".RelativeTo(Fixture.ServerPublicFolder);
ModListMetaData.ToJson(metadataPath);
await ModListMetaData.ToJsonAsync(metadataPath);
return new Uri(MakeURL("test_mod_list_metadata.json"));
}

2
Wabbajack.Server/Controllers/AuthoredFiles.cs
View File

@ -99,7 +99,7 @@ namespace Wabbajack.BuildServer.Controllers
private async Task<FtpClient> GetBunnyCdnFtpClient()
{
var info = Utils.FromEncryptedJson<BunnyCdnFtpInfo>("bunny-cdn-ftp-info");
var info = await Utils.FromEncryptedJson<BunnyCdnFtpInfo>("bunny-cdn-ftp-info");
var client = new FtpClient(info.Hostname) {Credentials = new NetworkCredential(info.Username, info.Password)};
await client.ConnectAsync();
return client;

2
Wabbajack.Server/Services/ModListDownloader.cs
View File

@ -99,7 +99,7 @@ namespace Wabbajack.Server.Services
_maintainer.TryGetPath(list.DownloadMetadata.Hash, out var modlistPath);
ModList modlist;
await using (var fs = modlistPath.OpenRead())
await using (var fs = await modlistPath.OpenRead())
using (var zip = new ZipArchive(fs, ZipArchiveMode.Read))
await using (var entry = zip.GetEntry("modlist")?.Open())
{

10
Wabbajack.Server/Services/PatchBuilder.cs
View File

@ -70,10 +70,10 @@ namespace Wabbajack.Server.Services
using var sigFile = new TempFile();
using var patchFile = new TempFile();
await using var srcStream = srcPath.OpenShared();
await using var destStream = destPath.OpenShared();
await using var sigStream = sigFile.Path.Create();
await using var patchOutput = patchFile.Path.Create();
await using var srcStream = await srcPath.OpenShared();
await using var destStream = await destPath.OpenShared();
await using var sigStream = await sigFile.Path.Create();
await using var patchOutput = await patchFile.Path.Create();
OctoDiff.Create(destStream, srcStream, sigStream, patchOutput);
await patchOutput.DisposeAsync();
var size = patchFile.Path.Size;
@ -132,7 +132,7 @@ namespace Wabbajack.Server.Services
private async Task<FtpClient> GetBunnyCdnFtpClient()
{
var info = Utils.FromEncryptedJson<BunnyCdnFtpInfo>("bunny-cdn-ftp-info");
var info = await Utils.FromEncryptedJson<BunnyCdnFtpInfo>("bunny-cdn-ftp-info");
var client = new FtpClient(info.Hostname) {Credentials = new NetworkCredential(info.Username, info.Password)};
await client.ConnectAsync();
return client;

2
Wabbajack.Test/DownloaderTests.cs
View File

@ -436,7 +436,7 @@ namespace Wabbajack.Test
await converted.Download(new Archive(state: null!) { Name = "mod.zip" }, filename.Path);
await using var fs = filename.Path.OpenRead();
await using var fs = await filename.Path.OpenRead();
using var archive = new ZipArchive(fs);
var entries = archive.Entries.Select(e => e.FullName).ToList();
Assert.Equal(entries, new List<string> {@"Data\TestCK.esp", @"Data\TestCK.ini"});

6
Wabbajack.Test/EndToEndTests.cs
View File

@ -47,7 +47,7 @@ namespace Wabbajack.Test
public async Task CreateModlist()
{
var profile = utils.AddProfile("Default");
var mod = utils.AddMod();
var mod = await utils.AddMod();
await DownloadAndInstall(
"https://github.com/ModOrganizer2/modorganizer/releases/download/v2.2.1/Mod.Organizer.2.2.1.7z",
@ -79,7 +79,7 @@ namespace Wabbajack.Test
await CompileAndInstall(profile);
utils.VerifyAllFiles();
await utils.VerifyAllFiles();
await utils.InstallFolder.Combine(Consts.LOOTFolderFilesDir).DeleteDirectory();
@ -111,7 +111,7 @@ namespace Wabbajack.Test
private async Task<(AbsolutePath Download, AbsolutePath ModFolder)> DownloadAndInstall(Game game, int modId, string modName)
{
utils.AddMod(modName);
await 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));

10
Wabbajack.Test/MO2Tests.cs
View File

@ -27,8 +27,8 @@ namespace Wabbajack.Test
public async Task CheckValidInstallPath_HasModlist()
{
await using var tempDir = await TempFolder.Create();
await using var mo2 = tempDir.Dir.Combine("ModOrganizer.exe").Create();
await using var molist = tempDir.Dir.Combine(((RelativePath)"modlist")).WithExtension(Consts.ModListExtension).Create();
await using var mo2 = await tempDir.Dir.Combine("ModOrganizer.exe").Create();
await using var molist = await tempDir.Dir.Combine(((RelativePath)"modlist")).WithExtension(Consts.ModListExtension).Create();
Assert.False(MO2Installer.CheckValidInstallPath(tempDir.Dir, downloadFolder: null).Succeeded);
}
@ -36,7 +36,7 @@ namespace Wabbajack.Test
public async Task CheckValidInstallPath_ProperOverwrite()
{
await using var tempDir = await TempFolder.Create();
await using var tmp = tempDir.Dir.Combine(Consts.ModOrganizer2Exe).Create();
await using var tmp = await tempDir.Dir.Combine(Consts.ModOrganizer2Exe).Create();
Assert.True(MO2Installer.CheckValidInstallPath(tempDir.Dir, downloadFolder: null).Succeeded);
}
@ -46,7 +46,7 @@ namespace Wabbajack.Test
await using var tempDir = await TempFolder.Create();
await tempDir.Dir.DeleteDirectory();
tempDir.Dir.CreateDirectory();
await using var tmp = tempDir.Dir.Combine($"someFile.txt").Create();
await using var tmp = await tempDir.Dir.Combine($"someFile.txt").Create();
Assert.False(MO2Installer.CheckValidInstallPath(tempDir.Dir, downloadFolder: null).Succeeded);
}
@ -56,7 +56,7 @@ namespace Wabbajack.Test
await using var tempDir = await TempFolder.Create();
var downloadsFolder = tempDir.Dir.Combine("downloads");
downloadsFolder.CreateDirectory();
await using var tmp = tempDir.Dir.Combine($"downloads/someFile.txt").Create();
await using var tmp = await tempDir.Dir.Combine($"downloads/someFile.txt").Create();
Assert.True(MO2Installer.CheckValidInstallPath(tempDir.Dir, downloadFolder: downloadsFolder).Succeeded);
}
#endregion

132
Wabbajack.Test/SanityTests.cs
View File

@ -27,17 +27,17 @@ namespace Wabbajack.Test
{
var profile = utils.AddProfile();
var mod = utils.AddMod();
var testPex = utils.AddModFile(mod, @"Data\scripts\test.pex", 10);
var mod = await utils.AddMod();
var testPex = await utils.AddModFile(mod, @"Data\scripts\test.pex", 10);
await utils.Configure();
utils.AddManualDownload(
await utils.AddManualDownload(
new Dictionary<string, byte[]> {{"/baz/biz.pex", await testPex.ReadAllBytesAsync()}});
await CompileAndInstall(profile);
utils.VerifyInstalledFile(mod, @"Data\scripts\test.pex");
await utils.VerifyInstalledFile(mod, @"Data\scripts\test.pex");
}
[Fact]
@ -45,19 +45,19 @@ namespace Wabbajack.Test
{
var profile = utils.AddProfile();
var mod = utils.AddMod();
var testPex = utils.AddModFile(mod, @"Data\scripts\test.pex", 10);
var mod = await utils.AddMod();
var testPex = await utils.AddModFile(mod, @"Data\scripts\test.pex", 10);
await utils.Configure();
utils.AddManualDownload(
await utils.AddManualDownload(
new Dictionary<string, byte[]> {{"/baz/biz.pex", await testPex.ReadAllBytesAsync()}});
await utils.DownloadsFolder.Combine("some_other_file.7z").WriteAllTextAsync("random data");
await CompileAndInstall(profile);
utils.VerifyInstalledFile(mod, @"Data\scripts\test.pex");
await utils.VerifyInstalledFile(mod, @"Data\scripts\test.pex");
}
[Fact]
@ -65,17 +65,17 @@ namespace Wabbajack.Test
{
var profile = utils.AddProfile();
var mod = utils.AddMod();
var testPex = utils.AddGameFile(@"enbstuff\test.pex", 10);
var mod = await utils.AddMod();
var testPex = await utils.AddGameFile(@"enbstuff\test.pex", 10);
await utils.Configure();
utils.AddManualDownload(
await utils.AddManualDownload(
new Dictionary<string, byte[]> {{"/baz/biz.pex", await testPex.ReadAllBytesAsync()}});
await CompileAndInstall(profile);
utils.VerifyInstalledGameFile(@"enbstuff\test.pex");
await utils.VerifyInstalledGameFile(@"enbstuff\test.pex");
}
[Fact]
@ -83,14 +83,14 @@ namespace Wabbajack.Test
{
var profile = utils.AddProfile();
var mod = utils.AddMod();
var testPex = utils.AddGameFile(@"enbstuff\test.pex", 10);
var mod = await utils.AddMod();
var testPex = await utils.AddGameFile(@"enbstuff\test.pex", 10);
await utils.Configure();
utils.MO2Folder.Combine(Consts.GameFolderFilesDir).CreateDirectory();
utils.AddManualDownload(
await utils.AddManualDownload(
new Dictionary<string, byte[]> {{"/baz/biz.pex", await testPex.ReadAllBytesAsync()}});
await CompileAndInstall(profile);
@ -103,21 +103,21 @@ namespace Wabbajack.Test
{
var profile = utils.AddProfile();
var mod = utils.AddMod();
var testPex = utils.AddModFile(mod, @"Data\scripts\test.pex", 10);
var mod = await utils.AddMod();
var testPex = await utils.AddModFile(mod, @"Data\scripts\test.pex", 10);
// Make a copy to make sure it gets picked up and moved around.
testPex.CopyTo(testPex.WithExtension(new Extension(".copy")));
await testPex.CopyToAsync(testPex.WithExtension(new Extension(".copy")));
await utils.Configure();
utils.AddManualDownload(
await utils.AddManualDownload(
new Dictionary<string, byte[]> { { "/baz/biz.pex", await testPex.ReadAllBytesAsync() } });
await CompileAndInstall(profile);
utils.VerifyInstalledFile(mod, @"Data\scripts\test.pex");
utils.VerifyInstalledFile(mod, @"Data\scripts\test.pex.copy");
await utils.VerifyInstalledFile(mod, @"Data\scripts\test.pex");
await utils.VerifyInstalledFile(mod, @"Data\scripts\test.pex.copy");
}
[Fact]
@ -125,14 +125,14 @@ namespace Wabbajack.Test
{
var profile = utils.AddProfile();
var mod = utils.AddMod();
var unchanged = utils.AddModFile(mod, @"Data\scripts\unchanged.pex", 10);
var deleted = utils.AddModFile(mod, @"Data\scripts\deleted.pex", 10);
var modified = utils.AddModFile(mod, @"Data\scripts\modified.pex", 10);
var mod = await utils.AddMod();
var unchanged = await utils.AddModFile(mod, @"Data\scripts\unchanged.pex", 10);
var deleted = await utils.AddModFile(mod, @"Data\scripts\deleted.pex", 10);
var modified = await utils.AddModFile(mod, @"Data\scripts\modified.pex", 10);
await utils.Configure();
utils.AddManualDownload(
await utils.AddManualDownload(
new Dictionary<string, byte[]>
{
{ "/baz/unchanged.pex", await unchanged.ReadAllBytesAsync() },
@ -142,9 +142,9 @@ namespace Wabbajack.Test
await CompileAndInstall(profile);
utils.VerifyInstalledFile(mod, @"Data\scripts\unchanged.pex");
utils.VerifyInstalledFile(mod, @"Data\scripts\deleted.pex");
utils.VerifyInstalledFile(mod, @"Data\scripts\modified.pex");
await utils.VerifyInstalledFile(mod, @"Data\scripts\unchanged.pex");
await utils.VerifyInstalledFile(mod, @"Data\scripts\deleted.pex");
await utils.VerifyInstalledFile(mod, @"Data\scripts\modified.pex");
var unchangedPath = utils.PathOfInstalledFile(mod, @"Data\scripts\unchanged.pex");
var deletedPath = utils.PathOfInstalledFile(mod, @"Data\scripts\deleted.pex");
@ -170,9 +170,9 @@ namespace Wabbajack.Test
await CompileAndInstall(profile);
utils.VerifyInstalledFile(mod, @"Data\scripts\unchanged.pex");
utils.VerifyInstalledFile(mod, @"Data\scripts\deleted.pex");
utils.VerifyInstalledFile(mod, @"Data\scripts\modified.pex");
await utils.VerifyInstalledFile(mod, @"Data\scripts\unchanged.pex");
await utils.VerifyInstalledFile(mod, @"Data\scripts\deleted.pex");
await utils.VerifyInstalledFile(mod, @"Data\scripts\modified.pex");
Assert.Equal(unchangedModified, unchangedPath.LastModified);
Assert.NotEqual(modifiedModified, modifiedPath.LastModified);
@ -184,7 +184,7 @@ namespace Wabbajack.Test
public async Task SetScreenSizeTest()
{
var profile = utils.AddProfile();
var mod = utils.AddMod("dummy");
var mod = await utils.AddMod("dummy");
await utils.Configure();
await utils.MO2Folder.Combine("profiles", profile, "somegameprefs.ini").WriteAllLinesAsync(
@ -216,11 +216,11 @@ namespace Wabbajack.Test
public async Task UnmodifiedInlinedFilesArePulledFromArchives()
{
var profile = utils.AddProfile();
var mod = utils.AddMod();
var ini = utils.AddModFile(mod, @"foo.ini", 10);
var mod = await utils.AddMod();
var ini = await utils.AddModFile(mod, @"foo.ini", 10);
await utils.Configure();
utils.AddManualDownload(
await utils.AddManualDownload(
new Dictionary<string, byte[]> { { "/baz/biz.pex", await ini.ReadAllBytesAsync() } });
var modlist = await CompileAndInstall(profile);
@ -234,9 +234,9 @@ namespace Wabbajack.Test
public async Task ModifiedIniFilesArePatchedAgainstFileWithSameName()
{
var profile = utils.AddProfile();
var mod = utils.AddMod();
var ini = utils.AddModFile(mod, @"foo.ini", 10);
var meta = utils.AddModFile(mod, "meta.ini");
var mod = await utils.AddMod();
var ini = await utils.AddModFile(mod, @"foo.ini", 10);
var meta = await utils.AddModFile(mod, "meta.ini");
await utils.Configure();
@ -262,8 +262,8 @@ namespace Wabbajack.Test
public async Task CanPatchFilesSourcedFromBSAs()
{
var profile = utils.AddProfile();
var mod = utils.AddMod();
var file = utils.AddModFile(mod, @"baz.bin", 10);
var mod = await utils.AddMod();
var file = await utils.AddModFile(mod, @"baz.bin", 10);
await utils.Configure();
@ -287,7 +287,7 @@ namespace Wabbajack.Test
new Dictionary<string, byte[]> { { "/stuff/files.bsa", await tempFile.Path.ReadAllBytesAsync() } });
await CompileAndInstall(profile);
utils.VerifyInstalledFile(mod, @"baz.bin");
await utils.VerifyInstalledFile(mod, @"baz.bin");
}
@ -295,8 +295,8 @@ namespace Wabbajack.Test
public async Task CanNoMatchIncludeFilesFromBSAs()
{
var profile = utils.AddProfile();
var mod = utils.AddMod();
var file = utils.AddModFile(mod, @"baz.bsa", 10);
var mod = await utils.AddMod();
var file = await utils.AddModFile(mod, @"baz.bsa", 10);
await file.Parent.Combine("meta.ini").WriteAllLinesAsync(new[]
{
@ -334,7 +334,7 @@ namespace Wabbajack.Test
new Dictionary<string, byte[]> { { "/stuff/matching_file_data.bin", tempFileData } });
await CompileAndInstall(profile);
utils.VerifyInstalledFile(mod, @"baz.bsa");
await utils.VerifyInstalledFile(mod, @"baz.bsa");
}
@ -342,8 +342,8 @@ namespace Wabbajack.Test
public async Task CanInstallFilesFromBSAAndBSA()
{
var profile = utils.AddProfile();
var mod = utils.AddMod();
var file = utils.AddModFile(mod, @"baz.bin", 128);
var mod = await utils.AddMod();
var file = await utils.AddModFile(mod, @"baz.bin", 128);
await utils.Configure();
@ -362,14 +362,14 @@ namespace Wabbajack.Test
}, new MemoryStream(await file.ReadAllBytesAsync()));
await bsa.Build(tempFile.Path);
}
tempFile.Path.CopyTo(file.Parent.Combine("bsa_data.bsa"));
await tempFile.Path.CopyToAsync(file.Parent.Combine("bsa_data.bsa"));
var archive = utils.AddManualDownload(
new Dictionary<string, byte[]> { { "/stuff/files.bsa", await tempFile.Path.ReadAllBytesAsync() } });
await CompileAndInstall(profile);
utils.VerifyInstalledFile(mod, @"baz.bin");
utils.VerifyInstalledFile(mod, @"bsa_data.bsa");
await utils.VerifyInstalledFile(mod, @"baz.bin");
await utils.VerifyInstalledFile(mod, @"bsa_data.bsa");
}
@ -377,8 +377,8 @@ namespace Wabbajack.Test
public async Task CanRecreateBSAsFromFilesSourcedInOtherBSAs()
{
var profile = utils.AddProfile();
var mod = utils.AddMod();
var file = utils.AddModFile(mod, @"baz.bsa", 10);
var mod = await utils.AddMod();
var file = await utils.AddModFile(mod, @"baz.bsa", 10);
await utils.Configure();
@ -414,7 +414,7 @@ namespace Wabbajack.Test
await CompileAndInstall(profile);
utils.VerifyInstalledFile(mod, @"baz.bsa");
await utils.VerifyInstalledFile(mod, @"baz.bsa");
}
@ -425,8 +425,8 @@ namespace Wabbajack.Test
Consts.TestMode = false;
var profile = utils.AddProfile();
var mod = utils.AddMod();
var skyrimExe = utils.AddModFile(mod, @"Data\test.exe", 10);
var mod = await utils.AddMod();
var skyrimExe = await utils.AddModFile(mod, @"Data\test.exe", 10);
var gameFolder = Consts.GameFolderFilesDir.RelativeTo(utils.MO2Folder);
gameFolder.CreateDirectory();
@ -459,18 +459,18 @@ namespace Wabbajack.Test
{
var profile = utils.AddProfile();
var mod = utils.AddMod();
var testPex = utils.AddModFile(mod, @"Data\scripts\test.pex", 10);
var mod = await utils.AddMod();
var testPex = await utils.AddModFile(mod, @"Data\scripts\test.pex", 10);
await utils.Configure();
await utils.AddModFile(mod, "meta.ini").WriteAllLinesAsync(new[]
await (await utils.AddModFile(mod, "meta.ini")).WriteAllLinesAsync(new[]
{
"[General]", "notes= fsdaf WABBAJACK_NOMATCH_INCLUDE fadsfsad",
});
await CompileAndInstall(profile);
utils.VerifyInstalledFile(mod, @"Data\scripts\test.pex");
await utils.VerifyInstalledFile(mod, @"Data\scripts\test.pex");
}
@ -483,11 +483,11 @@ namespace Wabbajack.Test
{
var profile = utils.AddProfile();
var enabledMod = utils.AddMod();
var enabledTestPex = utils.AddModFile(enabledMod, @"Data\scripts\enabledTestPex.pex", 10);
var enabledMod = await utils.AddMod();
var enabledTestPex = await utils.AddModFile(enabledMod, @"Data\scripts\enabledTestPex.pex", 10);
var disabledMod = utils.AddMod();
var disabledTestPex = utils.AddModFile(disabledMod, @"Data\scripts\disabledTestPex.pex", 10);
var disabledMod = await utils.AddMod();
var disabledTestPex = await utils.AddModFile(disabledMod, @"Data\scripts\disabledTestPex.pex", 10);
await disabledMod.RelativeTo(utils.ModsFolder).Combine("meta.ini").WriteAllLinesAsync(
"[General]",
@ -499,7 +499,7 @@ namespace Wabbajack.Test
(enabledMod, true)
});
utils.AddManualDownload(
await utils.AddManualDownload(
new Dictionary<string, byte[]>
{
{"/file1.pex", await enabledTestPex.ReadAllBytesAsync()},
@ -508,8 +508,8 @@ namespace Wabbajack.Test
await CompileAndInstall(profile);
utils.VerifyInstalledFile(enabledMod, @"Data\scripts\enabledTestPex.pex");
utils.VerifyInstalledFile(disabledMod, @"Data\scripts\disabledTestPex.pex");
await utils.VerifyInstalledFile(enabledMod, @"Data\scripts\enabledTestPex.pex");
await utils.VerifyInstalledFile(disabledMod, @"Data\scripts\disabledTestPex.pex");
var modlistTxt = await utils.InstallFolder.Combine("profiles", profile, "modlist.txt").ReadAllLinesAsync();
Assert.Equal(new string[]

49
Wabbajack.Test/TestUtils.cs
View File

@ -57,7 +57,7 @@ namespace Wabbajack.Test
{
Profiles.Do(profile =>
{
MO2Folder.Combine("profiles", profile, "modlist.txt").WriteAllLines(
MO2Folder.Combine("profiles", profile, "modlist.txt").WriteAllLinesAsync(
Mods.Select(s => $"+{s}").ToArray());
});
}
@ -65,7 +65,7 @@ namespace Wabbajack.Test
{
Profiles.Do(profile =>
{
MO2Folder.Combine("profiles", profile, "modlist.txt").WriteAllLines(
MO2Folder.Combine("profiles", profile, "modlist.txt").WriteAllLinesAsync(
enabledMods.Select(s => $"{(s.IsEnabled ? "+" : "-")}{s.ModName}").ToArray());
});
}
@ -79,17 +79,14 @@ namespace Wabbajack.Test
return profile_name;
}
public string AddMod(string name = null)
public async Task<string> AddMod(string name = null)
{
lock (this)
{
string mod_name = name ?? RandomName();
var mod_folder = MO2Folder.Combine(Consts.MO2ModFolderName, (RelativePath)mod_name);
mod_folder.CreateDirectory();
mod_folder.Combine("meta.ini").WriteAllText("[General]");
Mods.Add(mod_name);
return mod_name;
}
string mod_name = name ?? RandomName();
var mod_folder = MO2Folder.Combine(Consts.MO2ModFolderName, (RelativePath)mod_name);
mod_folder.CreateDirectory();
await mod_folder.Combine("meta.ini").WriteAllTextAsync("[General]");
Mods.Add(mod_name);
return mod_name;
}
/// <summary>
@ -100,17 +97,15 @@ namespace Wabbajack.Test
/// <param name="path"></param>
/// <param name="random_fill"></param>
/// <returns></returns>
public AbsolutePath AddModFile(string mod_name, string path, int random_fill=128)
public async Task<AbsolutePath> AddModFile(string mod_name, string path, int random_fill=128)
{
var full_path = ModsFolder.Combine(mod_name, path);
full_path.Parent.CreateDirectory();
GenerateRandomFileData(full_path, random_fill);
await GenerateRandomFileData(full_path, random_fill);
return full_path;
}
public void GenerateRandomFileData(AbsolutePath full_path, int random_fill)
public async Task GenerateRandomFileData(AbsolutePath full_path, int random_fill)
{
byte[] bytes = new byte[0];
if (random_fill != 0)
@ -118,7 +113,7 @@ namespace Wabbajack.Test
bytes = new byte[random_fill];
RNG.NextBytes(bytes);
}
full_path.WriteAllBytes(bytes);
await full_path.WriteAllBytesAsync(bytes);
}
public static byte[] RandomData(int? size = null, int maxSize = 1024)
@ -162,20 +157,20 @@ namespace Wabbajack.Test
return data;
}
public string AddManualDownload(Dictionary<string, byte[]> contents)
public async ValueTask<string> AddManualDownload(Dictionary<string, byte[]> contents)
{
var name = RandomName() + ".zip";
using FileStream fs = DownloadsFolder.Combine(name).Create();
await using FileStream fs = await DownloadsFolder.Combine(name).Create();
using ZipArchive archive = new ZipArchive(fs, ZipArchiveMode.Create);
contents.Do(kv =>
{
var entry = archive.CreateEntry(kv.Key);
using (var os = entry.Open())
os.Write(kv.Value, 0, kv.Value.Length);
using var os = entry.Open();
os.Write(kv.Value, 0, kv.Value.Length);
});
DownloadsFolder.Combine(name + Consts.MetaFileExtension).WriteAllLines(
await DownloadsFolder.Combine(name + Consts.MetaFileExtension).WriteAllLinesAsync(
"[General]",
"manualURL=<TESTING>"
);
@ -227,7 +222,7 @@ namespace Wabbajack.Test
return InstallFolder.Combine((string)Consts.MO2ModFolderName, mod, file);
}
public void VerifyAllFiles(bool gameFileShouldNotExistInGameFolder = true)
public async ValueTask VerifyAllFiles(bool gameFileShouldNotExistInGameFolder = true)
{
if (gameFileShouldNotExistInGameFolder)
{
@ -264,16 +259,16 @@ namespace Wabbajack.Test
if (!skipExtensions.Contains(srcFile.Extension))
{
Assert.Equal(srcFile.Size, destFile.Size);
Assert.Equal(srcFile.FileHash(), destFile.FileHash());
Assert.Equal(await srcFile.FileHashAsync(), await destFile.FileHashAsync());
}
}
}
public AbsolutePath AddGameFile(string path, int i)
public async ValueTask<AbsolutePath> AddGameFile(string path, int i)
{
var fullPath = GameFolder.Combine(path);
fullPath.Parent.CreateDirectory();
GenerateRandomFileData(fullPath, i);
await GenerateRandomFileData(fullPath, i);
return fullPath;
}
}

14
Wabbajack.Test/ZEditIntegrationTests.cs
View File

@ -15,13 +15,13 @@ namespace Wabbajack.Test
public async Task CanCreatezEditPatches()
{
var profile = utils.AddProfile();
var moda = utils.AddMod();
var modb = utils.AddMod();
var moddest = utils.AddMod();
var srca = utils.AddModFile(moda, @"srca.esp", 10);
var srcb = utils.AddModFile(moda, @"srcb.esp.mohidden", 10);
var srcc = utils.AddModFile(modb, @"optional\srcc.esp", 10);
var dest = utils.AddModFile(moddest, @"merged.esp", 20);
var moda = await utils.AddMod();
var modb = await utils.AddMod();
var moddest = await utils.AddMod();
var srca = await utils.AddModFile(moda, @"srca.esp", 10);
var srcb = await utils.AddModFile(moda, @"srcb.esp.mohidden", 10);
var srcc = await utils.AddModFile(modb, @"optional\srcc.esp", 10);
var dest = await utils.AddModFile(moddest, @"merged.esp", 20);
var srcs = new List<string> {srca, srcb, srcc};

19
Wabbajack/Settings.cs
View File

@ -4,6 +4,7 @@ using System.Collections.Generic;
using System.IO;
using System.Reactive;
using System.Reactive.Subjects;
using System.Threading.Tasks;
using Wabbajack.Common;
using Wabbajack.Common.Serialization.Json;
using Wabbajack.Lib;
@ -29,20 +30,19 @@ namespace Wabbajack
[JsonIgnore]
public IObservable<Unit> SaveSignal => _saveSignal;
public static bool TryLoadTypicalSettings(out MainSettings settings)
public static async ValueTask<(MainSettings settings, bool loaded)> TryLoadTypicalSettings()
{
if (!Consts.SettingsFile.Exists)
{
settings = default;
return false;
return default;
}
// Version check
try
{
settings = Consts.SettingsFile.FromJson<MainSettings>();
var settings = Consts.SettingsFile.FromJson<MainSettings>();
if (settings.Version == Consts.SettingsVersion)
return true;
return (settings, true);
}
catch (Exception ex)
{
@ -52,14 +52,13 @@ namespace Wabbajack
var backup = Consts.SettingsFile.AppendToName("-backup");
backup.Delete();
Consts.SettingsFile.CopyTo(backup);
await Consts.SettingsFile.CopyToAsync(backup);
Consts.SettingsFile.Delete();
settings = default;
return false;
return default;
}
public static void SaveSettings(MainSettings settings)
public static async ValueTask SaveSettings(MainSettings settings)
{
settings._saveSignal.OnNext(Unit.Default);
@ -68,7 +67,7 @@ namespace Wabbajack
//settings._saveSignal.OnCompleted();
//await settings._saveSignal;
settings.ToJson(Consts.SettingsFile);
await settings.ToJsonAsync(Consts.SettingsFile);
}
}

6
Wabbajack/View Models/Gallery/ModListMetadataVM.cs
View File

@ -144,11 +144,11 @@ namespace Wabbajack
.Unit()
.StartWith(Unit.Default)
.FlowSwitch(_parent.WhenAny(x => x.IsActive))
.Select(_ =>
.SelectAsync(async _ =>
{
try
{
return !metadata.NeedsDownload(Location);
return !(await metadata.NeedsDownload(Location));
}
catch (Exception)
{
@ -185,7 +185,7 @@ namespace Wabbajack
var downloader = DownloadDispatcher.ResolveArchive(Metadata.Links.Download);
var result = await downloader.Download(new Archive(state: null!) { Name = Metadata.Title, Size = Metadata.DownloadMetadata?.Size ?? 0 }, Location);
// Want to rehash to current file, even if failed?
Location.FileHashCached();
await Location.FileHashCachedAsync();
tcs.SetResult(result);
}
catch (Exception ex)

2
Wabbajack/View Models/MainWindowVM.cs
View File

@ -199,7 +199,7 @@ namespace Wabbajack
Settings.PosY = MainWindow.Top;
Settings.Width = MainWindow.Width;
Settings.Height = MainWindow.Height;
MainSettings.SaveSettings(Settings);
MainSettings.SaveSettings(Settings).AsTask().Wait();
Application.Current.Shutdown();
}
}

15
Wabbajack/View Models/ModListVM.cs
View File

@ -45,19 +45,17 @@ namespace Wabbajack
ImageObservable = Observable.Return(Unit.Default)
// Download and retrieve bytes on background thread
.ObserveOn(RxApp.TaskpoolScheduler)
.Select(filePath =>
.SelectAsync(async filePath =>
{
try
{
using var fs = ModListPath.OpenShared();
await using var fs = await ModListPath.OpenShared();
using var ar = new ZipArchive(fs, ZipArchiveMode.Read);
var ms = new MemoryStream();
var entry = ar.GetEntry("modlist-image.png");
if (entry == null) return default(MemoryStream);
using (var e = entry.Open())
{
e.CopyTo(ms);
}
await using var e = entry.Open();
e.CopyTo(ms);
return ms;
}
catch (Exception ex)
@ -82,10 +80,7 @@ namespace Wabbajack
}
})
// If ever would return null, show WJ logo instead
.Select(x =>
{
return x ?? ResourceLinks.WabbajackLogoNoText.Value;
})
.Select(x => x ?? ResourceLinks.WabbajackLogoNoText.Value)
.Replay(1)
.RefCount();
}

3
Wabbajack/Views/MainWindow.xaml.cs
View File

@ -45,8 +45,9 @@ namespace Wabbajack
Warmup();
var (settings, loadedSettings) = MainSettings.TryLoadTypicalSettings().AsTask().Result;
// Load settings
if (CLIArguments.NoSettings || !MainSettings.TryLoadTypicalSettings(out var settings))
if (CLIArguments.NoSettings || !loadedSettings)
{
_settings = new MainSettings
{