StabilityMatrix/StabilityMatrix.Avalonia/ViewModels/Progress/PackageInstallProgressItemViewModel.cs

using System;
using System.Linq;
using System.Threading.Tasks;
using Avalonia.Threading;
using FluentAvalonia.UI.Controls;
using StabilityMatrix.Avalonia.Controls;
using StabilityMatrix.Avalonia.ViewModels.Base;
using StabilityMatrix.Avalonia.Views.Dialogs;
using StabilityMatrix.Core.Helper;
using StabilityMatrix.Core.Models.PackageModification;
using StabilityMatrix.Core.Models.Progress;

namespace StabilityMatrix.Avalonia.ViewModels.Progress;

public class PackageInstallProgressItemViewModel : ProgressItemViewModelBase
{
    private readonly IPackageModificationRunner packageModificationRunner;
    private BetterContentDialog? dialog;

    public PackageInstallProgressItemViewModel(
        IPackageModificationRunner packageModificationRunner,
        bool hideCloseButton = false
    )
    {
        this.packageModificationRunner = packageModificationRunner;
        Id = packageModificationRunner.Id;
        Name = packageModificationRunner.CurrentStep?.ProgressTitle;
        Progress.Value = packageModificationRunner.CurrentProgress.Percentage;
        Progress.Text = packageModificationRunner.ConsoleOutput.LastOrDefault();
        Progress.IsIndeterminate = packageModificationRunner.CurrentProgress.IsIndeterminate;
        Progress.HideCloseButton = hideCloseButton;

        Progress.Console.StartUpdates();

        Progress.Console.Post(
            string.Join(Environment.NewLine, packageModificationRunner.ConsoleOutput)
        );

        packageModificationRunner.ProgressChanged += PackageModificationRunnerOnProgressChanged;
    }

    private void PackageModificationRunnerOnProgressChanged(object? sender, ProgressReport e)
    {
        Progress.Value = e.Percentage;
        Progress.Description = e.Message;
        Progress.IsIndeterminate = e.IsIndeterminate;
        Progress.Text = packageModificationRunner.CurrentStep?.ProgressTitle;
        Name = packageModificationRunner.CurrentStep?.ProgressTitle;
        Failed = packageModificationRunner.Failed;

        if (string.IsNullOrWhiteSpace(e.Message) || e.Message.Contains("Downloading..."))
            return;

        Progress.Console.PostLine(e.Message);
        EventManager.Instance.OnScrollToBottomRequested();

        if (
            e is { Message: not null, Percentage: >= 100 }
            && e.Message.Contains(
                packageModificationRunner.ModificationCompleteMessage ?? "Package Install Complete"
            )
            && Progress.CloseWhenFinished
        )
        {
            Dispatcher.UIThread.Post(() => dialog?.Hide());
        }

        if (Failed)
        {
            Progress.Text = "Package Modification Failed";
            Name = "Package Modification Failed";
        }
    }

    public async Task ShowProgressDialog()
    {
        Progress.CloseWhenFinished = true;
        dialog = new BetterContentDialog
        {
            MaxDialogWidth = 900,
            MinDialogWidth = 900,
            DefaultButton = ContentDialogButton.Close,
            IsPrimaryButtonEnabled = false,
            IsSecondaryButtonEnabled = false,
            IsFooterVisible = false,
            Content = new PackageModificationDialog { DataContext = Progress }
        };
        EventManager.Instance.OnToggleProgressFlyout();
        await dialog.ShowAsync();
    }
}