using System; using System.Diagnostics; using System.Linq; using System.Threading.Tasks; using Newtonsoft.Json; using Wabbajack.Common; using Wabbajack.Lib.Http; namespace Wabbajack.Lib { public class LauncherUpdater { public static Lazy CommonFolder = new (() => { var entryPoint = AbsolutePath.EntryPoint; // If we're not in a folder that looks like a version, abort if (!Version.TryParse(entryPoint.FileName.ToString(), out var version)) { return entryPoint; } // If we're not in a folder that has Wabbajack.exe in the parent folder, abort if (!entryPoint.Parent.Combine(Consts.AppName).WithExtension(new Extension(".exe")).IsFile) { return entryPoint; } return entryPoint.Parent; }); public static async Task Run() { if (CommonFolder.Value == AbsolutePath.EntryPoint) { Utils.Log("Outside of standard install folder, not updating"); return; } var version = Version.Parse(AbsolutePath.EntryPoint.FileName.ToString()); var oldVersions = CommonFolder.Value .EnumerateDirectories() .Select(f => Version.TryParse(f.FileName.ToString(), out var ver) ? (ver, f) : default) .Where(f => f != default) .Where(f => f.ver < version) .Select(f => f!) .OrderByDescending(f => f) .Skip(2) .ToArray(); foreach (var (_, path) in oldVersions) { Utils.Log($"Deleting old Wabbajack version at: {path}"); await path.DeleteDirectory(); } var release = (await GetReleases()) .Select(release => Version.TryParse(release.Tag, out version) ? (version, release) : default) .Where(r => r != default) .OrderByDescending(r => r.version) .Select(r => { var (version, release) = r; var asset = release.Assets.FirstOrDefault(a => a.Name == "Wabbajack.exe"); return asset != default ? (version, release, asset) : default; }) .FirstOrDefault(); var launcherFolder = AbsolutePath.EntryPoint.Parent; var exePath = launcherFolder.Combine("Wabbajack.exe"); var launcherVersion = FileVersionInfo.GetVersionInfo(exePath.ToString()); if (release != default && release.version > Version.Parse(launcherVersion.FileVersion!)) { Utils.Log($"Updating Launcher from {launcherVersion.FileVersion} to {release.version}"); var tempPath = launcherFolder.Combine("Wabbajack.exe.temp"); var client = new Client(); client.UseChromeUserAgent(); await client.DownloadAsync(release.asset.BrowserDownloadUrl!, tempPath); if (tempPath.Size != release.asset.Size) { Utils.Log( $"Downloaded launcher did not match expected size: {tempPath.Size} expected {release.asset.Size}"); return; } if (exePath.Exists) await exePath.DeleteAsync(); await tempPath.MoveToAsync(exePath); Utils.Log("Finished updating wabbajack"); await Metrics.Send("updated_launcher", $"{launcherVersion.FileVersion} -> {release.version}"); } } private static async Task GetReleases() { Utils.Log("Getting new Wabbajack version list"); var client = new Client(); client.UseChromeUserAgent(); return await client.GetJsonAsync(Consts.GITHUB_REPO_RELEASES.ToString()); } class Release { [JsonProperty("tag_name")] public string Tag { get; set; } = ""; [JsonProperty("assets")] public Asset[] Assets { get; set; } = Array.Empty(); } class Asset { [JsonProperty("browser_download_url")] public Uri? BrowserDownloadUrl { get; set; } [JsonProperty("name")] public string Name { get; set; } = ""; [JsonProperty("size")] public long Size { get; set; } = 0; } } }