@ -1,29 +1,22 @@
using System ;
using System.Collections.Immutable ;
using System.Collections.ObjectModel ;
using System.IO ;
using System.Linq ;
using System.Threading.Tasks ;
using AsyncAwaitBestPractices ;
using Avalonia.Controls ;
using Avalonia.Controls.Notifications ;
using Avalonia.Threading ;
using CommunityToolkit.Mvvm.ComponentModel ;
using CommunityToolkit.Mvvm.Input ;
using FluentAvalonia.UI.Controls ;
using Microsoft.Extensions.Logging ;
using Polly ;
using StabilityMatrix.Avalonia.Controls ;
using StabilityMatrix.Avalonia.Services ;
using StabilityMatrix.Avalonia.ViewModels.Base ;
using StabilityMatrix.Avalonia.ViewModels.Dialogs ;
using StabilityMatrix.Avalonia.ViewModels.PackageManager ;
using StabilityMatrix.Avalonia.Views ;
using StabilityMatrix.Avalonia.Views.Dialogs ;
using StabilityMatrix.Core.Attributes ;
using StabilityMatrix.Core.Helper ;
using StabilityMatrix.Core.Helper.Factory ;
using StabilityMatrix.Core.Models ;
using StabilityMatrix.Core.Models.Packages ;
using StabilityMatrix.Core.Models.Progress ;
using StabilityMatrix.Core.Services ;
using Symbol = FluentIcons . Common . Symbol ;
using SymbolIconSource = FluentIcons . FluentAvalonia . SymbolIconSource ;
@ -37,313 +30,46 @@ namespace StabilityMatrix.Avalonia.ViewModels;
[View(typeof(PackageManagerPage))]
public partial class PackageManagerViewModel : PageViewModelBase
{
private readonly ILogger < PackageManagerViewModel > logger ;
private readonly ISettingsManager settingsManager ;
private readonly IPackageFactory packageFactory ;
private readonly INotificationService notificationService ;
private readonly ServiceManager < ViewModelBase > dialogFactory ;
private const int MinutesToWaitForUpdateCheck = 6 0 ;
public PackageManagerViewModel ( ILogger < PackageManagerViewModel > logger ,
ISettingsManager settingsManager , IPackageFactory packageFactory ,
INotificationService notificationService , ServiceManager < ViewModelBase > dialogFactory )
public PackageManagerViewModel ( ISettingsManager settingsManager , IPackageFactory packageFactory ,
ServiceManager < ViewModelBase > dialogFactory )
{
this . logger = logger ;
this . settingsManager = settingsManager ;
this . packageFactory = packageFactory ;
this . notificationService = notificationService ;
this . dialogFactory = dialogFactory ;
Packages =
new ObservableCollection < InstalledPackage > ( settingsManager . Settings . InstalledPackages ) ;
SelectedPackage = Packages . FirstOrDefault ( ) ;
EventManager . Instance . InstalledPackagesChanged + = OnInstalledPackagesChanged ;
}
[ObservableProperty]
private InstalledPackage ? selectedPackage ;
[ObservableProperty, NotifyPropertyChangedFor(nameof(ProgressBarVisibility))]
private int progressValue ;
[ObservableProperty]
private string progressText = string . Empty ;
[ObservableProperty, NotifyPropertyChangedFor(nameof(ProgressBarVisibility))]
private bool isIndeterminate ;
[ObservableProperty]
private bool isUninstalling ;
[ObservableProperty, NotifyPropertyChangedFor(nameof(SelectedPackage))]
private bool updateAvailable ;
[ObservableProperty] private string installButtonText ;
[ObservableProperty] private bool installButtonEnabled ;
public bool ProgressBarVisibility = > ProgressValue > 0 | | IsIndeterminate ;
public ObservableCollection < InstalledPackage > Packages { get ; }
[ObservableProperty] private ObservableCollection < PackageCardViewModel > packages ;
public override bool CanNavigateNext { get ; protected set ; } = true ;
public override bool CanNavigatePrevious { get ; protected set ; }
public override string Title = > "Packages" ;
public override IconSource IconSource = > new SymbolIconSource { Symbol = Symbol . Box , IsFilled = true } ;
public override async Task OnLoadedAsync ( )
{
if ( ! Design . IsDesignMode )
{
await CheckUpdates ( ) ;
}
var installedPackages = settingsManager . Settings . InstalledPackages ;
SelectedPackage = installedPackages . FirstOrDefault ( x = >
x . Id = = settingsManager . Settings . ActiveInstalledPackageId ) ;
SelectedPackage ? ? = installedPackages . FirstOrDefault ( ) ;
if ( Design . IsDesignMode ) return ;
InstallButtonEnabled = SelectedPackage ! = null ;
}
private async Task CheckUpdates ( )
{
var installedPackages = settingsManager . Settings . InstalledPackages ;
Packages . Clear ( ) ;
foreach ( var packageToUpdate in installedPackages )
{
var basePackage = packageFactory . FindPackageByName ( packageToUpdate . PackageName ) ;
if ( basePackage = = null ) continue ;
var canCheckUpdate = packageToUpdate . LastUpdateCheck = = null | |
packageToUpdate . LastUpdateCheck . Value . AddMinutes ( MinutesToWaitForUpdateCheck ) <
DateTimeOffset . Now ;
if ( canCheckUpdate )
Packages = new ObservableCollection < PackageCardViewModel > ( installedPackages . Select (
package = > dialogFactory . Get < PackageCardViewModel > ( vm = >
{
var hasUpdate = await basePackage . CheckForUpdates ( packageToUpdate ) ;
packageToUpdate . UpdateAvailable = hasUpdate ;
packageToUpdate . LastUpdateCheck = DateTimeOffset . Now ;
settingsManager . SetLastUpdateCheck ( packageToUpdate ) ;
}
Packages . Add ( packageToUpdate ) ;
}
}
public async Task Launch ( )
{
if ( SelectedPackage = = null ) return ;
if ( InstallButtonText = = "Launch" )
{
settingsManager . Transaction ( s = > s . ActiveInstalledPackageId = SelectedPackage . Id ) ;
EventManager . Instance . RequestPageChange ( typeof ( LaunchPageViewModel ) ) ;
EventManager . Instance . OnPackageLaunchRequested ( SelectedPackage . Id ) ;
}
else
{
await UpdateSelectedPackage ( ) ;
}
}
public async Task Uninstall ( )
{
// In design mode, just remove the package from the list
if ( Design . IsDesignMode )
{
if ( SelectedPackage ! = null )
{
Packages . Remove ( SelectedPackage ) ;
}
return ;
}
vm . Package = package ;
return vm ;
} ) ) ) ;
if ( SelectedPackage ? . LibraryPath = = null )
foreach ( var package in Packages )
{
logger . LogError ( "No package selected to uninstall" ) ;
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 )
{
IsUninstalling = true ;
InstallButtonEnabled = false ;
var deleteTask = DeleteDirectoryAsync ( Path . Combine ( settingsManager . LibraryDir ,
SelectedPackage . 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 {SelectedPackage.DisplayName} uninstalled" ,
NotificationType . Success ) ) ;
settingsManager . Transaction ( settings = >
{
settings . RemoveInstalledPackageAndUpdateActive ( SelectedPackage ) ;
} ) ;
}
IsUninstalling = false ;
InstallButtonEnabled = true ;
await OnLoadedAsync ( ) ;
await package . OnLoadedAsync ( ) ;
}
}
partial void OnSelectedPackageChanged ( InstalledPackage ? value )
{
if ( value is null ) return ;
UpdateAvailable = value . UpdateAvailable ;
InstallButtonText = value . UpdateAvailable ? "Update" : "Launch" ;
}
/// <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 ( 5 0 * Math . Pow ( 2 , attempt ) ) ,
onRetry : ( exception , calculatedWaitDuration ) = >
{
logger . LogWarning (
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 . LogInformation ( "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 ) ;
}
}
private async Task UpdateSelectedPackage ( )
{
if ( SelectedPackage = = null ) return ;
var installedPackage = SelectedPackage ;
var package = packageFactory . FindPackageByName ( installedPackage . PackageName ) ;
if ( package = = null )
{
logger . LogError ( "Could not find package {SelectedPackagePackageName}" ,
installedPackage . PackageName ) ;
return ;
}
ProgressText = $"Updating {installedPackage.DisplayName} to latest version..." ;
package . InstallLocation = installedPackage . FullPath ! ;
var progress = new Progress < ProgressReport > ( progress = >
{
var percent = Convert . ToInt32 ( progress . Percentage ) ;
ProgressValue = percent ;
IsIndeterminate = progress . IsIndeterminate ;
ProgressText = $"Updating {SelectedPackage.DisplayName} to latest version... {percent:N0}%" ;
EventManager . Instance . OnGlobalProgressChanged ( percent ) ;
} ) ;
var updateResult = await package . Update ( installedPackage , progress ) ;
if ( string . IsNullOrWhiteSpace ( updateResult ) )
{
var errorMsg =
$"There was an error updating {installedPackage.DisplayName}. Please try again later." ;
if ( installedPackage . 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 ( installedPackage . Id , updateResult ) ;
notificationService . Show ( "Update complete" ,
$"{installedPackage.DisplayName} has been updated to the latest version." ,
NotificationType . Success ) ;
installedPackage . UpdateAvailable = false ;
UpdateAvailable = false ;
InstallButtonText = "Launch" ;
await OnLoadedAsync ( ) ;
await DelayedClearProgress ( TimeSpan . FromSeconds ( 3 ) ) ;
}
[RelayCommand]
private async Task ShowInstallDialog ( )
public async Task ShowInstallDialog ( )
{
var viewModel = dialogFactory . Get < InstallerViewModel > ( ) ;
viewModel . AvailablePackages = packageFactory . GetAllAvailablePackages ( ) . ToImmutableArray ( ) ;
@ -366,13 +92,7 @@ public partial class PackageManagerViewModel : PageViewModelBase
await dialog . ShowAsync ( ) ;
await OnLoadedAsync ( ) ;
}
private async Task DelayedClearProgress ( TimeSpan delay )
{
await Task . Delay ( delay ) ;
ProgressText = string . Empty ;
ProgressValue = 0 ;
IsIndeterminate = false ;
}
private void OnInstalledPackagesChanged ( object? sender , EventArgs e ) = >
OnLoadedAsync ( ) . SafeFireAndForget ( ) ;
}
Ionite
1 year ago