BirthdayBot/ApplicationCommands/BirthdayModule.cs

196 lines
8.7 KiB
C#
Raw Normal View History

2022-02-25 02:02:20 +00:00
using BirthdayBot.Data;
using Discord.Interactions;
using System.Text;
2022-02-24 04:33:24 +00:00
namespace BirthdayBot.ApplicationCommands;
2022-03-10 23:37:34 +00:00
[Group("birthday", HelpCmdBirthday)]
[CommandContextType(InteractionContextType.Guild)]
2022-02-24 04:33:24 +00:00
public class BirthdayModule : BotModuleBase {
2022-03-10 23:37:34 +00:00
public const string HelpCmdBirthday = "Commands relating to birthdays.";
public const string HelpCmdSetDate = "Sets or updates your birthday.";
public const string HelpCmdSetZone = "Sets or updates your time zone if your birthday is already set.";
public const string HelpCmdRemove = "Removes your birthday information from this bot.";
public const string HelpCmdGet = "Gets a user's birthday.";
public const string HelpCmdNearest = "Get a list of users who recently had or will have a birthday.";
public const string ErrNotSetFk = $":x: The bot has not yet been set up. Please configure a birthday role."; // foreign key violation
2022-02-24 04:33:24 +00:00
2022-02-25 04:53:08 +00:00
// Note that these methods have largely been copied to BirthdayOverrideModule. Changes here should be reflected there as needed.
2022-02-24 04:33:24 +00:00
[Group("set", "Subcommands for setting birthday information.")]
public class SubCmdsBirthdaySet : BotModuleBase {
2022-03-10 23:37:34 +00:00
[SlashCommand("date", HelpCmdSetDate)]
2022-02-24 04:33:24 +00:00
public async Task CmdSetBday([Summary(description: HelpOptDate)] string date,
2022-03-10 23:37:34 +00:00
[Summary(description: HelpOptZone)] string? zone = null) {
2022-02-24 04:33:24 +00:00
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;
}
string? inzone = null;
if (zone != null) {
try {
inzone = ParseTimeZone(zone);
} catch (FormatException e) {
await ReplyAsync(e.Message).ConfigureAwait(false);
return;
}
}
2022-03-20 08:07:17 +00:00
using var db = new BotDatabaseContext();
var user = ((SocketGuildUser)Context.User).GetUserEntryOrNew(db);
if (user.IsNew) db.UserEntries.Add(user);
user.BirthMonth = inmonth;
user.BirthDay = inday;
user.TimeZone = inzone ?? user.TimeZone;
try {
await db.SaveChangesAsync();
} catch (Microsoft.EntityFrameworkCore.DbUpdateException e)
when (e.InnerException is Npgsql.PostgresException ex && ex.SqlState == Npgsql.PostgresErrorCodes.ForeignKeyViolation) {
await RespondAsync(ErrNotSetFk);
return;
}
2022-02-24 04:33:24 +00:00
var response = $":white_check_mark: Your birthday has been set to **{FormatDate(inmonth, inday)}**";
if (inzone != null) response += $" at time zone **{inzone}**";
response += ".";
if (user.TimeZone == null)
response += "\n(Tip: The `/birthday set timezone` command ensures your birthday is recognized just in time!)";
await RespondAsync(response).ConfigureAwait(false);
2022-02-24 04:33:24 +00:00
}
2022-03-10 23:37:34 +00:00
[SlashCommand("timezone", HelpCmdSetZone)]
2022-02-24 04:33:24 +00:00
public async Task CmdSetZone([Summary(description: HelpOptZone)] string zone) {
2022-03-20 08:07:17 +00:00
using var db = new BotDatabaseContext();
var user = ((SocketGuildUser)Context.User).GetUserEntryOrNew(db);
if (user.IsNew) {
2022-02-25 04:53:08 +00:00
await RespondAsync(":x: You do not have a birthday set.", ephemeral: true).ConfigureAwait(false);
2022-02-24 04:33:24 +00:00
return;
}
2022-03-20 08:07:17 +00:00
string newzone;
2022-02-24 04:33:24 +00:00
try {
2022-03-20 08:07:17 +00:00
newzone = ParseTimeZone(zone);
2022-02-24 04:33:24 +00:00
} catch (FormatException e) {
await RespondAsync(e.Message, ephemeral: true).ConfigureAwait(false);
return;
}
2022-03-20 08:07:17 +00:00
user.TimeZone = newzone;
await db.SaveChangesAsync();
await RespondAsync($":white_check_mark: Your time zone has been set to **{newzone}**.").ConfigureAwait(false);
2022-02-24 04:33:24 +00:00
}
}
2022-03-10 23:37:34 +00:00
[SlashCommand("remove", HelpCmdRemove)]
2022-02-24 04:33:24 +00:00
public async Task CmdRemove() {
2022-03-20 08:07:17 +00:00
using var db = new BotDatabaseContext();
var user = ((SocketGuildUser)Context.User).GetUserEntryOrNew(db);
if (!user.IsNew) {
db.UserEntries.Remove(user);
await db.SaveChangesAsync();
await RespondAsync(":white_check_mark: Your birthday in this server has been removed.");
2022-02-24 04:33:24 +00:00
} else {
2022-02-25 04:53:08 +00:00
await RespondAsync(":white_check_mark: Your birthday is not registered.")
.ConfigureAwait(false);
2022-02-24 04:33:24 +00:00
}
}
2022-02-25 02:02:20 +00:00
[SlashCommand("get", "Gets a user's birthday.")]
public async Task CmdGetBday([Summary(description: "Optional: The user's birthday to look up.")] SocketGuildUser? user = null) {
2022-03-20 08:07:17 +00:00
using var db = new BotDatabaseContext();
var isSelf = user is null;
if (isSelf) user = (SocketGuildUser)Context.User;
2022-02-25 02:02:20 +00:00
2022-03-20 08:07:17 +00:00
var targetdata = user!.GetUserEntryOrNew(db);
if (targetdata.IsNew) {
if (isSelf) await RespondAsync(":x: You do not have your birthday registered.", ephemeral: true).ConfigureAwait(false);
2022-02-25 02:02:20 +00:00
else await RespondAsync(":x: The given user does not have their birthday registered.", ephemeral: true).ConfigureAwait(false);
return;
}
2022-03-11 03:41:45 +00:00
await RespondAsync($"{Common.FormatName(user!, false)}: `{FormatDate(targetdata.BirthMonth, targetdata.BirthDay)}`" +
2022-02-25 02:02:20 +00:00
(targetdata.TimeZone == null ? "" : $" - {targetdata.TimeZone}")).ConfigureAwait(false);
}
// "Recent and upcoming birthdays"
// The 'recent' bit removes time zone ambiguity and spares us from extra time zone processing here
// TODO stop being lazy
2022-03-10 23:37:34 +00:00
[SlashCommand("show-nearest", HelpCmdNearest)]
2022-02-25 02:02:20 +00:00
public async Task CmdShowNearest() {
if (!await HasMemberCacheAsync(Context.Guild).ConfigureAwait(false)) {
2022-03-11 03:41:45 +00:00
await RespondAsync(MemberCacheEmptyError, ephemeral: true).ConfigureAwait(false);
2022-02-25 02:02:20 +00:00
return;
}
var now = DateTimeOffset.UtcNow;
var search = DateIndex(now.Month, now.Day) - 8; // begin search 8 days prior to current date UTC
if (search <= 0) search = 366 - Math.Abs(search);
2022-03-20 08:07:17 +00:00
var query = GetSortedUserList(Context.Guild);
2022-02-25 02:02:20 +00:00
// TODO pagination instead of this workaround
2022-09-01 04:09:10 +00:00
var hasOutputOneLine = false;
2022-02-25 02:02:20 +00:00
// First output is shown as an interaction response, followed then as regular channel messages
async Task doOutput(string msg) {
if (!hasOutputOneLine) {
await RespondAsync(msg).ConfigureAwait(false);
hasOutputOneLine = true;
} else {
2022-03-11 03:41:45 +00:00
await ReplyAsync(msg).ConfigureAwait(false);
2022-02-25 02:02:20 +00:00
}
}
var output = new StringBuilder();
var resultCount = 0;
output.AppendLine("Recent and upcoming birthdays:");
2022-09-01 04:09:10 +00:00
for (var count = 0; count <= 21; count++) { // cover 21 days total (7 prior, current day, 14 upcoming)
2022-02-25 02:02:20 +00:00
var results = from item in query
where item.DateIndex == search
select item;
// push up search by 1 now, in case we back out early
search += 1;
if (search > 366) search = 1; // wrap to beginning of year
if (!results.Any()) continue; // back out early
resultCount += results.Count();
// Build sorted name list
var names = new List<string>();
foreach (var item in results) {
names.Add(item.DisplayName);
}
names.Sort(StringComparer.OrdinalIgnoreCase);
var first = true;
output.AppendLine();
output.Append($"● `{Common.MonthNames[results.First().BirthMonth]}-{results.First().BirthDay:00}`: ");
foreach (var item in names) {
// If the output is starting to fill up, send out this message and prepare a new one.
if (output.Length > 800) {
await doOutput(output.ToString()).ConfigureAwait(false);
output.Clear();
first = true;
output.Append($"● `{Common.MonthNames[results.First().BirthMonth]}-{results.First().BirthDay:00}`: ");
}
if (first) first = false;
else output.Append(", ");
output.Append(item);
}
}
if (resultCount == 0)
await RespondAsync(
"There are no recent or upcoming birthdays (within the last 7 days and/or next 14 days).")
.ConfigureAwait(false);
else
await doOutput(output.ToString()).ConfigureAwait(false);
}
2022-02-24 04:33:24 +00:00
}