You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

RpcDMChannel.cs 6.8 kB

8 years ago
8 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126
  1. using Discord.Rest;
  2. using System;
  3. using System.Collections.Generic;
  4. using System.Collections.Immutable;
  5. using System.IO;
  6. using System.Linq;
  7. using System.Threading.Tasks;
  8. using Model = Discord.API.Rpc.Channel;
  9. namespace Discord.Rpc
  10. {
  11. [DebuggerDisplay(@"{DebuggerDisplay,nq}")]
  12. public class RpcDMChannel : RpcChannel, IRpcMessageChannel, IRpcPrivateChannel, IDMChannel
  13. {
  14. public IReadOnlyCollection<RpcMessage> CachedMessages { get; private set; }
  15. internal RpcDMChannel(DiscordRpcClient discord, ulong id)
  16. : base(discord, id)
  17. {
  18. }
  19. internal static new RpcDMChannel Create(DiscordRpcClient discord, Model model)
  20. {
  21. var entity = new RpcDMChannel(discord, model.Id);
  22. entity.Update(model);
  23. return entity;
  24. }
  25. internal override void Update(Model model)
  26. {
  27. base.Update(model);
  28. CachedMessages = model.Messages.Select(x => RpcMessage.Create(Discord, Id, x)).ToImmutableArray();
  29. }
  30. public Task CloseAsync(RequestOptions options = null)
  31. => ChannelHelper.DeleteAsync(this, Discord, options);
  32. //TODO: Use RPC cache
  33. public Task<RestMessage> GetMessageAsync(ulong id, RequestOptions options = null)
  34. => ChannelHelper.GetMessageAsync(this, Discord, id, options);
  35. public IAsyncEnumerable<IReadOnlyCollection<RestMessage>> GetMessagesAsync(int limit = DiscordConfig.MaxMessagesPerBatch, RequestOptions options = null)
  36. => ChannelHelper.GetMessagesAsync(this, Discord, null, Direction.Before, limit, options);
  37. public IAsyncEnumerable<IReadOnlyCollection<RestMessage>> GetMessagesAsync(ulong fromMessageId, Direction dir, int limit = DiscordConfig.MaxMessagesPerBatch, RequestOptions options = null)
  38. => ChannelHelper.GetMessagesAsync(this, Discord, fromMessageId, dir, limit, options);
  39. public IAsyncEnumerable<IReadOnlyCollection<RestMessage>> GetMessagesAsync(IMessage fromMessage, Direction dir, int limit = DiscordConfig.MaxMessagesPerBatch, RequestOptions options = null)
  40. => ChannelHelper.GetMessagesAsync(this, Discord, fromMessage.Id, dir, limit, options);
  41. public Task<IReadOnlyCollection<RestMessage>> GetPinnedMessagesAsync(RequestOptions options = null)
  42. => ChannelHelper.GetPinnedMessagesAsync(this, Discord, options);
  43. public Task<RestUserMessage> SendMessageAsync(string text, bool isTTS = false, Embed embed = null, RequestOptions options = null)
  44. => ChannelHelper.SendMessageAsync(this, Discord, text, isTTS, embed, options);
  45. #if FILESYSTEM
  46. public Task<RestUserMessage> SendFileAsync(string filePath, string text, bool isTTS = false, RequestOptions options = null)
  47. => ChannelHelper.SendFileAsync(this, Discord, filePath, text, isTTS, options);
  48. #endif
  49. public Task<RestUserMessage> SendFileAsync(Stream stream, string filename, string text, bool isTTS = false, RequestOptions options = null)
  50. => ChannelHelper.SendFileAsync(this, Discord, stream, filename, text, isTTS, options);
  51. public Task TriggerTypingAsync(RequestOptions options = null)
  52. => ChannelHelper.TriggerTypingAsync(this, Discord, options);
  53. public IDisposable EnterTypingState(RequestOptions options = null)
  54. => ChannelHelper.EnterTypingState(this, Discord, options);
  55. public override string ToString() => Id.ToString();
  56. private string DebuggerDisplay => $"({Id}, DM)";
  57. //IDMChannel
  58. IUser IDMChannel.Recipient { get { throw new NotSupportedException(); } }
  59. //IPrivateChannel
  60. IReadOnlyCollection<IUser> IPrivateChannel.Recipients { get { throw new NotSupportedException(); } }
  61. //IMessageChannel
  62. async Task<IMessage> IMessageChannel.GetMessageAsync(ulong id, CacheMode mode, RequestOptions options)
  63. {
  64. if (mode == CacheMode.AllowDownload)
  65. return await GetMessageAsync(id, options).ConfigureAwait(false);
  66. else
  67. return null;
  68. }
  69. IAsyncEnumerable<IReadOnlyCollection<IMessage>> IMessageChannel.GetMessagesAsync(int limit, CacheMode mode, RequestOptions options)
  70. {
  71. if (mode == CacheMode.AllowDownload)
  72. return GetMessagesAsync(limit, options);
  73. else
  74. return AsyncEnumerable.Empty<IReadOnlyCollection<IMessage>>();
  75. }
  76. IAsyncEnumerable<IReadOnlyCollection<IMessage>> IMessageChannel.GetMessagesAsync(ulong fromMessageId, Direction dir, int limit, CacheMode mode, RequestOptions options)
  77. {
  78. if (mode == CacheMode.AllowDownload)
  79. return GetMessagesAsync(fromMessageId, dir, limit, options);
  80. else
  81. return AsyncEnumerable.Empty<IReadOnlyCollection<IMessage>>();
  82. }
  83. IAsyncEnumerable<IReadOnlyCollection<IMessage>> IMessageChannel.GetMessagesAsync(IMessage fromMessage, Direction dir, int limit, CacheMode mode, RequestOptions options)
  84. {
  85. if (mode == CacheMode.AllowDownload)
  86. return GetMessagesAsync(fromMessage, dir, limit, options);
  87. else
  88. return AsyncEnumerable.Empty<IReadOnlyCollection<IMessage>>();
  89. }
  90. async Task<IReadOnlyCollection<IMessage>> IMessageChannel.GetPinnedMessagesAsync(RequestOptions options)
  91. => await GetPinnedMessagesAsync(options).ConfigureAwait(false);
  92. #if FILESYSTEM
  93. async Task<IUserMessage> IMessageChannel.SendFileAsync(string filePath, string text, bool isTTS, RequestOptions options)
  94. => await SendFileAsync(filePath, text, isTTS, options).ConfigureAwait(false);
  95. #endif
  96. async Task<IUserMessage> IMessageChannel.SendFileAsync(Stream stream, string filename, string text, bool isTTS, RequestOptions options)
  97. => await SendFileAsync(stream, filename, text, isTTS, options).ConfigureAwait(false);
  98. async Task<IUserMessage> IMessageChannel.SendMessageAsync(string text, bool isTTS, Embed embed, RequestOptions options)
  99. => await SendMessageAsync(text, isTTS, embed, options).ConfigureAwait(false);
  100. IDisposable IMessageChannel.EnterTypingState(RequestOptions options)
  101. => EnterTypingState(options);
  102. //IChannel
  103. string IChannel.Name { get { throw new NotSupportedException(); } }
  104. Task<IUser> IChannel.GetUserAsync(ulong id, CacheMode mode, RequestOptions options)
  105. {
  106. throw new NotSupportedException();
  107. }
  108. IAsyncEnumerable<IReadOnlyCollection<IUser>> IChannel.GetUsersAsync(CacheMode mode, RequestOptions options)
  109. {
  110. throw new NotSupportedException();
  111. }
  112. }
  113. }