forked from TeamInklings/Octobot
321 lines
12 KiB
C#
321 lines
12 KiB
C#
using System.Text;
|
||
using Boyfriend.Commands;
|
||
using Boyfriend.Data;
|
||
using Discord;
|
||
using Discord.Commands;
|
||
using Discord.WebSocket;
|
||
|
||
namespace Boyfriend;
|
||
|
||
public sealed class CommandProcessor {
|
||
private static readonly string Mention = $"<@{Boyfriend.Client.CurrentUser.Id}>";
|
||
private static readonly TimeSpan Infinity = TimeSpan.FromMilliseconds(-1);
|
||
|
||
public static readonly ICommand[] Commands = {
|
||
new BanCommand(), new ClearCommand(), new HelpCommand(),
|
||
new KickCommand(), new MuteCommand(), new PingCommand(),
|
||
new RemindCommand(), new SettingsCommand(), new UnbanCommand(),
|
||
new UnmuteCommand()
|
||
};
|
||
|
||
private readonly StringBuilder _stackedPrivateFeedback = new();
|
||
private readonly StringBuilder _stackedPublicFeedback = new();
|
||
private readonly StringBuilder _stackedReplyMessage = new();
|
||
private readonly List<Task> _tasks = new();
|
||
|
||
public readonly SocketCommandContext Context;
|
||
|
||
public bool ConfigWriteScheduled = false;
|
||
|
||
public CommandProcessor(SocketUserMessage message) {
|
||
Context = new SocketCommandContext(Boyfriend.Client, message);
|
||
}
|
||
|
||
public async Task HandleCommandAsync() {
|
||
var guild = Context.Guild;
|
||
var data = GuildData.Get(guild);
|
||
Utils.SetCurrentLanguage(guild);
|
||
|
||
var list = Context.Message.Content.Split("\n");
|
||
var cleanList = Context.Message.CleanContent.Split("\n");
|
||
for (var i = 0; i < list.Length; i++)
|
||
_tasks.Add(RunCommandOnLine(list[i], cleanList[i], data.Preferences["Prefix"]));
|
||
|
||
try { Task.WaitAll(_tasks.ToArray()); } catch (AggregateException e) {
|
||
foreach (var ex in e.InnerExceptions)
|
||
await Boyfriend.Log(
|
||
new LogMessage(
|
||
LogSeverity.Error, nameof(CommandProcessor),
|
||
"Exception while executing commands", ex));
|
||
}
|
||
|
||
_tasks.Clear();
|
||
|
||
if (ConfigWriteScheduled) await data.Save(true);
|
||
|
||
SendFeedbacks();
|
||
}
|
||
|
||
private async Task RunCommandOnLine(string line, string cleanLine, string prefix) {
|
||
var prefixed = line.StartsWith(prefix);
|
||
if (!prefixed && !line.StartsWith(Mention)) return;
|
||
foreach (var command in Commands) {
|
||
var lineNoMention = line.Remove(0, prefixed ? prefix.Length : Mention.Length);
|
||
if (!command.Aliases.Contains(lineNoMention.Trim().Split()[0])) continue;
|
||
|
||
var args = lineNoMention.Trim().Split().Skip(1).ToArray();
|
||
var cleanArgs = cleanLine.Split().Skip(lineNoMention.StartsWith(" ") ? 2 : 1).ToArray();
|
||
await command.RunAsync(this, args, cleanArgs);
|
||
if (_stackedReplyMessage.Length > 0) _ = Context.Channel.TriggerTypingAsync();
|
||
return;
|
||
}
|
||
}
|
||
|
||
public void Reply(string response, string? customEmoji = null) {
|
||
Utils.SafeAppendToBuilder(
|
||
_stackedReplyMessage, $"{customEmoji ?? ReplyEmojis.Success} {response}",
|
||
Context.Message);
|
||
}
|
||
|
||
public void Audit(string action, bool isPublic = true) {
|
||
var format = $"*[{Context.User.Mention}: {action}]*";
|
||
var data = GuildData.Get(Context.Guild);
|
||
if (isPublic) Utils.SafeAppendToBuilder(_stackedPublicFeedback, format, data.PublicFeedbackChannel);
|
||
Utils.SafeAppendToBuilder(_stackedPrivateFeedback, format, data.PrivateFeedbackChannel);
|
||
if (_tasks.Count is 0) SendFeedbacks(false);
|
||
}
|
||
|
||
private void SendFeedbacks(bool reply = true) {
|
||
var hasReply = _stackedReplyMessage.Length > 0;
|
||
if (reply && hasReply)
|
||
_ = Context.Message.ReplyAsync(_stackedReplyMessage.ToString(), false, null, AllowedMentions.None);
|
||
|
||
var data = GuildData.Get(Context.Guild);
|
||
var adminChannel = data.PrivateFeedbackChannel;
|
||
var systemChannel = data.PublicFeedbackChannel;
|
||
if (_stackedPrivateFeedback.Length > 0
|
||
&& adminChannel is not null
|
||
&& (adminChannel.Id != Context.Message.Channel.Id || !hasReply)) {
|
||
_ = Utils.SilentSendAsync(adminChannel, _stackedPrivateFeedback.ToString());
|
||
_stackedPrivateFeedback.Clear();
|
||
}
|
||
|
||
if (_stackedPublicFeedback.Length > 0
|
||
&& systemChannel is not null
|
||
&& systemChannel.Id != adminChannel?.Id
|
||
&& (systemChannel.Id != Context.Message.Channel.Id || !hasReply)) {
|
||
_ = Utils.SilentSendAsync(systemChannel, _stackedPublicFeedback.ToString());
|
||
_stackedPublicFeedback.Clear();
|
||
}
|
||
}
|
||
|
||
public string? GetRemaining(string[] from, int startIndex, string? argument) {
|
||
if (startIndex >= from.Length && argument is not null)
|
||
Utils.SafeAppendToBuilder(
|
||
_stackedReplyMessage,
|
||
$"{ReplyEmojis.MissingArgument} {Utils.GetMessage($"Missing{argument}")}", Context.Message);
|
||
else return string.Join(" ", from, startIndex, from.Length - startIndex);
|
||
return null;
|
||
}
|
||
|
||
public (ulong Id, SocketUser? User)? GetUser(string[] args, string[] cleanArgs, int index) {
|
||
if (index >= args.Length) {
|
||
Utils.SafeAppendToBuilder(
|
||
_stackedReplyMessage, $"{ReplyEmojis.MissingArgument} {Messages.MissingUser}",
|
||
Context.Message);
|
||
return null;
|
||
}
|
||
|
||
var mention = Utils.ParseMention(args[index]);
|
||
if (mention is 0) {
|
||
Utils.SafeAppendToBuilder(
|
||
_stackedReplyMessage,
|
||
$"{ReplyEmojis.InvalidArgument} {string.Format(Messages.InvalidUser, Utils.Wrap(cleanArgs[index]))}",
|
||
Context.Message);
|
||
return null;
|
||
}
|
||
|
||
var exists = Utils.UserExists(mention);
|
||
if (!exists) {
|
||
Utils.SafeAppendToBuilder(
|
||
_stackedReplyMessage,
|
||
$"{ReplyEmojis.Error} {string.Format(Messages.UserNotFound, Utils.Wrap(cleanArgs[index]))}",
|
||
Context.Message);
|
||
return null;
|
||
}
|
||
|
||
return (mention, Boyfriend.Client.GetUser(mention));
|
||
}
|
||
|
||
public bool HasPermission(GuildPermission permission) {
|
||
if (!Context.Guild.CurrentUser.GuildPermissions.Has(permission)) {
|
||
Utils.SafeAppendToBuilder(
|
||
_stackedReplyMessage,
|
||
$"{ReplyEmojis.NoPermission} {Utils.GetMessage($"BotCannot{permission}")}",
|
||
Context.Message);
|
||
return false;
|
||
}
|
||
|
||
if (!GetMember().GuildPermissions.Has(permission)
|
||
&& Context.Guild.OwnerId != Context.User.Id) {
|
||
Utils.SafeAppendToBuilder(
|
||
_stackedReplyMessage,
|
||
$"{ReplyEmojis.NoPermission} {Utils.GetMessage($"UserCannot{permission}")}",
|
||
Context.Message);
|
||
return false;
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|
||
private SocketGuildUser GetMember() {
|
||
return GetMember(Context.User.Id)!;
|
||
}
|
||
|
||
public SocketGuildUser? GetMember(ulong id) {
|
||
return Context.Guild.GetUser(id);
|
||
}
|
||
|
||
public SocketGuildUser? GetMember(string[] args, int index) {
|
||
if (index >= args.Length) {
|
||
Utils.SafeAppendToBuilder(
|
||
_stackedReplyMessage, $"{ReplyEmojis.MissingArgument} {Messages.MissingMember}",
|
||
Context.Message);
|
||
return null;
|
||
}
|
||
|
||
var member = Context.Guild.GetUser(Utils.ParseMention(args[index]));
|
||
if (member is null)
|
||
Utils.SafeAppendToBuilder(
|
||
_stackedReplyMessage,
|
||
$"{ReplyEmojis.InvalidArgument} {Messages.InvalidMember}",
|
||
Context.Message);
|
||
return member;
|
||
}
|
||
|
||
public ulong? GetBan(string[] args, int index) {
|
||
if (index >= args.Length) {
|
||
Utils.SafeAppendToBuilder(
|
||
_stackedReplyMessage, $"{ReplyEmojis.MissingArgument} {Messages.MissingUser}",
|
||
Context.Message);
|
||
return null;
|
||
}
|
||
|
||
var id = Utils.ParseMention(args[index]);
|
||
if (Context.Guild.GetBanAsync(id) is null) {
|
||
Utils.SafeAppendToBuilder(_stackedReplyMessage, Messages.UserNotBanned, Context.Message);
|
||
return null;
|
||
}
|
||
|
||
return id;
|
||
}
|
||
|
||
public int? GetNumberRange(string[] args, int index, int min, int max, string? argument) {
|
||
if (index >= args.Length) {
|
||
Utils.SafeAppendToBuilder(
|
||
_stackedReplyMessage,
|
||
$"{ReplyEmojis.MissingArgument} {string.Format(Messages.MissingNumber, min.ToString(), max.ToString())}",
|
||
Context.Message);
|
||
return null;
|
||
}
|
||
|
||
if (!int.TryParse(args[index], out var i)) {
|
||
Utils.SafeAppendToBuilder(
|
||
_stackedReplyMessage,
|
||
$"{ReplyEmojis.InvalidArgument} {string.Format(Utils.GetMessage($"{argument}Invalid"), min.ToString(), max.ToString(), Utils.Wrap(args[index]))}",
|
||
Context.Message);
|
||
return null;
|
||
}
|
||
|
||
if (argument is null) return i;
|
||
if (i < min) {
|
||
Utils.SafeAppendToBuilder(
|
||
_stackedReplyMessage,
|
||
$"{ReplyEmojis.InvalidArgument} {string.Format(Utils.GetMessage($"{argument}TooSmall"), min.ToString())}",
|
||
Context.Message);
|
||
return null;
|
||
}
|
||
|
||
if (i > max) {
|
||
Utils.SafeAppendToBuilder(
|
||
_stackedReplyMessage,
|
||
$"{ReplyEmojis.InvalidArgument} {string.Format(Utils.GetMessage($"{argument}TooLarge"), max.ToString())}",
|
||
Context.Message);
|
||
return null;
|
||
}
|
||
|
||
return i;
|
||
}
|
||
|
||
public static TimeSpan GetTimeSpan(string[] args, int index) {
|
||
if (index >= args.Length) return Infinity;
|
||
var chars = args[index].AsSpan();
|
||
var numberBuilder = Boyfriend.StringBuilder;
|
||
int days = 0, hours = 0, minutes = 0, seconds = 0;
|
||
foreach (var c in chars)
|
||
if (char.IsDigit(c)) { numberBuilder.Append(c); } else {
|
||
if (numberBuilder.Length is 0) return Infinity;
|
||
switch (c) {
|
||
case 'd' or 'D' or 'д' or 'Д':
|
||
days += int.Parse(numberBuilder.ToString());
|
||
numberBuilder.Clear();
|
||
break;
|
||
case 'h' or 'H' or 'ч' or 'Ч':
|
||
hours += int.Parse(numberBuilder.ToString());
|
||
numberBuilder.Clear();
|
||
break;
|
||
case 'm' or 'M' or 'м' or 'М':
|
||
minutes += int.Parse(numberBuilder.ToString());
|
||
numberBuilder.Clear();
|
||
break;
|
||
case 's' or 'S' or 'с' or 'С':
|
||
seconds += int.Parse(numberBuilder.ToString());
|
||
numberBuilder.Clear();
|
||
break;
|
||
default: return Infinity;
|
||
}
|
||
}
|
||
|
||
numberBuilder.Clear();
|
||
return new TimeSpan(days, hours, minutes, seconds);
|
||
}
|
||
|
||
public bool CanInteractWith(SocketGuildUser user, string action) {
|
||
if (Context.User.Id == user.Id) {
|
||
Utils.SafeAppendToBuilder(
|
||
_stackedReplyMessage,
|
||
$"{ReplyEmojis.CantInteract} {Utils.GetMessage($"UserCannot{action}Themselves")}", Context.Message);
|
||
return false;
|
||
}
|
||
|
||
if (Context.Guild.CurrentUser.Id == user.Id) {
|
||
Utils.SafeAppendToBuilder(
|
||
_stackedReplyMessage,
|
||
$"{ReplyEmojis.CantInteract} {Utils.GetMessage($"UserCannot{action}Bot")}", Context.Message);
|
||
return false;
|
||
}
|
||
|
||
if (Context.Guild.Owner.Id == user.Id) {
|
||
Utils.SafeAppendToBuilder(
|
||
_stackedReplyMessage,
|
||
$"{ReplyEmojis.CantInteract} {Utils.GetMessage($"UserCannot{action}Owner")}", Context.Message);
|
||
return false;
|
||
}
|
||
|
||
if (Context.Guild.CurrentUser.Hierarchy <= user.Hierarchy) {
|
||
Utils.SafeAppendToBuilder(
|
||
_stackedReplyMessage,
|
||
$"{ReplyEmojis.CantInteract} {Utils.GetMessage($"BotCannot{action}Target")}", Context.Message);
|
||
return false;
|
||
}
|
||
|
||
if (Context.Guild.Owner.Id != Context.User.Id && GetMember().Hierarchy <= user.Hierarchy) {
|
||
Utils.SafeAppendToBuilder(
|
||
_stackedReplyMessage,
|
||
$"{ReplyEmojis.CantInteract} {Utils.GetMessage($"UserCannot{action}Target")}", Context.Message);
|
||
return false;
|
||
}
|
||
|
||
return true;
|
||
}
|
||
}
|