diff --git a/ModAssistant/App.xaml.cs b/ModAssistant/App.xaml.cs index e8a7ae7..ac397d0 100644 --- a/ModAssistant/App.xaml.cs +++ b/ModAssistant/App.xaml.cs @@ -162,18 +162,18 @@ namespace ModAssistant break; case "--register": - if (args.Length < 2 || string.IsNullOrEmpty(args[1])) + if (args.Length < 3 || string.IsNullOrEmpty(args[1])) { Utils.SendNotify(string.Format((string)Current.FindResource("App:InvalidArgument"), "--register")); } else { - OneClickInstaller.Register(args[1], true); + OneClickInstaller.Register(args[1], true, args[2]); } Update = false; GUI = false; - args = Shift(args, 2); + args = Shift(args, 3); break; case "--unregister": diff --git a/ModAssistant/Classes/OneClickInstaller.cs b/ModAssistant/Classes/OneClickInstaller.cs index 631705a..6ab19d2 100644 --- a/ModAssistant/Classes/OneClickInstaller.cs +++ b/ModAssistant/Classes/OneClickInstaller.cs @@ -58,7 +58,7 @@ namespace ModAssistant 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 @@ namespace ModAssistant 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 @@ namespace ModAssistant } 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 index b2215cd..dfe605c 100644 --- a/ModAssistant/Pages/Options.xaml.cs +++ b/ModAssistant/Pages/Options.xaml.cs @@ -137,7 +137,7 @@ namespace ModAssistant.Pages 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 @@ namespace ModAssistant.Pages 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 @@ namespace ModAssistant.Pages } 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)