/// <summary>
/// Removes a binding of a specific badword.
/// </summary>
/// <param name="global">global == true means across the entire guild, else restricted to the channel</param>
/// <param name="context">context of the command</param>
/// <param name="badWord">word to be unbanned</param>
/// <returns></returns>
public async Task <BindingStatus> RemoveBinding(bool global, ICommandContext context, string badWord)
{
using (var db = new BadWordContext(dbOptions))
{
if (global)
{
var badWordDbEntry = new ServerBadWord
{
Entry = badWord,
ServerId = context.Guild.Id,
};
if (!await db.BadWordServerBindings.AnyAsync(b => b.ServerId == context.Guild.Id && b.BadWords.Any(e => e == badWordDbEntry)))
{
return(BindingStatus.NotExisting);
}
foreach (var server in db.BadWordServerBindings.Include(b => b.BadWords).Where(b => b.ServerId == context.Guild.Id))
{
db.BadWordServerBindings.Update(server);
server.BadWords.Remove(badWordDbEntry);
}
await db.SaveChangesAsync();
return(BindingStatus.Removed);
}
else
{
var badWordDbEntry = new ChannelBadWord
{
Entry = badWord,
ServerId = context.Guild.Id,
ChannelId = context.Channel.Id,
};
if (!await db.BadWordChannelBindings.AnyAsync(b => b.ChannelId == context.Guild.Id && b.BadWords.Any(e => e == badWordDbEntry)))
{
return(BindingStatus.NotExisting);
}
foreach (var channel in db.BadWordChannelBindings.Include(b => b.BadWords).Where(b => b.ChannelId == context.Channel.Id))
{
db.BadWordChannelBindings.Update(channel);
channel.BadWords.Remove(badWordDbEntry);
}
await db.SaveChangesAsync();
return(BindingStatus.Removed);
}
}
}
/// <summary>
/// Lists the bindings of every channel of the guild and the guild specified in context.
/// </summary>
/// <param name="context">context of the command</param>
/// <returns>A dictionary with key of the binding and a indicator of the scope of the binding, and value being the banned words for that binding.</returns>
public IDictionary <(object entity, bool isGuild), IEnumerable <string> > ListBindings(ICommandContext context)
{
using (var db = new BadWordContext(dbOptions))
{
var bindingList = new Dictionary <(object entity, bool isGuild), IEnumerable <string> >();
foreach (var server in db.BadWordServerBindings.Include(b => b.BadWords).Where(b => b.ServerId == context.Guild.Id))
{
bindingList.Add((server, true), server.BadWords.Select(b => b.Entry).ToList());
}
foreach (var channel in db.BadWordChannelBindings.Include(b => b.BadWords).Where(b => b.ServerId == context.Guild.Id))
{
bindingList.Add((channel, false), channel.BadWords.Select(b => b.Entry).ToList());
}
return(bindingList);
}
}
private async Task Handler(SocketMessage arg)
{
if (!(arg.Channel is ITextChannel guildChannel))
{
return;
}
if (arg.Content.StartsWith($"{(string)Shinoa.Config["global"]["command_prefix"]}badword"))
{
return;
}
if (arg.Author.IsBot)
{
return;
}
using (var db = new BadWordContext(dbOptions))
{
var toBeDeleted = false;
var channelBindings = db.BadWordChannelBindings.Where(b => b.ChannelId == arg.Channel.Id);
{
var serverBindings = db.BadWordServerBindings.Where(b => b.ServerId == guildChannel.GuildId);
if (serverBindings.Any())
{
var badwords = serverBindings.SelectMany(b => b.BadWords.Select(w => w.Entry));
if (badwords.Any(w => arg.Content.ToLowerInvariant().RemovePunctuation().Contains(w.ToLowerInvariant())))
{
toBeDeleted = true;
}
}
}
if (channelBindings.Any())
{
var badwords = channelBindings.SelectMany(b => b.BadWords.Select(w => w.Entry));
if (badwords.Any(w => arg.Content.ToLowerInvariant().RemovePunctuation().Contains(w.ToLowerInvariant())))
{
toBeDeleted = true;
}
}
if (toBeDeleted)
{
await arg.DeleteAsync();
}
}
}
private async Task <bool> RemoveGuildBinding(IGuild guild)
{
using (var db = new BadWordContext(dbOptions))
{
var bindings = await db.BadWordServerBindings.Where(b => b.ServerId == guild.Id).ToListAsync();
if (!bindings.Any())
{
return(false);
}
foreach (var guildBinding in bindings)
{
db.Remove(guildBinding);
}
await db.SaveChangesAsync();
return(true);
}
}
private async Task <bool> RemoveChannelBinding(ITextChannel channel)
{
using (var db = new BadWordContext(dbOptions))
{
var bindings = await db.BadWordChannelBindings.Where(b => b.ChannelId == channel.Id).ToListAsync();
if (!bindings.Any())
{
return(false);
}
foreach (var channelBinding in bindings)
{
db.Remove(channelBinding);
}
await db.SaveChangesAsync();
return(true);
}
}
/// <summary>
/// Adds a binding of a specific badword.
/// </summary>
/// <param name="global">global == true means across the entire guild, else restricted to the channel</param>
/// <param name="context">context of the command</param>
/// <param name="badWord">word to be banned</param>
/// <returns></returns>
public async Task <BindingStatus> AddBinding(bool global, ICommandContext context, string badWord)
{
using (var db = new BadWordContext(dbOptions))
{
if (global)
{
var badWordDbEntry = new ServerBadWord
{
Entry = badWord,
ServerId = context.Guild.Id,
};
if (await db.BadWordServerBindings.AnyAsync(b => b.ServerId == context.Guild.Id && b.BadWords.Any(e => e == badWordDbEntry)))
{
return(BindingStatus.AlreadyExists);
}
if (!await db.BadWordServerBindings.AnyAsync(b => b.ServerId == context.Guild.Id))
{
var serverDbEntry = new BadWordServerBinding
{
ServerId = context.Guild.Id,
BadWords = new List <ServerBadWord>
{
badWordDbEntry,
},
};
await db.BadWordServerBindings.AddAsync(serverDbEntry);
}
else
{
foreach (var server in db.BadWordServerBindings.Include(b => b.BadWords).Where(b => b.ServerId == context.Guild.Id))
{
db.BadWordServerBindings.Update(server);
server.BadWords.Add(badWordDbEntry);
}
}
await db.SaveChangesAsync();
return(BindingStatus.Added);
}
else
{
var badWordDbEntry = new ChannelBadWord
{
Entry = badWord,
ChannelId = context.Channel.Id,
ServerId = context.Guild.Id,
};
if (await db.BadWordChannelBindings.AnyAsync(b => b.ChannelId == context.Guild.Id && b.BadWords.Any(e => e == badWordDbEntry)))
{
return(BindingStatus.AlreadyExists);
}
if (!await db.BadWordChannelBindings.AnyAsync(b => b.ChannelId == context.Guild.Id))
{
var serverDbEntry = new BadWordChannelBinding
{
ChannelId = context.Guild.Id,
ServerId = context.Guild.Id,
BadWords = new List <ChannelBadWord>
{
badWordDbEntry,
},
};
await db.BadWordChannelBindings.AddAsync(serverDbEntry);
}
else
{
foreach (var channel in db.BadWordChannelBindings.Include(b => b.BadWords).Where(b => b.ChannelId == context.Channel.Id))
{
db.BadWordChannelBindings.Update(channel);
channel.BadWords.Add(badWordDbEntry);
}
}
await db.SaveChangesAsync();
return(BindingStatus.Added);
}
}
}
