Add description for OneClick registry keys

Adds a description to protocol handler registry keys when setting up OneClick install.
bsmg · Jan 19, 2021 · 140dde1 · 140dde1
1 parent 3b9dc22
commit 140dde1
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 7 deletions.
diff --git a/ModAssistant/App.xaml.cs b/ModAssistant/App.xaml.cs
@@ -168,12 +168,12 @@ private async Task ArgumentHandler(string[] args)
                         }
                         else
                         {
-                            OneClickInstaller.Register(args[1], true);
+                            OneClickInstaller.Register(args[1], true, args[2]);
                         }
 
                         Update = false;
                         GUI = false;
-                        args = Shift(args, 2);
+                        args = Shift(args, args.Length);
                         break;
 
                     case "--unregister":

diff --git a/ModAssistant/Classes/OneClickInstaller.cs b/ModAssistant/Classes/OneClickInstaller.cs
@@ -58,7 +58,7 @@ private static async Task Playlist(Uri uri)
             await API.Playlists.DownloadAll(uri);
         }
 
-        public static void Register(string Protocol, bool Background = false)
+        public static void Register(string Protocol, bool Background = false, string Description = null)
         {
             if (IsRegistered(Protocol) == true)
                 return;
@@ -75,6 +75,10 @@ public static void Register(string Protocol, bool Background = false)
 
                     if (ProtocolKey.GetValue("OneClick-Provider", "").ToString() != "ModAssistant")
                     {
+                        if (Description != null)
+                        {
+                            ProtocolKey.SetValue("", Description, RegistryValueKind.String);
+                        }
                         ProtocolKey.SetValue("URL Protocol", "", RegistryValueKind.String);
                         ProtocolKey.SetValue("OneClick-Provider", "ModAssistant", RegistryValueKind.String);
                         CommandKey.SetValue("", $"\"{Utils.ExePath}\" \"--install\" \"%1\"");
@@ -84,7 +88,7 @@ public static void Register(string Protocol, bool Background = false)
                 }
                 else
                 {
-                    Utils.StartAsAdmin($"\"--register\" \"{Protocol}\"");
+                    Utils.StartAsAdmin($"\"--register\" \"{Protocol}\" \"{Description}\"");
                 }
             }
             catch (Exception e)

diff --git a/ModAssistant/Pages/Options.xaml.cs b/ModAssistant/Pages/Options.xaml.cs
@@ -137,7 +137,7 @@ private void CloseWindowOnFinish_Unchecked(object sender, RoutedEventArgs e)
 
         public void ModelSaberProtocolHandler_Checked(object sender, RoutedEventArgs e)
         {
-            OneClickInstaller.Register("modelsaber");
+            OneClickInstaller.Register("modelsaber", Description: "URL:ModelSaber OneClick Install");
         }
 
         public void ModelSaberProtocolHandler_Unchecked(object sender, RoutedEventArgs e)
@@ -147,7 +147,7 @@ public void ModelSaberProtocolHandler_Unchecked(object sender, RoutedEventArgs e
 
         public void BeatSaverProtocolHandler_Checked(object sender, RoutedEventArgs e)
         {
-            OneClickInstaller.Register("beatsaver");
+            OneClickInstaller.Register("beatsaver", Description: "URL:BeatSaver OneClick Install");
         }
 
         public void BeatSaverProtocolHandler_Unchecked(object sender, RoutedEventArgs e)
@@ -156,7 +156,7 @@ public void BeatSaverProtocolHandler_Unchecked(object sender, RoutedEventArgs e)
         }
         public void PlaylistsProtocolHandler_Checked(object sender, RoutedEventArgs e)
         {
-            OneClickInstaller.Register("bsplaylist");
+            OneClickInstaller.Register("bsplaylist", Description: "URL:BeatSaver Playlist OneClick Install");
         }
 
         public void PlaylistsProtocolHandler_Unchecked(object sender, RoutedEventArgs e)