StabilityMatrix/StabilityMatrix.Core/Models/PackageModification/UpdatePackageStep.cs

using StabilityMatrix.Core.Models.Packages;
using StabilityMatrix.Core.Models.Progress;
using StabilityMatrix.Core.Processes;
using StabilityMatrix.Core.Services;

namespace StabilityMatrix.Core.Models.PackageModification;

public class UpdatePackageStep : IPackageStep
{
    private readonly ISettingsManager settingsManager;
    private readonly InstalledPackage installedPackage;
    private readonly DownloadPackageVersionOptions versionOptions;
    private readonly BasePackage basePackage;

    public UpdatePackageStep(
        ISettingsManager settingsManager,
        InstalledPackage installedPackage,
        DownloadPackageVersionOptions versionOptions,
        BasePackage basePackage
    )
    {
        this.settingsManager = settingsManager;
        this.installedPackage = installedPackage;
        this.versionOptions = versionOptions;
        this.basePackage = basePackage;
    }

    public async Task ExecuteAsync(IProgress<ProgressReport>? progress = null)
    {
        var torchVersion = installedPackage.PreferredTorchVersion ?? basePackage.GetRecommendedTorchVersion();

        void OnConsoleOutput(ProcessOutput output)
        {
            progress?.Report(
                new ProgressReport
                {
                    IsIndeterminate = true,
                    Message = output.Text,
                    ProcessOutput = output
                }
            );
        }

        var updateResult = await basePackage
            .Update(
                installedPackage,
                torchVersion,
                versionOptions,
                progress,
                onConsoleOutput: OnConsoleOutput
            )
            .ConfigureAwait(false);

        await using (settingsManager.BeginTransaction())
        {
            installedPackage.Version = updateResult;
            installedPackage.UpdateAvailable = false;
        }
    }

    public string ProgressTitle => $"Updating {installedPackage.DisplayName}";
}