Merge pull request #732 from TheTrackerCouncil/auto-download-updates

Download Windows setup files

Author: MattEqualsCoder

src/TrackerCouncil.Smz3.UI/Services/MainWindowService.cs

@@ -1,6 +1,8 @@
 using System;
+using System.Diagnostics;
 using System.IO;
 using System.Linq;
+using System.Net.Http;
 using System.Threading.Tasks;
 using AppImageManager;
 using Avalonia.Threading;
@@ -212,6 +214,70 @@ await Dispatcher.UIThread.InvokeAsync(async () =>
         trackerSpriteService.LoadSprites();
     }
 
+    public async Task<string?> InstallWindowsUpdate(string url)
+    {
+        var filename = Path.GetFileName(new Uri(url).AbsolutePath);
+        var localPath = Path.Combine(Path.GetTempPath(), filename);
+
+        logger.LogInformation("Downloading {Url} to {LocalPath}", url, localPath);
+
+        var response = await DownloadFileAsyncAttempt(url, localPath);
+
+        if (!response.Item1)
+        {
+            logger.LogInformation("Download failed: {Error}", response.Item2);
+            return response.Item2;
+        }
+
+        try
+        {
+            logger.LogInformation("Launching setup file");
+
+            var psi = new ProcessStartInfo
+            {
+                FileName = localPath,
+                UseShellExecute = true,
+                RedirectStandardOutput = false,
+                RedirectStandardError = false,
+                RedirectStandardInput = false,
+                CreateNoWindow = true
+            };
+
+            Process.Start(psi);
+            return null;
+        }
+        catch (Exception e)
+        {
+            return "Failed to start setup file";
+        }
+    }
+
+    private static async Task<(bool, string?)> DownloadFileAsyncAttempt(string url, string target, int attemptNumber = 0, int totalAttempts = 3)
+    {
+
+        using var httpClient = new HttpClient();
+
+        try
+        {
+            await using var downloadStream = await httpClient.GetStreamAsync(url);
+            await using var fileStream = new FileStream(target, FileMode.Create);
+            await downloadStream.CopyToAsync(fileStream);
+            return (true, null);
+        }
+        catch (Exception ex)
+        {
+            if (attemptNumber < totalAttempts)
+            {
+                await Task.Delay(TimeSpan.FromSeconds(attemptNumber));
+                return await DownloadFileAsyncAttempt(url, target, attemptNumber + 1, totalAttempts);
+            }
+            else
+            {
+                return (false, $"Download failed: {ex.Message}");
+            }
+        }
+    }
+
     private async Task CheckForUpdates()
     {
         if (!_options.GeneralOptions.CheckForUpdatesOnStartup)
@@ -235,6 +301,12 @@ private async Task CheckForUpdates()
                         .FirstOrDefault(x => x.Url.ToLower().EndsWith(".appimage"))?.Url;
                     _model.DisplayDownloadLink = !string.IsNullOrEmpty(_model.NewVersionDownloadUrl);
                 }
+                else if (OperatingSystem.IsWindows())
+                {
+                    _model.NewVersionDownloadUrl = gitHubRelease.Asset
+                        .FirstOrDefault(x => x.Url.ToLower().EndsWith(".exe"))?.Url;
+                    _model.DisplayDownloadLink = !string.IsNullOrEmpty(_model.NewVersionDownloadUrl);
+                }
             }
         }
         catch (Exception ex)

src/TrackerCouncil.Smz3.UI/Views/MainWindow.axaml.cs

@@ -137,7 +137,7 @@ private async void DownloadReleaseButton_OnClick(object? sender, RoutedEventArgs
     {
         try
         {
-            if (string.IsNullOrEmpty(_model.NewVersionDownloadUrl))
+            if (string.IsNullOrEmpty(_model.NewVersionDownloadUrl) || _service == null)
             {
                 return;
             }
@@ -162,11 +162,22 @@ private async void DownloadReleaseButton_OnClick(object? sender, RoutedEventArgs
                     await MessageWindow.ShowErrorDialog("Failed downloading AppImage");
                 }
             }
+            else
+            {
+                var result = await _service.InstallWindowsUpdate(_model.NewVersionDownloadUrl);
+                if (!string.IsNullOrEmpty(result))
+                {
+                    await MessageWindow.ShowErrorDialog(result);
+                }
+                else
+                {
+                    Close();
+                }
+            }
         }
-        catch (Exception exception)
+        catch
         {
-            Console.WriteLine(exception);
-            throw;
+            await MessageWindow.ShowErrorDialog("Failed downloading update");
         }
     }
 }