mirror of
https://github.com/NoiTheCat/BirthdayBot.git
synced 2024-11-21 13:54:36 +00:00
Add overriding commands
This commit is contained in:
parent
07343735a0
commit
1728285683
2 changed files with 68 additions and 174 deletions
68
ApplicationCommands/BirthdayOverrideModule.cs
Normal file
68
ApplicationCommands/BirthdayOverrideModule.cs
Normal file
|
|
@ -0,0 +1,68 @@
|
||||||
|
using BirthdayBot.Data;
|
||||||
|
using Discord.Interactions;
|
||||||
|
|
||||||
|
namespace BirthdayBot.ApplicationCommands;
|
||||||
|
|
||||||
|
[RequireContext(ContextType.Guild)]
|
||||||
|
[RequireBotModerator]
|
||||||
|
[Group("override", "Run certain commands on behalf of other users.")]
|
||||||
|
public class BirthdayOverrideModule : BotModuleBase {
|
||||||
|
const string HelpOptOvTarget = "The user whose data to modify.";
|
||||||
|
|
||||||
|
// Note that these methods have largely been copied from BirthdayModule. Changes there should be reflected here as needed.
|
||||||
|
// TODO possible to use a common base class for shared functionality instead?
|
||||||
|
|
||||||
|
[SlashCommand("set-birthday", HelpPfxModOnly + "Set a user's birthday on their behalf.")]
|
||||||
|
public async Task OvSetBirthday([Summary(description: HelpOptOvTarget)]SocketGuildUser target,
|
||||||
|
[Summary(description: HelpOptDate)]string date) {
|
||||||
|
int inmonth, inday;
|
||||||
|
try {
|
||||||
|
(inmonth, inday) = ParseDate(date);
|
||||||
|
} catch (FormatException e) {
|
||||||
|
// Our parse method's FormatException has its message to send out to Discord.
|
||||||
|
await RespondAsync(e.Message, ephemeral: true).ConfigureAwait(false);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
var user = await target.GetConfigAsync().ConfigureAwait(false);
|
||||||
|
await user.UpdateAsync(inmonth, inday, user.TimeZone).ConfigureAwait(false);
|
||||||
|
|
||||||
|
await RespondAsync($":white_check_mark: {Common.FormatName(target, false)}'s birthday has been set to " +
|
||||||
|
$"**{FormatDate(inmonth, inday)}**.").ConfigureAwait(false);
|
||||||
|
}
|
||||||
|
|
||||||
|
[SlashCommand("set-timezone", HelpPfxModOnly + "Set a user's time zone on their behalf.")]
|
||||||
|
public async Task OvSetTimezone([Summary(description: HelpOptOvTarget)]SocketGuildUser target,
|
||||||
|
[Summary(description: HelpOptZone)]string zone) {
|
||||||
|
var user = await target.GetConfigAsync().ConfigureAwait(false);
|
||||||
|
if (!user.IsKnown) {
|
||||||
|
await RespondAsync($":x: {Common.FormatName(target, false)} does not have a birthday set.")
|
||||||
|
.ConfigureAwait(false);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
string inzone;
|
||||||
|
try {
|
||||||
|
inzone = ParseTimeZone(zone);
|
||||||
|
} catch (FormatException e) {
|
||||||
|
await RespondAsync(e.Message, ephemeral: true).ConfigureAwait(false);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
await user.UpdateAsync(user.BirthMonth, user.BirthDay, inzone).ConfigureAwait(false);
|
||||||
|
await RespondAsync($":white_check_mark: {Common.FormatName(target, false)}'s time zone has been set to " +
|
||||||
|
$"**{inzone}**.").ConfigureAwait(false);
|
||||||
|
}
|
||||||
|
|
||||||
|
[SlashCommand("remove-birthday", HelpPfxModOnly + "Remove a user's birthday information on their behalf.")]
|
||||||
|
public async Task OvRemove([Summary(description: HelpOptOvTarget)]SocketGuildUser target) {
|
||||||
|
var user = await target.GetConfigAsync().ConfigureAwait(false);
|
||||||
|
if (user.IsKnown) {
|
||||||
|
await user.DeleteAsync().ConfigureAwait(false);
|
||||||
|
await RespondAsync($":white_check_mark: {Common.FormatName(target, false)}'s birthday in this server has been removed.")
|
||||||
|
.ConfigureAwait(false);
|
||||||
|
} else {
|
||||||
|
await RespondAsync($":white_check_mark: {Common.FormatName(target, false)}'s birthday is not registered.")
|
||||||
|
.ConfigureAwait(false);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -1,174 +0,0 @@
|
||||||
using BirthdayBot.Data;
|
|
||||||
|
|
||||||
namespace BirthdayBot.ApplicationCommands;
|
|
||||||
|
|
||||||
internal class RegistrationOverrideCommands : BotApplicationCommand {
|
|
||||||
private delegate Task SubCommandHandler(GuildConfiguration gconf, SocketSlashCommand arg, Dictionary<string, object> subparam);
|
|
||||||
|
|
||||||
#region Help strings
|
|
||||||
public const string HelpOverride = "Run certain commands on behalf of other users.";
|
|
||||||
|
|
||||||
const string HelpOptTarget = "The user whose data to modify.";
|
|
||||||
#endregion
|
|
||||||
|
|
||||||
public override IEnumerable<ApplicationCommandProperties> GetCommands() => new ApplicationCommandProperties[] {
|
|
||||||
new SlashCommandBuilder()
|
|
||||||
.WithName("override")
|
|
||||||
.WithDescription(HelpPfxModOnly + HelpOverride)
|
|
||||||
.AddOption(new SlashCommandOptionBuilder()
|
|
||||||
.WithName("set-birthday")
|
|
||||||
.WithDescription(HelpPfxModOnly + "Sets or updates a user's birthday on their behalf.")
|
|
||||||
.WithType(ApplicationCommandOptionType.SubCommand)
|
|
||||||
.AddOption(new SlashCommandOptionBuilder()
|
|
||||||
.WithName("target")
|
|
||||||
.WithType(ApplicationCommandOptionType.User)
|
|
||||||
.WithDescription(HelpOptTarget)
|
|
||||||
.WithRequired(true)
|
|
||||||
).AddOption(new SlashCommandOptionBuilder()
|
|
||||||
.WithName("date")
|
|
||||||
.WithType(ApplicationCommandOptionType.String)
|
|
||||||
.WithDescription(RegistrationCommands.HelpOptDate)
|
|
||||||
.WithRequired(true)
|
|
||||||
)
|
|
||||||
).AddOption(new SlashCommandOptionBuilder()
|
|
||||||
.WithName("set-timezone")
|
|
||||||
.WithDescription(HelpPfxModOnly + "Sets or updates a user's time zone on their behalf.")
|
|
||||||
.WithType(ApplicationCommandOptionType.SubCommand)
|
|
||||||
.AddOption(new SlashCommandOptionBuilder()
|
|
||||||
.WithName("target")
|
|
||||||
.WithType(ApplicationCommandOptionType.User)
|
|
||||||
.WithDescription(HelpOptTarget)
|
|
||||||
.WithRequired(true)
|
|
||||||
).AddOption(new SlashCommandOptionBuilder()
|
|
||||||
.WithName("zone")
|
|
||||||
.WithType(ApplicationCommandOptionType.String)
|
|
||||||
.WithDescription(RegistrationCommands.HelpOptZone)
|
|
||||||
.WithRequired(true)
|
|
||||||
)
|
|
||||||
).AddOption(new SlashCommandOptionBuilder()
|
|
||||||
.WithName("remove-timezone")
|
|
||||||
.WithDescription(HelpPfxModOnly + "Removes a user's time zone on their behalf.")
|
|
||||||
.WithType(ApplicationCommandOptionType.SubCommand)
|
|
||||||
.AddOption(new SlashCommandOptionBuilder()
|
|
||||||
.WithName("target")
|
|
||||||
.WithType(ApplicationCommandOptionType.User)
|
|
||||||
.WithDescription(HelpOptTarget)
|
|
||||||
.WithRequired(true)
|
|
||||||
)
|
|
||||||
).AddOption(new SlashCommandOptionBuilder()
|
|
||||||
.WithName("remove-birthday")
|
|
||||||
.WithDescription(HelpPfxModOnly + "Removes a user's data from the bot on their behalf.")
|
|
||||||
.WithType(ApplicationCommandOptionType.SubCommand)
|
|
||||||
.AddOption(new SlashCommandOptionBuilder()
|
|
||||||
.WithName("target")
|
|
||||||
.WithType(ApplicationCommandOptionType.User)
|
|
||||||
.WithDescription(HelpOptTarget)
|
|
||||||
.WithRequired(true)
|
|
||||||
)
|
|
||||||
).Build()
|
|
||||||
};
|
|
||||||
public override CommandResponder? GetHandlerFor(string commandName) => commandName switch {
|
|
||||||
"override" => CmdOverride,
|
|
||||||
_ => null
|
|
||||||
};
|
|
||||||
|
|
||||||
/*
|
|
||||||
* Personally, this confuses me. So here are some notes:
|
|
||||||
* arg.Data.Options contains the SubCommand and only the SubCommand. Its name is the subcommand's name.
|
|
||||||
* arg.Data.Options.Options then contains the options. "target" and others, all within the same collection.
|
|
||||||
*/
|
|
||||||
private Task CmdOverride(ShardInstance instance, GuildConfiguration gconf, SocketSlashCommand arg) {
|
|
||||||
SubCommandHandler? subh = arg.Data.Options.First().Name switch {
|
|
||||||
"set-birthday" => SubCmdSetBd,
|
|
||||||
"set-timezone" => SubCmdSetTz,
|
|
||||||
"remove-timezone" => SubCmdDelTz,
|
|
||||||
"remove-birthday" => SubCmdDelBd,
|
|
||||||
_ => null
|
|
||||||
};
|
|
||||||
|
|
||||||
if (subh == null) {
|
|
||||||
instance.Log($"{nameof(RegistrationOverrideCommands)}", $"Encountered unknown subcommand {arg.Data.Name}");
|
|
||||||
return arg.RespondAsync(ShardInstance.UnknownCommandError, ephemeral: true);
|
|
||||||
}
|
|
||||||
|
|
||||||
var subparam = ((SocketSlashCommandDataOption)arg.Data.Options.First()).Options.ToDictionary(o => o.Name, o => o.Value);
|
|
||||||
return subh(gconf, arg, subparam);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Note that the following subcommands have largely been copied from RegistrationCommands.
|
|
||||||
// Any changes made there should be reflected here, if appropriate.
|
|
||||||
// TODO A common base class might be more appropriate...
|
|
||||||
|
|
||||||
private async Task SubCmdSetBd(GuildConfiguration gconf, SocketSlashCommand arg, Dictionary<string, object> subparam) {
|
|
||||||
var target = (SocketGuildUser)subparam["target"];
|
|
||||||
int inmonth, inday;
|
|
||||||
try {
|
|
||||||
(inmonth, inday) = ParseDate((string)subparam["date"]);
|
|
||||||
} catch (FormatException e) {
|
|
||||||
// Our parse method's FormatException has its message to send out to Discord.
|
|
||||||
arg.RespondAsync(e.Message).Wait();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool known;
|
|
||||||
try {
|
|
||||||
var user = await GuildUserConfiguration.LoadAsync(gconf.GuildId, target.Id).ConfigureAwait(false);
|
|
||||||
known = user.IsKnown;
|
|
||||||
await user.UpdateAsync(inmonth, inday, user.TimeZone).ConfigureAwait(false);
|
|
||||||
} catch (Exception ex) {
|
|
||||||
Program.Log("Error", ex.ToString());
|
|
||||||
arg.RespondAsync(ShardInstance.InternalError).Wait();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
await arg.RespondAsync($":white_check_mark: {target}'s birthday has been " +
|
|
||||||
$"{ (known ? "updated to" : "recorded as") } **{inday:00}-{Common.MonthNames[inmonth]}**.").ConfigureAwait(false);
|
|
||||||
}
|
|
||||||
|
|
||||||
private async Task SubCmdSetTz(GuildConfiguration gconf, SocketSlashCommand arg, Dictionary<string, object> subparam) {
|
|
||||||
var target = (SocketGuildUser)subparam["target"];
|
|
||||||
var user = await GuildUserConfiguration.LoadAsync(gconf.GuildId, target.Id).ConfigureAwait(false);
|
|
||||||
if (!user.IsKnown) {
|
|
||||||
await arg.RespondAsync(":x: The user must have a birthday set before you can use this command.",
|
|
||||||
ephemeral: true).ConfigureAwait(false);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
bool hasZone = user.TimeZone != null;
|
|
||||||
|
|
||||||
string inZone;
|
|
||||||
try {
|
|
||||||
inZone = ParseTimeZone((string)subparam["zone"]);
|
|
||||||
} catch (Exception e) {
|
|
||||||
arg.RespondAsync(e.Message).Wait();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
await user.UpdateAsync(user.BirthMonth, user.BirthDay, inZone).ConfigureAwait(false);
|
|
||||||
|
|
||||||
await arg.RespondAsync($":white_check_mark: {target}'s time zone has been { (hasZone ? "updated" : "set") } to **{inZone}**.")
|
|
||||||
.ConfigureAwait(false);
|
|
||||||
}
|
|
||||||
|
|
||||||
private async Task SubCmdDelTz(GuildConfiguration gconf, SocketSlashCommand arg, Dictionary<string, object> subparam) {
|
|
||||||
var target = (SocketGuildUser)subparam["target"];
|
|
||||||
var u = await GuildUserConfiguration.LoadAsync(gconf.GuildId, target.Id).ConfigureAwait(false);
|
|
||||||
if (!u.IsKnown) {
|
|
||||||
await arg.RespondAsync(":white_check_mark: User is not registered. Nothing to remove.");
|
|
||||||
} else if (u.TimeZone is null) {
|
|
||||||
await arg.RespondAsync(":white_check_mark: User does not have zone registered. Nothing to remove.");
|
|
||||||
} else {
|
|
||||||
await u.UpdateAsync(u.BirthMonth, u.BirthDay, null);
|
|
||||||
await arg.RespondAsync($":white_check_mark: {target}'s time zone information has been removed.");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private async Task SubCmdDelBd(GuildConfiguration gconf, SocketSlashCommand arg, Dictionary<string, object> subparam) {
|
|
||||||
var target = (SocketGuildUser)subparam["target"];
|
|
||||||
var u = await GuildUserConfiguration.LoadAsync(gconf.GuildId, target.Id).ConfigureAwait(false);
|
|
||||||
if (u.IsKnown) {
|
|
||||||
await u.DeleteAsync().ConfigureAwait(false);
|
|
||||||
await arg.RespondAsync($":white_check_mark: {target}'s birthday information has been removed.");
|
|
||||||
} else {
|
|
||||||
await arg.RespondAsync(":white_check_mark: User is not registered. Nothing to remove.");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Loading…
Reference in a new issue