2017-11-17 20:13:47 +00:00
|
|
|
|
using Discord;
|
|
|
|
|
using Discord.WebSocket;
|
|
|
|
|
using Newtonsoft.Json.Linq;
|
|
|
|
|
using System.Text;
|
|
|
|
|
using System.Threading.Tasks;
|
|
|
|
|
|
|
|
|
|
namespace Noikoio.RegexBot.Module.DMLogger
|
|
|
|
|
{
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// Listens for and logs direct messages sent to the bot.
|
|
|
|
|
/// The function of this module should be transparent to the user, and thus no configuration is needed.
|
|
|
|
|
/// </summary>
|
|
|
|
|
class DMLogger : BotModule
|
|
|
|
|
{
|
|
|
|
|
public override string Name => nameof(DMLogger);
|
|
|
|
|
|
|
|
|
|
public DMLogger(DiscordSocketClient client) : base(client)
|
|
|
|
|
{
|
|
|
|
|
client.MessageReceived += Client_MessageReceived;
|
|
|
|
|
client.MessageUpdated += Client_MessageUpdated;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private async Task Client_MessageReceived(SocketMessage arg)
|
|
|
|
|
{
|
|
|
|
|
if (!(arg.Channel is IDMChannel)) return;
|
2017-11-18 01:45:00 +00:00
|
|
|
|
if (arg.Author.IsBot) return;
|
2017-11-17 20:13:47 +00:00
|
|
|
|
|
|
|
|
|
await ProcessMessage(arg, false);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private async Task Client_MessageUpdated(Cacheable<IMessage, ulong> arg1, SocketMessage arg2, ISocketMessageChannel arg3)
|
|
|
|
|
{
|
|
|
|
|
if (!(arg2.Channel is IDMChannel)) return;
|
2017-11-18 01:45:00 +00:00
|
|
|
|
if (arg2.Author.IsBot) return;
|
2017-11-17 20:13:47 +00:00
|
|
|
|
|
|
|
|
|
await ProcessMessage(arg2, true);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public override Task<object> ProcessConfiguration(JToken configSection) => Task.FromResult<object>(null);
|
|
|
|
|
|
|
|
|
|
private async Task ProcessMessage(SocketMessage arg, bool edited)
|
|
|
|
|
{
|
|
|
|
|
var result = new StringBuilder();
|
|
|
|
|
result.Append(arg.Author.ToString() + (edited ? "(edit) " : "") + ": ");
|
|
|
|
|
if (!string.IsNullOrWhiteSpace(arg.Content))
|
|
|
|
|
{
|
|
|
|
|
if (arg.Content.Contains("\n")) result.AppendLine(); // If multi-line, show sender on separate line
|
|
|
|
|
result.AppendLine(arg.Content);
|
|
|
|
|
}
|
|
|
|
|
foreach (var i in arg.Attachments) result.AppendLine($"[Attachment: {i.Url}]");
|
|
|
|
|
|
|
|
|
|
await Log(result.ToString().TrimEnd(new char[] { ' ', '\r', '\n' }));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|