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

namespace StabilityMatrix.Core.Models.PackageModification;

public class InstallPackageStep : IPackageStep
{
    private readonly BasePackage package;
    private readonly TorchVersion torchVersion;
    private readonly string installPath;

    public InstallPackageStep(BasePackage package, TorchVersion torchVersion, string installPath)
    {
        this.package = package;
        this.torchVersion = torchVersion;
        this.installPath = installPath;
    }

    public async Task ExecuteAsync(IProgress<ProgressReport>? progress = null)
    {
        void OnConsoleOutput(ProcessOutput output)
        {
            progress?.Report(new ProgressReport { IsIndeterminate = true, Message = output.Text });
        }

        await package
            .InstallPackage(installPath, torchVersion, progress, OnConsoleOutput)
            .ConfigureAwait(false);
    }

    public string ProgressTitle => "Installing package...";
}