From 187b3d9b8f710d187609c2ec70638eff09d7c971 Mon Sep 17 00:00:00 2001
From: pikminmario500 <pikminmario500@outlook.com>
Date: Tue, 13 Aug 2024 01:40:13 +0200
Subject: [PATCH] add process priority (sync)

---
 Bloxstrap/Bootstrapper.cs                     | 21 ++++++
 Bloxstrap/Enums/PriorityClass.cs              | 11 +++
 Bloxstrap/Models/Settings.cs                  |  1 +
 Bloxstrap/Resources/Strings.Designer.cs       | 72 +++++++++++++++++++
 Bloxstrap/Resources/Strings.resx              | 24 +++++++
 .../Settings/Pages/BehaviourPage.xaml         | 12 ++++
 .../ViewModels/Settings/BehaviourViewModel.cs |  8 +++
 7 files changed, 149 insertions(+)
 create mode 100644 Bloxstrap/Enums/PriorityClass.cs

diff --git a/Bloxstrap/Bootstrapper.cs b/Bloxstrap/Bootstrapper.cs
index 97b7dbb..cfc962c 100644
--- a/Bloxstrap/Bootstrapper.cs
+++ b/Bloxstrap/Bootstrapper.cs
@@ -374,6 +374,27 @@ namespace Bloxstrap
                 }
             }
 
+            Process[] processes = Process.GetProcessesByName("RobloxPlayerBeta");
+#if STUDIO_FEATURES
+            if (_launchMode == LaunchMode.Studio)
+                processes = Process.GetProcessesByName("RobloxStudioBeta");
+#endif
+
+            ProcessPriorityClass[] PriorityThing = {
+                ProcessPriorityClass.Idle,
+                ProcessPriorityClass.BelowNormal,
+                ProcessPriorityClass.Normal,
+                ProcessPriorityClass.AboveNormal,
+                ProcessPriorityClass.High
+            };
+
+            foreach (Process proc in processes)
+            {
+                proc.PriorityClass = PriorityThing[(int)App.Settings.Prop.ChoosePriorityClass];
+
+                App.Logger.WriteLine(LOG_IDENT, $"Launching with priority '{proc.PriorityClass}'");
+            }
+
             // event fired, wait for 3 seconds then close
             await Task.Delay(3000);
             Dialog?.CloseBootstrapper();
diff --git a/Bloxstrap/Enums/PriorityClass.cs b/Bloxstrap/Enums/PriorityClass.cs
new file mode 100644
index 0000000..3d10285
--- /dev/null
+++ b/Bloxstrap/Enums/PriorityClass.cs
@@ -0,0 +1,11 @@
+namespace Bloxstrap.Enums
+{
+    public enum PriorityClasses
+    {
+        Idle,
+        BelowNormal,
+        Normal,
+        AboveNormal,
+        High
+    }
+}
\ No newline at end of file
diff --git a/Bloxstrap/Models/Settings.cs b/Bloxstrap/Models/Settings.cs
index 3bc2580..0384b69 100644
--- a/Bloxstrap/Models/Settings.cs
+++ b/Bloxstrap/Models/Settings.cs
@@ -10,6 +10,7 @@ namespace Bloxstrap.Models
         public string BootstrapperTitle { get; set; } = App.ProjectName;
         public string BootstrapperIconCustomLocation { get; set; } = "";
         public Theme Theme { get; set; } = Theme.Default;
+        public PriorityClasses ChoosePriorityClass { get; set; } = PriorityClasses.Normal;
         public bool CheckForUpdates { get; set; } = true;
         public bool CreateDesktopIcon { get; set; } = true;
         public bool ConfirmLaunches { get; set; } = false;
diff --git a/Bloxstrap/Resources/Strings.Designer.cs b/Bloxstrap/Resources/Strings.Designer.cs
index bf92399..b5f86d3 100644
--- a/Bloxstrap/Resources/Strings.Designer.cs
+++ b/Bloxstrap/Resources/Strings.Designer.cs
@@ -2963,5 +2963,77 @@ namespace Bloxstrap.Resources {
                 return ResourceManager.GetString("Uninstaller.Uninstall", resourceCulture);
             }
         }
+
+        /// <summary>
+        ///   Looks up a localized string similar to Change the priority class Roblox uses..
+        /// </summary>
+        public static string Menu_Behaviour_ChangePriorityClass_Description {
+            get {
+                return ResourceManager.GetString("Menu.Behaviour.ChangePriorityClass.Description", resourceCulture);
+            }
+        }
+
+        /// <summary>
+        ///   Looks up a localized string similar to Set priority.
+        /// </summary>
+        public static string Menu_Behaviour_ChangePriorityClass_Title {
+            get {
+                return ResourceManager.GetString("Menu.Behaviour.ChangePriorityClass.Title", resourceCulture);
+            }
+        }
+
+        /// <summary>
+        ///   Looks up a localized string similar to Low.
+        /// </summary>
+        public static string Enums_PriorityClasses_Idle {
+            get {
+                return ResourceManager.GetString("Enums.PriorityClasses.Idle", resourceCulture);
+            }
+        }
+
+        /// <summary>
+        ///   Looks up a localized string similar to Below normal.
+        /// </summary>
+        public static string Enums_PriorityClasses_BelowNormal {
+            get {
+                return ResourceManager.GetString("Enums.PriorityClasses.BelowNormal", resourceCulture);
+            }
+        }
+
+        /// <summary>
+        ///   Looks up a localized string similar to Normal.
+        /// </summary>
+        public static string Enums_PriorityClasses_Normal {
+            get {
+                return ResourceManager.GetString("Enums.PriorityClasses.Normal", resourceCulture);
+            }
+        }
+
+        /// <summary>
+        ///   Looks up a localized string similar to Above normal.
+        /// </summary>
+        public static string Enums_PriorityClasses_AboveNormal {
+            get {
+                return ResourceManager.GetString("Enums.PriorityClasses.AboveNormal", resourceCulture);
+            }
+        }
+
+        /// <summary>
+        ///   Looks up a localized string similar to High.
+        /// </summary>
+        public static string Enums_PriorityClasses_High {
+            get {
+                return ResourceManager.GetString("Enums.PriorityClasses.High", resourceCulture);
+            }
+        }
+
+        /// <summary>
+        ///   Looks up a localized string similar to Realtime.
+        /// </summary>
+        public static string Enums_PriorityClasses_RealTime {
+            get {
+                return ResourceManager.GetString("Enums.PriorityClasses.RealTime", resourceCulture);
+            }
+        }
     }
 }
diff --git a/Bloxstrap/Resources/Strings.resx b/Bloxstrap/Resources/Strings.resx
index 234c918..82f5b44 100644
--- a/Bloxstrap/Resources/Strings.resx
+++ b/Bloxstrap/Resources/Strings.resx
@@ -1120,4 +1120,28 @@ If not, then please report this exception to the maintainers of this fork. Do NO
   <data name="Menu.Shortcuts.Function.Description" xml:space="preserve">
     <value>Create shortcuts for quick access to specific functions. These will all be placed on the Desktop.</value>
   </data>
+  <data name="Menu.Behaviour.ChangePriorityClass.Title" xml:space="preserve">
+    <value>Set priority</value>
+  </data>
+  <data name="Menu.Behaviour.ChangePriorityClass.Description" xml:space="preserve">
+    <value>Change the priority class Roblox uses.</value>
+  </data>
+  <data name="Enums.PriorityClasses.Idle" xml:space="preserve">
+    <value>Low</value>
+  </data>
+  <data name="Enums.PriorityClasses.BelowNormal" xml:space="preserve">
+    <value>Below normal</value>
+  </data>
+  <data name="Enums.PriorityClasses.Normal" xml:space="preserve">
+    <value>Normal</value>
+  </data>
+  <data name="Enums.PriorityClasses.AboveNormal" xml:space="preserve">
+    <value>Above normal</value>
+  </data>
+  <data name="Enums.PriorityClasses.High" xml:space="preserve">
+    <value>High</value>
+  </data>
+  <data name="Enums.PriorityClasses.RealTime" xml:space="preserve">
+    <value>Realtime</value>
+  </data>
 </root>
\ No newline at end of file
diff --git a/Bloxstrap/UI/Elements/Settings/Pages/BehaviourPage.xaml b/Bloxstrap/UI/Elements/Settings/Pages/BehaviourPage.xaml
index 9bb7ef7..c902631 100644
--- a/Bloxstrap/UI/Elements/Settings/Pages/BehaviourPage.xaml
+++ b/Bloxstrap/UI/Elements/Settings/Pages/BehaviourPage.xaml
@@ -17,6 +17,18 @@
     <StackPanel Margin="0,0,14,14">
         <TextBlock Margin="0,0,0,8" Text="{x:Static resources:Strings.Menu_Behaviour_Description}" FontSize="14" Foreground="{DynamicResource TextFillColorSecondaryBrush}" />
 
+        <controls:OptionControl
+            Header="{x:Static resources:Strings.Menu_Behaviour_ChangePriorityClass_Title}"
+            Description="{x:Static resources:Strings.Menu_Behaviour_ChangePriorityClass_Description}">
+            <ComboBox Width="200" Padding="10,5,10,5" ItemsSource="{Binding PriorityClassesList, Mode=OneTime}" Text="{Binding ChoosePriorityClass, Mode=TwoWay}">
+                <ComboBox.ItemTemplate>
+                    <DataTemplate>
+                        <TextBlock Text="{Binding Path=., Converter={StaticResource EnumNameConverter}}" />
+                    </DataTemplate>
+                </ComboBox.ItemTemplate>
+            </ComboBox>
+        </controls:OptionControl>
+
         <controls:OptionControl 
             Header="{x:Static resources:Strings.Menu_Behaviour_CreateDesktopIcon_Title}"
             Description="{x:Static resources:Strings.Menu_Behaviour_CreateDesktopIcon_Description}">
diff --git a/Bloxstrap/UI/ViewModels/Settings/BehaviourViewModel.cs b/Bloxstrap/UI/ViewModels/Settings/BehaviourViewModel.cs
index dc2945e..8260cbe 100644
--- a/Bloxstrap/UI/ViewModels/Settings/BehaviourViewModel.cs
+++ b/Bloxstrap/UI/ViewModels/Settings/BehaviourViewModel.cs
@@ -5,6 +5,14 @@
         private string _oldPlayerVersionGuid = "";
         private string _oldStudioVersionGuid = "";
 
+        public IEnumerable<PriorityClasses> PriorityClassesList { get; } = Enum.GetValues(typeof(PriorityClasses)).Cast<PriorityClasses>();
+
+        public PriorityClasses ChoosePriorityClass
+        {
+            get => App.Settings.Prop.ChoosePriorityClass;
+            set => App.Settings.Prop.ChoosePriorityClass = value;
+        }
+
         public bool CreateDesktopIcon
         {
             get => App.Settings.Prop.CreateDesktopIcon;