JT
1 year ago
17 changed files with 229 additions and 83 deletions
@ -1,6 +1,13 @@ |
|||||||
namespace StabilityMatrix.Helper; |
using System; |
||||||
|
using System.Threading.Tasks; |
||||||
|
using StabilityMatrix.Models; |
||||||
|
|
||||||
|
namespace StabilityMatrix.Helper; |
||||||
|
|
||||||
public interface IUpdateHelper |
public interface IUpdateHelper |
||||||
{ |
{ |
||||||
void StartCheckingForUpdates(); |
Task StartCheckingForUpdates(); |
||||||
} |
|
||||||
|
Task DownloadUpdate(UpdateInfo updateInfo, |
||||||
|
IProgress<ProgressReport> progress); |
||||||
|
} |
||||||
|
|||||||
@ -1,51 +1,114 @@ |
|||||||
using System; |
using System; |
||||||
using System.IO; |
using System.IO; |
||||||
|
using System.Net.Http; |
||||||
|
using System.Reflection; |
||||||
|
using System.Text.Json; |
||||||
|
using System.Threading.Tasks; |
||||||
using System.Windows.Threading; |
using System.Windows.Threading; |
||||||
using AutoUpdaterDotNET; |
|
||||||
using Microsoft.Extensions.Logging; |
using Microsoft.Extensions.Logging; |
||||||
|
using StabilityMatrix.Models; |
||||||
|
using StabilityMatrix.Services; |
||||||
|
|
||||||
namespace StabilityMatrix.Helper; |
namespace StabilityMatrix.Helper; |
||||||
|
|
||||||
public class UpdateHelper : IUpdateHelper |
public class UpdateHelper : IUpdateHelper |
||||||
{ |
{ |
||||||
private readonly ILogger<UpdateHelper> logger; |
private readonly ILogger<UpdateHelper> logger; |
||||||
|
private readonly IHttpClientFactory httpClientFactory; |
||||||
|
private readonly IDownloadService downloadService; |
||||||
private readonly DispatcherTimer timer = new(); |
private readonly DispatcherTimer timer = new(); |
||||||
|
|
||||||
|
private static readonly string UpdateFolder = |
||||||
|
Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "Update"); |
||||||
|
|
||||||
public UpdateHelper(ILogger<UpdateHelper> logger) |
public static readonly string ExecutablePath = |
||||||
|
Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "Update", "StabilityMatrix.exe"); |
||||||
|
|
||||||
|
public UpdateHelper(ILogger<UpdateHelper> logger, IHttpClientFactory httpClientFactory, |
||||||
|
IDownloadService downloadService) |
||||||
{ |
{ |
||||||
this.logger = logger; |
this.logger = logger; |
||||||
|
this.httpClientFactory = httpClientFactory; |
||||||
|
this.downloadService = downloadService; |
||||||
|
|
||||||
timer.Interval = TimeSpan.FromMinutes(5); |
timer.Interval = TimeSpan.FromMinutes(5); |
||||||
timer.Tick += (_, _) => |
timer.Tick += async (_, _) => { await CheckForUpdate(); }; |
||||||
{ |
|
||||||
CheckForUpdate(); |
|
||||||
}; |
|
||||||
AutoUpdater.CheckForUpdateEvent += AutoUpdaterOnCheckForUpdateEvent; |
|
||||||
} |
} |
||||||
|
|
||||||
public void StartCheckingForUpdates() |
public async Task StartCheckingForUpdates() |
||||||
{ |
{ |
||||||
timer.IsEnabled = true; |
timer.IsEnabled = true; |
||||||
timer.Start(); |
timer.Start(); |
||||||
CheckForUpdate(); |
await CheckForUpdate(); |
||||||
} |
} |
||||||
|
|
||||||
private void CheckForUpdate() |
public async Task DownloadUpdate(UpdateInfo updateInfo, |
||||||
|
IProgress<ProgressReport> progress) |
||||||
{ |
{ |
||||||
AutoUpdater.DownloadPath = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "Update"); |
var downloadUrl = updateInfo.DownloadUrl; |
||||||
AutoUpdater.ExecutablePath = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "Update", "StabilityMatrix.exe"); |
|
||||||
// TODO: make this github url? |
Directory.CreateDirectory(UpdateFolder); |
||||||
AutoUpdater.Start("https://cdn.lykos.ai/update.xml"); |
|
||||||
|
// download the file from URL |
||||||
|
await downloadService.DownloadToFileAsync(downloadUrl, ExecutablePath, progress: progress, |
||||||
|
httpClientName: "UpdateClient"); |
||||||
} |
} |
||||||
|
|
||||||
private void AutoUpdaterOnCheckForUpdateEvent(UpdateInfoEventArgs args) |
private async Task CheckForUpdate() |
||||||
{ |
{ |
||||||
if (args.Error == null && args.IsUpdateAvailable) |
var httpClient = httpClientFactory.CreateClient("UpdateClient"); |
||||||
|
var response = await httpClient.GetAsync("https://cdn.lykos.ai/update.json"); |
||||||
|
if (!response.IsSuccessStatusCode) |
||||||
|
{ |
||||||
|
logger.LogError("Error while checking for update"); |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
var updateInfo = |
||||||
|
await JsonSerializer.DeserializeAsync<UpdateInfo>( |
||||||
|
await response.Content.ReadAsStreamAsync()); |
||||||
|
|
||||||
|
if (updateInfo == null) |
||||||
|
{ |
||||||
|
logger.LogError("UpdateInfo is null"); |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
if (updateInfo.Version == Utilities.GetAppVersion()) |
||||||
|
{ |
||||||
|
logger.LogInformation("No update available"); |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
// check if update is newer |
||||||
|
var updateVersion = updateInfo.Version.Split('.'); |
||||||
|
var currentVersion = Utilities.GetAppVersion().Split('.'); |
||||||
|
if (updateVersion.Length != 4 || currentVersion.Length != 4) |
||||||
|
{ |
||||||
|
logger.LogError("Invalid version format"); |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
var updateVersionInt = new int[4]; |
||||||
|
var currentVersionInt = new int[4]; |
||||||
|
for (var i = 0; i < 4; i++) |
||||||
{ |
{ |
||||||
EventManager.Instance.OnUpdateAvailable(args); |
if (int.TryParse(updateVersion[i], out updateVersionInt[i]) && |
||||||
|
int.TryParse(currentVersion[i], out currentVersionInt[i])) continue; |
||||||
|
logger.LogError("Invalid version format"); |
||||||
|
return; |
||||||
} |
} |
||||||
else if (args.Error != null) |
|
||||||
|
// check if update is newer |
||||||
|
for (var i = 0; i < 4; i++) |
||||||
{ |
{ |
||||||
logger.LogError(args.Error, "Error while checking for update"); |
if (updateVersionInt[i] <= currentVersionInt[i]) continue; |
||||||
|
|
||||||
|
logger.LogInformation("Update available"); |
||||||
|
EventManager.Instance.OnUpdateAvailable(updateInfo); |
||||||
|
return; |
||||||
} |
} |
||||||
} |
} |
||||||
|
|
||||||
|
|
||||||
} |
} |
||||||
|
|||||||
@ -0,0 +1,15 @@ |
|||||||
|
using System.Reflection; |
||||||
|
|
||||||
|
namespace StabilityMatrix.Helper; |
||||||
|
|
||||||
|
public static class Utilities |
||||||
|
{ |
||||||
|
public static string GetAppVersion() |
||||||
|
{ |
||||||
|
var assembly = Assembly.GetExecutingAssembly(); |
||||||
|
var version = assembly.GetName().Version; |
||||||
|
return version == null |
||||||
|
? "(Unknown)" |
||||||
|
: $"{version.Major}.{version.Minor}.{version.Build}.{version.Revision}"; |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,15 @@ |
|||||||
|
using System.Text.Json.Serialization; |
||||||
|
|
||||||
|
namespace StabilityMatrix.Models; |
||||||
|
|
||||||
|
public class UpdateInfo |
||||||
|
{ |
||||||
|
[JsonPropertyName("version")] |
||||||
|
public string Version { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("url")] |
||||||
|
public string DownloadUrl { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("changelog")] |
||||||
|
public string ChangelogUrl { get; set; } |
||||||
|
} |
||||||
Loading…
Reference in new issue