664 lines
30 KiB
C#
664 lines
30 KiB
C#
using Discord.Audio;
|
|
using Discord.Rest;
|
|
using System;
|
|
using System.Collections.Concurrent;
|
|
using System.Collections.Generic;
|
|
using System.Collections.Immutable;
|
|
using System.Diagnostics;
|
|
using System.Linq;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using ChannelModel = Discord.API.Channel;
|
|
using EmojiUpdateModel = Discord.API.Gateway.GuildEmojiUpdateEvent;
|
|
using ExtendedModel = Discord.API.Gateway.ExtendedGuild;
|
|
using GuildSyncModel = Discord.API.Gateway.GuildSyncEvent;
|
|
using MemberModel = Discord.API.GuildMember;
|
|
using Model = Discord.API.Guild;
|
|
using PresenceModel = Discord.API.Presence;
|
|
using RoleModel = Discord.API.Role;
|
|
using UserModel = Discord.API.User;
|
|
using VoiceStateModel = Discord.API.VoiceState;
|
|
|
|
namespace Discord.WebSocket
|
|
{
|
|
public class SocketGuild : SocketEntity<ulong>, IGuild
|
|
{
|
|
private readonly SemaphoreSlim _audioLock;
|
|
private TaskCompletionSource<bool> _syncPromise, _downloaderPromise;
|
|
private TaskCompletionSource<AudioClient> _audioConnectPromise;
|
|
private ConcurrentHashSet<ulong> _channels;
|
|
private ConcurrentDictionary<ulong, SocketGuildUser> _members;
|
|
private ConcurrentDictionary<ulong, SocketRole> _roles;
|
|
private ConcurrentDictionary<ulong, SocketVoiceState> _voiceStates;
|
|
private ImmutableArray<GuildEmote> _emotes;
|
|
private ImmutableArray<string> _features;
|
|
private AudioClient _audioClient;
|
|
|
|
public string Name { get; private set; }
|
|
public int AFKTimeout { get; private set; }
|
|
public bool IsEmbeddable { get; private set; }
|
|
public VerificationLevel VerificationLevel { get; private set; }
|
|
public MfaLevel MfaLevel { get; private set; }
|
|
public DefaultMessageNotifications DefaultMessageNotifications { get; private set; }
|
|
public int MemberCount { get; internal set; }
|
|
public int DownloadedMemberCount { get; private set; }
|
|
internal bool IsAvailable { get; private set; }
|
|
public bool IsConnected { get; internal set; }
|
|
|
|
internal ulong? AFKChannelId { get; private set; }
|
|
internal ulong? EmbedChannelId { get; private set; }
|
|
public ulong OwnerId { get; private set; }
|
|
public SocketGuildUser Owner => GetUser(OwnerId);
|
|
public string VoiceRegionId { get; private set; }
|
|
public string IconId { get; private set; }
|
|
public string SplashId { get; private set; }
|
|
|
|
public DateTimeOffset CreatedAt => SnowflakeUtils.FromSnowflake(Id);
|
|
public SocketTextChannel DefaultChannel => GetTextChannel(Id);
|
|
public string IconUrl => CDN.GetGuildIconUrl(Id, IconId);
|
|
public string SplashUrl => CDN.GetGuildSplashUrl(Id, SplashId);
|
|
public bool HasAllMembers => MemberCount == DownloadedMemberCount;// _downloaderPromise.Task.IsCompleted;
|
|
public bool IsSynced => _syncPromise.Task.IsCompleted;
|
|
public Task SyncPromise => _syncPromise.Task;
|
|
public Task DownloaderPromise => _downloaderPromise.Task;
|
|
public IAudioClient AudioClient => _audioClient;
|
|
public SocketVoiceChannel AFKChannel
|
|
{
|
|
get
|
|
{
|
|
var id = AFKChannelId;
|
|
return id.HasValue ? GetVoiceChannel(id.Value) : null;
|
|
}
|
|
}
|
|
public SocketGuildChannel EmbedChannel
|
|
{
|
|
get
|
|
{
|
|
var id = EmbedChannelId;
|
|
return id.HasValue ? GetChannel(id.Value) : null;
|
|
}
|
|
}
|
|
public IReadOnlyCollection<SocketTextChannel> TextChannels
|
|
=> Channels.Select(x => x as SocketTextChannel).Where(x => x != null).ToImmutableArray();
|
|
public IReadOnlyCollection<SocketVoiceChannel> VoiceChannels
|
|
=> Channels.Select(x => x as SocketVoiceChannel).Where(x => x != null).ToImmutableArray();
|
|
public SocketGuildUser CurrentUser => _members.TryGetValue(Discord.CurrentUser.Id, out SocketGuildUser member) ? member : null;
|
|
public SocketRole EveryoneRole => GetRole(Id);
|
|
public IReadOnlyCollection<SocketGuildChannel> Channels
|
|
{
|
|
get
|
|
{
|
|
var channels = _channels;
|
|
var state = Discord.State;
|
|
return channels.Select(x => state.GetChannel(x) as SocketGuildChannel).Where(x => x != null).ToReadOnlyCollection(channels);
|
|
}
|
|
}
|
|
public IReadOnlyCollection<GuildEmote> Emotes => _emotes;
|
|
public IReadOnlyCollection<string> Features => _features;
|
|
public IReadOnlyCollection<SocketGuildUser> Users => _members.ToReadOnlyCollection();
|
|
public IReadOnlyCollection<SocketRole> Roles => _roles.ToReadOnlyCollection();
|
|
|
|
internal SocketGuild(DiscordSocketClient client, ulong id)
|
|
: base(client, id)
|
|
{
|
|
_audioLock = new SemaphoreSlim(1, 1);
|
|
_emotes = ImmutableArray.Create<GuildEmote>();
|
|
_features = ImmutableArray.Create<string>();
|
|
}
|
|
internal static SocketGuild Create(DiscordSocketClient discord, ClientState state, ExtendedModel model)
|
|
{
|
|
var entity = new SocketGuild(discord, model.Id);
|
|
entity.Update(state, model);
|
|
return entity;
|
|
}
|
|
internal void Update(ClientState state, ExtendedModel model)
|
|
{
|
|
IsAvailable = !(model.Unavailable ?? false);
|
|
if (!IsAvailable)
|
|
{
|
|
if (_channels == null)
|
|
_channels = new ConcurrentHashSet<ulong>();
|
|
if (_members == null)
|
|
_members = new ConcurrentDictionary<ulong, SocketGuildUser>();
|
|
if (_roles == null)
|
|
_roles = new ConcurrentDictionary<ulong, SocketRole>();
|
|
/*if (Emojis == null)
|
|
_emojis = ImmutableArray.Create<Emoji>();
|
|
if (Features == null)
|
|
_features = ImmutableArray.Create<string>();*/
|
|
_syncPromise = new TaskCompletionSource<bool>();
|
|
_downloaderPromise = new TaskCompletionSource<bool>();
|
|
return;
|
|
}
|
|
|
|
Update(state, model as Model);
|
|
|
|
var channels = new ConcurrentHashSet<ulong>(ConcurrentHashSet.DefaultConcurrencyLevel, (int)(model.Channels.Length * 1.05));
|
|
{
|
|
for (int i = 0; i < model.Channels.Length; i++)
|
|
{
|
|
var channel = SocketGuildChannel.Create(this, state, model.Channels[i]);
|
|
state.AddChannel(channel);
|
|
channels.TryAdd(channel.Id);
|
|
}
|
|
}
|
|
_channels = channels;
|
|
|
|
var members = new ConcurrentDictionary<ulong, SocketGuildUser>(ConcurrentHashSet.DefaultConcurrencyLevel, (int)(model.Members.Length * 1.05));
|
|
{
|
|
for (int i = 0; i < model.Members.Length; i++)
|
|
{
|
|
var member = SocketGuildUser.Create(this, state, model.Members[i]);
|
|
members.TryAdd(member.Id, member);
|
|
}
|
|
DownloadedMemberCount = members.Count;
|
|
|
|
for (int i = 0; i < model.Presences.Length; i++)
|
|
{
|
|
if (members.TryGetValue(model.Presences[i].User.Id, out SocketGuildUser member))
|
|
member.Update(state, model.Presences[i], true);
|
|
else
|
|
Debug.Assert(false);
|
|
}
|
|
}
|
|
_members = members;
|
|
MemberCount = model.MemberCount;
|
|
|
|
var voiceStates = new ConcurrentDictionary<ulong, SocketVoiceState>(ConcurrentHashSet.DefaultConcurrencyLevel, (int)(model.VoiceStates.Length * 1.05));
|
|
{
|
|
for (int i = 0; i < model.VoiceStates.Length; i++)
|
|
{
|
|
SocketVoiceChannel channel = null;
|
|
if (model.VoiceStates[i].ChannelId.HasValue)
|
|
channel = state.GetChannel(model.VoiceStates[i].ChannelId.Value) as SocketVoiceChannel;
|
|
var voiceState = SocketVoiceState.Create(channel, model.VoiceStates[i]);
|
|
voiceStates.TryAdd(model.VoiceStates[i].UserId, voiceState);
|
|
}
|
|
}
|
|
_voiceStates = voiceStates;
|
|
|
|
_syncPromise = new TaskCompletionSource<bool>();
|
|
_downloaderPromise = new TaskCompletionSource<bool>();
|
|
if (Discord.ApiClient.AuthTokenType != TokenType.User)
|
|
{
|
|
var _ = _syncPromise.TrySetResultAsync(true);
|
|
/*if (!model.Large)
|
|
_ = _downloaderPromise.TrySetResultAsync(true);*/
|
|
}
|
|
}
|
|
internal void Update(ClientState state, Model model)
|
|
{
|
|
AFKChannelId = model.AFKChannelId;
|
|
EmbedChannelId = model.EmbedChannelId;
|
|
AFKTimeout = model.AFKTimeout;
|
|
IsEmbeddable = model.EmbedEnabled;
|
|
IconId = model.Icon;
|
|
Name = model.Name;
|
|
OwnerId = model.OwnerId;
|
|
VoiceRegionId = model.Region;
|
|
SplashId = model.Splash;
|
|
VerificationLevel = model.VerificationLevel;
|
|
MfaLevel = model.MfaLevel;
|
|
DefaultMessageNotifications = model.DefaultMessageNotifications;
|
|
|
|
if (model.Emojis != null)
|
|
{
|
|
var emojis = ImmutableArray.CreateBuilder<GuildEmote>(model.Emojis.Length);
|
|
for (int i = 0; i < model.Emojis.Length; i++)
|
|
emojis.Add(model.Emojis[i].ToEntity());
|
|
_emotes = emojis.ToImmutable();
|
|
}
|
|
else
|
|
_emotes = ImmutableArray.Create<GuildEmote>();
|
|
|
|
if (model.Features != null)
|
|
_features = model.Features.ToImmutableArray();
|
|
else
|
|
_features = ImmutableArray.Create<string>();
|
|
|
|
var roles = new ConcurrentDictionary<ulong, SocketRole>(ConcurrentHashSet.DefaultConcurrencyLevel, (int)(model.Roles.Length * 1.05));
|
|
if (model.Roles != null)
|
|
{
|
|
for (int i = 0; i < model.Roles.Length; i++)
|
|
{
|
|
var role = SocketRole.Create(this, state, model.Roles[i]);
|
|
roles.TryAdd(role.Id, role);
|
|
}
|
|
}
|
|
_roles = roles;
|
|
}
|
|
internal void Update(ClientState state, GuildSyncModel model)
|
|
{
|
|
var members = new ConcurrentDictionary<ulong, SocketGuildUser>(ConcurrentHashSet.DefaultConcurrencyLevel, (int)(model.Members.Length * 1.05));
|
|
{
|
|
for (int i = 0; i < model.Members.Length; i++)
|
|
{
|
|
var member = SocketGuildUser.Create(this, state, model.Members[i]);
|
|
members.TryAdd(member.Id, member);
|
|
}
|
|
DownloadedMemberCount = members.Count;
|
|
|
|
for (int i = 0; i < model.Presences.Length; i++)
|
|
{
|
|
if (members.TryGetValue(model.Presences[i].User.Id, out SocketGuildUser member))
|
|
member.Update(state, model.Presences[i], true);
|
|
else
|
|
Debug.Assert(false);
|
|
}
|
|
}
|
|
_members = members;
|
|
|
|
var _ = _syncPromise.TrySetResultAsync(true);
|
|
/*if (!model.Large)
|
|
_ = _downloaderPromise.TrySetResultAsync(true);*/
|
|
}
|
|
|
|
internal void Update(ClientState state, EmojiUpdateModel model)
|
|
{
|
|
var emotes = ImmutableArray.CreateBuilder<GuildEmote>(model.Emojis.Length);
|
|
for (int i = 0; i < model.Emojis.Length; i++)
|
|
emotes.Add(model.Emojis[i].ToEntity());
|
|
_emotes = emotes.ToImmutable();
|
|
}
|
|
|
|
//General
|
|
public Task DeleteAsync(RequestOptions options = null)
|
|
=> GuildHelper.DeleteAsync(this, Discord, options);
|
|
|
|
public Task ModifyAsync(Action<GuildProperties> func, RequestOptions options = null)
|
|
=> GuildHelper.ModifyAsync(this, Discord, func, options);
|
|
public Task ModifyEmbedAsync(Action<GuildEmbedProperties> func, RequestOptions options = null)
|
|
=> GuildHelper.ModifyEmbedAsync(this, Discord, func, options);
|
|
public Task ReorderChannelsAsync(IEnumerable<ReorderChannelProperties> args, RequestOptions options = null)
|
|
=> GuildHelper.ReorderChannelsAsync(this, Discord, args, options);
|
|
public Task ReorderRolesAsync(IEnumerable<ReorderRoleProperties> args, RequestOptions options = null)
|
|
=> GuildHelper.ReorderRolesAsync(this, Discord, args, options);
|
|
|
|
public Task LeaveAsync(RequestOptions options = null)
|
|
=> GuildHelper.LeaveAsync(this, Discord, options);
|
|
|
|
//Bans
|
|
public Task<IReadOnlyCollection<RestBan>> GetBansAsync(RequestOptions options = null)
|
|
=> GuildHelper.GetBansAsync(this, Discord, options);
|
|
|
|
public Task AddBanAsync(IUser user, int pruneDays = 0, RequestOptions options = null)
|
|
=> GuildHelper.AddBanAsync(this, Discord, user.Id, pruneDays, options);
|
|
public Task AddBanAsync(ulong userId, int pruneDays = 0, RequestOptions options = null)
|
|
=> GuildHelper.AddBanAsync(this, Discord, userId, pruneDays, options);
|
|
|
|
public Task RemoveBanAsync(IUser user, RequestOptions options = null)
|
|
=> GuildHelper.RemoveBanAsync(this, Discord, user.Id, options);
|
|
public Task RemoveBanAsync(ulong userId, RequestOptions options = null)
|
|
=> GuildHelper.RemoveBanAsync(this, Discord, userId, options);
|
|
|
|
//Channels
|
|
public SocketGuildChannel GetChannel(ulong id)
|
|
{
|
|
var channel = Discord.State.GetChannel(id) as SocketGuildChannel;
|
|
if (channel?.Guild.Id == Id)
|
|
return channel;
|
|
return null;
|
|
}
|
|
public SocketTextChannel GetTextChannel(ulong id)
|
|
=> GetChannel(id) as SocketTextChannel;
|
|
public SocketVoiceChannel GetVoiceChannel(ulong id)
|
|
=> GetChannel(id) as SocketVoiceChannel;
|
|
public Task<RestTextChannel> CreateTextChannelAsync(string name, RequestOptions options = null)
|
|
=> GuildHelper.CreateTextChannelAsync(this, Discord, name, options);
|
|
public Task<RestVoiceChannel> CreateVoiceChannelAsync(string name, RequestOptions options = null)
|
|
=> GuildHelper.CreateVoiceChannelAsync(this, Discord, name, options);
|
|
internal SocketGuildChannel AddChannel(ClientState state, ChannelModel model)
|
|
{
|
|
var channel = SocketGuildChannel.Create(this, state, model);
|
|
_channels.TryAdd(model.Id);
|
|
state.AddChannel(channel);
|
|
return channel;
|
|
}
|
|
internal SocketGuildChannel RemoveChannel(ClientState state, ulong id)
|
|
{
|
|
if (_channels.TryRemove(id))
|
|
return state.RemoveChannel(id) as SocketGuildChannel;
|
|
return null;
|
|
}
|
|
|
|
//Integrations
|
|
public Task<IReadOnlyCollection<RestGuildIntegration>> GetIntegrationsAsync(RequestOptions options = null)
|
|
=> GuildHelper.GetIntegrationsAsync(this, Discord, options);
|
|
public Task<RestGuildIntegration> CreateIntegrationAsync(ulong id, string type, RequestOptions options = null)
|
|
=> GuildHelper.CreateIntegrationAsync(this, Discord, id, type, options);
|
|
|
|
//Invites
|
|
public Task<IReadOnlyCollection<RestInviteMetadata>> GetInvitesAsync(RequestOptions options = null)
|
|
=> GuildHelper.GetInvitesAsync(this, Discord, options);
|
|
|
|
//Roles
|
|
public SocketRole GetRole(ulong id)
|
|
{
|
|
if (_roles.TryGetValue(id, out SocketRole value))
|
|
return value;
|
|
return null;
|
|
}
|
|
public Task<RestRole> CreateRoleAsync(string name, GuildPermissions? permissions = default(GuildPermissions?), Color? color = default(Color?),
|
|
bool isHoisted = false, RequestOptions options = null)
|
|
=> GuildHelper.CreateRoleAsync(this, Discord, name, permissions, color, isHoisted, options);
|
|
internal SocketRole AddRole(RoleModel model)
|
|
{
|
|
var role = SocketRole.Create(this, Discord.State, model);
|
|
_roles[model.Id] = role;
|
|
return role;
|
|
}
|
|
internal SocketRole RemoveRole(ulong id)
|
|
{
|
|
if (_roles.TryRemove(id, out SocketRole role))
|
|
return role;
|
|
return null;
|
|
}
|
|
|
|
//Users
|
|
public SocketGuildUser GetUser(ulong id)
|
|
{
|
|
if (_members.TryGetValue(id, out SocketGuildUser member))
|
|
return member;
|
|
return null;
|
|
}
|
|
public Task<int> PruneUsersAsync(int days = 30, bool simulate = false, RequestOptions options = null)
|
|
=> GuildHelper.PruneUsersAsync(this, Discord, days, simulate, options);
|
|
|
|
internal SocketGuildUser AddOrUpdateUser(UserModel model)
|
|
{
|
|
if (_members.TryGetValue(model.Id, out SocketGuildUser member))
|
|
member.GlobalUser?.Update(Discord.State, model);
|
|
else
|
|
{
|
|
member = SocketGuildUser.Create(this, Discord.State, model);
|
|
member.GlobalUser.AddRef();
|
|
_members[member.Id] = member;
|
|
DownloadedMemberCount++;
|
|
}
|
|
return member;
|
|
}
|
|
internal SocketGuildUser AddOrUpdateUser(MemberModel model)
|
|
{
|
|
if (_members.TryGetValue(model.User.Id, out SocketGuildUser member))
|
|
member.Update(Discord.State, model);
|
|
else
|
|
{
|
|
member = SocketGuildUser.Create(this, Discord.State, model);
|
|
member.GlobalUser.AddRef();
|
|
_members[member.Id] = member;
|
|
DownloadedMemberCount++;
|
|
}
|
|
return member;
|
|
}
|
|
internal SocketGuildUser AddOrUpdateUser(PresenceModel model)
|
|
{
|
|
if (_members.TryGetValue(model.User.Id, out SocketGuildUser member))
|
|
member.Update(Discord.State, model, false);
|
|
else
|
|
{
|
|
member = SocketGuildUser.Create(this, Discord.State, model);
|
|
member.GlobalUser.AddRef();
|
|
_members[member.Id] = member;
|
|
DownloadedMemberCount++;
|
|
}
|
|
return member;
|
|
}
|
|
internal SocketGuildUser RemoveUser(ulong id)
|
|
{
|
|
if (_members.TryRemove(id, out SocketGuildUser member))
|
|
{
|
|
DownloadedMemberCount--;
|
|
member.GlobalUser.RemoveRef(Discord);
|
|
return member;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
public async Task DownloadUsersAsync()
|
|
{
|
|
await Discord.DownloadUsersAsync(new[] { this }).ConfigureAwait(false);
|
|
}
|
|
internal void CompleteDownloadUsers()
|
|
{
|
|
_downloaderPromise.TrySetResultAsync(true);
|
|
}
|
|
|
|
//Voice States
|
|
internal async Task<SocketVoiceState> AddOrUpdateVoiceStateAsync(ClientState state, VoiceStateModel model)
|
|
{
|
|
var voiceChannel = state.GetChannel(model.ChannelId.Value) as SocketVoiceChannel;
|
|
var before = GetVoiceState(model.UserId) ?? SocketVoiceState.Default;
|
|
var after = SocketVoiceState.Create(voiceChannel, model);
|
|
_voiceStates[model.UserId] = after;
|
|
|
|
if (_audioClient != null && before.VoiceChannel?.Id != after.VoiceChannel?.Id)
|
|
{
|
|
if (model.UserId == CurrentUser.Id)
|
|
{
|
|
if (after.VoiceChannel != null && _audioClient.ChannelId != after.VoiceChannel?.Id)
|
|
{
|
|
_audioClient.ChannelId = after.VoiceChannel.Id;
|
|
await RepopulateAudioStreamsAsync().ConfigureAwait(false);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
await _audioClient.RemoveInputStreamAsync(model.UserId).ConfigureAwait(false); //User changed channels, end their stream
|
|
if (CurrentUser.VoiceChannel != null && after.VoiceChannel?.Id == CurrentUser.VoiceChannel?.Id)
|
|
await _audioClient.CreateInputStreamAsync(model.UserId).ConfigureAwait(false);
|
|
}
|
|
}
|
|
|
|
return after;
|
|
}
|
|
internal SocketVoiceState? GetVoiceState(ulong id)
|
|
{
|
|
if (_voiceStates.TryGetValue(id, out SocketVoiceState voiceState))
|
|
return voiceState;
|
|
return null;
|
|
}
|
|
internal async Task<SocketVoiceState?> RemoveVoiceStateAsync(ulong id)
|
|
{
|
|
if (_voiceStates.TryRemove(id, out SocketVoiceState voiceState))
|
|
{
|
|
if (_audioClient != null)
|
|
await _audioClient.RemoveInputStreamAsync(id).ConfigureAwait(false); //User changed channels, end their stream
|
|
return voiceState;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
//Audio
|
|
internal AudioInStream GetAudioStream(ulong userId)
|
|
{
|
|
return _audioClient?.GetInputStream(userId);
|
|
}
|
|
internal async Task<IAudioClient> ConnectAudioAsync(ulong channelId, bool selfDeaf, bool selfMute, Action<IAudioClient> configAction)
|
|
{
|
|
selfDeaf = false;
|
|
selfMute = false;
|
|
|
|
TaskCompletionSource<AudioClient> promise;
|
|
|
|
await _audioLock.WaitAsync().ConfigureAwait(false);
|
|
try
|
|
{
|
|
await DisconnectAudioInternalAsync().ConfigureAwait(false);
|
|
promise = new TaskCompletionSource<AudioClient>();
|
|
_audioConnectPromise = promise;
|
|
|
|
if (_audioClient == null)
|
|
{
|
|
var audioClient = new AudioClient(this, Discord.GetAudioId(), channelId);
|
|
audioClient.Disconnected += async ex =>
|
|
{
|
|
if (!promise.Task.IsCompleted)
|
|
{
|
|
try { audioClient.Dispose(); } catch { }
|
|
_audioClient = null;
|
|
if (ex != null)
|
|
await promise.TrySetExceptionAsync(ex);
|
|
else
|
|
await promise.TrySetCanceledAsync();
|
|
return;
|
|
}
|
|
};
|
|
audioClient.Connected += () =>
|
|
{
|
|
var _ = promise.TrySetResultAsync(_audioClient);
|
|
return Task.Delay(0);
|
|
};
|
|
configAction?.Invoke(audioClient);
|
|
_audioClient = audioClient;
|
|
}
|
|
|
|
await Discord.ApiClient.SendVoiceStateUpdateAsync(Id, channelId, selfDeaf, selfMute).ConfigureAwait(false);
|
|
}
|
|
catch (Exception)
|
|
{
|
|
await DisconnectAudioInternalAsync().ConfigureAwait(false);
|
|
throw;
|
|
}
|
|
finally
|
|
{
|
|
_audioLock.Release();
|
|
}
|
|
|
|
try
|
|
{
|
|
var timeoutTask = Task.Delay(15000);
|
|
if (await Task.WhenAny(promise.Task, timeoutTask) == timeoutTask)
|
|
throw new TimeoutException();
|
|
return await promise.Task.ConfigureAwait(false);
|
|
}
|
|
catch (Exception)
|
|
{
|
|
await DisconnectAudioAsync().ConfigureAwait(false);
|
|
throw;
|
|
}
|
|
}
|
|
|
|
internal async Task DisconnectAudioAsync()
|
|
{
|
|
await _audioLock.WaitAsync().ConfigureAwait(false);
|
|
try
|
|
{
|
|
await DisconnectAudioInternalAsync().ConfigureAwait(false);
|
|
}
|
|
finally
|
|
{
|
|
_audioLock.Release();
|
|
}
|
|
}
|
|
private async Task DisconnectAudioInternalAsync()
|
|
{
|
|
_audioConnectPromise?.TrySetCanceledAsync(); //Cancel any previous audio connection
|
|
_audioConnectPromise = null;
|
|
if (_audioClient != null)
|
|
await _audioClient.StopAsync().ConfigureAwait(false);
|
|
await Discord.ApiClient.SendVoiceStateUpdateAsync(Id, null, false, false).ConfigureAwait(false);
|
|
_audioClient = null;
|
|
}
|
|
internal async Task FinishConnectAudio(string url, string token)
|
|
{
|
|
//TODO: Mem Leak: Disconnected/Connected handlers arent cleaned up
|
|
var voiceState = GetVoiceState(Discord.CurrentUser.Id).Value;
|
|
|
|
await _audioLock.WaitAsync().ConfigureAwait(false);
|
|
try
|
|
{
|
|
await RepopulateAudioStreamsAsync().ConfigureAwait(false);
|
|
await _audioClient.StartAsync(url, Discord.CurrentUser.Id, voiceState.VoiceSessionId, token).ConfigureAwait(false);
|
|
}
|
|
catch (OperationCanceledException)
|
|
{
|
|
await DisconnectAudioInternalAsync().ConfigureAwait(false);
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
await _audioConnectPromise.SetExceptionAsync(e).ConfigureAwait(false);
|
|
await DisconnectAudioInternalAsync().ConfigureAwait(false);
|
|
}
|
|
finally
|
|
{
|
|
_audioLock.Release();
|
|
}
|
|
}
|
|
|
|
internal async Task RepopulateAudioStreamsAsync()
|
|
{
|
|
await _audioClient.ClearInputStreamsAsync().ConfigureAwait(false); //We changed channels, end all current streams
|
|
if (CurrentUser.VoiceChannel != null)
|
|
{
|
|
foreach (var pair in _voiceStates)
|
|
{
|
|
if (pair.Value.VoiceChannel?.Id == CurrentUser.VoiceChannel?.Id && pair.Key != CurrentUser.Id)
|
|
await _audioClient.CreateInputStreamAsync(pair.Key).ConfigureAwait(false);
|
|
}
|
|
}
|
|
}
|
|
|
|
public override string ToString() => Name;
|
|
private string DebuggerDisplay => $"{Name} ({Id})";
|
|
internal SocketGuild Clone() => MemberwiseClone() as SocketGuild;
|
|
|
|
//IGuild
|
|
ulong? IGuild.AFKChannelId => AFKChannelId;
|
|
IAudioClient IGuild.AudioClient => null;
|
|
bool IGuild.Available => true;
|
|
ulong IGuild.DefaultChannelId => Id;
|
|
ulong? IGuild.EmbedChannelId => EmbedChannelId;
|
|
IRole IGuild.EveryoneRole => EveryoneRole;
|
|
IReadOnlyCollection<IRole> IGuild.Roles => Roles;
|
|
|
|
async Task<IReadOnlyCollection<IBan>> IGuild.GetBansAsync(RequestOptions options)
|
|
=> await GetBansAsync(options).ConfigureAwait(false);
|
|
|
|
Task<IReadOnlyCollection<IGuildChannel>> IGuild.GetChannelsAsync(CacheMode mode, RequestOptions options)
|
|
=> Task.FromResult<IReadOnlyCollection<IGuildChannel>>(Channels);
|
|
Task<IGuildChannel> IGuild.GetChannelAsync(ulong id, CacheMode mode, RequestOptions options)
|
|
=> Task.FromResult<IGuildChannel>(GetChannel(id));
|
|
Task<IReadOnlyCollection<ITextChannel>> IGuild.GetTextChannelsAsync(CacheMode mode, RequestOptions options)
|
|
=> Task.FromResult<IReadOnlyCollection<ITextChannel>>(TextChannels);
|
|
Task<ITextChannel> IGuild.GetTextChannelAsync(ulong id, CacheMode mode, RequestOptions options)
|
|
=> Task.FromResult<ITextChannel>(GetTextChannel(id));
|
|
Task<IReadOnlyCollection<IVoiceChannel>> IGuild.GetVoiceChannelsAsync(CacheMode mode, RequestOptions options)
|
|
=> Task.FromResult<IReadOnlyCollection<IVoiceChannel>>(VoiceChannels);
|
|
Task<IVoiceChannel> IGuild.GetVoiceChannelAsync(ulong id, CacheMode mode, RequestOptions options)
|
|
=> Task.FromResult<IVoiceChannel>(GetVoiceChannel(id));
|
|
Task<IVoiceChannel> IGuild.GetAFKChannelAsync(CacheMode mode, RequestOptions options)
|
|
=> Task.FromResult<IVoiceChannel>(AFKChannel);
|
|
Task<ITextChannel> IGuild.GetDefaultChannelAsync(CacheMode mode, RequestOptions options)
|
|
=> Task.FromResult<ITextChannel>(DefaultChannel);
|
|
Task<IGuildChannel> IGuild.GetEmbedChannelAsync(CacheMode mode, RequestOptions options)
|
|
=> Task.FromResult<IGuildChannel>(EmbedChannel);
|
|
async Task<ITextChannel> IGuild.CreateTextChannelAsync(string name, RequestOptions options)
|
|
=> await CreateTextChannelAsync(name, options).ConfigureAwait(false);
|
|
async Task<IVoiceChannel> IGuild.CreateVoiceChannelAsync(string name, RequestOptions options)
|
|
=> await CreateVoiceChannelAsync(name, options).ConfigureAwait(false);
|
|
|
|
async Task<IReadOnlyCollection<IGuildIntegration>> IGuild.GetIntegrationsAsync(RequestOptions options)
|
|
=> await GetIntegrationsAsync(options).ConfigureAwait(false);
|
|
async Task<IGuildIntegration> IGuild.CreateIntegrationAsync(ulong id, string type, RequestOptions options)
|
|
=> await CreateIntegrationAsync(id, type, options).ConfigureAwait(false);
|
|
|
|
async Task<IReadOnlyCollection<IInviteMetadata>> IGuild.GetInvitesAsync(RequestOptions options)
|
|
=> await GetInvitesAsync(options).ConfigureAwait(false);
|
|
|
|
IRole IGuild.GetRole(ulong id)
|
|
=> GetRole(id);
|
|
async Task<IRole> IGuild.CreateRoleAsync(string name, GuildPermissions? permissions, Color? color, bool isHoisted, RequestOptions options)
|
|
=> await CreateRoleAsync(name, permissions, color, isHoisted, options).ConfigureAwait(false);
|
|
|
|
Task<IReadOnlyCollection<IGuildUser>> IGuild.GetUsersAsync(CacheMode mode, RequestOptions options)
|
|
=> Task.FromResult<IReadOnlyCollection<IGuildUser>>(Users);
|
|
Task<IGuildUser> IGuild.GetUserAsync(ulong id, CacheMode mode, RequestOptions options)
|
|
=> Task.FromResult<IGuildUser>(GetUser(id));
|
|
Task<IGuildUser> IGuild.GetCurrentUserAsync(CacheMode mode, RequestOptions options)
|
|
=> Task.FromResult<IGuildUser>(CurrentUser);
|
|
Task<IGuildUser> IGuild.GetOwnerAsync(CacheMode mode, RequestOptions options)
|
|
=> Task.FromResult<IGuildUser>(Owner);
|
|
Task IGuild.DownloadUsersAsync() { throw new NotSupportedException(); }
|
|
}
|
|
}
|