feat: go through some TODOs

2021-11-26 22:02:58 -05:00
parent 04d78e3348
commit 4450ae4214
21 changed files with 193 additions and 250 deletions
--- a/PluralKit.Bot/Commands/GroupMember.cs
+++ b/PluralKit.Bot/Commands/GroupMember.cs
@@ -0,0 +1,157 @@
+using System.Text;
+
+using Myriad.Builders;
+
+using PluralKit.Core;
+
+namespace PluralKit.Bot;
+
+public class GroupMember
+{
+    private readonly IDatabase _db;
+    private readonly ModelRepository _repo;
+
+    public GroupMember(IDatabase db, ModelRepository repo)
+    {
+        _db = db;
+        _repo = repo;
+    }
+
+    public async Task AddRemoveGroups(Context ctx, PKMember target, Groups.AddRemoveOperation op)
+    {
+        ctx.CheckSystem().CheckOwnMember(target);
+
+        var groups = (await ctx.ParseGroupList(ctx.System.Id))
+            .Select(g => g.Id)
+            .Distinct()
+            .ToList();
+
+        var existingGroups = (await _repo.GetMemberGroups(target.Id).ToListAsync())
+            .Select(g => g.Id)
+            .Distinct()
+            .ToList();
+
+        List<GroupId> toAction;
+
+        if (op == Groups.AddRemoveOperation.Add)
+        {
+            toAction = groups
+                .Where(group => !existingGroups.Contains(group))
+                .ToList();
+
+            await _repo.AddGroupsToMember(target.Id, toAction);
+        }
+        else if (op == Groups.AddRemoveOperation.Remove)
+        {
+            toAction = groups
+                .Where(group => existingGroups.Contains(group))
+                .ToList();
+
+            await _repo.RemoveGroupsFromMember(target.Id, toAction);
+        }
+        else
+        {
+            return; // otherwise toAction "may be unassigned"
+        }
+
+        await ctx.Reply(GroupMemberUtils.GenerateResponse(op, 1, groups.Count, toAction.Count,
+            groups.Count - toAction.Count));
+    }
+
+    public async Task ListMemberGroups(Context ctx, PKMember target)
+    {
+        var pctx = ctx.LookupContextFor(target.System);
+
+        var groups = await _repo.GetMemberGroups(target.Id)
+            .Where(g => g.Visibility.CanAccess(pctx))
+            .OrderBy(g => g.Name, StringComparer.InvariantCultureIgnoreCase)
+            .ToListAsync();
+
+        var description = "";
+        var msg = "";
+
+        if (groups.Count == 0)
+            description = "This member has no groups.";
+        else
+            description = string.Join("\n", groups.Select(g => $"[`{g.Hid}`] **{g.DisplayName ?? g.Name}**"));
+
+        if (pctx == LookupContext.ByOwner)
+        {
+            msg +=
+                $"\n\nTo add this member to one or more groups, use `pk;m {target.Reference()} group add <group> [group 2] [group 3...]`";
+            if (groups.Count > 0)
+                msg +=
+                    $"\nTo remove this member from one or more groups, use `pk;m {target.Reference()} group remove <group> [group 2] [group 3...]`";
+        }
+
+        await ctx.Reply(msg, new EmbedBuilder().Title($"{target.Name}'s groups").Description(description).Build());
+    }
+
+    public async Task AddRemoveMembers(Context ctx, PKGroup target, Groups.AddRemoveOperation op)
+    {
+        ctx.CheckOwnGroup(target);
+
+        var members = (await ctx.ParseMemberList(ctx.System.Id))
+            .Select(m => m.Id)
+            .Distinct()
+            .ToList();
+
+        var existingMembersInGroup = (await _db.Execute(conn => conn.QueryMemberList(target.System,
+                new DatabaseViewsExt.MemberListQueryOptions { GroupFilter = target.Id })))
+            .Select(m => m.Id.Value)
+            .Distinct()
+            .ToHashSet();
+
+        List<MemberId> toAction;
+
+        if (op == Groups.AddRemoveOperation.Add)
+        {
+            toAction = members
+                .Where(m => !existingMembersInGroup.Contains(m.Value))
+                .ToList();
+            await _repo.AddMembersToGroup(target.Id, toAction);
+        }
+        else if (op == Groups.AddRemoveOperation.Remove)
+        {
+            toAction = members
+                .Where(m => existingMembersInGroup.Contains(m.Value))
+                .ToList();
+            await _repo.RemoveMembersFromGroup(target.Id, toAction);
+        }
+        else
+        {
+            return; // otherwise toAction "may be undefined"
+        }
+
+        await ctx.Reply(GroupMemberUtils.GenerateResponse(op, members.Count, 1, toAction.Count,
+            members.Count - toAction.Count));
+    }
+
+    public async Task ListGroupMembers(Context ctx, PKGroup target)
+    {
+        var targetSystem = await GetGroupSystem(ctx, target);
+        ctx.CheckSystemPrivacy(targetSystem, target.ListPrivacy);
+
+        var opts = ctx.ParseMemberListOptions(ctx.LookupContextFor(target.System));
+        opts.GroupFilter = target.Id;
+
+        var title = new StringBuilder($"Members of {target.DisplayName ?? target.Name} (`{target.Hid}`) in ");
+        if (targetSystem.Name != null)
+            title.Append($"{targetSystem.Name} (`{targetSystem.Hid}`)");
+        else
+            title.Append($"`{targetSystem.Hid}`");
+        if (opts.Search != null)
+            title.Append($" matching **{opts.Search}**");
+
+        await ctx.RenderMemberList(ctx.LookupContextFor(target.System), target.System, title.ToString(),
+            target.Color, opts);
+    }
+
+    private async Task<PKSystem> GetGroupSystem(Context ctx, PKGroup target)
+    {
+        var system = ctx.System;
+        if (system?.Id == target.System)
+            return system;
+        return await _repo.GetSystem(target.System)!;
+    }
+}