68 lines
2.5 KiB
C#
68 lines
2.5 KiB
C#
using System;
|
|
using System.Collections.Concurrent;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using Discord;
|
|
using Discord.WebSocket;
|
|
using NadekoBot.Common.ModuleBehaviors;
|
|
using NadekoBot.Extensions;
|
|
using NadekoBot.Modules.Administration.Common;
|
|
using NadekoBot.Services;
|
|
using NadekoBot.Services.Database.Models;
|
|
using NLog;
|
|
|
|
namespace NadekoBot.Modules.Administration.Services
|
|
{
|
|
public class SlowmodeService : IEarlyBlocker, INService
|
|
{
|
|
public ConcurrentDictionary<ulong, Ratelimiter> RatelimitingChannels = new ConcurrentDictionary<ulong, Ratelimiter>();
|
|
public ConcurrentDictionary<ulong, HashSet<ulong>> IgnoredRoles = new ConcurrentDictionary<ulong, HashSet<ulong>>();
|
|
public ConcurrentDictionary<ulong, HashSet<ulong>> IgnoredUsers = new ConcurrentDictionary<ulong, HashSet<ulong>>();
|
|
|
|
private readonly Logger _log;
|
|
private readonly DiscordSocketClient _client;
|
|
|
|
public SlowmodeService(DiscordSocketClient client, IEnumerable<GuildConfig> gcs)
|
|
{
|
|
_log = LogManager.GetCurrentClassLogger();
|
|
_client = client;
|
|
|
|
IgnoredRoles = new ConcurrentDictionary<ulong, HashSet<ulong>>(
|
|
gcs.ToDictionary(x => x.GuildId,
|
|
x => new HashSet<ulong>(x.SlowmodeIgnoredRoles.Select(y => y.RoleId))));
|
|
|
|
IgnoredUsers = new ConcurrentDictionary<ulong, HashSet<ulong>>(
|
|
gcs.ToDictionary(x => x.GuildId,
|
|
x => new HashSet<ulong>(x.SlowmodeIgnoredUsers.Select(y => y.UserId))));
|
|
}
|
|
|
|
public async Task<bool> TryBlockEarly(IGuild guild, IUserMessage usrMsg)
|
|
{
|
|
if (guild == null)
|
|
return false;
|
|
try
|
|
{
|
|
var channel = usrMsg?.Channel as SocketTextChannel;
|
|
|
|
if (channel == null || usrMsg == null || usrMsg.IsAuthor(_client))
|
|
return false;
|
|
if (!RatelimitingChannels.TryGetValue(channel.Id, out Ratelimiter limiter))
|
|
return false;
|
|
|
|
if (limiter.CheckUserRatelimit(usrMsg.Author.Id, channel.Guild.Id, usrMsg.Author as SocketGuildUser))
|
|
{
|
|
await usrMsg.DeleteAsync();
|
|
return true;
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_log.Warn(ex);
|
|
|
|
}
|
|
return false;
|
|
}
|
|
}
|
|
}
|