Move member random to Random.cs, add group random and group member random

2020-11-22 11:58:23 -05:00
parent 45b87bd132
commit e3274d5d26
6 changed files with 92 additions and 28 deletions
--- a/PluralKit.Bot/Commands/Random.cs
+++ b/PluralKit.Bot/Commands/Random.cs
@@ -0,0 +1,79 @@
+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)));
+        }
+    }
+}