85 lines
3.7 KiB
C#
85 lines
3.7 KiB
C#
using Discord.Rest;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Collections.Immutable;
|
|
using System.Linq;
|
|
|
|
namespace Discord.WebSocket
|
|
{
|
|
internal static class SocketChannelHelper
|
|
{
|
|
public static IAsyncEnumerable<IReadOnlyCollection<IMessage>> GetMessagesAsync(ISocketMessageChannel channel, DiscordSocketClient discord, MessageCache messages,
|
|
ulong? fromMessageId, Direction dir, int limit, CacheMode mode, RequestOptions options)
|
|
{
|
|
if (dir == Direction.Around)
|
|
throw new NotImplementedException(); //TODO: Impl
|
|
|
|
IReadOnlyCollection<SocketMessage> cachedMessages = null;
|
|
IAsyncEnumerable<IReadOnlyCollection<IMessage>> result = null;
|
|
|
|
if (dir == Direction.After && fromMessageId == null)
|
|
return AsyncEnumerable.Empty<IReadOnlyCollection<IMessage>>();
|
|
|
|
if (dir == Direction.Before || mode == CacheMode.CacheOnly)
|
|
{
|
|
if (messages != null) //Cache enabled
|
|
cachedMessages = messages.GetMany(fromMessageId, dir, limit);
|
|
else
|
|
cachedMessages = ImmutableArray.Create<SocketMessage>();
|
|
result = ImmutableArray.Create(cachedMessages).ToAsyncEnumerable<IReadOnlyCollection<IMessage>>();
|
|
}
|
|
|
|
if (dir == Direction.Before)
|
|
{
|
|
limit -= cachedMessages.Count;
|
|
if (mode == CacheMode.CacheOnly || limit <= 0)
|
|
return result;
|
|
|
|
//Download remaining messages
|
|
ulong? minId = cachedMessages.Count > 0 ? cachedMessages.Min(x => x.Id) : fromMessageId;
|
|
var downloadedMessages = ChannelHelper.GetMessagesAsync(channel, discord, minId, dir, limit, options);
|
|
return result.Concat(downloadedMessages);
|
|
}
|
|
else
|
|
{
|
|
if (mode == CacheMode.CacheOnly)
|
|
return result;
|
|
|
|
//Dont use cache in this case
|
|
return ChannelHelper.GetMessagesAsync(channel, discord, fromMessageId, dir, limit, options);
|
|
}
|
|
}
|
|
public static IReadOnlyCollection<SocketMessage> GetCachedMessages(SocketChannel channel, DiscordSocketClient discord, MessageCache messages,
|
|
ulong? fromMessageId, Direction dir, int limit)
|
|
{
|
|
if (messages != null) //Cache enabled
|
|
return messages.GetMany(fromMessageId, dir, limit);
|
|
else
|
|
return ImmutableArray.Create<SocketMessage>();
|
|
}
|
|
|
|
public static void AddMessage(ISocketMessageChannel channel, DiscordSocketClient discord,
|
|
SocketMessage msg)
|
|
{
|
|
switch (channel)
|
|
{
|
|
case SocketDMChannel dmChannel: dmChannel.AddMessage(msg); break;
|
|
case SocketGroupChannel groupChannel: groupChannel.AddMessage(msg); break;
|
|
case SocketTextChannel textChannel: textChannel.AddMessage(msg); break;
|
|
default: throw new NotSupportedException("Unexpected ISocketMessageChannel type");
|
|
}
|
|
}
|
|
public static SocketMessage RemoveMessage(ISocketMessageChannel channel, DiscordSocketClient discord,
|
|
ulong id)
|
|
{
|
|
switch (channel)
|
|
{
|
|
case SocketDMChannel dmChannel: return dmChannel.RemoveMessage(id);
|
|
case SocketGroupChannel groupChannel: return groupChannel.RemoveMessage(id);
|
|
case SocketTextChannel textChannel: return textChannel.RemoveMessage(id);
|
|
default: throw new NotSupportedException("Unexpected ISocketMessageChannel type");
|
|
}
|
|
}
|
|
}
|
|
}
|