Unify upgrade process and add update progress bar

Source/LibationUiBase/Upgrader.cs

@@ -0,0 +1,148 @@
+using AppScaffolding;
+using Dinah.Core.Net.Http;
+using LibationFileManager;
+using System;
+using System.IO;
+using System.Net.Http;
+using System.Threading.Tasks;
+
+namespace LibationUiBase
+{
+	public class UpgradeEventArgs
+	{
+		public UpgradeProperties UpgradeProperties { get; internal init; }
+		public bool CapUpgrade { get; internal init; }
+		private bool _ignore = false;
+		private bool _installUpdate = true;
+		public bool Ignore
+		{
+			get => _ignore;
+			set
+			{
+				_ignore = value;
+				_installUpdate &= !Ignore;
+			}
+		}
+		public bool InstallUpdate
+		{
+			get => _installUpdate;
+			set
+			{
+				_installUpdate = value;
+				_ignore &= !InstallUpdate;
+			}
+		}
+	}
+
+	public class Upgrader
+	{
+		public event EventHandler DownloadBegin;
+		public event EventHandler<DownloadProgress> DownloadProgress;
+		public event EventHandler<bool> DownloadCompleted;
+
+		public async Task CheForUpgradeAsync(Func<UpgradeEventArgs,Task> upgradeAvailableHandler)
+		{
+			try
+			{
+				var upgradeProperties = await Task.Run(LibationScaffolding.GetLatestRelease);
+				if (upgradeProperties is null) return;
+
+				const string ignoreUpdate = "IgnoreUpdate";
+				var config = Configuration.Instance;
+
+				if (config.GetString(propertyName: ignoreUpdate) == upgradeProperties.LatestRelease.ToString())
+					return;
+
+				var interop = InteropFactory.Create();
+
+				if (!interop.CanUpdate)
+					Serilog.Log.Logger.Information("Can't perform update automatically");
+
+				var upgradeEventArgs = new UpgradeEventArgs
+				{
+					UpgradeProperties = upgradeProperties,
+					CapUpgrade = interop.CanUpdate
+				};
+
+				await upgradeAvailableHandler(upgradeEventArgs);
+
+				if (upgradeEventArgs.Ignore)
+					config.SetString(upgradeProperties.LatestRelease.ToString(), ignoreUpdate);
+
+				if (!upgradeEventArgs.InstallUpdate) return;
+
+				//Download the update file in the background,
+				DownloadBegin?.Invoke(this, EventArgs.Empty);
+				string updateBundle = await DownloadUpgradeAsync(upgradeProperties);
+
+				if (string.IsNullOrEmpty(updateBundle) || !File.Exists(updateBundle))
+				{
+					DownloadCompleted?.Invoke(this, false);
+				}
+				else
+				{
+					DownloadCompleted?.Invoke(this, true);
+
+					//Install the update
+					Serilog.Log.Logger.Information($"Begin running auto-updater");
+					interop.InstallUpdate(updateBundle);
+					Serilog.Log.Logger.Information($"Completed running auto-updater");
+				}
+			}
+			catch (Exception ex)
+			{
+				Serilog.Log.Logger.Error(ex, "An error occured while checking for app updates.");
+			}
+		}
+
+		private async Task<string> DownloadUpgradeAsync(UpgradeProperties upgradeProperties)
+		{
+			if (upgradeProperties.ZipUrl is null)
+			{
+				Serilog.Log.Logger.Warning("Download link for new version not found");
+				return null;
+			}
+
+			//Silently download the update in the background, save it to a temp file.
+
+			var zipFile = Path.Combine(Path.GetTempPath(), Path.GetFileName(upgradeProperties.ZipUrl));
+
+			Serilog.Log.Logger.Information($"Downloading {zipFile}");
+
+			try
+			{
+				using var dlClient = new HttpClient();
+				using var response = await dlClient.GetAsync(upgradeProperties.ZipUrl, HttpCompletionOption.ResponseHeadersRead);
+				using var dlStream = await response.Content.ReadAsStreamAsync();
+				using var tempFile = File.OpenWrite(zipFile);
+
+				int read;
+				long totalRead = 0;
+				Memory<byte> buffer = new byte[128 * 1024];
+				long contentLength = response.Content.Headers.ContentLength ?? 0;
+
+				while ((read = await dlStream.ReadAsync(buffer)) > 0)
+				{
+					await tempFile.WriteAsync(buffer[..read]);
+					totalRead += read;
+
+					DownloadProgress?.Invoke(
+						this,
+						new DownloadProgress
+						{
+							BytesReceived = totalRead,
+							TotalBytesToReceive = contentLength,
+							ProgressPercentage = contentLength > 0 ? 100d * totalRead / contentLength : 0
+						});
+				}
+
+				return zipFile;
+			}
+			catch (Exception ex)
+			{
+				Serilog.Log.Logger.Error(ex, "Failed to download the update: {pdate}", upgradeProperties.ZipUrl);
+				return null;
+			}
+		}
+	}
+}