Only prompt channel change once during enrollment

bloxstraplabs · Jul 16, 2023 · 7caa59e · 7caa59e
1 parent 8a4adae
commit 7caa59e
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/Bloxstrap/Models/State.cs b/Bloxstrap/Models/State.cs
@@ -2,6 +2,7 @@
 {
     public class State
     {
+        public string LastEnrolledChannel { get; set; } = "";
         public string VersionGuid { get; set; } = "";
         public List<string> ModManifest { get; set; } = new();
     }

diff --git a/Bloxstrap/ProtocolHandler.cs b/Bloxstrap/ProtocolHandler.cs
@@ -55,7 +55,7 @@ public static string ParseUri(string protocol)
                 if (key == "channel" && !String.IsNullOrEmpty(val))
                 {
                     channelArgPresent = true;
-                    ChangeChannel(val);
+                    EnrollChannel(val);
 
                     // we'll set the arg when launching
                     continue;
@@ -65,7 +65,7 @@ public static string ParseUri(string protocol)
             }
 
             if (!channelArgPresent)
-                ChangeChannel(RobloxDeployment.DefaultChannel);
+                EnrollChannel(RobloxDeployment.DefaultChannel);
 
             return commandLine.ToString();
         }
@@ -80,9 +80,12 @@ public static void ChangeChannel(string channel)
 
             if (App.Settings.Prop.ChannelChangeMode != ChannelChangeMode.Automatic)
             {
+                if (channel == App.State.Prop.LastEnrolledChannel)
+                    return;
+
                 MessageBoxResult result = Controls.ShowMessageBox(
                     $"Roblox is attempting to set your channel to {channel}, however your current preferred channel is {App.Settings.Prop.Channel}.\n\n" +
-                    $"Would you like to switch channels from {App.Settings.Prop.Channel} to {channel}?",
+                    $"Would you like to switch your preferred channel to {channel}?",
                     MessageBoxImage.Question,
                     MessageBoxButton.YesNo
                 );
@@ -95,6 +98,13 @@ public static void ChangeChannel(string channel)
             App.Settings.Prop.Channel = channel;
         }
 
+        public static void EnrollChannel(string channel)
+        {
+            ChangeChannel(channel);
+            App.State.Prop.LastEnrolledChannel = channel;
+            App.State.Save();
+        }
+
         public static void Register(string key, string name, string handler)
         {
             string handlerArgs = $"\"{handler}\" %1";