From 5c1d16b95bbeff62c6b52ae08a27c6eec121b58b Mon Sep 17 00:00:00 2001
From: JT <mohnjiles@gmail.com>
Date: Thu, 25 May 2023 18:42:43 -0700
Subject: [PATCH] Fix initial ListView selection (now shows highlight like it's
 selected)

---
 StabilityMatrix/App.xaml.cs                   |  1 +
 StabilityMatrix/Helper/ProcessRunner.cs       |  3 ++
 StabilityMatrix/InstallPage.xaml              | 33 +++++++++++------
 StabilityMatrix/InstallPage.xaml.cs           | 11 +++---
 StabilityMatrix/Models/Packages/A3WebUI.cs    | 29 ++++++++++++---
 StabilityMatrix/PyVenvRunner.cs               |  6 ++--
 .../ViewModels/InstallerViewModel.cs          | 36 ++++++++++++++-----
 7 files changed, 88 insertions(+), 31 deletions(-)

diff --git a/StabilityMatrix/App.xaml.cs b/StabilityMatrix/App.xaml.cs
index 99a39d9b..1f7736ef 100644
--- a/StabilityMatrix/App.xaml.cs
+++ b/StabilityMatrix/App.xaml.cs
@@ -31,6 +31,7 @@ namespace StabilityMatrix
             serviceCollection.AddTransient<MainWindowViewModel>();
             serviceCollection.AddSingleton<SettingsViewModel>();
             serviceCollection.AddSingleton<LaunchViewModel>();
+            serviceCollection.AddSingleton<InstallerViewModel>();
             serviceCollection.AddSingleton<IContentDialogService, ContentDialogService>();
             serviceCollection.AddSingleton<ISnackbarService, SnackbarService>();
             serviceCollection.AddSingleton<ISettingsManager, SettingsManager>();
diff --git a/StabilityMatrix/Helper/ProcessRunner.cs b/StabilityMatrix/Helper/ProcessRunner.cs
index 109f033d..104e9ac9 100644
--- a/StabilityMatrix/Helper/ProcessRunner.cs
+++ b/StabilityMatrix/Helper/ProcessRunner.cs
@@ -36,11 +36,13 @@ public static class ProcessRunner
         process.StartInfo.Arguments = arguments;
         process.StartInfo.UseShellExecute = false;
         process.StartInfo.RedirectStandardOutput = true;
+        process.StartInfo.RedirectStandardError = true;
         process.StartInfo.CreateNoWindow = true;
 
         if (outputDataReceived != null)
         {
             process.OutputDataReceived += (_, args) => outputDataReceived(args.Data);
+            process.ErrorDataReceived += (_, args) => outputDataReceived(args.Data);
         }
 
         process.Start();
@@ -48,6 +50,7 @@ public static class ProcessRunner
         if (outputDataReceived != null)
         {
             process.BeginOutputReadLine();
+            process.BeginErrorReadLine();
         }
 
         return process;
diff --git a/StabilityMatrix/InstallPage.xaml b/StabilityMatrix/InstallPage.xaml
index 7c0f5c0d..1885a09f 100644
--- a/StabilityMatrix/InstallPage.xaml
+++ b/StabilityMatrix/InstallPage.xaml
@@ -20,14 +20,32 @@
     </Page.DataContext>
     
     <Grid Margin="16">
-        <StackPanel Orientation="Horizontal" Margin="16" HorizontalAlignment="Left" Height="400">
-            <ListView
-                      ItemsSource="{Binding Packages}"
+        <Grid.RowDefinitions>
+            <RowDefinition Height="Auto"/>
+            <RowDefinition Height="*"/>
+        </Grid.RowDefinitions>
+        
+        <StackPanel Grid.Row="0" Orientation="Vertical">
+            <Button Content="Install" Command="{Binding InstallCommand}" Width="100" Height="50" HorizontalAlignment="Center"/>
+            <TextBlock Text="{Binding InstalledText}" HorizontalAlignment="Center" Padding="8"/>
+            <ProgressBar Visibility="{Binding ProgressBarVisibility, FallbackValue=Visible}" Value="{Binding ProgressValue, FallbackValue=10}"
+                         IsIndeterminate="{Binding IsIndeterminate, FallbackValue=False}"
+                         Maximum="100" Width="500"/>
+        </StackPanel>
+        
+        <StackPanel Grid.Row="1" Orientation="Horizontal" Margin="16" HorizontalAlignment="Left" Height="400">
+            <ListView ItemsSource="{Binding Packages}"
                       SelectedItem="{Binding SelectedPackage, Mode=TwoWay}">
+                
+                <ListView.Style>
+                    <Style TargetType="ListView">
+                        <Setter Property="Background" Value="#191919"/>
+                    </Style>
+                </ListView.Style>
             
                 <ListView.ItemTemplate>
                     <DataTemplate DataType="{x:Type models:BasePackage}">
-                            <StackPanel VerticalAlignment="Top" Margin="10,0,0,0">
+                            <StackPanel VerticalAlignment="Top" Margin="8">
                                 <TextBlock Text="{Binding DisplayName}" Margin="0,5,0,5" />
                                 <TextBlock Text="{Binding ByAuthor}" Margin="0,0,0,5" />
                             </StackPanel>
@@ -45,12 +63,5 @@
                 </ui:Hyperlink>
             </StackPanel>
         </StackPanel>
-        <StackPanel Orientation="Vertical">
-            <Button Content="Install" Command="{Binding InstallCommand}" Width="100" Height="50" HorizontalAlignment="Center"/>
-            <TextBlock Text="{Binding InstalledText}" HorizontalAlignment="Center" Padding="8"/>
-            <ProgressBar Visibility="{Binding ProgressBarVisibility}" Value="{Binding ProgressValue, FallbackValue=10}"
-                         IsIndeterminate="{Binding IsIndeterminate, FallbackValue=True}"
-                         Maximum="100" Width="500"/>
-        </StackPanel>
     </Grid>
 </Page>
diff --git a/StabilityMatrix/InstallPage.xaml.cs b/StabilityMatrix/InstallPage.xaml.cs
index 4c915c1a..fa24efbb 100644
--- a/StabilityMatrix/InstallPage.xaml.cs
+++ b/StabilityMatrix/InstallPage.xaml.cs
@@ -12,15 +12,18 @@ namespace StabilityMatrix
     /// </summary>
     public sealed partial class InstallPage : Page
     {
-        public InstallPage()
-        { 
+        private readonly InstallerViewModel viewModel;
+
+        public InstallPage(InstallerViewModel viewModel)
+        {
+            this.viewModel = viewModel;
             InitializeComponent();
-            DataContext = new InstallerViewModel();
+            DataContext = viewModel;
         }
 
         private async void InstallPage_OnLoaded(object sender, RoutedEventArgs e)
         {
-            await ((InstallerViewModel) DataContext).OnLoaded(); 
+            await viewModel.OnLoaded(); 
         }
     }
 }
diff --git a/StabilityMatrix/Models/Packages/A3WebUI.cs b/StabilityMatrix/Models/Packages/A3WebUI.cs
index 46294f47..2a719f69 100644
--- a/StabilityMatrix/Models/Packages/A3WebUI.cs
+++ b/StabilityMatrix/Models/Packages/A3WebUI.cs
@@ -3,6 +3,7 @@ using System.Diagnostics;
 using System.IO;
 using System.Net.Http;
 using System.Net.Http.Headers;
+using System.Threading;
 using System.Threading.Tasks;
 using Refit;
 using StabilityMatrix.Api;
@@ -30,9 +31,20 @@ public class A3WebUI: BasePackage
         using var client = new HttpClient {Timeout = TimeSpan.FromMinutes(5)};
         client.DefaultRequestHeaders.UserAgent.Add(new ProductInfoHeaderValue("StabilityMatrix", "1.0"));
         await using var file = new FileStream(DownloadLocation, FileMode.Create, FileAccess.Write, FileShare.None);
-        using var response = await client.GetAsync(downloadUrl, HttpCompletionOption.ResponseHeadersRead);
-        var length = response.Content.Headers.ContentLength;
 
+        long contentLength = 0;
+        var retryCount = 0;
+        var response = await client.GetAsync(downloadUrl, HttpCompletionOption.ResponseHeadersRead);
+        while (contentLength == 0 && retryCount++ < 5)
+        {
+            response = await client.GetAsync(downloadUrl, HttpCompletionOption.ResponseHeadersRead);
+            contentLength = response.Content.Headers.ContentLength ?? 0;
+            Debug.WriteLine("Retrying...");
+            Thread.Sleep(50);
+        }
+
+        var isIndeterminate = contentLength == 0;
+        
         await using var stream = await response.Content.ReadAsStreamAsync();
         var totalBytesRead = 0;
         while (true)
@@ -44,9 +56,16 @@ public class A3WebUI: BasePackage
             
             totalBytesRead += bytesRead;
 
-            var progress = (int) (totalBytesRead * 100d / length);
-            Debug.WriteLine($"Progress; {progress}");
-            OnDownloadProgressChanged(progress);
+            if (isIndeterminate)
+            {
+                OnDownloadProgressChanged(-1);
+            }
+            else
+            {
+                var progress = (int) (totalBytesRead * 100d / contentLength);
+                Debug.WriteLine($"Progress; {progress}");
+                OnDownloadProgressChanged(progress);
+            }
         }
 
         await file.FlushAsync();
diff --git a/StabilityMatrix/PyVenvRunner.cs b/StabilityMatrix/PyVenvRunner.cs
index 640bcf7e..e7c9b0cc 100644
--- a/StabilityMatrix/PyVenvRunner.cs
+++ b/StabilityMatrix/PyVenvRunner.cs
@@ -53,11 +53,13 @@ public class PyVenvRunner: IDisposable
         // Create venv
         var venvProc = ProcessRunner.StartProcess(PyRunner.ExePath, "-m virtualenv " + RootPath);
         await venvProc.WaitForExitAsync();
+        
         // Check return code
         var returnCode = venvProc.ExitCode;
         if (returnCode != 0)
         {
-           var output = await venvProc.StandardOutput.ReadToEndAsync(); 
+           var output = await venvProc.StandardOutput.ReadToEndAsync();
+           output += await venvProc.StandardError.ReadToEndAsync();
            throw new InvalidOperationException($"Venv creation failed with code {returnCode}: {output}");
         }
     }
@@ -81,4 +83,4 @@ public class PyVenvRunner: IDisposable
         Process?.Dispose();
         GC.SuppressFinalize(this);
     }
-}
\ No newline at end of file
+}
diff --git a/StabilityMatrix/ViewModels/InstallerViewModel.cs b/StabilityMatrix/ViewModels/InstallerViewModel.cs
index 7941560f..bf55c23a 100644
--- a/StabilityMatrix/ViewModels/InstallerViewModel.cs
+++ b/StabilityMatrix/ViewModels/InstallerViewModel.cs
@@ -12,7 +12,7 @@ using CommunityToolkit.Mvvm.Input;
 
 namespace StabilityMatrix.ViewModels;
 
-internal class InstallerViewModel : INotifyPropertyChanged
+public class InstallerViewModel : INotifyPropertyChanged
 {
     private string installedText;
     private int progressValue;
@@ -23,19 +23,20 @@ internal class InstallerViewModel : INotifyPropertyChanged
     {
         InstalledText = "shrug";
         ProgressValue = 0;
+        Packages = new ObservableCollection<BasePackage>
+        {
+            new A3WebUI(),
+            new DankDiffusion()
+        };
+        SelectedPackage = Packages[0];
     }
 
     public Task OnLoaded()
     {
-        SelectedPackage = Packages.First();
         return Task.CompletedTask;
     }
     
-    public static ObservableCollection<BasePackage> Packages => new()
-    {
-        new A3WebUI(),
-        new DankDiffusion()
-    };
+    public ObservableCollection<BasePackage> Packages { get; }
 
     public string InstalledText
     {
@@ -81,9 +82,26 @@ internal class InstallerViewModel : INotifyPropertyChanged
         }
     }
 
-    public Visibility ProgressBarVisibility => ProgressValue > 0 ? Visibility.Visible : Visibility.Collapsed;
+    public Visibility ProgressBarVisibility => ProgressValue > 0 || IsIndeterminate ? Visibility.Visible : Visibility.Collapsed;
 
-    public AsyncRelayCommand InstallCommand => new(async () => await SelectedPackage.DownloadPackage());
+    public AsyncRelayCommand InstallCommand => new(async () =>
+    {
+        SelectedPackage.DownloadProgressChanged += (_, progress) =>
+        {
+            if (progress == -1)
+            {
+                IsIndeterminate = true;
+                ProgressValue = 1;
+            }
+            else
+            {
+                IsIndeterminate = false;
+                ProgressValue = progress;
+            }
+        };
+        SelectedPackage.DownloadComplete += (_, _) => InstalledText = "Download Complete";
+        await SelectedPackage.DownloadPackage();
+    });
     private async Task<bool> InstallGitIfNecessary()
     {
         var gitOutput = await ProcessRunner.GetProcessOutputAsync("git", "--version");