71 lines
3.1 KiB
C#
71 lines
3.1 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(SocketChannel channel, DiscordSocketClient discord, MessageCache messages,
|
|
ulong? fromMessageId, Direction dir, int limit, CacheMode mode)
|
|
{
|
|
IReadOnlyCollection<SocketMessage> cachedMessages;
|
|
IAsyncEnumerable<IReadOnlyCollection<IMessage>> result;
|
|
|
|
if (messages != null) //Cache enabled
|
|
cachedMessages = messages.GetMany(fromMessageId, dir, limit);
|
|
else
|
|
cachedMessages = ImmutableArray.Create<SocketMessage>();
|
|
|
|
result = ImmutableArray.Create(cachedMessages).ToAsyncEnumerable<IReadOnlyCollection<IMessage>>();
|
|
limit -= cachedMessages.Count;
|
|
if (mode == CacheMode.CacheOnly || limit <= 0)
|
|
return result;
|
|
|
|
if (dir == Direction.Before)
|
|
fromMessageId = cachedMessages.Min(x => x.Id);
|
|
else
|
|
fromMessageId = cachedMessages.Max(x => x.Id);
|
|
var downloadedMessages = ChannelHelper.GetMessagesAsync(channel, discord, fromMessageId, dir, limit);
|
|
return result.Concat(downloadedMessages);
|
|
}
|
|
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)
|
|
{
|
|
//C#7 Candidate for pattern matching
|
|
if (channel is SocketDMChannel)
|
|
(channel as SocketDMChannel).AddMessage(msg);
|
|
else if (channel is SocketGroupChannel)
|
|
(channel as SocketGroupChannel).AddMessage(msg);
|
|
else if (channel is SocketTextChannel)
|
|
(channel as SocketTextChannel).AddMessage(msg);
|
|
else
|
|
throw new NotSupportedException("Unexpected ISocketMessageChannel type");
|
|
}
|
|
public static SocketMessage RemoveMessage(ISocketMessageChannel channel, DiscordSocketClient discord,
|
|
ulong id)
|
|
{
|
|
//C#7 Candidate for pattern matching
|
|
if (channel is SocketDMChannel)
|
|
return (channel as SocketDMChannel).RemoveMessage(id);
|
|
else if (channel is SocketGroupChannel)
|
|
return (channel as SocketGroupChannel).RemoveMessage(id);
|
|
else if (channel is SocketTextChannel)
|
|
return (channel as SocketTextChannel).RemoveMessage(id);
|
|
else
|
|
throw new NotSupportedException("Unexpected ISocketMessageChannel type");
|
|
}
|
|
}
|
|
}
|