Discord.Net/src/Discord.Net.Interactions/RestInteractionModuleBase.cs

using Discord.Rest;
using System;
using System.Threading.Tasks;

namespace Discord.Interactions
{
    /// <summary>
    ///     Provides a base class for a Rest based command module to inherit from.
    /// </summary>
    /// <typeparam name="T">Type of interaction context to be injected into the module.</typeparam>
    public abstract class RestInteractionModuleBase<T> : InteractionModuleBase<T>
        where T : class, IInteractionContext
    {
        /// <summary>
        ///     Gets or sets the underlying Interaction Service.
        /// </summary>
        public InteractionService InteractionService { get; set; }

        /// <summary>
        ///     Defer a Rest based Discord Interaction using the <see cref="InteractionServiceConfig.RestResponseCallback"/> delegate.
        /// </summary>
        /// <param name="ephemeral"><see langword="true"/> if the response should be hidden to everyone besides the invoker of the command, otherwise <see langword="false"/>.</param>
        /// <param name="options">The request options for this response.</param>
        /// <returns>
        ///     A Task representing the operation of creating the interaction response.
        /// </returns>
        /// <exception cref="InvalidOperationException">Thrown if the interaction isn't a type of <see cref="RestInteraction"/>.</exception>
        protected override async Task DeferAsync(bool ephemeral = false, RequestOptions options = null)
        {
            if (Context.Interaction is not RestInteraction restInteraction)
                throw new InvalidOperationException($"Invalid interaction type. Interaction must be a type of {nameof(RestInteraction)} in order to execute this method");

            var payload = restInteraction.Defer(ephemeral, options);

            if (Context is IRestInteractionContext restContext && restContext.InteractionResponseCallback != null)
                await restContext.InteractionResponseCallback.Invoke(payload).ConfigureAwait(false);
            else
                await InteractionService._restResponseCallback(Context, payload).ConfigureAwait(false);
        }

        /// <summary>
        ///     Respond to a Rest based Discord Interaction using the <see cref="InteractionServiceConfig.RestResponseCallback"/> delegate.
        /// </summary>
        /// <param name="text">The text of the message to be sent.</param>
        /// <param name="embeds">A array of embeds to send with this response. Max 10.</param>
        /// <param name="isTTS"><see langword="true"/> if the message should be read out by a text-to-speech reader, otherwise <see langword="false"/>.</param>
        /// <param name="ephemeral"><see langword="true"/> if the response should be hidden to everyone besides the invoker of the command, otherwise <see langword="false"/>.</param>
        /// <param name="allowedMentions">The allowed mentions for this response.</param>
        /// <param name="options">The request options for this response.</param>
        /// <param name="components">A <see cref="MessageComponent"/> to be sent with this response.</param>
        /// <param name="embed">A single embed to send with this response. If this is passed alongside an array of embeds, the single embed will be ignored.</param>
        /// <returns>
        ///     A Task representing the operation of creating the interaction response.
        /// </returns>
        /// <exception cref="InvalidOperationException">Thrown if the interaction isn't a type of <see cref="RestInteraction"/>.</exception>
        protected override async Task RespondAsync(string text = null, Embed[] embeds = null, bool isTTS = false, bool ephemeral = false, AllowedMentions allowedMentions = null, RequestOptions options = null, MessageComponent components = null, Embed embed = null)
        {
            if (Context.Interaction is not RestInteraction restInteraction)
                throw new InvalidOperationException($"Invalid interaction type. Interaction must be a type of {nameof(RestInteraction)} in order to execute this method");

            var payload = restInteraction.Respond(text, embeds, isTTS, ephemeral, allowedMentions, components, embed, options);

            if (Context is IRestInteractionContext restContext && restContext.InteractionResponseCallback != null)
                await restContext.InteractionResponseCallback.Invoke(payload).ConfigureAwait(false);
            else
                await InteractionService._restResponseCallback(Context, payload).ConfigureAwait(false);
        }

        /// <summary>
        ///     Responds to the interaction with a modal.
        /// </summary>
        /// <param name="modal">The modal to respond with.</param>
        /// <param name="options">The request options for this <see langword="async"/> request.</param>
        /// <returns>A string that contains json to write back to the incoming http request.</returns>
        /// <exception cref="TimeoutException"></exception>
        /// <exception cref="InvalidOperationException"></exception>
        protected override async Task RespondWithModalAsync(Modal modal, RequestOptions options = null)
        {
            if (Context.Interaction is not RestInteraction restInteraction)
                throw new InvalidOperationException($"Invalid interaction type. Interaction must be a type of {nameof(RestInteraction)} in order to execute this method");

            var payload = restInteraction.RespondWithModal(modal, options);

            if (Context is IRestInteractionContext restContext && restContext.InteractionResponseCallback != null)
                await restContext.InteractionResponseCallback.Invoke(payload).ConfigureAwait(false);
            else
                await InteractionService._restResponseCallback(Context, payload).ConfigureAwait(false);
        }

        protected override async Task RespondWithModalAsync<T>(string customId, RequestOptions options = null)
        {
            if (Context.Interaction is not RestInteraction restInteraction)
                throw new InvalidOperationException($"Invalid interaction type. Interaction must be a type of {nameof(RestInteraction)} in order to execute this method");

            var payload = restInteraction.RespondWithModal<T>(customId, options);

            if (Context is IRestInteractionContext restContext && restContext.InteractionResponseCallback != null)
                await restContext.InteractionResponseCallback.Invoke(payload).ConfigureAwait(false);
            else
                await InteractionService._restResponseCallback(Context, payload).ConfigureAwait(false);
        }
    }
}