From cc63c1e239918fe116774d1139892d505503a126 Mon Sep 17 00:00:00 2001
From: JT <mohnjiles@gmail.com>
Date: Tue, 23 May 2023 22:19:42 -0700
Subject: [PATCH] Added helpers to deal with hardware info & A1111 ui settings.
 also removed unused usings

---
 StabilityMatrix/App.xaml.cs                   | 15 --------
 StabilityMatrix/Helper/A1111Helper.cs         | 29 ++++++++++++++++
 StabilityMatrix/Helper/HardwareHelper.cs      | 34 +++++++++++++++++++
 StabilityMatrix/Helper/ProcessRunner.cs       | 10 +++---
 StabilityMatrix/InstallPage.xaml              |  3 +-
 StabilityMatrix/InstallPage.xaml.cs           | 14 --------
 StabilityMatrix/LaunchPage.xaml.cs            | 11 ------
 StabilityMatrix/MainWindow.xaml.cs            | 13 +------
 .../ViewModels/InstallerViewModel.cs          | 34 +++++++++++++------
 9 files changed, 93 insertions(+), 70 deletions(-)
 create mode 100644 StabilityMatrix/Helper/A1111Helper.cs
 create mode 100644 StabilityMatrix/Helper/HardwareHelper.cs

diff --git a/StabilityMatrix/App.xaml.cs b/StabilityMatrix/App.xaml.cs
index ccb2a215..a2a9b217 100644
--- a/StabilityMatrix/App.xaml.cs
+++ b/StabilityMatrix/App.xaml.cs
@@ -1,22 +1,7 @@
 using Microsoft.UI.Xaml;
-using Microsoft.UI.Xaml.Controls;
-using Microsoft.UI.Xaml.Controls.Primitives;
-using Microsoft.UI.Xaml.Data;
-using Microsoft.UI.Xaml.Input;
-using Microsoft.UI.Xaml.Media;
-using Microsoft.UI.Xaml.Navigation;
-using Microsoft.UI.Xaml.Shapes;
 using System;
-using System.Collections.Generic;
 using System.Diagnostics;
-using System.IO;
-using System.Linq;
-using System.Runtime.InteropServices.WindowsRuntime;
-using Windows.ApplicationModel;
-using Windows.ApplicationModel.Activation;
 using Windows.ApplicationModel.Core;
-using Windows.Foundation;
-using Windows.Foundation.Collections;
 
 namespace StabilityMatrix
 {
diff --git a/StabilityMatrix/Helper/A1111Helper.cs b/StabilityMatrix/Helper/A1111Helper.cs
new file mode 100644
index 00000000..27f0f0c0
--- /dev/null
+++ b/StabilityMatrix/Helper/A1111Helper.cs
@@ -0,0 +1,29 @@
+using System;
+
+namespace StabilityMatrix.Helper;
+
+public static class A1111Helper
+{
+    public static string GetVramOption()
+    {
+        var vramGb = HardwareHelper.GetGpuMemoryBytes() / 1024 / 1024 / 1024;
+
+        return vramGb switch
+        {
+            < 4 => "--lowvram",
+            < 8 => "--medvram",
+            _ => string.Empty
+        };
+    }
+    
+    public static string GetXformersOption()
+    {
+        var gpuName = HardwareHelper.GetGpuChipName();
+        if (gpuName.Contains("NVIDIA", StringComparison.OrdinalIgnoreCase))
+        {
+            return "--xformers";
+        }
+        
+        return string.Empty;
+    }
+}
diff --git a/StabilityMatrix/Helper/HardwareHelper.cs b/StabilityMatrix/Helper/HardwareHelper.cs
new file mode 100644
index 00000000..12c075ac
--- /dev/null
+++ b/StabilityMatrix/Helper/HardwareHelper.cs
@@ -0,0 +1,34 @@
+using System;
+using Microsoft.Win32;
+
+namespace StabilityMatrix.Helper;
+
+public class HardwareHelper
+{
+    public static ulong GetGpuMemoryBytes()
+    {
+        var registry = Registry.LocalMachine;
+        var key = registry.OpenSubKey(@"SYSTEM\CurrentControlSet\Control\Class\{4d36e968-e325-11ce-bfc1-08002be10318}\0000", false);
+        if (key == null)
+        {
+            return 0;
+        }
+        
+        var vram = key.GetValue("HardwareInformation.qwMemorySize");
+        var vramLong = Convert.ToUInt64(vram);
+        return vramLong;
+    }
+
+    public static string GetGpuChipName()
+    {
+        var registry = Registry.LocalMachine;
+        var key = registry.OpenSubKey(@"SYSTEM\CurrentControlSet\Control\Class\{4d36e968-e325-11ce-bfc1-08002be10318}\0000", false);
+        if (key == null)
+        {
+            return "Unknown";
+        }
+        
+        var gpuName = key.GetValue("HardwareInformation.ChipType");
+        return gpuName?.ToString() ?? "Unknown";
+    }
+}
diff --git a/StabilityMatrix/Helper/ProcessRunner.cs b/StabilityMatrix/Helper/ProcessRunner.cs
index 5cfeb0ac..377acf24 100644
--- a/StabilityMatrix/Helper/ProcessRunner.cs
+++ b/StabilityMatrix/Helper/ProcessRunner.cs
@@ -1,13 +1,11 @@
-using System;
-using System.Diagnostics;
-using System.IO;
+using System.Diagnostics;
 using System.Threading.Tasks;
 
 namespace StabilityMatrix.Helper;
 
 public static class ProcessRunner
 {
-    public static async Task<string> RunProcessAsync(string fileName, string arguments)
+    public static async Task<string> GetProcessOutputAsync(string fileName, string arguments)
     {
         using var process = new Process();
         process.StartInfo.FileName = fileName;
@@ -23,10 +21,10 @@ public static class ProcessRunner
         return output;
     }
 
-    public static Process RunProcess(string fileName, string arguments)
+    public static Process StartProcess(string fileName, string arguments)
 
     {
-        using var process = new Process();
+        var process = new Process();
         process.StartInfo.FileName = fileName;
         process.StartInfo.Arguments = arguments;
         process.StartInfo.UseShellExecute = false;
diff --git a/StabilityMatrix/InstallPage.xaml b/StabilityMatrix/InstallPage.xaml
index 31f7d3e8..da83e9cf 100644
--- a/StabilityMatrix/InstallPage.xaml
+++ b/StabilityMatrix/InstallPage.xaml
@@ -42,7 +42,8 @@
         <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}" 
+            <ProgressBar Visibility="{Binding ProgressBarVisibility}" Value="{Binding ProgressValue, FallbackValue=10}"
+                         IsIndeterminate="{Binding IsIndeterminate, FallbackValue=True}"
                          Maximum="100" Width="500" Style="{ThemeResource DefaultProgressBarStyle}"/>
         </StackPanel>
     </Grid>
diff --git a/StabilityMatrix/InstallPage.xaml.cs b/StabilityMatrix/InstallPage.xaml.cs
index 004f5419..4a31b739 100644
--- a/StabilityMatrix/InstallPage.xaml.cs
+++ b/StabilityMatrix/InstallPage.xaml.cs
@@ -1,21 +1,7 @@
 // Copyright (c) Microsoft Corporation and Contributors.
 // Licensed under the MIT License.
 
-using Microsoft.UI.Xaml;
 using Microsoft.UI.Xaml.Controls;
-using Microsoft.UI.Xaml.Controls.Primitives;
-using Microsoft.UI.Xaml.Data;
-using Microsoft.UI.Xaml.Input;
-using Microsoft.UI.Xaml.Media;
-using Microsoft.UI.Xaml.Navigation;
-using System;
-using System.Collections.Generic;
-using System.Collections.ObjectModel;
-using System.IO;
-using System.Linq;
-using System.Runtime.InteropServices.WindowsRuntime;
-using Windows.Foundation;
-using Windows.Foundation.Collections;
 using StabilityMatrix.ViewModels;
 
 // To learn more about WinUI, the WinUI project structure,
diff --git a/StabilityMatrix/LaunchPage.xaml.cs b/StabilityMatrix/LaunchPage.xaml.cs
index 763704fd..0460fc62 100644
--- a/StabilityMatrix/LaunchPage.xaml.cs
+++ b/StabilityMatrix/LaunchPage.xaml.cs
@@ -1,17 +1,6 @@
 using Microsoft.UI.Xaml;
 using Microsoft.UI.Xaml.Controls;
-using Microsoft.UI.Xaml.Controls.Primitives;
-using Microsoft.UI.Xaml.Data;
-using Microsoft.UI.Xaml.Input;
-using Microsoft.UI.Xaml.Media;
-using Microsoft.UI.Xaml.Navigation;
 using System;
-using System.Collections.Generic;
-using System.IO;
-using System.Linq;
-using System.Runtime.InteropServices.WindowsRuntime;
-using Windows.Foundation;
-using Windows.Foundation.Collections;
 
 // To learn more about WinUI, the WinUI project structure,
 // and more about our project templates, see: http://aka.ms/winui-project-info.
diff --git a/StabilityMatrix/MainWindow.xaml.cs b/StabilityMatrix/MainWindow.xaml.cs
index d7cae787..f332842f 100644
--- a/StabilityMatrix/MainWindow.xaml.cs
+++ b/StabilityMatrix/MainWindow.xaml.cs
@@ -1,18 +1,7 @@
 using Microsoft.UI.Xaml;
 using Microsoft.UI.Xaml.Controls;
-using Microsoft.UI.Xaml.Controls.Primitives;
-using Microsoft.UI.Xaml.Data;
-using Microsoft.UI.Xaml.Input;
-using Microsoft.UI.Xaml.Media;
-using Microsoft.UI.Xaml.Navigation;
 using System;
-using System.Collections.Generic;
-using System.IO;
 using System.Linq;
-using System.Runtime.InteropServices.WindowsRuntime;
-using Windows.Foundation;
-using Windows.Foundation.Collections;
-using WinRT.Interop;
 
 namespace StabilityMatrix
 {
@@ -69,4 +58,4 @@ namespace StabilityMatrix
             MainNavigationView.SelectedItem = item;
         }
     }
-}
\ No newline at end of file
+}
diff --git a/StabilityMatrix/ViewModels/InstallerViewModel.cs b/StabilityMatrix/ViewModels/InstallerViewModel.cs
index 91d75669..b52e814a 100644
--- a/StabilityMatrix/ViewModels/InstallerViewModel.cs
+++ b/StabilityMatrix/ViewModels/InstallerViewModel.cs
@@ -1,15 +1,10 @@
 using System.Collections.ObjectModel;
 using System.Threading.Tasks;
-using Windows.UI.Popups;
 using CommunityToolkit.Mvvm.Input;
 using StabilityMatrix.Helper;
 using StabilityMatrix.Models;
-using System;
-using System.Collections.Generic;
 using System.ComponentModel;
-using System.Diagnostics;
 using System.Runtime.CompilerServices;
-using CommunityToolkit.Mvvm.ComponentModel;
 using Microsoft.UI.Xaml;
 
 namespace StabilityMatrix.ViewModels;
@@ -18,6 +13,7 @@ internal class InstallerViewModel : INotifyPropertyChanged
 {
     private string installedText;
     private int progressValue;
+    private bool isIndeterminate;
 
     public InstallerViewModel()
     {
@@ -53,31 +49,47 @@ internal class InstallerViewModel : INotifyPropertyChanged
         }
     }
 
+    public bool IsIndeterminate
+    {
+        get => isIndeterminate;
+        set
+        {
+            if (value == isIndeterminate) return;
+            isIndeterminate = value;
+            OnPropertyChanged();
+        }
+    }
+
     public Visibility ProgressBarVisibility => ProgressValue > 0 ? Visibility.Visible : Visibility.Collapsed;
 
     public AsyncRelayCommand InstallCommand => new(InstallGitIfNecessary);
     private async Task InstallGitIfNecessary()
     {
-        var gitOutput = await ProcessRunner.RunProcessAsync("git", "--version");
+        var gitOutput = await ProcessRunner.GetProcessOutputAsync("git", "--version");
         if (gitOutput.Contains("git version 2"))
         {
             InstalledText = "Installed";
+            IsIndeterminate = false;
             ProgressValue = 100;
             return;
         }
 
         InstalledText = "Not Installed";
-        var installProcess = ProcessRunner.RunProcess("Assets\\Git-2.40.1-64-bit.exe", "/SILENT /NORESTART");
-        while (!installProcess.StandardOutput.EndOfStream && !installProcess.HasExited)
-        {
-            Debug.WriteLine(await installProcess.StandardOutput.ReadLineAsync());
-        }
+        IsIndeterminate = true;
+        using var installProcess = ProcessRunner.StartProcess("Assets\\Git-2.40.1-64-bit.exe", "/VERYSILENT /NORESTART");
+        await installProcess.WaitForExitAsync();
 
         if (installProcess.ExitCode == 0)
         {
             InstalledText = "Git successfully installed";
             ProgressValue = 100;
         }
+        else
+        {
+            InstalledText = $"There was an error installing Git: {installProcess.ExitCode}";
+            ProgressValue = 0;
+        }
+        IsIndeterminate = false;
     }
 
     public event PropertyChangedEventHandler PropertyChanged;