Add pk;member group

PluralKit.Bot/Commands/CommandTree.cs

@@ -39,6 +39,9 @@ namespace PluralKit.Bot
         public static Command MemberProxy = new Command("member proxy", "member <member> proxy [add|remove] [example proxy]", "Changes, adds, or removes a member's proxy tags");
         public static Command MemberDelete = new Command("member delete", "member <member> delete", "Deletes a member");
         public static Command MemberAvatar = new Command("member avatar", "member <member> avatar [url|@mention]", "Changes a member's avatar");
+        public static Command MemberGroups = new Command("member group", "member <member> group", "Shows the groups a member is in");
+        public static Command MemberGroupAdd = new Command("member group", "member <member> group add <group> [group 2] [group 3...]", "Adds a member to one or more groups");
+        public static Command MemberGroupRemove = new Command("member group", "member <member> group remove <group> [group 2] [group 3...]", "Removes a member from one or more groups");
         public static Command MemberServerAvatar = new Command("member serveravatar", "member <member> serveravatar [url|@mention]", "Changes a member's avatar in the current server");
         public static Command MemberDisplayName = new Command("member displayname", "member <member> displayname [display name]", "Changes a member's display name");
         public static Command MemberServerName = new Command("member servername", "member <member> servername [server name]", "Changes a member's display name in the current server");
@@ -89,8 +92,8 @@ namespace PluralKit.Bot
 
         public static Command[] MemberCommands = {
             MemberInfo, MemberNew, MemberRename, MemberDisplayName, MemberServerName, MemberDesc, MemberPronouns,
-            MemberColor, MemberBirthday, MemberProxy, MemberKeepProxy, MemberDelete, MemberAvatar, MemberServerAvatar, MemberPrivacy,
-            MemberRandom
+            MemberColor, MemberBirthday, MemberProxy, MemberKeepProxy, MemberGroups, MemberGroupAdd, MemberGroupRemove,
+            MemberDelete, MemberAvatar, MemberServerAvatar, MemberPrivacy, MemberRandom
         };
 
         public static Command[] GroupCommands =
@@ -328,6 +331,13 @@ namespace PluralKit.Bot
                 await ctx.Execute<MemberEdit>(MemberDelete, m => m.Delete(ctx, target));
             else if (ctx.Match("avatar", "profile", "picture", "icon", "image", "pfp", "pic"))
                 await ctx.Execute<MemberAvatar>(MemberAvatar, m => m.Avatar(ctx, target));
+            else if (ctx.Match("group", "groups"))
+                if (ctx.Match("add", "a"))
+                    await ctx.Execute<MemberGroup>(MemberGroupAdd, m => m.AddRemove(ctx, target, Groups.AddRemoveOperation.Add));
+                else if (ctx.Match("remove", "rem"))
+                    await ctx.Execute<MemberGroup>(MemberGroupRemove, m => m.AddRemove(ctx, target, Groups.AddRemoveOperation.Remove));
+                else 
+                    await ctx.Execute<MemberGroup>(MemberGroups, m => m.List(ctx, target));
             else if (ctx.Match("serveravatar", "servericon", "serverimage", "serverpfp", "serverpic", "savatar", "spic", "guildavatar", "guildpic", "guildicon", "sicon"))
                 await ctx.Execute<MemberAvatar>(MemberServerAvatar, m => m.ServerAvatar(ctx, target));
             else if (ctx.Match("displayname", "dn", "dname", "nick", "nickname", "dispname"))

PluralKit.Bot/Commands/MemberGroup.cs

@@ -0,0 +1,102 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Threading.Tasks;
+
+using DSharpPlus.Entities;
+
+using PluralKit.Core;
+
+namespace PluralKit.Bot
+{
+    public class MemberGroup
+    {
+        private readonly IDatabase _db;
+        private readonly ModelRepository _repo;
+
+        public MemberGroup(IDatabase db, ModelRepository repo)
+        {
+            _db = db;
+            _repo = repo;
+        }
+
+        private String groupTerm(int groups) => groups == 1 ? "group" : "groups";
+        
+        private String Response(List<GroupId> groupList, List<GroupId> actionedOn, Groups.AddRemoveOperation op)
+        {
+            var opStr = op == Groups.AddRemoveOperation.Add ? "added to" : "removed from";
+            var inStr = op == Groups.AddRemoveOperation.Add ? "in" : "not in";
+            var notActionedOn = groupList.Count - actionedOn.Count;
+
+            if (notActionedOn == 0)
+                return $"{Emojis.Success} Member {opStr} {groupTerm(actionedOn.Count)}.";
+            else
+                return $"{Emojis.Success} Member {opStr} {actionedOn.Count} {groupTerm(actionedOn.Count)} (member already {inStr} {notActionedOn} {groupTerm(notActionedOn)}).";
+        }
+
+        public async Task AddRemove(Context ctx, PKMember target, Groups.AddRemoveOperation op)
+        {
+            ctx.CheckSystem().CheckOwnMember(target);
+
+            var groups = (await ctx.ParseGroupList(ctx.System.Id))
+                .Select(g => g.Id)
+                .ToList();
+
+            await using var conn = await _db.Obtain();
+            var existingGroups = (await _repo.GetMemberGroups(conn, target.Id).ToListAsync())
+                .Select(g => g.Id)
+                .ToList();
+
+            List<GroupId> toAction;
+
+            if (op == Groups.AddRemoveOperation.Add)
+            {
+                toAction = groups
+                    .Where(group => !existingGroups.Contains(group))
+                    .ToList();
+
+                await _repo.AddGroupsToMember(conn, target.Id, toAction);
+            }
+            else if (op == Groups.AddRemoveOperation.Remove)
+            {
+                toAction = groups
+                    .Where(group => existingGroups.Contains(group))
+                    .ToList();
+
+                await _repo.RemoveGroupsFromMember(conn, target.Id, toAction);
+            }
+            else return; // otherwise toAction "may be unassigned"
+
+            await ctx.Reply(Response(groups, toAction, op));
+        }
+
+        public async Task List(Context ctx, PKMember target)
+        {
+            await using var conn = await _db.Obtain();
+
+            var pctx = ctx.LookupContextFor(target.System);
+
+            var groups = await _repo.GetMemberGroups(conn, 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, embed: (new DiscordEmbedBuilder().WithTitle($"{target.Name}'s groups").WithDescription(description)).Build());
+        }
+    }
+}