JT
1 year ago
committed by
GitHub
16 changed files with 608 additions and 490 deletions
@ -0,0 +1,289 @@ |
|||||||
|
using System; |
||||||
|
using System.IO; |
||||||
|
using System.Linq; |
||||||
|
using System.Threading.Tasks; |
||||||
|
using AsyncAwaitBestPractices; |
||||||
|
using Avalonia.Controls.Notifications; |
||||||
|
using Avalonia.Threading; |
||||||
|
using CommunityToolkit.Mvvm.ComponentModel; |
||||||
|
using FluentAvalonia.UI.Controls; |
||||||
|
using NLog; |
||||||
|
using Polly; |
||||||
|
using StabilityMatrix.Avalonia.Services; |
||||||
|
using StabilityMatrix.Core.Helper; |
||||||
|
using StabilityMatrix.Core.Helper.Factory; |
||||||
|
using StabilityMatrix.Core.Models; |
||||||
|
using StabilityMatrix.Core.Models.Progress; |
||||||
|
using StabilityMatrix.Core.Processes; |
||||||
|
using StabilityMatrix.Core.Services; |
||||||
|
|
||||||
|
namespace StabilityMatrix.Avalonia.ViewModels.PackageManager; |
||||||
|
|
||||||
|
public partial class PackageCardViewModel : ProgressViewModel |
||||||
|
{ |
||||||
|
private readonly IPackageFactory packageFactory; |
||||||
|
private readonly INotificationService notificationService; |
||||||
|
private readonly ISettingsManager settingsManager; |
||||||
|
private readonly Logger logger = LogManager.GetCurrentClassLogger(); |
||||||
|
|
||||||
|
[ObservableProperty] private InstalledPackage? package; |
||||||
|
[ObservableProperty] private Uri cardImage; |
||||||
|
[ObservableProperty] private bool isUpdateAvailable; |
||||||
|
[ObservableProperty] private string installedVersion; |
||||||
|
|
||||||
|
public PackageCardViewModel(IPackageFactory packageFactory, |
||||||
|
INotificationService notificationService, ISettingsManager settingsManager) |
||||||
|
{ |
||||||
|
this.packageFactory = packageFactory; |
||||||
|
this.notificationService = notificationService; |
||||||
|
this.settingsManager = settingsManager; |
||||||
|
} |
||||||
|
|
||||||
|
partial void OnPackageChanged(InstalledPackage? value) |
||||||
|
{ |
||||||
|
if (string.IsNullOrWhiteSpace(value?.PackageName)) |
||||||
|
return; |
||||||
|
|
||||||
|
var basePackage = packageFactory[value.PackageName]; |
||||||
|
CardImage = basePackage?.PreviewImageUri ?? Assets.NoImage; |
||||||
|
InstalledVersion = value.DisplayVersion ?? "Unknown"; |
||||||
|
} |
||||||
|
|
||||||
|
public override async Task OnLoadedAsync() |
||||||
|
{ |
||||||
|
IsUpdateAvailable = await HasUpdate(); |
||||||
|
} |
||||||
|
|
||||||
|
public void Launch() |
||||||
|
{ |
||||||
|
if (Package == null) |
||||||
|
return; |
||||||
|
|
||||||
|
settingsManager.Transaction(s => s.ActiveInstalledPackageId = Package.Id); |
||||||
|
EventManager.Instance.RequestPageChange(typeof(LaunchPageViewModel)); |
||||||
|
EventManager.Instance.OnPackageLaunchRequested(Package.Id); |
||||||
|
} |
||||||
|
|
||||||
|
public async Task Uninstall() |
||||||
|
{ |
||||||
|
if (Package?.LibraryPath == null) |
||||||
|
{ |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
var dialog = new ContentDialog |
||||||
|
{ |
||||||
|
Title = "Are you sure?", |
||||||
|
Content = "This will delete all folders in the package directory, including any generated images in that directory as well as any files you may have added.", |
||||||
|
PrimaryButtonText = "Yes, delete it", |
||||||
|
CloseButtonText = "No, keep it", |
||||||
|
DefaultButton = ContentDialogButton.Primary |
||||||
|
}; |
||||||
|
var result = await dialog.ShowAsync(); |
||||||
|
|
||||||
|
if (result == ContentDialogResult.Primary) |
||||||
|
{ |
||||||
|
Text = "Uninstalling..."; |
||||||
|
IsIndeterminate = true; |
||||||
|
Value = -1; |
||||||
|
|
||||||
|
var deleteTask = DeleteDirectoryAsync(Path.Combine(settingsManager.LibraryDir, |
||||||
|
Package.LibraryPath)); |
||||||
|
var taskResult = await notificationService.TryAsync(deleteTask, |
||||||
|
"Some files could not be deleted. Please close any open files in the package directory and try again."); |
||||||
|
if (taskResult.IsSuccessful) |
||||||
|
{ |
||||||
|
notificationService.Show(new Notification("Success", |
||||||
|
$"Package {Package.DisplayName} uninstalled", |
||||||
|
NotificationType.Success)); |
||||||
|
|
||||||
|
settingsManager.Transaction(settings => |
||||||
|
{ |
||||||
|
settings.RemoveInstalledPackageAndUpdateActive(Package); |
||||||
|
}); |
||||||
|
|
||||||
|
EventManager.Instance.OnInstalledPackagesChanged(); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
public async Task Update() |
||||||
|
{ |
||||||
|
if (Package == null) return; |
||||||
|
var basePackage = packageFactory[Package.PackageName!]; |
||||||
|
if (basePackage == null) |
||||||
|
{ |
||||||
|
logger.Error("Could not find package {SelectedPackagePackageName}", |
||||||
|
Package.PackageName); |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
Text = $"Updating {Package.DisplayName}"; |
||||||
|
basePackage.InstallLocation = Package.FullPath!; |
||||||
|
var progress = new Progress<ProgressReport>(progress => |
||||||
|
{ |
||||||
|
var percent = Convert.ToInt32(progress.Percentage); |
||||||
|
|
||||||
|
Value = percent; |
||||||
|
IsIndeterminate = progress.IsIndeterminate; |
||||||
|
Text = $"Updating {Package.DisplayName}"; |
||||||
|
|
||||||
|
EventManager.Instance.OnGlobalProgressChanged(percent); |
||||||
|
}); |
||||||
|
|
||||||
|
var updateResult = await basePackage.Update(Package, progress); |
||||||
|
|
||||||
|
if (string.IsNullOrWhiteSpace(updateResult)) |
||||||
|
{ |
||||||
|
var errorMsg = |
||||||
|
$"There was an error updating {Package.DisplayName}. Please try again later."; |
||||||
|
|
||||||
|
if (Package.PackageName == "automatic") |
||||||
|
{ |
||||||
|
errorMsg = errorMsg.Replace("Please try again later.", |
||||||
|
"Please stash any changes before updating, or manually update the package."); |
||||||
|
} |
||||||
|
|
||||||
|
// there was an error |
||||||
|
notificationService.Show(new Notification("Error updating package", |
||||||
|
errorMsg, NotificationType.Error)); |
||||||
|
} |
||||||
|
|
||||||
|
settingsManager.UpdatePackageVersionNumber(Package.Id, updateResult); |
||||||
|
notificationService.Show("Update complete", |
||||||
|
$"{Package.DisplayName} has been updated to the latest version.", |
||||||
|
NotificationType.Success); |
||||||
|
|
||||||
|
|
||||||
|
Dispatcher.UIThread.Invoke(() => |
||||||
|
{ |
||||||
|
using (settingsManager.BeginTransaction()) |
||||||
|
{ |
||||||
|
Package.UpdateAvailable = false; |
||||||
|
} |
||||||
|
IsUpdateAvailable = false; |
||||||
|
IsIndeterminate = false; |
||||||
|
InstalledVersion = settingsManager.Settings.InstalledPackages |
||||||
|
.First(x => x.Id == Package.Id).DisplayVersion ?? "Unknown"; |
||||||
|
Value = 0; |
||||||
|
}); |
||||||
|
} |
||||||
|
|
||||||
|
public async Task OpenFolder() |
||||||
|
{ |
||||||
|
if (string.IsNullOrWhiteSpace(Package?.FullPath)) |
||||||
|
return; |
||||||
|
|
||||||
|
await ProcessRunner.OpenFolderBrowser(Package.FullPath); |
||||||
|
} |
||||||
|
|
||||||
|
private async Task<bool> HasUpdate() |
||||||
|
{ |
||||||
|
if (Package == null) |
||||||
|
return false; |
||||||
|
|
||||||
|
var basePackage = packageFactory[Package.PackageName!]; |
||||||
|
if (basePackage == null) |
||||||
|
return false; |
||||||
|
|
||||||
|
var canCheckUpdate = Package.LastUpdateCheck == null || |
||||||
|
Package.LastUpdateCheck < DateTime.Now.AddMinutes(-15); |
||||||
|
|
||||||
|
if (!canCheckUpdate) |
||||||
|
{ |
||||||
|
return Package.UpdateAvailable; |
||||||
|
} |
||||||
|
|
||||||
|
try |
||||||
|
{ |
||||||
|
var hasUpdate = await basePackage.CheckForUpdates(Package); |
||||||
|
Package.UpdateAvailable = hasUpdate; |
||||||
|
Package.LastUpdateCheck = DateTimeOffset.Now; |
||||||
|
settingsManager.SetLastUpdateCheck(Package); |
||||||
|
return hasUpdate; |
||||||
|
} |
||||||
|
catch (Exception e) |
||||||
|
{ |
||||||
|
logger.Error(e, $"Error checking {Package.PackageName} for updates"); |
||||||
|
return false; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/// <summary> |
||||||
|
/// Deletes a directory and all of its contents recursively. |
||||||
|
/// Uses Polly to retry the deletion if it fails, up to 5 times with an exponential backoff. |
||||||
|
/// </summary> |
||||||
|
/// <param name="targetDirectory"></param> |
||||||
|
private Task DeleteDirectoryAsync(string targetDirectory) |
||||||
|
{ |
||||||
|
var policy = Policy.Handle<IOException>() |
||||||
|
.WaitAndRetryAsync(3, attempt => TimeSpan.FromMilliseconds(50 * Math.Pow(2, attempt)), |
||||||
|
onRetry: (exception, calculatedWaitDuration) => |
||||||
|
{ |
||||||
|
logger.Warn( |
||||||
|
exception, |
||||||
|
"Deletion of {TargetDirectory} failed. Retrying in {CalculatedWaitDuration}", |
||||||
|
targetDirectory, calculatedWaitDuration); |
||||||
|
}); |
||||||
|
|
||||||
|
return policy.ExecuteAsync(async () => |
||||||
|
{ |
||||||
|
await Task.Run(() => |
||||||
|
{ |
||||||
|
DeleteDirectory(targetDirectory); |
||||||
|
}); |
||||||
|
}); |
||||||
|
} |
||||||
|
|
||||||
|
private void DeleteDirectory(string targetDirectory) |
||||||
|
{ |
||||||
|
// Skip if directory does not exist |
||||||
|
if (!Directory.Exists(targetDirectory)) |
||||||
|
{ |
||||||
|
return; |
||||||
|
} |
||||||
|
// For junction points, delete with recursive false |
||||||
|
if (new DirectoryInfo(targetDirectory).LinkTarget != null) |
||||||
|
{ |
||||||
|
logger.Info("Removing junction point {TargetDirectory}", targetDirectory); |
||||||
|
try |
||||||
|
{ |
||||||
|
Directory.Delete(targetDirectory, false); |
||||||
|
return; |
||||||
|
} |
||||||
|
catch (IOException ex) |
||||||
|
{ |
||||||
|
throw new IOException($"Failed to delete junction point {targetDirectory}", ex); |
||||||
|
} |
||||||
|
} |
||||||
|
// Recursively delete all subdirectories |
||||||
|
var subdirectoryEntries = Directory.GetDirectories(targetDirectory); |
||||||
|
foreach (var subdirectoryPath in subdirectoryEntries) |
||||||
|
{ |
||||||
|
DeleteDirectory(subdirectoryPath); |
||||||
|
} |
||||||
|
// Delete all files in the directory |
||||||
|
var fileEntries = Directory.GetFiles(targetDirectory); |
||||||
|
foreach (var filePath in fileEntries) |
||||||
|
{ |
||||||
|
try |
||||||
|
{ |
||||||
|
File.SetAttributes(filePath, FileAttributes.Normal); |
||||||
|
File.Delete(filePath); |
||||||
|
} |
||||||
|
catch (IOException ex) |
||||||
|
{ |
||||||
|
throw new IOException($"Failed to delete file {filePath}", ex); |
||||||
|
} |
||||||
|
} |
||||||
|
// Delete the target directory itself |
||||||
|
try |
||||||
|
{ |
||||||
|
Directory.Delete(targetDirectory, false); |
||||||
|
} |
||||||
|
catch (IOException ex) |
||||||
|
{ |
||||||
|
throw new IOException($"Failed to delete directory {targetDirectory}", ex); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue