Discord.Net/src/Discord.Net.WebSocket/Interactions/SocketInteractionContext.cs

using Discord.WebSocket;
using System.Collections.Generic;
using System.Collections.Immutable;

namespace Discord.Interactions
{
    /// <summary>
    ///     Represents a Web-Socket based context of an <see cref="IDiscordInteraction"/>.
    /// </summary>
    public class SocketInteractionContext<TInteraction> : IInteractionContext, IRouteMatchContainer
        where TInteraction : SocketInteraction
    {
        /// <summary>
        ///     Gets the <see cref="DiscordSocketClient"/> that the command will be executed with.
        /// </summary>
        public DiscordSocketClient Client { get; }

        /// <summary>
        ///     Gets the <see cref="SocketGuild"/> the command originated from.
        /// </summary>
        /// <remarks>
        ///     Will be null if the command is from a DM Channel.
        /// </remarks>
        public SocketGuild Guild { get; }

        /// <summary>
        ///     Gets the <see cref="ISocketMessageChannel"/> the command originated from.
        /// </summary>
        public ISocketMessageChannel Channel { get; }

        /// <summary>
        ///     Gets the <see cref="SocketUser"/> who executed the command.
        /// </summary>
        public SocketUser User { get; }

        /// <summary>
        ///     Gets the <see cref="SocketInteraction"/> the command was recieved with.
        /// </summary>
        public TInteraction Interaction { get; }

        /// <inheritdoc cref="IRouteMatchContainer.SegmentMatches"/>
        public IReadOnlyCollection<IRouteSegmentMatch> SegmentMatches { get; private set; }

        /// <summary>
        ///     Initializes a new <see cref="SocketInteractionContext{TInteraction}"/>. 
        /// </summary>
        /// <param name="client">The underlying client.</param>
        /// <param name="interaction">The underlying interaction.</param>
        public SocketInteractionContext(DiscordSocketClient client, TInteraction interaction)
        {
            Client = client;
            Channel = interaction.Channel;
            Guild = (interaction.User as SocketGuildUser)?.Guild.Value;
            User = interaction.User;
            Interaction = interaction;
        }

        /// <inheritdoc/>
        public void SetSegmentMatches(IEnumerable<IRouteSegmentMatch> segmentMatches) => SegmentMatches = segmentMatches.ToImmutableArray();

        //IRouteMatchContainer
        /// <inheritdoc/>
        IEnumerable<IRouteSegmentMatch> IRouteMatchContainer.SegmentMatches => SegmentMatches;

        // IInteractionContext
        /// <inheritdoc/>
        IDiscordClient IInteractionContext.Client => Client;

        /// <inheritdoc/>
        IGuild IInteractionContext.Guild => Guild;

        /// <inheritdoc/>
        IMessageChannel IInteractionContext.Channel => Channel;

        /// <inheritdoc/>
        IUser IInteractionContext.User => User;

        /// <inheritdoc/>
        IDiscordInteraction IInteractionContext.Interaction => Interaction;
    }

    /// <summary>
    ///     Represents a Web-Socket based context of an <see cref="IDiscordInteraction"/>
    /// </summary>
    public class SocketInteractionContext : SocketInteractionContext<SocketInteraction>
    {
        /// <summary>
        ///     Initializes a new <see cref="SocketInteractionContext"/> 
        /// </summary>
        /// <param name="client">The underlying client</param>
        /// <param name="interaction">The underlying interaction</param>
        public SocketInteractionContext(DiscordSocketClient client, SocketInteraction interaction) : base(client, interaction) { }
    }
}