PluralKit/PluralKit.Bot/Commands/Random.cs

using System.Linq;
using System.Threading.Tasks;

using PluralKit.Core;

namespace PluralKit.Bot
{
    public class Random
    {
        private readonly IDatabase _db;
        private readonly ModelRepository _repo;
        private readonly EmbedService _embeds;

        private readonly global::System.Random randGen = new global::System.Random();

        public Random(EmbedService embeds, IDatabase db, ModelRepository repo)
        {
            _embeds = embeds;
            _db = db;
            _repo = repo;
        }

        // todo: get postgresql to return one random member/group instead of querying all members/groups

        public async Task Member(Context ctx)
        {
            ctx.CheckSystem();

            var members = await _db.Execute(c =>
            {
                if (ctx.MatchFlag("all", "a"))
                    return _repo.GetSystemMembers(c, ctx.System.Id);
                return _repo.GetSystemMembers(c, ctx.System.Id)
                    .Where(m => m.MemberVisibility == PrivacyLevel.Public);
            }).ToListAsync();
            
            if (members == null || !members.Any())
                throw new PKError("Your system has no members! Please create at least one member before using this command.");

            var randInt = randGen.Next(members.Count);
            await ctx.Reply(embed: await _embeds.CreateMemberEmbed(ctx.System, members[randInt], ctx.Guild, ctx.LookupContextFor(ctx.System)));
        }

        public async Task Group(Context ctx)
        {
            ctx.CheckSystem();

            var groups = await _db.Execute(c => c.QueryGroupList(ctx.System.Id));
            if (!ctx.MatchFlag("all", "a"))
                groups = groups.Where(g => g.Visibility == PrivacyLevel.Public);

            if (groups == null || !groups.Any())
                throw new PKError("Your system has no groups! Please create at least one group before using this command.");

            var randInt = randGen.Next(groups.Count());
            await ctx.Reply(embed: await _embeds.CreateGroupEmbed(ctx, ctx.System, groups.ToArray()[randInt]));
        }

        public async Task GroupMember(Context ctx, PKGroup group)
        {
            var opts = ctx.ParseMemberListOptions(ctx.LookupContextFor(group.System));
            opts.GroupFilter = group.Id;

            await using var conn = await _db.Obtain();
            var members = await conn.QueryMemberList(ctx.System.Id, opts.ToQueryOptions());

            if (members == null || !members.Any())
                throw new PKError("This group has no members! Please add at least one member to this group before using this command.");

            if (!ctx.MatchFlag("all", "a"))
                members = members.Where(g => g.MemberVisibility == PrivacyLevel.Public);

            var ms = members.ToList();

            var randInt = randGen.Next(ms.Count);
            await ctx.Reply(embed: await _embeds.CreateMemberEmbed(ctx.System, ms[randInt], ctx.Guild, ctx.LookupContextFor(ctx.System)));
        }
    }
}