feat: premium ID changes

PluralKit.Bot/CommandMeta/CommandHelp.cs

@@ -22,6 +22,7 @@ public partial class CommandTree
     public static Command SystemFrontHistory = new Command("system fronthistory", "system [system] fronthistory", "Shows a system's front history");
     public static Command SystemFrontPercent = new Command("system frontpercent", "system [system] frontpercent [timespan]", "Shows a system's front breakdown");
     public static Command SystemId = new Command("system id", "system [system] id", "Prints your system's id.");
+    public static Command SystemIdChange = new Command("system changeid", "system [system] changeid <id>", "PREMIUM: Changes your system's ID.");
     public static Command SystemPrivacy = new Command("system privacy", "system [system] privacy <name|avatar|description|members|fronter|fronthistory|all> <public|private>", "Changes your system's privacy settings");
     public static Command ConfigTimezone = new Command("config timezone", "config timezone [timezone]", "Changes your system's time zone");
     public static Command ConfigPing = new Command("config ping", "config ping [on|off]", "Changes your system's ping preferences");
@@ -61,6 +62,7 @@ public partial class CommandTree
     public static Command MemberServerKeepProxy = new Command("member server keepproxy", "member <member> serverkeepproxy [on|off|clear]", "Sets whether to include a member's proxy tags when proxying in the current server.");
     public static Command MemberRandom = new Command("system random", "system [system] random", "Shows the info card of a randomly selected member in a system.");
     public static Command MemberId = new Command("member id", "member [member] id", "Prints a member's id.");
+    public static Command MemberIdChange = new Command("member changeid", "member [member] changeid <id>", "PREMIUM: Changes a member's ID.");
     public static Command MemberPrivacy = new Command("member privacy", "member <member> privacy <name|description|birthday|pronouns|proxy|metadata|visibility|all> <public|private>", "Changes a members's privacy settings");
     public static Command GroupInfo = new Command("group", "group <name>", "Looks up information about a group");
     public static Command GroupNew = new Command("group new", "group new <name>", "Creates a new group");
@@ -73,6 +75,7 @@ public partial class CommandTree
     public static Command GroupAdd = new Command("group add", "group <group> add <member> [member 2] [member 3...]", "Adds one or more members to a group");
     public static Command GroupRemove = new Command("group remove", "group <group> remove <member> [member 2] [member 3...]", "Removes one or more members from a group");
     public static Command GroupId = new Command("group id", "group [group] id", "Prints a group's id.");
+    public static Command GroupIdChange = new Command("group changeid", "group [group] changeid <id>", "PREMIUM: Changes a group's ID.");
     public static Command GroupPrivacy = new Command("group privacy", "group <group> privacy <name|description|icon|metadata|visibility|all> <public|private>", "Changes a group's privacy settings");
     public static Command GroupBannerImage = new Command("group banner", "group <group> banner [url]", "Set the group's banner image");
     public static Command GroupIcon = new Command("group icon", "group <group> icon [url|@mention]", "Changes a group's icon");

PluralKit.Bot/CommandMeta/CommandTree.cs

@@ -184,6 +184,8 @@ public partial class CommandTree
             await ctx.Execute<Admin>(Admin, a => a.SystemDelete(ctx));
         else if (ctx.Match("pe", "premiumexpiry", "premium"))
             await ctx.Execute<Admin>(Admin, a => a.PremiumExpiry(ctx));
+        else if (ctx.Match("pid", "premiumidchange", "premiumid"))
+            await ctx.Execute<Admin>(Admin, a => a.PremiumIdChangeAllowance(ctx));
         else if (ctx.Match("sendmsg", "sendmessage"))
             await ctx.Execute<Admin>(Admin, a => a.SendAdminMessage(ctx));
         else if (ctx.Match("al", "abuselog"))
@@ -320,6 +322,8 @@ public partial class CommandTree
             await ctx.CheckSystem(target).Execute<SystemEdit>(SystemDelete, m => m.Delete(ctx, target));
         else if (ctx.Match("id"))
             await ctx.CheckSystem(target).Execute<System>(SystemId, m => m.DisplayId(ctx, target));
+        else if (ctx.Match("changeid", "updateid"))
+            await ctx.CheckSystem(target).Execute<SystemEdit>(SystemIdChange, m => m.ChangeId(ctx, target));
         else if (ctx.Match("random", "rand", "r"))
             if (ctx.Match("group", "g") || ctx.MatchFlag("group", "g"))
                 await ctx.CheckSystem(target).Execute<Random>(GroupRandom, r => r.Group(ctx, target));
@@ -395,6 +399,8 @@ public partial class CommandTree
             await ctx.Execute<MemberEdit>(MemberServerKeepProxy, m => m.ServerKeepProxy(ctx, target));
         else if (ctx.Match("id"))
             await ctx.Execute<Member>(MemberId, m => m.DisplayId(ctx, target));
+        else if (ctx.Match("changeid", "updateid"))
+            await ctx.Execute<MemberEdit>(MemberIdChange, m => m.ChangeId(ctx, target));
         else if (ctx.Match("privacy"))
             await ctx.Execute<MemberEdit>(MemberPrivacy, m => m.Privacy(ctx, target, null));
         else if (ctx.Match("private", "hidden", "hide"))
@@ -457,6 +463,8 @@ public partial class CommandTree
                 await ctx.Execute<Groups>(GroupColor, g => g.GroupColor(ctx, target));
             else if (ctx.Match("id"))
                 await ctx.Execute<Groups>(GroupId, g => g.DisplayId(ctx, target));
+            else if (ctx.Match("changeid", "updateid"))
+                await ctx.Execute<Groups>(GroupIdChange, g => g.ChangeId(ctx, target));
             else if (!ctx.HasNext())
                 await ctx.Execute<Groups>(GroupInfo, g => g.ShowGroupCard(ctx, target));
             else

PluralKit.Bot/Commands/Admin.cs

@@ -416,12 +416,15 @@ public class Admin
         if (target == null)
             throw new PKError("Unknown system.");
 
-        await ctx.Reply(null, await CreateEmbed(ctx, target));
         if (!ctx.HasNext())
+        {
+            await ctx.Reply(null, await CreateEmbed(ctx, target));
             return;
+        }
 
         if (ctx.Match("lifetime", "staff"))
         {
+            await ctx.Reply(null, await CreateEmbed(ctx, target));
             if (!await ctx.PromptYesNo($"Grant system `{target.Hid}` lifetime premium?", "Grant"))
                 throw new PKError("Premium entitlement change cancelled.");
 
@@ -434,6 +437,7 @@ public class Admin
         }
         else if (ctx.Match("none", "clear"))
         {
+            await ctx.Reply(null, await CreateEmbed(ctx, target));
             if (!await ctx.PromptYesNo($"Clear premium entitlements for system `{target.Hid}`?", "Clear"))
                 throw new PKError("Premium entitlement change cancelled.");
 
@@ -465,6 +469,7 @@ public class Admin
                 time = result.Value.ToInstant();
             }
 
+            await ctx.Reply(null, await CreateEmbed(ctx, target));
             if (!await ctx.PromptYesNo($"Change premium expiry for system `{target.Hid}` to <t:{time?.ToUnixTimeSeconds()}>?", "Change"))
                 throw new PKError("Premium entitlement change cancelled.");
 
@@ -477,6 +482,37 @@ public class Admin
         }
     }
 
+    public async Task PremiumIdChangeAllowance(Context ctx)
+    {
+        ctx.AssertBotAdmin();
+
+        var target = await ctx.MatchSystem();
+        if (target == null)
+            throw new PKError("Unknown system.");
+
+        if (!ctx.HasNext())
+        {
+            await ctx.Reply(null, await CreateEmbed(ctx, target));
+            return;
+        }
+
+        var config = await ctx.Repository.GetSystemConfig(target.Id);
+        var newAllowanceStr = ctx.PopArgument().ToLower().Replace(",", null).Replace("k", "000");
+        if (!int.TryParse(newAllowanceStr, out var newAllowance))
+            throw new PKError($"Couldn't parse `{newAllowanceStr}` as number.");
+
+        await ctx.Reply(null, await CreateEmbed(ctx, target));
+        if (!await ctx.PromptYesNo($"Update premium ID change allowance from **{config.PremiumIdChangesRemaining}** to **{newAllowance}**?", "Update"))
+            throw new PKError("ID change allowance cancelled.");
+
+        await ctx.Repository.UpdateSystemConfig(target.Id, new SystemConfigPatch
+        {
+            PremiumIdChangesRemaining = newAllowance,
+        });
+
+        await ctx.Reply($"{Emojis.Success} Premium entitlement changed.");
+    }
+
     public async Task AbuseLogCreate(Context ctx)
     {
         var denyBotUsage = ctx.MatchFlag("deny", "deny-bot-usage");

PluralKit.Bot/Commands/Groups.cs

@@ -4,6 +4,8 @@ using System.Text.RegularExpressions;
 using Myriad.Builders;
 using Myriad.Types;
 
+using NodaTime;
+
 using Newtonsoft.Json.Linq;
 
 using PluralKit.Core;
@@ -99,6 +101,38 @@ public class Groups
         await ctx.Reply(replyStr, eb.Build());
     }
 
+    public async Task ChangeId(Context ctx, PKGroup target)
+    {
+        ctx.CheckSystem().CheckOwnGroup(target);
+        if (!ctx.Premium)
+            throw Errors.PremiumExclusiveCommand();
+
+        var input = ctx.PopArgument();
+        if (!input.TryParseHid(out var newHid))
+            throw new PKError($"Invalid new member ID `{input}`.");
+
+        var existingGroup = await ctx.Repository.GetGroupByHid(newHid);
+        if (existingGroup != null)
+            throw new PKError($"Another group already exists with ID `{newHid.DisplayHid(ctx.Config)}`.");
+
+        if (ctx.Config.PremiumIdChangesRemaining < 1)
+            throw new PKError("You do not have enough available ID changes to do this.");
+        if ((await ctx.Repository.GetHidChangelogCountForDate(ctx.System.Id, SystemClock.Instance.GetCurrentInstant().InUtc().Date)) >= Limits.PremiumDailyHidChanges)
+            throw new PKError($"You have already changed {Limits.PremiumDailyHidChanges} IDs today. Please try again tomorrow.");
+
+        if (!await ctx.PromptYesNo($"Change ID for group **{target.NameFor(ctx)}** (`{target.DisplayHid(ctx.Config)}`) to `{newHid.DisplayHid(ctx.Config)}`?", "Change"))
+            throw new PKError("ID change cancelled.");
+
+        if (!await ctx.Repository.TryUpdateSystemConfigForIdChange(ctx.System.Id))
+            throw new PKError("You do not have enough available ID changes to do this.");
+
+        await ctx.Repository.CreateHidChangelog(ctx.System.Id, ctx.Message.Author.Id, "group", target.Hid, newHid);
+        await ctx.Repository.UpdateGroup(target.Id, new GroupPatch { Hid = newHid });
+
+        var newConfig = await ctx.Repository.GetSystemConfig(ctx.System.Id);
+        await ctx.Reply($"{Emojis.Success} Group ID changed to `{newHid.DisplayHid(ctx.Config)}`. You have **{newConfig.PremiumIdChangesRemaining}** ID changes remaining.");
+    }
+
     public async Task RenameGroup(Context ctx, PKGroup target)
     {
         ctx.CheckOwnGroup(target);

PluralKit.Bot/Commands/MemberEdit.cs

@@ -21,6 +21,38 @@ public class MemberEdit
         _avatarHosting = avatarHosting;
     }
 
+    public async Task ChangeId(Context ctx, PKMember target)
+    {
+        ctx.CheckSystem().CheckOwnMember(target);
+        if (!ctx.Premium)
+            throw Errors.PremiumExclusiveCommand();
+
+        var input = ctx.PopArgument();
+        if (!input.TryParseHid(out var newHid))
+            throw new PKError($"Invalid new member ID `{input}`.");
+
+        var existingMember = await ctx.Repository.GetMemberByHid(newHid);
+        if (existingMember != null)
+            throw new PKError($"Another member already exists with ID `{newHid.DisplayHid(ctx.Config)}`.");
+
+        if (ctx.Config.PremiumIdChangesRemaining < 1)
+            throw new PKError("You do not have enough available ID changes to do this.");
+        if ((await ctx.Repository.GetHidChangelogCountForDate(ctx.System.Id, SystemClock.Instance.GetCurrentInstant().InUtc().Date)) >= Limits.PremiumDailyHidChanges)
+            throw new PKError($"You have already changed {Limits.PremiumDailyHidChanges} IDs today. Please try again tomorrow.");
+
+        if (!await ctx.PromptYesNo($"Change ID  for member **{target.NameFor(ctx)}** (`{target.DisplayHid(ctx.Config)}`) to `{newHid.DisplayHid(ctx.Config)}`?", "Change"))
+            throw new PKError("ID change cancelled.");
+
+        if (!await ctx.Repository.TryUpdateSystemConfigForIdChange(ctx.System.Id))
+            throw new PKError("You do not have enough available ID changes to do this.");
+
+        await ctx.Repository.CreateHidChangelog(ctx.System.Id, ctx.Message.Author.Id, "member", target.Hid, newHid);
+        await ctx.Repository.UpdateMember(target.Id, new MemberPatch { Hid = newHid });
+
+        var newConfig = await ctx.Repository.GetSystemConfig(ctx.System.Id);
+        await ctx.Reply($"{Emojis.Success} Member ID changed to `{newHid.DisplayHid(ctx.Config)}`. You have **{newConfig.PremiumIdChangesRemaining}** ID changes remaining.");
+    }
+
     public async Task Name(Context ctx, PKMember target)
     {
         var format = ctx.MatchFormat();

PluralKit.Bot/Commands/Misc.cs

@@ -54,7 +54,57 @@ public class Misc
             }
         }
 
-        await ctx.Reply(message + $"\n\nManage your subscription at <{_botConfig.PremiumDashboardUrl}>");
+        List<MessageComponent> components = [
+            new MessageComponent()
+            {
+                Type = ComponentType.Text,
+                Content = message,
+            },
+        ];
+
+        if (ctx.Premium)
+        {
+            var hidChangesLeftToday = Limits.PremiumDailyHidChanges - await ctx.Repository.GetHidChangelogCountForDate(ctx.System.Id, SystemClock.Instance.GetCurrentInstant().InUtc().Date);
+            var limitMessage = $"You have **{ctx.Config.PremiumIdChangesRemaining}** ID changes available, of which you can use **{hidChangesLeftToday}** today.";
+
+            components.Add(new()
+            {
+                Type = ComponentType.Separator,
+            });
+            components.Add(new()
+            {
+                Type = ComponentType.Text,
+                Content = limitMessage,
+            });
+        }
+
+        await ctx.Reply(components: [
+            new()
+            {
+                Type = ComponentType.Container,
+                Components = [
+                    new()
+                    {
+                        Type = ComponentType.Text,
+                        Content = $"## {(_botConfig.PremiumSubscriberEmoji != null ? $"<:premium_subscriber:{_botConfig.PremiumSubscriberEmoji}>" : "\u2729")} PluralKit Premium",
+                    },
+                    ..components,
+                ],
+            },
+            new()
+            {
+                Type = ComponentType.ActionRow,
+                Components = [
+                    new()
+                    {
+                        Type = ComponentType.Button,
+                        Style = ButtonStyle.Link,
+                        Label = "Manage your subscription",
+                        Url = _botConfig.PremiumDashboardUrl,
+                    },
+                ],
+            },
+        ]);
     }
 
     public async Task Invite(Context ctx)

PluralKit.Bot/Commands/SystemEdit.cs

@@ -9,6 +9,8 @@ using Myriad.Types;
 
 using Newtonsoft.Json;
 
+using NodaTime;
+
 using PluralKit.Core;
 using SqlKata.Compilers;
 
@@ -20,13 +22,47 @@ public class SystemEdit
     private readonly DataFileService _dataFiles;
     private readonly PrivateChannelService _dmCache;
     private readonly AvatarHostingService _avatarHosting;
+    private readonly BotConfig _botConfig;
 
-    public SystemEdit(DataFileService dataFiles, HttpClient client, PrivateChannelService dmCache, AvatarHostingService avatarHosting)
+    public SystemEdit(DataFileService dataFiles, HttpClient client, PrivateChannelService dmCache, AvatarHostingService avatarHosting, BotConfig botConfig)
     {
         _dataFiles = dataFiles;
         _client = client;
         _dmCache = dmCache;
         _avatarHosting = avatarHosting;
+        _botConfig = botConfig;
+    }
+
+    public async Task ChangeId(Context ctx, PKSystem target)
+    {
+        ctx.CheckSystem().CheckOwnSystem(target);
+        if (!ctx.Premium)
+            throw Errors.PremiumExclusiveCommand();
+
+        var input = ctx.PopArgument();
+        if (!input.TryParseHid(out var newHid))
+            throw new PKError($"Invalid new system ID `{input}`.");
+
+        var existingSystem = await ctx.Repository.GetSystemByHid(newHid);
+        if (existingSystem != null)
+            throw new PKError($"Another system already exists with ID `{newHid.DisplayHid(ctx.Config)}`.");
+
+        if (ctx.Config.PremiumIdChangesRemaining < 1)
+            throw new PKError("You do not have enough available ID changes to do this.");
+        if ((await ctx.Repository.GetHidChangelogCountForDate(target.Id, SystemClock.Instance.GetCurrentInstant().InUtc().Date)) >= Limits.PremiumDailyHidChanges)
+            throw new PKError($"You have already changed {Limits.PremiumDailyHidChanges} IDs today. Please try again tomorrow.");
+
+        if (!await ctx.PromptYesNo($"Change your system ID to `{newHid.DisplayHid(ctx.Config)}`?", "Change"))
+            throw new PKError("ID change cancelled.");
+
+        if (!await ctx.Repository.TryUpdateSystemConfigForIdChange(target.Id))
+            throw new PKError("You do not have enough available ID changes to do this.");
+
+        await ctx.Repository.CreateHidChangelog(target.Id, ctx.Message.Author.Id, "system", target.Hid, newHid);
+        await ctx.Repository.UpdateSystem(target.Id, new SystemPatch { Hid = newHid });
+
+        var newConfig = await ctx.Repository.GetSystemConfig(target.Id);
+        await ctx.Reply($"{Emojis.Success} System ID changed to `{newHid.DisplayHid(ctx.Config)}`. You have **{newConfig.PremiumIdChangesRemaining}** ID changes remaining.");
     }
 
     public async Task Name(Context ctx, PKSystem target)

PluralKit.Bot/Errors.cs

@@ -185,4 +185,6 @@ public static class Errors
         new($"Channel \"{channelString}\" not found or is not in this server.");
 
     public static PKError InteractionWrongAccount(ulong user) => new($"This prompt is only available for <@{user}>");
+
+    public static PKError PremiumExclusiveCommand() => new("This command is only available for PluralKit Premium subscribers.");
 }

PluralKit.Bot/Utils/ModelUtils.cs

@@ -31,6 +31,7 @@ public static class ModelUtils
     public static string DisplayHid(this PKSystem system, SystemConfig? cfg = null, bool isList = false) => HidTransform(system.Hid, cfg, isList);
     public static string DisplayHid(this PKGroup group, SystemConfig? cfg = null, bool isList = false, bool shouldPad = true) => HidTransform(group.Hid, cfg, isList, shouldPad);
     public static string DisplayHid(this PKMember member, SystemConfig? cfg = null, bool isList = false, bool shouldPad = true) => HidTransform(member.Hid, cfg, isList, shouldPad);
+    public static string DisplayHid(this string hid, SystemConfig? cfg = null, bool isList = false, bool shouldPad = true) => HidTransform(hid, cfg, isList, shouldPad);
     private static string HidTransform(string hid, SystemConfig? cfg = null, bool isList = false, bool shouldPad = true) =>
         HidUtils.HidTransform(
             hid,