From 9a52d05637e5f68a57f12a14d14172044e78ca8d Mon Sep 17 00:00:00 2001
From: Quin Lynch <49576606+quinchs@users.noreply.github.com>
Date: Fri, 14 Jan 2022 07:52:43 -0400
Subject: [PATCH] Add search methods to InteractionService (#2041)

---
 .../InteractionService.cs                     | 54 +++++++++++++++++++
 1 file changed, 54 insertions(+)

diff --git a/src/Discord.Net.Interactions/InteractionService.cs b/src/Discord.Net.Interactions/InteractionService.cs
index ac96f08ec..b394109a5 100644
--- a/src/Discord.Net.Interactions/InteractionService.cs
+++ b/src/Discord.Net.Interactions/InteractionService.cs
@@ -588,6 +588,60 @@ namespace Discord.Interactions
             return true;
         }
 
+        /// <summary>
+        ///     Search the registered slash commands using a <see cref="ISlashCommandInteraction"/>.
+        /// </summary>
+        /// <param name="slashCommandInteraction">Interaction entity to perform the search with.</param>
+        /// <returns>
+        ///     The search result. When successful, result contains the found <see cref="SlashCommandInfo"/>.
+        /// </returns>
+        public SearchResult<SlashCommandInfo> SearchSlashCommand(ISlashCommandInteraction slashCommandInteraction)
+            => _slashCommandMap.GetCommand(slashCommandInteraction.Data.GetCommandKeywords());
+
+        /// <summary>
+        ///     Search the registered slash commands using a <see cref="IComponentInteraction"/>.
+        /// </summary>
+        /// <param name="componentInteraction">Interaction entity to perform the search with.</param>
+        /// <returns>
+        ///     The search result. When successful, result contains the found <see cref="ComponentCommandInfo"/>.
+        /// </returns>
+        public SearchResult<ComponentCommandInfo> SearchComponentCommand(IComponentInteraction componentInteraction)
+            => _componentCommandMap.GetCommand(componentInteraction.Data.CustomId);
+
+        /// <summary>
+        ///     Search the registered slash commands using a <see cref="IUserCommandInteraction"/>.
+        /// </summary>
+        /// <param name="userCommandInteraction">Interaction entity to perform the search with.</param>
+        /// <returns>
+        ///     The search result. When successful, result contains the found <see cref="ContextCommandInfo"/>.
+        /// </returns>
+        public SearchResult<ContextCommandInfo> SearchUserCommand(IUserCommandInteraction userCommandInteraction)
+            => _contextCommandMaps[ApplicationCommandType.User].GetCommand(userCommandInteraction.Data.Name);
+
+        /// <summary>
+        ///     Search the registered slash commands using a <see cref="IMessageCommandInteraction"/>.
+        /// </summary>
+        /// <param name="messageCommandInteraction">Interaction entity to perform the search with.</param>
+        /// <returns>
+        ///     The search result. When successful, result contains the found <see cref="ContextCommandInfo"/>.
+        /// </returns>
+        public SearchResult<ContextCommandInfo> SearchMessageCommand(IMessageCommandInteraction messageCommandInteraction)
+            => _contextCommandMaps[ApplicationCommandType.Message].GetCommand(messageCommandInteraction.Data.Name);
+
+        /// <summary>
+        ///     Search the registered slash commands using a <see cref="IAutocompleteInteraction"/>.
+        /// </summary>
+        /// <param name="autocompleteInteraction">Interaction entity to perform the search with.</param>
+        /// <returns>
+        ///     The search result. When successful, result contains the found <see cref="AutocompleteCommandInfo"/>.
+        /// </returns>
+        public SearchResult<AutocompleteCommandInfo> SearchAutocompleteCommand(IAutocompleteInteraction autocompleteInteraction)
+        {
+            var keywords = autocompleteInteraction.Data.GetCommandKeywords();
+            keywords.Add(autocompleteInteraction.Data.Current.Name);
+            return _autocompleteCommandMap.GetCommand(keywords);
+        }
+
         /// <summary>
         ///     Execute a Command from a given <see cref="IInteractionContext"/>.
         /// </summary>