From 227071258cf2a3de86fd859c0204bb18318c0d1a Mon Sep 17 00:00:00 2001
From: JT <mohnjiles@gmail.com>
Date: Wed, 31 May 2023 00:41:55 -0700
Subject: [PATCH 1/8] setup one-click install foundations

---
 StabilityMatrix/App.xaml.cs                   |   2 +
 StabilityMatrix/Helper/EventManager.cs        |   2 +
 StabilityMatrix/Helper/IPrerequisiteHelper.cs |   9 +
 StabilityMatrix/Helper/PrerequisiteHelper.cs  |  38 ++++
 StabilityMatrix/LaunchOptionsDialog.xaml.cs   |   2 +
 StabilityMatrix/MainWindow.xaml               |  67 +++++--
 .../ViewModels/InstallerViewModel.cs          |   2 +-
 StabilityMatrix/ViewModels/LaunchViewModel.cs |  11 +-
 .../ViewModels/MainWindowViewModel.cs         | 165 ++++++++++++++++--
 9 files changed, 272 insertions(+), 26 deletions(-)
 create mode 100644 StabilityMatrix/Helper/IPrerequisiteHelper.cs
 create mode 100644 StabilityMatrix/Helper/PrerequisiteHelper.cs

diff --git a/StabilityMatrix/App.xaml.cs b/StabilityMatrix/App.xaml.cs
index c59d55a6..8049ba93 100644
--- a/StabilityMatrix/App.xaml.cs
+++ b/StabilityMatrix/App.xaml.cs
@@ -9,6 +9,7 @@ using Microsoft.Extensions.Configuration;
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Logging;
 using NLog.Extensions.Logging;
+using NLog.Layouts;
 using Polly;
 using Polly.Contrib.WaitAndRetry;
 using Polly.Extensions.Http;
@@ -82,6 +83,7 @@ namespace StabilityMatrix
             serviceCollection.AddSingleton<BasePackage, VladAutomatic>();
             serviceCollection.AddSingleton<ISnackbarService, SnackbarService>();
             serviceCollection.AddSingleton<ISettingsManager, SettingsManager>();
+            serviceCollection.AddSingleton<IPrerequisiteHelper, PrerequisiteHelper>();
             serviceCollection.AddSingleton<IDialogErrorHandler, DialogErrorHandler>();
             serviceCollection.AddMemoryCache();
             serviceCollection.AddSingleton<IGithubApiCache, GithubApiCache>();
diff --git a/StabilityMatrix/Helper/EventManager.cs b/StabilityMatrix/Helper/EventManager.cs
index 06ebed05..2a892ae6 100644
--- a/StabilityMatrix/Helper/EventManager.cs
+++ b/StabilityMatrix/Helper/EventManager.cs
@@ -13,6 +13,8 @@ public class EventManager
     
     public event EventHandler<int>? GlobalProgressChanged;
     public event EventHandler<Type>? PageChangeRequested;
+    public event EventHandler? InstalledPackagesChanged;
     public void OnGlobalProgressChanged(int progress) => GlobalProgressChanged?.Invoke(this, progress);
     public void RequestPageChange(Type pageType) => PageChangeRequested?.Invoke(this, pageType);
+    public void OnInstalledPackagesChanged() => InstalledPackagesChanged?.Invoke(this, EventArgs.Empty);
 }
diff --git a/StabilityMatrix/Helper/IPrerequisiteHelper.cs b/StabilityMatrix/Helper/IPrerequisiteHelper.cs
new file mode 100644
index 00000000..f999a374
--- /dev/null
+++ b/StabilityMatrix/Helper/IPrerequisiteHelper.cs
@@ -0,0 +1,9 @@
+using System.Diagnostics;
+using System.Threading.Tasks;
+
+namespace StabilityMatrix.Helper;
+
+public interface IPrerequisiteHelper
+{
+    Task<Process?> InstallGitIfNecessary();
+}
\ No newline at end of file
diff --git a/StabilityMatrix/Helper/PrerequisiteHelper.cs b/StabilityMatrix/Helper/PrerequisiteHelper.cs
new file mode 100644
index 00000000..3e6076bf
--- /dev/null
+++ b/StabilityMatrix/Helper/PrerequisiteHelper.cs
@@ -0,0 +1,38 @@
+using System;
+using System.Diagnostics;
+using System.Threading.Tasks;
+using Microsoft.Extensions.Logging;
+
+namespace StabilityMatrix.Helper;
+
+public class PrerequisiteHelper : IPrerequisiteHelper
+{
+    private readonly ILogger<PrerequisiteHelper> logger;
+
+    public PrerequisiteHelper(ILogger<PrerequisiteHelper> logger)
+    {
+        this.logger = logger;
+    }
+    
+    public async Task<Process?> InstallGitIfNecessary()
+    {
+        try
+        {
+            var gitOutput = await ProcessRunner.GetProcessOutputAsync("git", "--version");
+            if (gitOutput.Contains("git version 2"))
+            {
+                return default;
+            }
+        }
+        catch (Exception e)
+        {
+            logger.LogError(e, "Error running git: ");
+        }
+
+        var installProcess =
+            ProcessRunner.StartProcess("Assets\\Git-2.40.1-64-bit.exe", "/VERYSILENT /NORESTART");
+        installProcess.OutputDataReceived += (sender, args) => { logger.LogDebug(args.Data); };
+
+        return installProcess;
+    }
+}
diff --git a/StabilityMatrix/LaunchOptionsDialog.xaml.cs b/StabilityMatrix/LaunchOptionsDialog.xaml.cs
index 63d0a7ba..a9d521d8 100644
--- a/StabilityMatrix/LaunchOptionsDialog.xaml.cs
+++ b/StabilityMatrix/LaunchOptionsDialog.xaml.cs
@@ -1,4 +1,5 @@
 using System.Collections.Generic;
+using System.Diagnostics;
 using System.Windows;
 using System.Windows.Controls;
 using System.Windows.Input;
@@ -19,6 +20,7 @@ public partial class LaunchOptionsDialog : ContentDialog
         dialogService.GetContentPresenter())
     {
         this.viewModel = viewModel;
+        var sw = new Stopwatch();
         InitializeComponent();
         DataContext = viewModel;
     }
diff --git a/StabilityMatrix/MainWindow.xaml b/StabilityMatrix/MainWindow.xaml
index 6e450f6b..e5fbb67c 100644
--- a/StabilityMatrix/MainWindow.xaml
+++ b/StabilityMatrix/MainWindow.xaml
@@ -8,6 +8,8 @@
     Title="Stability Matrix"
     Width="1100"
     WindowBackdropType="Mica"
+    d:DesignHeight="500"
+    d:DesignWidth="700"
     d:DataContext="{d:DesignInstance Type=viewModels:MainWindowViewModel,
                                      IsDesignTimeCreatable=True}"
     mc:Ignorable="d"
@@ -18,7 +20,16 @@
     xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006"
     xmlns:ui="http://schemas.lepo.co/wpfui/2022/xaml"
     xmlns:viewModels="clr-namespace:StabilityMatrix.ViewModels"
-    xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml">
+    xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
+    xmlns:converters="clr-namespace:StabilityMatrix.Converters">
+
+    <ui:FluentWindow.Resources>
+        <BooleanToVisibilityConverter x:Key="BoolToVisConverter" />
+        <converters:ValueConverterGroup x:Key="InvertAndVisibilitate">
+            <converters:BoolNegationConverter />
+            <BooleanToVisibilityConverter />
+        </converters:ValueConverterGroup>
+    </ui:FluentWindow.Resources>
 
     <ui:FluentWindow.TaskbarItemInfo>
         <TaskbarItemInfo ProgressState="{Binding ProgressState}" ProgressValue="{Binding ProgressValue}" />
@@ -35,7 +46,10 @@
                 <TextBlock Margin="16,8" Text="Stability Matrix" />
             </ui:TitleBar.Header>
         </ui:TitleBar>
-        <Grid Grid.Row="1" Visibility="{Binding AdvancedModeVisibility, FallbackValue=Visible}">
+        <Grid Grid.Row="1"
+              Visibility="{Binding IsAdvancedMode, 
+                            Converter={StaticResource BoolToVisConverter}, 
+                            FallbackValue=Hidden}">
             <ui:NavigationView
                 IsBackButtonVisible="Collapsed"
                 PaneClosed="RootNavigation_OnPaneClosed"
@@ -81,16 +95,47 @@
                 </ui:NavigationView.ContentOverlay>
             </ui:NavigationView>
         </Grid>
-        <Grid Grid.Row="1" Visibility="{Binding SimpleModeVisibility, FallbackValue=Hidden}">
-            <ui:Button
-                Content="Install"
-                FontSize="36"
-                HorizontalAlignment="Center"
-                Padding="16" />
-            <Button
+        <Grid Grid.Row="1"
+              Visibility="{Binding IsAdvancedMode,
+               Converter={StaticResource InvertAndVisibilitate}, 
+                FallbackValue=Visible}">
+            <Grid.RowDefinitions>
+                <RowDefinition Height="Auto"/>
+                <RowDefinition Height="Auto"/>
+                <RowDefinition Height="1*"/>
+                <RowDefinition Height="0.25*"/>
+            </Grid.RowDefinitions>
+            <TextBlock Text="{Binding HeaderText}"
+                       TextWrapping="Wrap"
+                       VerticalAlignment="Top" HorizontalAlignment="Center"
+                       FontSize="36" FontWeight="Light" Margin="8"/>
+            <TextBlock Grid.Row="1"
+                       Text="{Binding SubHeaderText}"
+                       TextWrapping="Wrap" TextAlignment="Center"
+                       VerticalAlignment="Top" HorizontalAlignment="Center"
+                       FontSize="16" FontWeight="Light" Margin="8">
+            </TextBlock>
+            
+            <StackPanel Grid.Row="2" VerticalAlignment="Center">
+                <ProgressBar Maximum="100" Value="{Binding OneClickInstallProgress}"
+                             Visibility="{Binding ProgressBarVisibility}"
+                             IsIndeterminate="{Binding IsIndeterminate}"
+                             Margin="16"/>
+                <ui:Button
+                    Visibility="{Binding ShowInstallButton, Converter={StaticResource BoolToVisConverter}}"
+                    Content="{Binding InstallButtonText, FallbackValue=Install}"
+                    Command="{Binding InstallCommand}"
+                    FontSize="32"
+                    HorizontalAlignment="Center"
+                    Margin="16"
+                    Padding="16, 8, 16, 8" />
+            </StackPanel>
+            
+            <Button Grid.Row="3"
                 Background="Transparent"
-                Command="{Binding GoAdvancedModeCommand}"
-                Content="Advanced Options"
+                Visibility="{Binding ShowInstallButton, Converter={StaticResource BoolToVisConverter}}"
+                Command="{Binding ToggleAdvancedModeCommand}"
+                Content="Skip first-time setup"
                 FontSize="14"
                 HorizontalAlignment="Center"
                 Margin="16"
diff --git a/StabilityMatrix/ViewModels/InstallerViewModel.cs b/StabilityMatrix/ViewModels/InstallerViewModel.cs
index 65043f8d..23431cb2 100644
--- a/StabilityMatrix/ViewModels/InstallerViewModel.cs
+++ b/StabilityMatrix/ViewModels/InstallerViewModel.cs
@@ -255,7 +255,7 @@ public partial class InstallerViewModel : ObservableObject
         settingsManager.SetActiveInstalledPackage(package);
     }
     
-    private Task<string?> DownloadPackage(string? version = null)
+    private Task<string?> DownloadPackage(string version)
     {
         SelectedPackage.DownloadProgressChanged += SelectedPackageOnProgressChanged;
         SelectedPackage.DownloadComplete += (_, _) => ProgressText = "Download Complete";
diff --git a/StabilityMatrix/ViewModels/LaunchViewModel.cs b/StabilityMatrix/ViewModels/LaunchViewModel.cs
index 278124d8..1ce41dd8 100644
--- a/StabilityMatrix/ViewModels/LaunchViewModel.cs
+++ b/StabilityMatrix/ViewModels/LaunchViewModel.cs
@@ -1,4 +1,5 @@
 using System;
+using System.Collections.Generic;
 using System.Collections.ObjectModel;
 using System.Diagnostics;
 using System.Linq;
@@ -15,6 +16,7 @@ using StabilityMatrix.Models.Packages;
 using StabilityMatrix.Python;
 using Wpf.Ui.Contracts;
 using Wpf.Ui.Controls.ContentDialogControl;
+using EventManager = StabilityMatrix.Helper.EventManager;
 
 namespace StabilityMatrix.ViewModels;
 
@@ -79,10 +81,17 @@ public partial class LaunchViewModel : ObservableObject
         this.settingsManager = settingsManager;
         this.packageFactory = packageFactory;
         SetProcessRunning(false);
+        
+        EventManager.Instance.InstalledPackagesChanged += OnInstalledPackagesChanged;
 
         ToastNotificationManagerCompat.OnActivated += ToastNotificationManagerCompatOnOnActivated;
     }
 
+    private void OnInstalledPackagesChanged(object? sender, EventArgs e)
+    {
+        OnLoaded();
+    }
+
     private void ToastNotificationManagerCompatOnOnActivated(ToastNotificationActivatedEventArgsCompat e)
     {
         if (e.Argument.StartsWith("http"))
@@ -222,7 +231,7 @@ public partial class LaunchViewModel : ObservableObject
     private void LoadPackages()
     {
         var packages = settingsManager.Settings.InstalledPackages;
-        if (!packages.Any())
+        if (!packages?.Any() ?? true)
         {
             return;
         }
diff --git a/StabilityMatrix/ViewModels/MainWindowViewModel.cs b/StabilityMatrix/ViewModels/MainWindowViewModel.cs
index f9f28745..5270d08e 100644
--- a/StabilityMatrix/ViewModels/MainWindowViewModel.cs
+++ b/StabilityMatrix/ViewModels/MainWindowViewModel.cs
@@ -1,11 +1,15 @@
 using System;
+using System.Diagnostics;
+using System.Linq;
 using System.Threading.Tasks;
 using System.Windows;
 using System.Windows.Threading;
 using System.Windows.Shell;
 using CommunityToolkit.Mvvm.ComponentModel;
 using CommunityToolkit.Mvvm.Input;
+using Microsoft.Extensions.Logging;
 using StabilityMatrix.Helper;
+using StabilityMatrix.Models;
 using Wpf.Ui.Appearance;
 using EventManager = StabilityMatrix.Helper.EventManager;
 
@@ -14,29 +18,171 @@ namespace StabilityMatrix.ViewModels;
 public partial class MainWindowViewModel : ObservableObject
 {
     private readonly ISettingsManager settingsManager;
+    private readonly IPackageFactory packageFactory;
+    private readonly IPrerequisiteHelper prerequisiteHelper;
+    private readonly ILogger<MainWindowViewModel> logger;
+    private const string DefaultPackageName = "stable-diffusion-webui";
 
-    public MainWindowViewModel(ISettingsManager settingsManager)
+    public MainWindowViewModel(ISettingsManager settingsManager, IPackageFactory packageFactory,
+        IPrerequisiteHelper prerequisiteHelper, ILogger<MainWindowViewModel> logger)
     {
         this.settingsManager = settingsManager;
+        this.packageFactory = packageFactory;
+        this.prerequisiteHelper = prerequisiteHelper;
+        this.logger = logger;
     }
 
+    [ObservableProperty] 
+    private bool isAdvancedMode;
+    
     [ObservableProperty]
-    private Visibility advancedModeVisibility;
+    private float progressValue;
 
-    [ObservableProperty]
-    private Visibility simpleModeVisibility;
+    [ObservableProperty] 
+    private string headerText;
     
     [ObservableProperty]
-    private float progressValue;
+    private string subHeaderText;
+
+    [ObservableProperty]
+    [NotifyPropertyChangedFor(nameof(ProgressBarVisibility))]
+    private int oneClickInstallProgress;
 
     [ObservableProperty] 
+    private bool isIndeterminate;
+
+    [ObservableProperty]   
     private TaskbarItemProgressState progressState;
 
+    [ObservableProperty]
+    private bool showInstallButton;
+
+    [ObservableProperty] 
+    private string installButtonText;
+    
+    public Visibility ProgressBarVisibility => ProgressValue > 0 ? Visibility.Visible : Visibility.Collapsed;
+
+    private string webUrl = string.Empty;
+
     public void OnLoaded()
     {
         SetTheme();
-        GoAdvancedMode();
+        ShowInstallButton = true;
         EventManager.Instance.GlobalProgressChanged += OnGlobalProgressChanged;
+
+        if (settingsManager.Settings.InstalledPackages.Any())
+        {
+            IsAdvancedMode = true;
+        }
+        else
+        {
+            IsAdvancedMode = false;
+            InstallButtonText = "Install";
+            HeaderText = "Click the Install button below to get started!";
+            SubHeaderText =
+                "This will install the latest version of Stable Diffusion WebUI by Automatic1111.\nIf you don't know what this means, don't worry, you'll be generating images soon!";
+        }
+    }
+    
+    [RelayCommand]
+    private void ToggleAdvancedMode()
+    {
+        IsAdvancedMode = !IsAdvancedMode;
+        EventManager.Instance.RequestPageChange(typeof(PackageManagerPage));
+    }
+
+    [RelayCommand]
+    private async Task Install()
+    {
+        if (InstallButtonText == "Install")
+        {
+            ShowInstallButton = false;
+            await DoInstall();
+            ShowInstallButton = true;
+        }
+        else
+        {
+            ToggleAdvancedMode();
+        }
+    }
+
+    private async Task DoInstall()
+    {
+        var a1111 = packageFactory.FindPackageByName(DefaultPackageName)!;
+        HeaderText = "Installing Stable Diffusion WebUI...";
+        // check git
+        var gitProcess = await prerequisiteHelper.InstallGitIfNecessary();
+        if (gitProcess != null) // git isn't installed
+        {
+            IsIndeterminate = true;
+            SubHeaderText = "Installing git...";
+            await gitProcess.WaitForExitAsync();
+            if (gitProcess.ExitCode != 0)
+            {
+                HeaderText = "Installation failed";
+                SubHeaderText = "Error installing git. Please try again later.";
+                OneClickInstallProgress = 0;
+                logger.LogError($"Git install failed with exit code {gitProcess.ExitCode}");
+            }
+        }
+
+        // get latest version & download & install
+        SubHeaderText = "Getting latest version...";
+        var latestVersion = await a1111.GetLatestVersion();
+        a1111.InstallLocation += "\\stable-diffusion-webui";
+        await DownloadPackage(a1111, latestVersion);
+        await InstallPackage(a1111);
+        
+        var package = new InstalledPackage
+        {
+            DisplayName = a1111.DisplayName,
+            Path = a1111.InstallLocation,
+            Id = Guid.NewGuid(),
+            PackageName = a1111.Name,
+            PackageVersion = latestVersion,
+            LaunchCommand = a1111.LaunchCommand,
+            LastUpdateCheck = DateTimeOffset.Now
+        };
+        settingsManager.AddInstalledPackage(package);
+        settingsManager.SetActiveInstalledPackage(package);
+        EventManager.Instance.OnInstalledPackagesChanged();
+        
+        HeaderText = "Installation complete!";
+        OneClickInstallProgress = 100;
+        SubHeaderText = "Proceeding to Launch page in 5 seconds...";
+        await Task.Delay(5000);
+        IsAdvancedMode = true;
+    }
+
+    private Task<string?> DownloadPackage(BasePackage selectedPackage, string version)
+    {
+        selectedPackage.DownloadProgressChanged += SelectedPackageOnProgressChanged;
+        selectedPackage.DownloadComplete += (_, _) => SubHeaderText = "Download Complete";
+        SubHeaderText = "Downloading package...";
+        return selectedPackage.DownloadPackage(version: version);
+    }
+
+    private async Task InstallPackage(BasePackage selectedPackage)
+    {
+        selectedPackage.InstallProgressChanged += SelectedPackageOnProgressChanged;
+        selectedPackage.InstallComplete += (_, _) => HeaderText = "Install Complete";
+        SubHeaderText = "Installing package...";
+        await selectedPackage.InstallPackage();
+    }
+    
+    private void SelectedPackageOnProgressChanged(object? sender, int progress)
+    {
+        if (progress == -1)
+        {
+            IsIndeterminate = true;
+        }
+        else
+        {
+            IsIndeterminate = false;
+            OneClickInstallProgress = progress;
+        }
+        
+        OnGlobalProgressChanged(this, progress);
     }
 
     /// <summary>
@@ -70,13 +216,6 @@ public partial class MainWindowViewModel : ObservableObject
         }
     }
 
-    [RelayCommand]
-    private void GoAdvancedMode()
-    {
-        AdvancedModeVisibility = Visibility.Visible;
-        SimpleModeVisibility = Visibility.Hidden;
-    }
-
     private void SetTheme()
     {
         var theme = settingsManager.Settings.Theme;

From c8b7f1558ebe61878e54a7c8318baf45e996af2c Mon Sep 17 00:00:00 2001
From: JT <mohnjiles@gmail.com>
Date: Wed, 31 May 2023 00:43:03 -0700
Subject: [PATCH 2/8] remove stopwatch

---
 StabilityMatrix/LaunchOptionsDialog.xaml.cs | 1 -
 1 file changed, 1 deletion(-)

diff --git a/StabilityMatrix/LaunchOptionsDialog.xaml.cs b/StabilityMatrix/LaunchOptionsDialog.xaml.cs
index a9d521d8..98b7b10d 100644
--- a/StabilityMatrix/LaunchOptionsDialog.xaml.cs
+++ b/StabilityMatrix/LaunchOptionsDialog.xaml.cs
@@ -20,7 +20,6 @@ public partial class LaunchOptionsDialog : ContentDialog
         dialogService.GetContentPresenter())
     {
         this.viewModel = viewModel;
-        var sw = new Stopwatch();
         InitializeComponent();
         DataContext = viewModel;
     }

From 53908aa252a95444ee4e18c95fcea805efcc9ba9 Mon Sep 17 00:00:00 2001
From: Ionite <dev@ionite.io>
Date: Wed, 31 May 2023 03:45:12 -0400
Subject: [PATCH 3/8] Add tool tip for launch button

---
 StabilityMatrix/LaunchPage.xaml               | 19 +++++++++++++++++++
 StabilityMatrix/ViewModels/LaunchViewModel.cs |  2 ++
 2 files changed, 21 insertions(+)

diff --git a/StabilityMatrix/LaunchPage.xaml b/StabilityMatrix/LaunchPage.xaml
index 238e8b5b..ab888e25 100644
--- a/StabilityMatrix/LaunchPage.xaml
+++ b/StabilityMatrix/LaunchPage.xaml
@@ -99,6 +99,25 @@
                     <ColumnDefinition Width="0.8*" />
                     <ColumnDefinition Width="0.2*" />
                 </Grid.ColumnDefinitions>
+                <ui:Flyout
+                    Background="{DynamicResource SystemAccentColorPrimaryBrush}"
+                    FontSize="18"
+                    Grid.Column="0"
+                    Grid.Row="0"
+                    IsOpen="{Binding isLaunchTeachingTipsOpen, Mode=TwoWay}"
+                    Placement="Bottom">
+                    <StackPanel Orientation="Horizontal">
+                        <Grid HorizontalAlignment="Left">
+                            <ui:SymbolIcon Symbol="ArrowCurveUpLeft20" />
+                        </Grid>
+                        <TextBlock
+                            HorizontalAlignment="Left"
+                            Text="Click here to get started!"
+                            TextWrapping="WrapWithOverflow"
+                            Width="280" />
+                    </StackPanel>
+
+                </ui:Flyout>
                 <ui:Button
                     Background="Green"
                     Command="{Binding LaunchCommand}"
diff --git a/StabilityMatrix/ViewModels/LaunchViewModel.cs b/StabilityMatrix/ViewModels/LaunchViewModel.cs
index 1ce41dd8..6d32b49c 100644
--- a/StabilityMatrix/ViewModels/LaunchViewModel.cs
+++ b/StabilityMatrix/ViewModels/LaunchViewModel.cs
@@ -41,6 +41,8 @@ public partial class LaunchViewModel : ObservableObject
 
     [ObservableProperty] private Visibility stopButtonVisibility;
 
+    [ObservableProperty] private bool isLaunchTeachingTipsOpen = false;
+
 
     private InstalledPackage? selectedPackage;
 

From 1e128d59ef14ef6a6715e19b3be0d9734bcd3d0c Mon Sep 17 00:00:00 2001
From: JT <mohnjiles@gmail.com>
Date: Wed, 31 May 2023 17:00:49 -0700
Subject: [PATCH 4/8] install prereqs & show tooltip after navigate

---
 StabilityMatrix/Helper/EventManager.cs        |  2 ++
 StabilityMatrix/LaunchPage.xaml               |  2 +-
 StabilityMatrix/ViewModels/LaunchViewModel.cs |  6 ++++
 .../ViewModels/MainWindowViewModel.cs         | 28 +++++++++++++++++--
 4 files changed, 34 insertions(+), 4 deletions(-)

diff --git a/StabilityMatrix/Helper/EventManager.cs b/StabilityMatrix/Helper/EventManager.cs
index 2a892ae6..6ddcf47b 100644
--- a/StabilityMatrix/Helper/EventManager.cs
+++ b/StabilityMatrix/Helper/EventManager.cs
@@ -14,7 +14,9 @@ public class EventManager
     public event EventHandler<int>? GlobalProgressChanged;
     public event EventHandler<Type>? PageChangeRequested;
     public event EventHandler? InstalledPackagesChanged;
+    public event EventHandler? OneClickInstallFinished;
     public void OnGlobalProgressChanged(int progress) => GlobalProgressChanged?.Invoke(this, progress);
     public void RequestPageChange(Type pageType) => PageChangeRequested?.Invoke(this, pageType);
     public void OnInstalledPackagesChanged() => InstalledPackagesChanged?.Invoke(this, EventArgs.Empty);
+    public void OnOneClickInstallFinished() => OneClickInstallFinished?.Invoke(this, EventArgs.Empty);
 }
diff --git a/StabilityMatrix/LaunchPage.xaml b/StabilityMatrix/LaunchPage.xaml
index ab888e25..adb0dfc6 100644
--- a/StabilityMatrix/LaunchPage.xaml
+++ b/StabilityMatrix/LaunchPage.xaml
@@ -104,7 +104,7 @@
                     FontSize="18"
                     Grid.Column="0"
                     Grid.Row="0"
-                    IsOpen="{Binding isLaunchTeachingTipsOpen, Mode=TwoWay}"
+                    IsOpen="{Binding IsLaunchTeachingTipsOpen, Mode=TwoWay}"
                     Placement="Bottom">
                     <StackPanel Orientation="Horizontal">
                         <Grid HorizontalAlignment="Left">
diff --git a/StabilityMatrix/ViewModels/LaunchViewModel.cs b/StabilityMatrix/ViewModels/LaunchViewModel.cs
index 6d32b49c..a9b230de 100644
--- a/StabilityMatrix/ViewModels/LaunchViewModel.cs
+++ b/StabilityMatrix/ViewModels/LaunchViewModel.cs
@@ -85,10 +85,16 @@ public partial class LaunchViewModel : ObservableObject
         SetProcessRunning(false);
         
         EventManager.Instance.InstalledPackagesChanged += OnInstalledPackagesChanged;
+        EventManager.Instance.OneClickInstallFinished += OnOneClickInstallFinished;
 
         ToastNotificationManagerCompat.OnActivated += ToastNotificationManagerCompatOnOnActivated;
     }
 
+    private void OnOneClickInstallFinished(object? sender, EventArgs e)
+    {
+        IsLaunchTeachingTipsOpen = true;
+    }
+
     private void OnInstalledPackagesChanged(object? sender, EventArgs e)
     {
         OnLoaded();
diff --git a/StabilityMatrix/ViewModels/MainWindowViewModel.cs b/StabilityMatrix/ViewModels/MainWindowViewModel.cs
index 5270d08e..29276e85 100644
--- a/StabilityMatrix/ViewModels/MainWindowViewModel.cs
+++ b/StabilityMatrix/ViewModels/MainWindowViewModel.cs
@@ -10,6 +10,7 @@ using CommunityToolkit.Mvvm.Input;
 using Microsoft.Extensions.Logging;
 using StabilityMatrix.Helper;
 using StabilityMatrix.Models;
+using StabilityMatrix.Python;
 using Wpf.Ui.Appearance;
 using EventManager = StabilityMatrix.Helper.EventManager;
 
@@ -21,15 +22,17 @@ public partial class MainWindowViewModel : ObservableObject
     private readonly IPackageFactory packageFactory;
     private readonly IPrerequisiteHelper prerequisiteHelper;
     private readonly ILogger<MainWindowViewModel> logger;
+    private readonly IPyRunner pyRunner;
     private const string DefaultPackageName = "stable-diffusion-webui";
 
     public MainWindowViewModel(ISettingsManager settingsManager, IPackageFactory packageFactory,
-        IPrerequisiteHelper prerequisiteHelper, ILogger<MainWindowViewModel> logger)
+        IPrerequisiteHelper prerequisiteHelper, ILogger<MainWindowViewModel> logger, IPyRunner pyRunner)
     {
         this.settingsManager = settingsManager;
         this.packageFactory = packageFactory;
         this.prerequisiteHelper = prerequisiteHelper;
         this.logger = logger;
+        this.pyRunner = pyRunner;
     }
 
     [ObservableProperty] 
@@ -126,6 +129,19 @@ public partial class MainWindowViewModel : ObservableObject
             }
         }
 
+        SubHeaderText = "Installing prerequisites...";
+        IsIndeterminate = true;
+        if (!PyRunner.PipInstalled)
+        {
+            await pyRunner.SetupPip();
+        }
+
+        if (!PyRunner.VenvInstalled)
+        {
+            await pyRunner.InstallPackage("virtualenv");
+        }
+        IsIndeterminate = false;
+
         // get latest version & download & install
         SubHeaderText = "Getting latest version...";
         var latestVersion = await a1111.GetLatestVersion();
@@ -149,9 +165,15 @@ public partial class MainWindowViewModel : ObservableObject
         
         HeaderText = "Installation complete!";
         OneClickInstallProgress = 100;
-        SubHeaderText = "Proceeding to Launch page in 5 seconds...";
-        await Task.Delay(5000);
+        SubHeaderText = "Proceeding to Launch page in 3 seconds...";
+        await Task.Delay(1000);
+        SubHeaderText = "Proceeding to Launch page in 2 seconds...";
+        await Task.Delay(1000);
+        SubHeaderText = "Proceeding to Launch page in 1 second...";
+        await Task.Delay(1000);
+        
         IsAdvancedMode = true;
+        EventManager.Instance.OnOneClickInstallFinished();
     }
 
     private Task<string?> DownloadPackage(BasePackage selectedPackage, string version)

From 20c1c8553784143d5cc0ac10e2d6805510143c55 Mon Sep 17 00:00:00 2001
From: Ionite <dev@ionite.io>
Date: Wed, 31 May 2023 20:22:56 -0400
Subject: [PATCH 5/8] DialogErrorHandler add support for void Tasks

---
 StabilityMatrix/Helper/DialogErrorHandler.cs  | 27 ++++++++++++++++++-
 StabilityMatrix/Helper/IDialogErrorHandler.cs |  8 +++++-
 2 files changed, 33 insertions(+), 2 deletions(-)

diff --git a/StabilityMatrix/Helper/DialogErrorHandler.cs b/StabilityMatrix/Helper/DialogErrorHandler.cs
index 01aa4194..a79fa047 100644
--- a/StabilityMatrix/Helper/DialogErrorHandler.cs
+++ b/StabilityMatrix/Helper/DialogErrorHandler.cs
@@ -42,7 +42,7 @@ public class DialogErrorHandler : IDialogErrorHandler
     }
     
     /// <summary>
-    /// Attempt to run the given action, showing a generic error snackbar if it fails.
+    /// Attempt to run the given task, showing a generic error snackbar if it fails.
     /// </summary>
     public async Task<TaskResult<T>> TryAsync<T>(Task<T> task, string message, LogLevel level = LogLevel.Error, int timeoutMilliseconds = 5000)
     {
@@ -62,4 +62,29 @@ public class DialogErrorHandler : IDialogErrorHandler
             };
         }
     }
+    
+    /// <summary>
+    /// Attempt to run the given void task, showing a generic error snackbar if it fails.
+    /// Return a TaskResult with true if the task succeeded, false if it failed.
+    /// </summary>
+    public async Task<TaskResult<bool>> TryAsync(Task task, string message, LogLevel level = LogLevel.Error, int timeoutMilliseconds = 5000)
+    {
+        try
+        {
+            await task;
+            return new TaskResult<bool>
+            {
+                Result = true
+            };
+        }
+        catch (Exception e)
+        {
+            ShowSnackbarAsync(message, level, timeoutMilliseconds);
+            return new TaskResult<bool>
+            {
+                Result = false,
+                Exception = e
+            };
+        }
+    }
 }
diff --git a/StabilityMatrix/Helper/IDialogErrorHandler.cs b/StabilityMatrix/Helper/IDialogErrorHandler.cs
index 6769a35e..9753ec71 100644
--- a/StabilityMatrix/Helper/IDialogErrorHandler.cs
+++ b/StabilityMatrix/Helper/IDialogErrorHandler.cs
@@ -12,7 +12,13 @@ public interface IDialogErrorHandler
     void ShowSnackbarAsync(string message, LogLevel level = LogLevel.Error, int timeoutMilliseconds = 5000);
 
     /// <summary>
-    /// Attempt to run the given action, showing a generic error snackbar if it fails.
+    /// Attempt to run the given task, showing a generic error snackbar if it fails.
     /// </summary>
     Task<TaskResult<T>> TryAsync<T>(Task<T> task, string message, LogLevel level = LogLevel.Error, int timeoutMilliseconds = 5000);
+    
+    /// <summary>
+    /// Attempt to run the given void task, showing a generic error snackbar if it fails.
+    /// Return a TaskResult with true if the task succeeded, false if it failed.
+    /// </summary>
+    Task<TaskResult<bool>> TryAsync(Task task, string message, LogLevel level = LogLevel.Error, int timeoutMilliseconds = 5000);
 }

From 46d1a1ce608aab3bb7ffe049a0964e3c19d7f62a Mon Sep 17 00:00:00 2001
From: Ionite <dev@ionite.io>
Date: Wed, 31 May 2023 20:23:20 -0400
Subject: [PATCH 6/8] Add DeleteDirectoryAsync and error dialog

---
 .../ViewModels/PackageManagerViewModel.cs     | 42 +++++++++++++++++--
 1 file changed, 39 insertions(+), 3 deletions(-)

diff --git a/StabilityMatrix/ViewModels/PackageManagerViewModel.cs b/StabilityMatrix/ViewModels/PackageManagerViewModel.cs
index a92c01cc..c9db7766 100644
--- a/StabilityMatrix/ViewModels/PackageManagerViewModel.cs
+++ b/StabilityMatrix/ViewModels/PackageManagerViewModel.cs
@@ -1,5 +1,7 @@
 using System;
+using System.Collections.Generic;
 using System.Collections.ObjectModel;
+using System.Diagnostics;
 using System.IO;
 using System.Threading.Tasks;
 using StabilityMatrix.Helper;
@@ -9,6 +11,7 @@ using System.Windows;
 using CommunityToolkit.Mvvm.ComponentModel;
 using CommunityToolkit.Mvvm.Input;
 using Microsoft.Extensions.Logging;
+using Polly;
 using Wpf.Ui.Contracts;
 using Wpf.Ui.Controls.ContentDialogControl;
 using EventManager = StabilityMatrix.Helper.EventManager;
@@ -22,6 +25,7 @@ public partial class PackageManagerViewModel : ObservableObject
     private readonly IPackageFactory packageFactory;
     private readonly IDialogFactory dialogFactory;
     private readonly IContentDialogService contentDialogService;
+    private readonly IDialogErrorHandler dialogErrorHandler;
     private const int MinutesToWaitForUpdateCheck = 60;
 
     [ObservableProperty]
@@ -51,13 +55,14 @@ public partial class PackageManagerViewModel : ObservableObject
     private bool updateAvailable;
 
     public PackageManagerViewModel(ILogger<PackageManagerViewModel> logger, ISettingsManager settingsManager,
-        IPackageFactory packageFactory, IDialogFactory dialogFactory, IContentDialogService contentDialogService)
+        IPackageFactory packageFactory, IDialogFactory dialogFactory, IContentDialogService contentDialogService, IDialogErrorHandler dialogErrorHandler)
     {
         this.logger = logger;
         this.settingsManager = settingsManager;
         this.packageFactory = packageFactory;
         this.dialogFactory = dialogFactory;
         this.contentDialogService = contentDialogService;
+        this.dialogErrorHandler = dialogErrorHandler;
 
         ProgressText = "shrug";
         InstallButtonText = "Install";
@@ -159,12 +164,43 @@ public partial class PackageManagerViewModel : ObservableObject
 
         if (result == ContentDialogResult.Primary)
         {
-            DeleteDirectory(SelectedPackage.Path);
-            settingsManager.RemoveInstalledPackage(SelectedPackage);
+            var deleteTask = DeleteDirectoryAsync(SelectedPackage.Path);
+            var taskResult = await dialogErrorHandler.TryAsync(deleteTask,
+                "Some files could not be deleted. Please close any open files in the package directory and try again.");
+            if (taskResult.IsSuccessful)
+            {
+                settingsManager.RemoveInstalledPackage(SelectedPackage);
+            }
             await OnLoaded();
         }
     }
     
+    /// <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.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)
     {
         // Delete all files in the directory

From 7e9a6d8bec7e30c8cca27163eb085a472b6ae8c6 Mon Sep 17 00:00:00 2001
From: JT <mohnjiles@gmail.com>
Date: Wed, 31 May 2023 17:28:05 -0700
Subject: [PATCH 7/8] Fix not removing packages on uninstall

---
 StabilityMatrix/Models/Packages/A3WebUI.cs            | 9 ++++++++-
 StabilityMatrix/ViewModels/LaunchViewModel.cs         | 1 +
 StabilityMatrix/ViewModels/PackageManagerViewModel.cs | 2 +-
 3 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/StabilityMatrix/Models/Packages/A3WebUI.cs b/StabilityMatrix/Models/Packages/A3WebUI.cs
index 7741ca74..5fed640d 100644
--- a/StabilityMatrix/Models/Packages/A3WebUI.cs
+++ b/StabilityMatrix/Models/Packages/A3WebUI.cs
@@ -3,6 +3,7 @@ using System.Collections.Generic;
 using System.Diagnostics;
 using System.IO;
 using System.Linq;
+using System.Text.RegularExpressions;
 using System.Threading.Tasks;
 using StabilityMatrix.Api;
 using StabilityMatrix.Helper;
@@ -100,7 +101,13 @@ public class A3WebUI : BaseGitPackage
 
             if (s.Contains("Running on", StringComparison.OrdinalIgnoreCase))
             {
-                WebUrl = s.Split(" ")[5];
+                var regex = new Regex(
+                    "(?:https?|ftp)://[-a-zA-Z0-9.]+(:(6553[0-5]|655[0-2][0-9]|65[0-4][0-9][0-9]|6[0-4][0-9][0-9][0-9]|\\d{2,4}|[1-9]))?");
+                var match = regex.Match(s);
+                if (match.Success)
+                {
+                    WebUrl = match.Value;
+                }
             }
 
             Debug.WriteLine($"process stdout: {s}");
diff --git a/StabilityMatrix/ViewModels/LaunchViewModel.cs b/StabilityMatrix/ViewModels/LaunchViewModel.cs
index a9b230de..e5c80c00 100644
--- a/StabilityMatrix/ViewModels/LaunchViewModel.cs
+++ b/StabilityMatrix/ViewModels/LaunchViewModel.cs
@@ -241,6 +241,7 @@ public partial class LaunchViewModel : ObservableObject
         var packages = settingsManager.Settings.InstalledPackages;
         if (!packages?.Any() ?? true)
         {
+            InstalledPackages.Clear();
             return;
         }
 
diff --git a/StabilityMatrix/ViewModels/PackageManagerViewModel.cs b/StabilityMatrix/ViewModels/PackageManagerViewModel.cs
index c9db7766..86b040fa 100644
--- a/StabilityMatrix/ViewModels/PackageManagerViewModel.cs
+++ b/StabilityMatrix/ViewModels/PackageManagerViewModel.cs
@@ -87,13 +87,13 @@ public partial class PackageManagerViewModel : ObservableObject
 
     public async Task OnLoaded()
     {
+        Packages.Clear();
         var installedPackages = settingsManager.Settings.InstalledPackages;
         if (installedPackages.Count == 0)
         {
             return;
         }
         
-        Packages.Clear();
         
         foreach (var packageToUpdate in installedPackages)
         {

From 0fd25a38f89bb45b47ef25910a63e587061f79b8 Mon Sep 17 00:00:00 2001
From: JT <mohnjiles@gmail.com>
Date: Wed, 31 May 2023 17:29:35 -0700
Subject: [PATCH 8/8] fix showing dummy values on last package uninstall

---
 StabilityMatrix/ViewModels/PackageManagerViewModel.cs | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/StabilityMatrix/ViewModels/PackageManagerViewModel.cs b/StabilityMatrix/ViewModels/PackageManagerViewModel.cs
index 86b040fa..88f28d39 100644
--- a/StabilityMatrix/ViewModels/PackageManagerViewModel.cs
+++ b/StabilityMatrix/ViewModels/PackageManagerViewModel.cs
@@ -91,6 +91,12 @@ public partial class PackageManagerViewModel : ObservableObject
         var installedPackages = settingsManager.Settings.InstalledPackages;
         if (installedPackages.Count == 0)
         {
+            SelectedPackage = new InstalledPackage
+            {
+                DisplayName = "Click \"Add Package\" to install a package"
+            };
+            InstallButtonVisibility = Visibility.Collapsed;
+            
             return;
         }