Added 'Unban' command type
This commit is contained in:
parent
c5229a22b2
commit
dc7725c7b0
2 changed files with 109 additions and 1 deletions
|
@ -48,7 +48,8 @@ namespace Noikoio.RegexBot.Module.ModTools
|
|||
// Define all command types and their corresponding Types here
|
||||
{ "ban", typeof(Commands.Ban) },
|
||||
{ "kick", typeof(Commands.Kick) },
|
||||
{ "say", typeof(Commands.Say) }
|
||||
{ "say", typeof(Commands.Say) },
|
||||
{ "unban", typeof(Commands.Unban) }
|
||||
});
|
||||
|
||||
public static CommandBase CreateInstance(ModTools root, JProperty def)
|
||||
|
|
107
Module/ModTools/Commands/Unban.cs
Normal file
107
Module/ModTools/Commands/Unban.cs
Normal file
|
@ -0,0 +1,107 @@
|
|||
using Discord;
|
||||
using Discord.WebSocket;
|
||||
using Newtonsoft.Json.Linq;
|
||||
using System;
|
||||
using System.Linq;
|
||||
using System.Text.RegularExpressions;
|
||||
using System.Threading.Tasks;
|
||||
|
||||
namespace Noikoio.RegexBot.Module.ModTools.Commands
|
||||
{
|
||||
class Unban : CommandBase
|
||||
{
|
||||
// No configuration.
|
||||
// TODO bring in some options from BanKick. Particularly custom success msg.
|
||||
// TODO when ModLogs fully implemented, add a reason?
|
||||
public Unban(ModTools l, string label, JObject conf) : base(l, label, conf) { }
|
||||
|
||||
#region Strings
|
||||
const string FailPrefix = ":x: **Unable to unban:** ";
|
||||
const string Fail403 = "I do not have the required permissions to perform that action.";
|
||||
const string Fail404 = "The target user is no longer available.";
|
||||
const string FailDefault = "An unknown error occurred. Notify the bot operator.";
|
||||
const string TargetNotFound = ":x: **Unable to determine the target user.**";
|
||||
const string Success = ":white_check_mark: Unbanned user **{0}**.";
|
||||
#endregion
|
||||
|
||||
// Usage: (command) (user query)
|
||||
public override async Task Invoke(SocketGuild g, SocketMessage msg)
|
||||
{
|
||||
// TODO oh god there's so much boilerplate copypasted from BanKick make it stop
|
||||
|
||||
string[] line = msg.Content.Split(new char[] { ' ' }, 3, StringSplitOptions.RemoveEmptyEntries);
|
||||
string targetstr;
|
||||
if (line.Length < 2)
|
||||
{
|
||||
await SendUsageMessage(msg, null);
|
||||
return;
|
||||
}
|
||||
targetstr = line[1];
|
||||
|
||||
// Getting SocketGuildUser target
|
||||
SocketGuildUser targetobj = null;
|
||||
|
||||
// Extract snowflake value from mention (if a mention was given)
|
||||
Match m = UserMention.Match(targetstr);
|
||||
if (m.Success) targetstr = m.Groups["snowflake"].Value;
|
||||
|
||||
EntityCache.CacheUser qres;
|
||||
try
|
||||
{
|
||||
qres = (await EntityCache.EntityCache.QueryAsync(g.Id, targetstr)).FirstOrDefault();
|
||||
}
|
||||
catch (Npgsql.NpgsqlException ex)
|
||||
{
|
||||
await Log("A database error occurred during user lookup: " + ex.Message);
|
||||
await msg.Channel.SendMessageAsync(FailPrefix + FailDefault);
|
||||
return;
|
||||
}
|
||||
|
||||
if (qres == null)
|
||||
{
|
||||
await SendUsageMessage(msg, TargetNotFound);
|
||||
return;
|
||||
}
|
||||
|
||||
ulong targetuid = qres.UserId;
|
||||
targetobj = g.GetUser(targetuid);
|
||||
string targetdisp = targetobj?.ToString() ?? $"ID {targetuid}";
|
||||
|
||||
// Do the action
|
||||
try
|
||||
{
|
||||
await g.RemoveBanAsync(targetuid);
|
||||
await msg.Channel.SendMessageAsync(string.Format(Success, targetdisp));
|
||||
}
|
||||
catch (Discord.Net.HttpException ex)
|
||||
{
|
||||
if (ex.HttpCode == System.Net.HttpStatusCode.Forbidden)
|
||||
{
|
||||
await msg.Channel.SendMessageAsync(FailPrefix + Fail403);
|
||||
}
|
||||
else if (ex.HttpCode == System.Net.HttpStatusCode.NotFound)
|
||||
{
|
||||
await msg.Channel.SendMessageAsync(FailPrefix + Fail404);
|
||||
}
|
||||
else
|
||||
{
|
||||
await msg.Channel.SendMessageAsync(FailPrefix + FailDefault);
|
||||
await Log(ex.ToString());
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private async Task SendUsageMessage(SocketMessage m, string message)
|
||||
{
|
||||
string desc = $"{this.Command} [user or user ID]\n";
|
||||
desc += "Unbans the given user, allowing them to rejoin the server.";
|
||||
|
||||
var usageEmbed = new EmbedBuilder()
|
||||
{
|
||||
Title = "Usage",
|
||||
Description = desc
|
||||
};
|
||||
await m.Channel.SendMessageAsync(message ?? "", embed: usageEmbed);
|
||||
}
|
||||
}
|
||||
}
|
Loading…
Reference in a new issue