cs-codex-dist-tests/Tools/BiblioTech/BaseCommand.cs

65 lines
2.1 KiB
C#
Raw Normal View History

2023-10-20 07:49:23 +00:00
using Discord.WebSocket;
2023-10-24 13:25:45 +00:00
using BiblioTech.Options;
2023-10-25 09:25:27 +00:00
using Discord;
2023-10-20 07:49:23 +00:00
namespace BiblioTech
{
public abstract class BaseCommand
{
public abstract string Name { get; }
public abstract string StartingMessage { get; }
2023-10-20 07:49:23 +00:00
public abstract string Description { get; }
public virtual CommandOption[] Options => Array.Empty<CommandOption>();
2023-10-20 07:49:23 +00:00
public async Task SlashCommandHandler(SocketSlashCommand command)
{
if (command.CommandName != Name) return;
try
{
var context = new CommandContext(command, command.Data.Options);
await command.RespondAsync(StartingMessage, ephemeral: IsEphemeral(context));
await Invoke(context);
}
catch (Exception ex)
{
2023-11-09 09:13:34 +00:00
if (IsInAdminChannel(command))
{
var msg = "Failed with exception: " + ex;
await command.FollowupAsync(msg.Substring(0, Math.Min(1900, msg.Length)));
}
else
{
await command.FollowupAsync("Something failed while trying to do that...", ephemeral: true);
}
Console.WriteLine(ex);
}
2023-10-20 07:49:23 +00:00
}
private bool IsEphemeral(CommandContext context)
{
if (IsSenderAdmin(context.Command) && IsInAdminChannel(context.Command)) return false;
return true;
}
2023-10-24 13:25:45 +00:00
protected abstract Task Invoke(CommandContext context);
2023-10-22 08:10:52 +00:00
protected bool IsSenderAdmin(SocketSlashCommand command)
{
2023-10-22 08:38:46 +00:00
return Program.AdminChecker.IsUserAdmin(command.User.Id);
2023-10-22 08:10:52 +00:00
}
protected bool IsInAdminChannel(SocketSlashCommand command)
{
return Program.AdminChecker.IsAdminChannel(command.Channel);
}
2023-10-25 09:25:27 +00:00
protected IUser GetUserFromCommand(UserOption userOption, CommandContext context)
2023-10-22 08:10:52 +00:00
{
2023-10-25 09:25:27 +00:00
var targetUser = userOption.GetUser(context);
if (IsSenderAdmin(context.Command) && targetUser != null) return targetUser;
return context.Command.User;
2023-10-24 12:07:15 +00:00
}
2023-10-20 07:49:23 +00:00
}
}