WorldTime/Commands/UserCommands.cs

using Discord.Interactions;
using System.Text;

namespace WorldTime.Commands;
public class UserCommands : CommandsBase {
    #region Help strings
    const string HelpHelp = "Displays a list of available bot commands.";
    const string HelpList = "Shows the current time for all recently active known users.";
    const string HelpSet = "Adds or updates your time zone to the bot.";
    const string HelpRemove = "Removes your time zone information from this bot.";
    
    #endregion

    [SlashCommand("help", HelpHelp)]
    [CommandContextType(InteractionContextType.Guild, InteractionContextType.BotDm)]
    public async Task CmdHelp() {
        var version = System.Reflection.Assembly.GetExecutingAssembly().GetName().Version!.ToString(3);
        var guildct = ShardedClient.Guilds.Count;
        using var db = DbContext;
        var uniquetz = db.GetDistinctZoneCount();
        await RespondAsync(embed: new EmbedBuilder() {
            Title = "Help & About",
            Description =
                $"World Time v{version} - Serving {guildct} communities across {uniquetz} time zones.\n\n"
                + "This bot is provided for free, without any paywalled 'premium' features. "
                + "If you've found this bot useful, please consider contributing via the "
                + "bot author's page on Ko-fi: https://ko-fi.com/noithecat.",
            Footer = new EmbedFooterBuilder() {
                IconUrl = Context.Client.CurrentUser.GetAvatarUrl(),
                Text = "World Time"
            }
        }.AddField(inline: false, name: "Commands", value:
            $"""
            `/help` - {HelpHelp}
            `/list` - {HelpList}
            `/set` - {HelpSet}
            `/remove` - {HelpRemove}
            """
        ).AddField(inline: false, name: "Admin commands", value:
            $"""
            `/config use-12hour` - {ConfigCommands.HelpUse12}
            `/config private-confirms` - {ConfigCommands.HelpPrivateConfirms}
            `/set-for` - {ConfigCommands.HelpSetFor}
            `/remove-for` - {ConfigCommands.HelpRemoveFor}
            """
        ).AddField(inline: false, name: "Zones", value:
            "This bot accepts zone names from the IANA Time Zone Database (a.k.a. Olson Database). " +
            "A useful tool to determine yours can be found at: https://zones.arilyn.cc/"
        ).Build());
    }

    [SlashCommand("list", HelpList)]
    [CommandContextType(InteractionContextType.Guild)]
    public async Task CmdList([Summary(description: "A specific user whose time to look up.")]SocketGuildUser? user = null) {
        if (!await AreUsersDownloadedAsync(Context.Guild)) {
            await RespondAsync(ErrNoUserCache, ephemeral: true);
            return;
        }

        if (user == null) {
            // No parameter - full listing
            await CmdListWithoutParamAsync();
        } else {
            // Has parameter - do single user listing
            await CmdListWithUserParamAsync(user);
        }
    }

    private async Task CmdListWithoutParamAsync() {
        // Called by CmdList
        using var db = DbContext;
        var userlist = db.GetGuildZones(Context.Guild.Id);
        if (userlist.Count == 0) {
            await RespondAsync(":x: Nothing to show. Register your time zones with the bot using the `/set` command.",
                ephemeral: true).ConfigureAwait(false);
            return;
        }
        
        // Generate time and zone names to be displayed, group with associated user IDs
        var sortedlist = new SortedDictionary<string, List<ulong>>();
        var ampm = db.GuildSettings.Where(s => s.GuildId == Context.Guild.Id).SingleOrDefault()?.Use12HourTime ?? false;
        foreach ((var area, List<ulong> users) in userlist.OrderByDescending(o => o.Value.Count)) {
            var areaprint = TzPrint(area, ampm);
            if (!sortedlist.ContainsKey(areaprint)) sortedlist[areaprint] = [];
            sortedlist[areaprint].AddRange(users);
        }

        const int MaxSingleLineLength = 750;
        const int MaxSingleOutputLength = 3000;

        // Build zone listings with users
        var outputlines = new List<string>();
        foreach ((var area, List<ulong> users) in sortedlist) {
            var buffer = new StringBuilder();
            buffer.Append(area[6..] + ": ");
            var empty = true;
            foreach (var userid in users) {
                var userinstance = Context.Guild.GetUser(userid);
                if (userinstance == null) continue;
                if (empty) empty = !empty;
                else buffer.Append(", ");
                var useradd = FormatName(userinstance);
                if (buffer.Length + useradd.Length > MaxSingleLineLength) {
                    buffer.Append("others...");
                    break;
                } else buffer.Append(useradd);
            }
            if (!empty) outputlines.Add(buffer.ToString());
        }

        // Prepare for output - send buffers out if they become too large
        outputlines.Sort();
        var hasOutputOneLine = false;
        // First output is shown as an interaction response, followed then as regular channel messages
        async Task doOutput(Embed msg) {
            if (!hasOutputOneLine) {
                await RespondAsync(embed: msg);
                hasOutputOneLine = true;
            } else {
                await ReplyAsync(embed: msg);
            }
        }

        var resultout = new StringBuilder();
        foreach (var line in outputlines) {
            if (resultout.Length + line.Length > MaxSingleOutputLength) {
                await doOutput(new EmbedBuilder().WithDescription(resultout.ToString()).Build());
                resultout.Clear();
            }
            if (resultout.Length > 0) resultout.AppendLine(); // avoids trailing newline by adding to the previous line
            resultout.Append(line);
        }
        if (resultout.Length > 0) {
            await doOutput(new EmbedBuilder().WithDescription(resultout.ToString()).Build());
        }
    }

    private async Task CmdListWithUserParamAsync(SocketGuildUser parameter) {
        // Called by CmdList
        using var db = DbContext;
        var result = db.GetUserZone(parameter);
        if (result == null) {
            var isself = Context.User.Id == parameter.Id;
            if (isself) await RespondAsync(":x: You do not have a time zone. Set it with `tz.set`.", ephemeral: true);
            else await RespondAsync(":x: The given user does not have a time zone set.", ephemeral: true);
            return;
        }

        var ampm = db.GuildSettings.Where(s => s.GuildId == Context.Guild.Id).SingleOrDefault()?.Use12HourTime ?? false;
        var resulttext = TzPrint(result, ampm)[6..] + ": " + FormatName(parameter);
        await RespondAsync(embed: new EmbedBuilder().WithDescription(resulttext).Build());
    }

    [SlashCommand("set", HelpSet)]
    [CommandContextType(InteractionContextType.Guild)]
    public async Task CmdSet([Summary(description: "The new time zone to set.")]string zone) {
        var parsedzone = ParseTimeZone(zone);
        if (parsedzone == null) {
            await RespondAsync(ErrInvalidZone, ephemeral: true);
            return;
        }
        using var db = DbContext;
        db.UpdateUser((SocketGuildUser)Context.User, parsedzone);
        await RespondAsync($":white_check_mark: Your time zone has been set to **{parsedzone}**.",
            ephemeral: db.GuildSettings.Where(r => r.GuildId == Context.Guild.Id).SingleOrDefault()?.EphemeralConfirm ?? false)
            .ConfigureAwait(false);
    }

    [SlashCommand("remove", HelpRemove)]
    [CommandContextType(InteractionContextType.Guild)]
    public async Task CmdRemove() {
        using var db = DbContext;
        var success = db.DeleteUser((SocketGuildUser)Context.User);
        if (success) await RespondAsync(":white_check_mark: Your zone has been removed.",
                ephemeral: db.GuildSettings.Where(r => r.GuildId == Context.Guild.Id).SingleOrDefault()?.EphemeralConfirm ?? false)
                .ConfigureAwait(false);
        else await RespondAsync(":x: You don't have a time zone set.",
                ephemeral: db.GuildSettings.Where(r => r.GuildId == Context.Guild.Id).SingleOrDefault()?.EphemeralConfirm ?? false)
                .ConfigureAwait(false);
    }
}