50 lines
1.6 KiB
C#
50 lines
1.6 KiB
C#
using System;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using Discord;
|
|
using Discord.Commands;
|
|
using Discord.WebSocket;
|
|
using NadekoBot.Common.Collections;
|
|
using NadekoBot.Core.Services;
|
|
using NLog;
|
|
|
|
namespace NadekoBot.Modules.Administration.Services
|
|
{
|
|
public class AdministrationService : INService
|
|
{
|
|
public readonly ConcurrentHashSet<ulong> DeleteMessagesOnCommand;
|
|
private readonly Logger _log;
|
|
private readonly NadekoBot _bot;
|
|
|
|
public AdministrationService(NadekoBot bot, CommandHandler cmdHandler)
|
|
{
|
|
_log = LogManager.GetCurrentClassLogger();
|
|
_bot = bot;
|
|
|
|
DeleteMessagesOnCommand = new ConcurrentHashSet<ulong>(bot.AllGuildConfigs.Where(g => g.DeleteMessageOnCommand).Select(g => g.GuildId));
|
|
cmdHandler.CommandExecuted += DelMsgOnCmd_Handler;
|
|
}
|
|
|
|
private Task DelMsgOnCmd_Handler(IUserMessage msg, CommandInfo cmd)
|
|
{
|
|
var _ = Task.Run(async () =>
|
|
{
|
|
try
|
|
{
|
|
var channel = msg.Channel as SocketTextChannel;
|
|
if (channel == null)
|
|
return;
|
|
if (DeleteMessagesOnCommand.Contains(channel.Guild.Id) && cmd.Name != "prune" && cmd.Name != "pick")
|
|
await msg.DeleteAsync().ConfigureAwait(false);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_log.Warn("Delmsgoncmd errored...");
|
|
_log.Warn(ex);
|
|
}
|
|
});
|
|
return Task.CompletedTask;
|
|
}
|
|
}
|
|
}
|