using System.Text; using System.Text.RegularExpressions; using Myriad.Builders; using Myriad.Rest.Exceptions; using Myriad.Rest.Types; using Myriad.Rest.Types.Requests; using Myriad.Types; using Newtonsoft.Json; using PluralKit.Core; namespace PluralKit.Bot; public class SystemEdit { private readonly HttpClient _client; private readonly DataFileService _dataFiles; private readonly PrivateChannelService _dmCache; public SystemEdit(DataFileService dataFiles, HttpClient client, PrivateChannelService dmCache) { _dataFiles = dataFiles; _client = client; _dmCache = dmCache; } public async Task Name(Context ctx, PKSystem target) { var isOwnSystem = target.Id == ctx.System?.Id; var noNameSetMessage = $"{(isOwnSystem ? "Your" : "This")} system does not have a name set."; if (isOwnSystem) noNameSetMessage += " Type `pk;system name ` to set one."; if (ctx.MatchRaw()) { if (target.Name != null) await ctx.Reply($"```\n{target.Name}\n```"); else await ctx.Reply(noNameSetMessage); return; } if (!ctx.HasNext(false)) { if (target.Name != null) await ctx.Reply( $"{(isOwnSystem ? "Your" : "This")} system's name is currently **{target.Name}**." + (isOwnSystem ? " Type `pk;system name -clear` to clear it." : "") + $" Using {target.Name.Length}/{Limits.MaxSystemNameLength} characters."); else await ctx.Reply(noNameSetMessage); return; } ctx.CheckSystem().CheckOwnSystem(target); if (ctx.MatchClear() && await ctx.ConfirmClear("your system's name")) { await ctx.Repository.UpdateSystem(target.Id, new SystemPatch { Name = null }); await ctx.Reply($"{Emojis.Success} System name cleared."); } else { var newSystemName = ctx.RemainderOrNull(false).NormalizeLineEndSpacing(); if (newSystemName.Length > Limits.MaxSystemNameLength) throw Errors.StringTooLongError("System name", newSystemName.Length, Limits.MaxSystemNameLength); await ctx.Repository.UpdateSystem(target.Id, new SystemPatch { Name = newSystemName }); await ctx.Reply($"{Emojis.Success} System name changed (using {newSystemName.Length}/{Limits.MaxSystemNameLength} characters)."); } } public async Task Description(Context ctx, PKSystem target) { ctx.CheckSystemPrivacy(target.Id, target.DescriptionPrivacy); var isOwnSystem = target.Id == ctx.System?.Id; var noDescriptionSetMessage = "This system does not have a description set."; if (isOwnSystem) noDescriptionSetMessage += " To set one, type `pk;s description `."; if (ctx.MatchRaw()) { if (target.Description == null) await ctx.Reply(noDescriptionSetMessage); else await ctx.Reply($"```\n{target.Description}\n```"); return; } if (!ctx.HasNext(false)) { if (target.Description == null) await ctx.Reply(noDescriptionSetMessage); else await ctx.Reply(embed: new EmbedBuilder() .Title("System description") .Description(target.Description) .Footer(new Embed.EmbedFooter( "To print the description with formatting, type `pk;s description -raw`." + (isOwnSystem ? " To clear it, type `pk;s description -clear`. To change it, type `pk;s description `." : "") + $" Using {target.Description.Length}/{Limits.MaxDescriptionLength} characters.")) .Build()); return; } ctx.CheckSystem().CheckOwnSystem(target); if (ctx.MatchClear() && await ctx.ConfirmClear("your system's description")) { await ctx.Repository.UpdateSystem(target.Id, new SystemPatch { Description = null }); await ctx.Reply($"{Emojis.Success} System description cleared."); } else { var newDescription = ctx.RemainderOrNull(false).NormalizeLineEndSpacing(); if (newDescription.Length > Limits.MaxDescriptionLength) throw Errors.StringTooLongError("Description", newDescription.Length, Limits.MaxDescriptionLength); await ctx.Repository.UpdateSystem(target.Id, new SystemPatch { Description = newDescription }); await ctx.Reply($"{Emojis.Success} System description changed (using {newDescription.Length}/{Limits.MaxDescriptionLength} characters)."); } } public async Task Color(Context ctx, PKSystem target) { var isOwnSystem = ctx.System?.Id == target.Id; var matchedRaw = ctx.MatchRaw(); var matchedClear = ctx.MatchClear(); if (!isOwnSystem || !(ctx.HasNext() || matchedClear)) { if (target.Color == null) await ctx.Reply( "This system does not have a color set." + (isOwnSystem ? " To set one, type `pk;system color `." : "")); else if (matchedRaw) await ctx.Reply("```\n#" + target.Color + "\n```"); else await ctx.Reply(embed: new EmbedBuilder() .Title("System color") .Color(target.Color.ToDiscordColor()) .Thumbnail(new Embed.EmbedThumbnail($"https://fakeimg.pl/256x256/{target.Color}/?text=%20")) .Description( $"This system's color is **#{target.Color}**." + (isOwnSystem ? " To clear it, type `pk;s color -clear`." : "")) .Build()); return; } ctx.CheckSystem().CheckOwnSystem(target); if (matchedClear) { await ctx.Repository.UpdateSystem(target.Id, new SystemPatch { Color = Partial.Null() }); await ctx.Reply($"{Emojis.Success} System color cleared."); } else { var color = ctx.RemainderOrNull(); if (color.StartsWith("#")) color = color.Substring(1); if (!Regex.IsMatch(color, "^[0-9a-fA-F]{6}$")) throw Errors.InvalidColorError(color); await ctx.Repository.UpdateSystem(target.Id, new SystemPatch { Color = Partial.Present(color.ToLowerInvariant()) }); await ctx.Reply(embed: new EmbedBuilder() .Title($"{Emojis.Success} System color changed.") .Color(color.ToDiscordColor()) .Thumbnail(new Embed.EmbedThumbnail($"https://fakeimg.pl/256x256/{color}/?text=%20")) .Build()); } } public async Task Tag(Context ctx, PKSystem target) { var isOwnSystem = ctx.System?.Id == target.Id; var noTagSetMessage = isOwnSystem ? "You currently have no system tag set. To set one, type `pk;s tag `." : "This system currently has no system tag set."; if (ctx.MatchRaw()) { if (target.Tag == null) await ctx.Reply(noTagSetMessage); else await ctx.Reply($"```\n{target.Tag}\n```"); return; } if (!ctx.HasNext(false)) { if (target.Tag == null) await ctx.Reply(noTagSetMessage); else await ctx.Reply($"{(isOwnSystem ? "Your" : "This system's")} current system tag is {target.Tag.AsCode()}." + (isOwnSystem ? "To change it, type `pk;s tag `. To clear it, type `pk;s tag -clear`." : "")); return; } ctx.CheckSystem().CheckOwnSystem(target); if (ctx.MatchClear() && await ctx.ConfirmClear("your system's tag")) { await ctx.Repository.UpdateSystem(target.Id, new SystemPatch { Tag = null }); await ctx.Reply($"{Emojis.Success} System tag cleared."); } else { var newTag = ctx.RemainderOrNull(false).NormalizeLineEndSpacing(); if (newTag != null) if (newTag.Length > Limits.MaxSystemTagLength) throw Errors.StringTooLongError("System tag", newTag.Length, Limits.MaxSystemTagLength); await ctx.Repository.UpdateSystem(target.Id, new SystemPatch { Tag = newTag }); await ctx.Reply( $"{Emojis.Success} System tag changed (using {newTag.Length}/{Limits.MaxSystemTagLength} characters). Member names will now end with {newTag.AsCode()} when proxied."); } } public async Task ServerTag(Context ctx, PKSystem target) { ctx.CheckSystem().CheckOwnSystem(target).CheckGuildContext(); var setDisabledWarning = $"{Emojis.Warn} Your system tag is currently **disabled** in this server. No tag will be applied when proxying.\nTo re-enable the system tag in the current server, type `pk;s servertag -enable`."; var settings = await ctx.Repository.GetSystemGuild(ctx.Guild.Id, target.Id); async Task Show(bool raw = false) { if (settings.Tag != null) { if (raw) { await ctx.Reply($"```{settings.Tag}```"); return; } var msg = $"Your current system tag in '{ctx.Guild.Name}' is {settings.Tag.AsCode()}"; if (!settings.TagEnabled) msg += ", but it is currently **disabled**. To re-enable it, type `pk;s servertag -enable`."; else msg += ". To change it, type `pk;s servertag `. To clear it, type `pk;s servertag -clear`."; await ctx.Reply(msg); return; } if (!settings.TagEnabled) await ctx.Reply( $"Your global system tag is {target.Tag}, but it is **disabled** in this server. To re-enable it, type `pk;s servertag -enable`"); else await ctx.Reply( $"You currently have no system tag specific to the server '{ctx.Guild.Name}'. To set one, type `pk;s servertag `. To disable the system tag in the current server, type `pk;s servertag -disable`."); } async Task Set() { var newTag = ctx.RemainderOrNull(false); if (newTag != null && newTag.Length > Limits.MaxSystemTagLength) throw Errors.StringTooLongError("System server tag", newTag.Length, Limits.MaxSystemTagLength); await ctx.Repository.UpdateSystemGuild(target.Id, ctx.Guild.Id, new SystemGuildPatch { Tag = newTag }); await ctx.Reply( $"{Emojis.Success} System server tag changed (using {newTag.Length}/{Limits.MaxSystemTagLength} characters). Member names will now end with {newTag.AsCode()} when proxied in the current server '{ctx.Guild.Name}'."); if (!settings.TagEnabled) await ctx.Reply(setDisabledWarning); } async Task Clear() { await ctx.Repository.UpdateSystemGuild(target.Id, ctx.Guild.Id, new SystemGuildPatch { Tag = null }); await ctx.Reply( $"{Emojis.Success} System server tag cleared. Member names will now end with the global system tag, if there is one set."); if (!settings.TagEnabled) await ctx.Reply(setDisabledWarning); } async Task EnableDisable(bool newValue) { await ctx.Repository.UpdateSystemGuild(target.Id, ctx.Guild.Id, new SystemGuildPatch { TagEnabled = newValue }); await ctx.Reply(PrintEnableDisableResult(newValue, newValue != settings.TagEnabled)); } string PrintEnableDisableResult(bool newValue, bool changedValue) { var opStr = newValue ? "enabled" : "disabled"; var str = ""; if (!changedValue) str = $"{Emojis.Note} The system tag is already {opStr} in this server."; else str = $"{Emojis.Success} System tag {opStr} in this server."; if (newValue) { if (settings.TagEnabled) { if (settings.Tag == null) str += " However, you do not have a system tag specific to this server. Messages will be proxied using your global system tag, if there is one set."; else str += $" Your current system tag in '{ctx.Guild.Name}' is {settings.Tag.AsCode()}."; } else { if (settings.Tag != null) str += $" Member names will now end with the server-specific tag {settings.Tag.AsCode()} when proxied in the current server '{ctx.Guild.Name}'."; else str += " Member names will now end with the global system tag when proxied in the current server, if there is one set."; } } return str; } if (ctx.MatchClear() && await ctx.ConfirmClear("your system's server tag")) await Clear(); else if (ctx.Match("disable") || ctx.MatchFlag("disable")) await EnableDisable(false); else if (ctx.Match("enable") || ctx.MatchFlag("enable")) await EnableDisable(true); else if (ctx.MatchRaw()) await Show(true); else if (!ctx.HasNext(false)) await Show(); else await Set(); } public async Task Pronouns(Context ctx, PKSystem target) { ctx.CheckSystemPrivacy(target.Id, target.PronounPrivacy); var isOwnSystem = ctx.System.Id == target.Id; var noPronounsSetMessage = "This system does not have pronouns set."; if (isOwnSystem) noPronounsSetMessage += " To set some, type `pk;system pronouns `"; if (ctx.MatchRaw()) { if (target.Pronouns == null) await ctx.Reply(noPronounsSetMessage); else await ctx.Reply($"```\n{target.Pronouns}\n```"); return; } if (!ctx.HasNext(false)) { if (target.Pronouns == null) await ctx.Reply(noPronounsSetMessage); else await ctx.Reply($"{(isOwnSystem ? "Your" : "This system's")} current pronouns are **{target.Pronouns}**.\nTo print the pronouns with formatting, type `pk;system pronouns -raw`." + (isOwnSystem ? " To clear them, type `pk;system pronouns -clear`." : "") + $" Using {target.Pronouns.Length}/{Limits.MaxPronounsLength} characters."); return; } ctx.CheckSystem().CheckOwnSystem(target); if (ctx.MatchClear() && await ctx.ConfirmClear("your system's pronouns")) { await ctx.Repository.UpdateSystem(target.Id, new SystemPatch { Pronouns = null }); await ctx.Reply($"{Emojis.Success} System pronouns cleared."); } else { var newPronouns = ctx.RemainderOrNull(false).NormalizeLineEndSpacing(); if (newPronouns != null) if (newPronouns.Length > Limits.MaxPronounsLength) throw Errors.StringTooLongError("Pronouns", newPronouns.Length, Limits.MaxPronounsLength); await ctx.Repository.UpdateSystem(target.Id, new SystemPatch { Pronouns = newPronouns }); await ctx.Reply( $"{Emojis.Success} System pronouns changed (using {newPronouns.Length}/{Limits.MaxPronounsLength} characters)."); } } public async Task Avatar(Context ctx, PKSystem target) { async Task ClearIcon() { ctx.CheckOwnSystem(target); await ctx.Repository.UpdateSystem(target.Id, new SystemPatch { AvatarUrl = null }); await ctx.Reply($"{Emojis.Success} System icon cleared."); } async Task SetIcon(ParsedImage img) { ctx.CheckOwnSystem(target); await AvatarUtils.VerifyAvatarOrThrow(_client, img.Url); await ctx.Repository.UpdateSystem(target.Id, new SystemPatch { AvatarUrl = img.Url }); var msg = img.Source switch { AvatarSource.User => $"{Emojis.Success} System icon changed to {img.SourceUser?.Username}'s avatar!\n{Emojis.Warn} If {img.SourceUser?.Username} changes their avatar, the system icon will need to be re-set.", AvatarSource.Url => $"{Emojis.Success} System icon changed to the image at the given URL.", AvatarSource.Attachment => $"{Emojis.Success} System icon changed to attached image.\n{Emojis.Warn} If you delete the message containing the attachment, the system icon will stop working.", _ => throw new ArgumentOutOfRangeException() }; // The attachment's already right there, no need to preview it. var hasEmbed = img.Source != AvatarSource.Attachment; await (hasEmbed ? ctx.Reply(msg, new EmbedBuilder().Image(new Embed.EmbedImage(img.Url)).Build()) : ctx.Reply(msg)); } async Task ShowIcon() { if ((target.AvatarUrl?.Trim() ?? "").Length > 0) { var eb = new EmbedBuilder() .Title("System icon") .Image(new Embed.EmbedImage(target.AvatarUrl.TryGetCleanCdnUrl())); if (target.Id == ctx.System?.Id) eb.Description("To clear, use `pk;system icon clear`."); await ctx.Reply(embed: eb.Build()); } else { throw new PKSyntaxError( "This system does not have an icon set. Set one by attaching an image to this command, or by passing an image URL or @mention."); } } if (target != null && target?.Id != ctx.System?.Id) { await ShowIcon(); return; } if (ctx.MatchClear() && await ctx.ConfirmClear("your system's icon")) await ClearIcon(); else if (await ctx.MatchImage() is { } img) await SetIcon(img); else await ShowIcon(); } public async Task BannerImage(Context ctx, PKSystem target) { ctx.CheckSystemPrivacy(target.Id, target.DescriptionPrivacy); var isOwnSystem = target.Id == ctx.System?.Id; if (!ctx.HasNext() && ctx.Message.Attachments.Length == 0) { if ((target.BannerImage?.Trim() ?? "").Length > 0) { var eb = new EmbedBuilder() .Title("System banner image") .Image(new Embed.EmbedImage(target.BannerImage)); if (isOwnSystem) eb.Description("To clear, use `pk;system banner clear`."); await ctx.Reply(embed: eb.Build()); } else { throw new PKSyntaxError("This system does not have a banner image set." + (isOwnSystem ? "Set one by attaching an image to this command, or by passing an image URL or @mention." : "")); } return; } ctx.CheckSystem().CheckOwnSystem(target); if (ctx.MatchClear() && await ctx.ConfirmClear("your system's banner image")) { await ctx.Repository.UpdateSystem(target.Id, new SystemPatch { BannerImage = null }); await ctx.Reply($"{Emojis.Success} System banner image cleared."); } else if (await ctx.MatchImage() is { } img) { await AvatarUtils.VerifyAvatarOrThrow(_client, img.Url, true); await ctx.Repository.UpdateSystem(target.Id, new SystemPatch { BannerImage = img.Url }); var msg = img.Source switch { AvatarSource.Url => $"{Emojis.Success} System banner image changed to the image at the given URL.", AvatarSource.Attachment => $"{Emojis.Success} System banner image changed to attached image.\n{Emojis.Warn} If you delete the message containing the attachment, the banner image will stop working.", AvatarSource.User => throw new PKError("Cannot set a banner image to an user's avatar."), _ => throw new ArgumentOutOfRangeException() }; // The attachment's already right there, no need to preview it. var hasEmbed = img.Source != AvatarSource.Attachment; await (hasEmbed ? ctx.Reply(msg, new EmbedBuilder().Image(new Embed.EmbedImage(img.Url)).Build()) : ctx.Reply(msg)); } } public async Task Delete(Context ctx, PKSystem target) { ctx.CheckSystem().CheckOwnSystem(target); await ctx.Reply( $"{Emojis.Warn} Are you sure you want to delete your system? If so, reply to this message with your system's ID (`{target.Hid}`).\n" + $"**Note: this action is permanent,** but you will get a copy of your system's data that can be re-imported into PluralKit at a later date sent to you in DMs." + " If you don't want this to happen, use `pk;s delete -no-export` instead."); if (!await ctx.ConfirmWithReply(target.Hid)) throw new PKError( $"System deletion cancelled. Note that you must reply with your system ID (`{target.Hid}`) *verbatim*."); // If the user confirms the deletion, export their system and send them the export file before actually // deleting their system, unless they specifically tell us not to do an export. var noExport = ctx.MatchFlag("ne", "no-export"); if (!noExport) { var json = await ctx.BusyIndicator(async () => { // Make the actual data file var data = await _dataFiles.ExportSystem(ctx.System); return JsonConvert.SerializeObject(data, Formatting.None); }); var stream = new MemoryStream(Encoding.UTF8.GetBytes(json)); try { var dm = await _dmCache.GetOrCreateDmChannel(ctx.Author.Id); var msg = await ctx.Rest.CreateMessage(dm, new MessageRequest { Content = $"{Emojis.Success} System deleted. If you want to set up your PluralKit system again, you can import the file below with `pk;import`." }, new[] { new MultipartFile("system.json", stream, null, null, null) }); await ctx.Rest.CreateMessage(dm, new MessageRequest { Content = $"<{msg.Attachments[0].Url}>" }); // If the original message wasn't posted in DMs, send a public reminder if (ctx.Channel.Type != Channel.ChannelType.Dm) await ctx.Reply($"{Emojis.Success} System deleted. Check your DMs for a copy of your system's data!"); } catch (ForbiddenException) { // If user has DMs closed, tell 'em to open them throw new PKError( $"I couldn't send you a DM with your system's data before deleting your system. Either make sure your DMs are open, or use `pk;s delete -no-export` to delete your system without exporting first."); } } else { await ctx.Reply($"{Emojis.Success} System deleted."); } // Now that we've sent the export data (or been told not to), we can safely delete the system await ctx.Repository.DeleteSystem(target.Id); } public async Task SystemProxy(Context ctx) { ctx.CheckSystem(); var guild = await ctx.MatchGuild() ?? ctx.Guild ?? throw new PKError("You must run this command in a server or pass a server ID."); var gs = await ctx.Repository.GetSystemGuild(guild.Id, ctx.System.Id); string serverText; if (guild.Id == ctx.Guild?.Id) serverText = $"this server ({guild.Name.EscapeMarkdown()})"; else serverText = $"the server {guild.Name.EscapeMarkdown()}"; if (!ctx.HasNext()) { if (gs.ProxyEnabled) await ctx.Reply( $"Proxying in {serverText} is currently **enabled** for your system. To disable it, type `pk;system proxy off`."); else await ctx.Reply( $"Proxying in {serverText} is currently **disabled** for your system. To enable it, type `pk;system proxy on`."); return; } var newValue = ctx.MatchToggle(); await ctx.Repository.UpdateSystemGuild(ctx.System.Id, guild.Id, new SystemGuildPatch { ProxyEnabled = newValue }); if (newValue) await ctx.Reply($"Message proxying in {serverText} is now **enabled** for your system."); else await ctx.Reply($"Message proxying in {serverText} is now **disabled** for your system."); } public async Task SystemPrivacy(Context ctx, PKSystem target) { ctx.CheckSystem().CheckOwnSystem(target); Task PrintEmbed() { var eb = new EmbedBuilder() .Title("Current privacy settings for your system") .Field(new Embed.Field("Description", target.DescriptionPrivacy.Explanation())) .Field(new Embed.Field("Pronouns", target.PronounPrivacy.Explanation())) .Field(new Embed.Field("Member list", target.MemberListPrivacy.Explanation())) .Field(new Embed.Field("Group list", target.GroupListPrivacy.Explanation())) .Field(new Embed.Field("Current fronter(s)", target.FrontPrivacy.Explanation())) .Field(new Embed.Field("Front/switch history", target.FrontHistoryPrivacy.Explanation())) .Description( "To edit privacy settings, use the command:\n`pk;system privacy `\n\n- `subject` is one of `description`, `list`, `front`, `fronthistory`, `groups`, or `all` \n- `level` is either `public` or `private`."); return ctx.Reply(embed: eb.Build()); } async Task SetLevel(SystemPrivacySubject subject, PrivacyLevel level) { await ctx.Repository.UpdateSystem(target.Id, new SystemPatch().WithPrivacy(subject, level)); var levelExplanation = level switch { PrivacyLevel.Public => "be able to query", PrivacyLevel.Private => "*not* be able to query", _ => "" }; var subjectStr = subject switch { SystemPrivacySubject.Description => "description", SystemPrivacySubject.Pronouns => "pronouns", SystemPrivacySubject.Front => "front", SystemPrivacySubject.FrontHistory => "front history", SystemPrivacySubject.MemberList => "member list", SystemPrivacySubject.GroupList => "group list", _ => "" }; var msg = $"System {subjectStr} privacy has been set to **{level.LevelName()}**. Other accounts will now {levelExplanation} your system {subjectStr}."; await ctx.Reply($"{Emojis.Success} {msg}"); } async Task SetAll(PrivacyLevel level) { await ctx.Repository.UpdateSystem(target.Id, new SystemPatch().WithAllPrivacy(level)); var msg = level switch { PrivacyLevel.Private => $"All system privacy settings have been set to **{level.LevelName()}**. Other accounts will now not be able to view your member list, group list, front history, or system description.", PrivacyLevel.Public => $"All system privacy settings have been set to **{level.LevelName()}**. Other accounts will now be able to view everything.", _ => "" }; await ctx.Reply($"{Emojis.Success} {msg}"); } if (!ctx.HasNext()) await PrintEmbed(); else if (ctx.Match("all")) await SetAll(ctx.PopPrivacyLevel()); else await SetLevel(ctx.PopSystemPrivacySubject(), ctx.PopPrivacyLevel()); } }