using Humanizer; using Myriad.Builders; using Myriad.Cache; using Myriad.Extensions; using Myriad.Rest; using Myriad.Rest.Exceptions; using Myriad.Types; using NodaTime; using PluralKit.Core; namespace PluralKit.Bot; public class EmbedService { public const string LEGACY_EMBED_WARNING = "\u26A0\uFE0F The \"legacy\" embeds for system/member/group cards are deprecated, and will be removed in future."; private readonly IDiscordCache _cache; private readonly IDatabase _db; private readonly ModelRepository _repo; private readonly DiscordApiClient _rest; private readonly BotConfig _config; private readonly CoreConfig _coreConfig; public EmbedService(IDatabase db, ModelRepository repo, IDiscordCache cache, DiscordApiClient rest, BotConfig config, CoreConfig coreConfig) { _db = db; _repo = repo; _cache = cache; _rest = rest; _config = config; _coreConfig = coreConfig; } private Task<(ulong Id, User? User)[]> GetUsers(IEnumerable ids) { async Task<(ulong Id, User? User)> Inner(ulong id) { var user = await _cache.GetOrFetchUser(_rest, id); return (id, user); } return Task.WhenAll(ids.Select(Inner)); } public async Task CreateSystemMessageComponents(Context cctx, PKSystem system, LookupContext ctx) { // Fetch/render info for all accounts simultaneously var accounts = await _repo.GetSystemAccounts(system.Id); var users = (await GetUsers(accounts)).Select(x => x.User?.NameAndMention() ?? $"(deleted account {x.Id})"); var linkedAccounts = new MessageComponent() { Type = ComponentType.Text, Content = "**Linked accounts:**\n" + string.Join("\n", users).Truncate(1000), }; var countctx = LookupContext.ByNonOwner; if (cctx.MatchFlag("a", "all")) { if (system.Id == cctx.System?.Id) countctx = LookupContext.ByOwner; else throw Errors.LookupNotAllowed; } var memberCount = await _repo.GetSystemMemberCount(system.Id, countctx == LookupContext.ByOwner ? null : PrivacyLevel.Public); var guildSettings = cctx.Guild != null ? await _repo.GetSystemGuild(cctx.Guild.Id, system.Id) : null; var avatar = system.AvatarFor(ctx); var headerText = ""; if (system.PronounPrivacy.CanAccess(ctx) && system.Pronouns != null) headerText += $"\n**Pronouns:** {system.Pronouns}"; if (system.Tag != null) headerText += $"\n**Tag:** {system.Tag.EscapeMarkdown()}"; if (cctx.Config != null && cctx.Config.CardShowColorHex && !system.Color.EmptyOrNull()) headerText += $"\n**Color:** #{system.Color}"; if (cctx.Guild != null) { if (guildSettings.Tag != null && guildSettings.TagEnabled) headerText += $"\n**Tag (in server '{cctx.Guild.Name}'):** {guildSettings.Tag.EscapeMarkdown()}"; if (!guildSettings.TagEnabled) headerText += $"\n**Tag (in server '{cctx.Guild.Name}'):** *(tag is disabled in this server)*"; } if (system.MemberListPrivacy.CanAccess(ctx)) { headerText += $"\n**Members:** {memberCount}"; if (system.Id == cctx.System?.Id) if (memberCount > 0) headerText += $" (see `{cctx.DefaultPrefix}system list`)"; else headerText += $" (add one with `{cctx.DefaultPrefix}member new`!)"; else if (memberCount > 0) headerText += $" (see `{cctx.DefaultPrefix}system {system.DisplayHid(cctx.Config)} list`)"; } List switchComponent = []; var latestSwitch = await _repo.GetLatestSwitch(system.Id); if (latestSwitch != null && system.FrontPrivacy.CanAccess(ctx)) { var switchMembers = await _db.Execute(conn => _repo.GetSwitchMembers(conn, latestSwitch.Id)).ToListAsync(); if (switchMembers.Count > 0) { var memberStr = string.Join(", ", switchMembers.Select(m => m.NameFor(ctx))); if (memberStr.Length > 200) memberStr = $"(too many to show, see `{cctx.DefaultPrefix}system {system.DisplayHid(cctx.Config)} fronters`)"; switchComponent.Add(new() { Type = ComponentType.Text, Content = $"**{"Current fronter".ToQuantity(switchMembers.Count, ShowQuantityAs.None)}:** {memberStr}", }); } } List descComponents = []; if (system.DescriptionFor(ctx) is { } desc && !string.IsNullOrWhiteSpace(desc)) { descComponents.Add(new() { Type = ComponentType.Separator, }); descComponents.Add(new() { Type = ComponentType.Text, Content = desc.NormalizeLineEndSpacing().Truncate(1024), }); } if (system.BannerPrivacy.CanAccess(ctx) && !string.IsNullOrWhiteSpace(system.BannerImage)) descComponents.Add(new() { Type = ComponentType.MediaGallery, Items = [new() { Media = new() { Url = system.BannerImage } }], }); var systemName = (cctx.Guild != null && guildSettings?.DisplayName != null) ? guildSettings?.DisplayName! : system.NameFor(ctx); var premiumText = ""; // TODO(iris): "\n\U0001F31F *PluralKit Premium supporter!*"; List header = [ new MessageComponent() { Type = ComponentType.Text, Content = $"### {systemName ?? $"`{system.DisplayHid(cctx.Config)}`"}{premiumText}", }, ]; if (!string.IsNullOrWhiteSpace(headerText)) header.Add(new MessageComponent() { Type = ComponentType.Text, Content = headerText, }); if (cctx.Guild != null) { var guildAvatar = guildSettings.AvatarUrl.TryGetCleanCdnUrl(); if (!string.IsNullOrWhiteSpace(guildAvatar)) avatar = guildAvatar; } if (!string.IsNullOrWhiteSpace(avatar)) header = [ new MessageComponent() { Type = ComponentType.Section, Components = [.. header], Accessory = new MessageComponent() { Type = ComponentType.Thumbnail, Media = new() { Url = avatar }, }, }, ]; return [ new MessageComponent() { Type = ComponentType.Container, AccentColor = system.Color?.ToDiscordColor(), Components = [ ..header, ..switchComponent, linkedAccounts, ..descComponents ], }, new MessageComponent() { Type = ComponentType.Section, Components = [ new MessageComponent() { Type = ComponentType.Text, Content = $"-# System ID: `{system.DisplayHid(cctx.Config)}`{cctx.PremiumEmoji}\n-# Created: {system.Created.FormatZoned(cctx.Zone)}", }, ], Accessory = new MessageComponent() { Type = ComponentType.Button, Style = ButtonStyle.Link, Label = "View on dashboard", Url = $"{_coreConfig.DashboardBaseUrl}/profile/s/{system.Hid}", }, }, ]; } public async Task CreateSystemEmbed(Context cctx, PKSystem system, LookupContext ctx) { // Fetch/render info for all accounts simultaneously var accounts = await _repo.GetSystemAccounts(system.Id); var users = (await GetUsers(accounts)).Select(x => x.User?.NameAndMention() ?? $"(deleted account {x.Id})"); var countctx = LookupContext.ByNonOwner; if (cctx.MatchFlag("a", "all")) { if (system.Id == cctx.System?.Id) countctx = LookupContext.ByOwner; else throw Errors.LookupNotAllowed; } var memberCount = await _repo.GetSystemMemberCount(system.Id, countctx == LookupContext.ByOwner ? null : PrivacyLevel.Public); var eb = new EmbedBuilder() .Title(system.NameFor(ctx)) .Footer(new Embed.EmbedFooter( $"System ID: {system.DisplayHid(cctx.Config)} | Created on {system.Created.FormatZoned(cctx.Zone)}")) .Color(system.Color?.ToDiscordColor()) .Url($"{_coreConfig.DashboardBaseUrl}/profile/s/{system.Hid}"); var avatar = system.AvatarFor(ctx); if (avatar != null) eb.Thumbnail(new Embed.EmbedThumbnail(avatar)); if (system.BannerPrivacy.CanAccess(ctx)) eb.Image(new Embed.EmbedImage(system.BannerImage)); var latestSwitch = await _repo.GetLatestSwitch(system.Id); if (latestSwitch != null && system.FrontPrivacy.CanAccess(ctx)) { var switchMembers = await _db.Execute(conn => _repo.GetSwitchMembers(conn, latestSwitch.Id)).ToListAsync(); if (switchMembers.Count > 0) { var memberStr = string.Join(", ", switchMembers.Select(m => m.NameFor(ctx))); if (memberStr.Length > 200) memberStr = $"[too many to show, see `{cctx.DefaultPrefix}system {system.DisplayHid(cctx.Config)} fronters`]"; eb.Field(new Embed.Field("Fronter".ToQuantity(switchMembers.Count, ShowQuantityAs.None), memberStr)); } } if (system.Tag != null) eb.Field(new Embed.Field("Tag", system.Tag.EscapeMarkdown(), true)); if (cctx.Guild != null) { var guildSettings = await _repo.GetSystemGuild(cctx.Guild.Id, system.Id); if (guildSettings.Tag != null && guildSettings.TagEnabled) eb.Field(new Embed.Field($"Tag (in server '{cctx.Guild.Name}')", guildSettings.Tag .EscapeMarkdown(), true)); if (!guildSettings.TagEnabled) eb.Field(new Embed.Field($"Tag (in server '{cctx.Guild.Name}')", "*(tag is disabled in this server)*")); if (guildSettings.DisplayName != null) eb.Title(guildSettings.DisplayName); var guildAvatar = guildSettings.AvatarUrl.TryGetCleanCdnUrl(); if (guildAvatar != null) { eb.Thumbnail(new Embed.EmbedThumbnail(guildAvatar)); var sysDesc = "*(this system has a server-specific avatar set"; if (avatar != null) sysDesc += $"; [click here]({system.AvatarUrl.TryGetCleanCdnUrl()}) to see their global avatar)*"; else sysDesc += ")*"; eb.Description(sysDesc); } } if (system.PronounPrivacy.CanAccess(ctx) && system.Pronouns != null) eb.Field(new Embed.Field("Pronouns", system.Pronouns, true)); if (!system.Color.EmptyOrNull()) eb.Field(new Embed.Field("Color", $"#{system.Color}", true)); eb.Field(new Embed.Field("Linked accounts", string.Join("\n", users).Truncate(1000), true)); if (system.MemberListPrivacy.CanAccess(ctx)) { if (memberCount > 0) eb.Field(new Embed.Field($"Members ({memberCount})", $"(see `{cctx.DefaultPrefix}system {system.DisplayHid(cctx.Config)} list` or `{cctx.DefaultPrefix}system {system.DisplayHid(cctx.Config)} list full`)", true)); else eb.Field(new Embed.Field($"Members ({memberCount})", $"Add one with `{cctx.DefaultPrefix}member new`!", true)); } if (system.DescriptionFor(ctx) is { } desc) eb.Field(new Embed.Field("Description", desc.NormalizeLineEndSpacing().Truncate(1024))); return eb.Build(); } public Embed CreateLoggedMessageEmbed(Message triggerMessage, Message proxiedMessage, string systemHid, PKMember member, string channelName, string oldContent = null) { // TODO: pronouns in ?-reacted response using this card var timestamp = DiscordUtils.SnowflakeToInstant(proxiedMessage.Id); var name = proxiedMessage.Author.Username; // sometimes Discord will just... not return the avatar hash with webhook messages var avatar = proxiedMessage.Author.Avatar != null ? proxiedMessage.Author.AvatarUrl() : member.WebhookAvatarFor(LookupContext.ByNonOwner); var embed = new EmbedBuilder() .Author(new Embed.EmbedAuthor($"#{channelName}: {name}", IconUrl: avatar)) .Thumbnail(new Embed.EmbedThumbnail(avatar)) .Description(proxiedMessage.Content?.NormalizeLineEndSpacing()) .Footer(new Embed.EmbedFooter( $"System ID: {systemHid} | Member ID: {member.Hid} | Sender: {triggerMessage.Author.Username}#{triggerMessage.Author.Discriminator} ({triggerMessage.Author.Id}) | Message ID: {proxiedMessage.Id} | Original Message ID: {triggerMessage.Id}")) .Timestamp(timestamp.ToDateTimeOffset().ToString("O")); if (oldContent == "") oldContent = "*no message content*"; if (oldContent != null) embed.Field(new Embed.Field("Old message", oldContent?.NormalizeLineEndSpacing().Truncate(1000))); return embed.Build(); } public async Task CreateMemberMessageComponents(PKSystem system, PKMember member, Guild guild, SystemConfig? ccfg, LookupContext ctx, DateTimeZone zone) { var name = member.NameFor(ctx); var systemGuildSettings = guild != null ? await _repo.GetSystemGuild(guild.Id, system.Id) : null; var systemName = (guild != null && systemGuildSettings?.DisplayName != null) ? systemGuildSettings?.DisplayName! : system.NameFor(ctx); var guildSettings = guild != null ? await _repo.GetMemberGuild(guild.Id, member.Id) : null; var guildDisplayName = guildSettings?.DisplayName; var webhook_avatar = guildSettings?.AvatarUrl ?? member.WebhookAvatarFor(ctx) ?? member.AvatarFor(ctx); var avatar = guildSettings?.AvatarUrl ?? member.AvatarFor(ctx); var groups = await _repo.GetMemberGroups(member.Id) .Where(g => g.Visibility.CanAccess(ctx)) .OrderBy(g => g.Name, StringComparer.InvariantCultureIgnoreCase) .ToListAsync(); var headerText = ""; if (member.MemberVisibility == PrivacyLevel.Private) headerText += "*(this member is hidden)*\n"; if (guildSettings?.AvatarUrl != null) if (member.AvatarFor(ctx) != null) headerText += $"*(this member has a server-specific avatar set; [click here]({member.AvatarUrl.TryGetCleanCdnUrl()}) to see the global avatar)*\n"; else headerText += "*(this member has a server-specific avatar set)*\n"; if (!member.DisplayName.EmptyOrNull() && member.NamePrivacy.CanAccess(ctx)) headerText += $"\n**Display name:** {member.DisplayName.Truncate(1024)}"; if (guild != null && guildDisplayName != null) headerText += $"\n**Server nickname (for '{guild.Name}'):** {guildDisplayName.Truncate(1024)}"; if (ccfg != null && ccfg.CardShowColorHex && !member.Color.EmptyOrNull()) headerText += $"\n**Color:** #{member.Color}"; if (member.PronounsFor(ctx) is { } pronouns && !string.IsNullOrWhiteSpace(pronouns)) headerText += $"\n**Pronouns:** {pronouns}"; if (member.BirthdayFor(ctx) != null) headerText += $"\n**Birthday:** {member.BirthdayString}"; if (member.MessageCountFor(ctx) is { } count && count > 0) headerText += $"\n**Message count:** {member.MessageCount}"; List extraData = []; if (member.HasProxyTags && member.ProxyPrivacy.CanAccess(ctx)) { extraData.Add(new MessageComponent { Type = ComponentType.Separator, }); extraData.Add(new MessageComponent { Type = ComponentType.Text, Content = $"**Proxy tags:**\n{member.ProxyTagsString("\n").Truncate(1024)}", }); } if (groups.Count > 0) { // More than 5 groups show in "compact" format without ID var content = groups.Count > 5 ? string.Join(", ", groups.Select(g => g.DisplayName ?? g.Name)) : string.Join("\n", groups.Select(g => $"[`{g.DisplayHid(ccfg, isList: true)}`] **{g.DisplayName ?? g.Name}**")); extraData.Add(new MessageComponent { Type = ComponentType.Separator, }); extraData.Add(new MessageComponent { Type = ComponentType.Text, Content = $"**Groups ({groups.Count}):**\n{content.Truncate(1000)}", }); } List descComponents = []; if (member.DescriptionFor(ctx) is { } desc && !string.IsNullOrWhiteSpace(desc)) { descComponents.Add(new() { Type = ComponentType.Separator, }); descComponents.Add(new() { Type = ComponentType.Text, Content = desc.NormalizeLineEndSpacing().Truncate(1024), }); } if (member.BannerPrivacy.CanAccess(ctx) && !string.IsNullOrWhiteSpace(member.BannerImage)) descComponents.Add(new() { Type = ComponentType.MediaGallery, Items = [new() { Media = new() { Url = member.BannerImage } }], }); List header = [ new MessageComponent() { Type = ComponentType.Text, Content = $"### {name}{(systemName != null ? $" ({systemName})" : "")}", }, ]; if (!string.IsNullOrWhiteSpace(headerText)) header.Add(new MessageComponent() { Type = ComponentType.Text, Content = headerText, }); if (!string.IsNullOrWhiteSpace(avatar)) header = [ new MessageComponent() { Type = ComponentType.Section, Components = [.. header], Accessory = new MessageComponent() { Type = ComponentType.Thumbnail, Media = new() { Url = avatar }, }, }, ]; return [ new MessageComponent() { Type = ComponentType.Container, AccentColor = member.Color?.ToDiscordColor(), Components = [ ..header, ..extraData, ..descComponents ], }, new MessageComponent() { Type = ComponentType.Section, Components = [ new MessageComponent() { Type = ComponentType.Text, Content = $"-# System ID: `{system.DisplayHid(ccfg)}` \u2219 Member ID: `{member.DisplayHid(ccfg)}`{(member.MetadataPrivacy.CanAccess(ctx) ? $"\n-# Created: {member.Created.FormatZoned(zone)}" : "")}", }, ], Accessory = new MessageComponent() { Type = ComponentType.Button, Style = ButtonStyle.Link, Label = "View on dashboard", Url = $"{_coreConfig.DashboardBaseUrl}/profile/m/{member.Hid}", }, }, ]; } public async Task CreateMemberEmbed(PKSystem system, PKMember member, Guild guild, SystemConfig? ccfg, LookupContext ctx, DateTimeZone zone) { // string FormatTimestamp(Instant timestamp) => DateTimeFormats.ZonedDateTimeFormat.Format(timestamp.InZone(system.Zone)); var name = member.NameFor(ctx); var systemGuildSettings = guild != null ? await _repo.GetSystemGuild(guild.Id, system.Id) : null; if (systemGuildSettings != null && systemGuildSettings.DisplayName != null) name = $"{name} ({systemGuildSettings.DisplayName})"; else if (system.NameFor(ctx) != null) name = $"{name} ({system.NameFor(ctx)})"; else name = $"{name}"; var guildSettings = guild != null ? await _repo.GetMemberGuild(guild.Id, member.Id) : null; var guildDisplayName = guildSettings?.DisplayName; var webhook_avatar = guildSettings?.AvatarUrl ?? member.WebhookAvatarFor(ctx) ?? member.AvatarFor(ctx); var avatar = guildSettings?.AvatarUrl ?? member.AvatarFor(ctx); var groups = await _repo.GetMemberGroups(member.Id) .Where(g => g.Visibility.CanAccess(ctx)) .OrderBy(g => g.Name, StringComparer.InvariantCultureIgnoreCase) .ToListAsync(); var eb = new EmbedBuilder() .Author(new Embed.EmbedAuthor(name, IconUrl: webhook_avatar.TryGetCleanCdnUrl(), Url: $"{_coreConfig.DashboardBaseUrl}/profile/m/{member.Hid}")) // .WithColor(member.ColorPrivacy.CanAccess(ctx) ? color : null) .Color(member.Color?.ToDiscordColor()) .Footer(new Embed.EmbedFooter( $"System ID: {system.DisplayHid(ccfg)} | Member ID: {member.DisplayHid(ccfg)} {(member.MetadataPrivacy.CanAccess(ctx) ? $"| Created on {member.Created.FormatZoned(zone)}" : "")}")); if (member.BannerPrivacy.CanAccess(ctx)) eb.Image(new Embed.EmbedImage(member.BannerImage)); var description = ""; if (member.MemberVisibility == PrivacyLevel.Private) description += "*(this member is hidden)*\n"; if (guildSettings?.AvatarUrl != null) if (member.AvatarFor(ctx) != null) description += $"*(this member has a server-specific avatar set; [click here]({member.AvatarUrl.TryGetCleanCdnUrl()}) to see the global avatar)*\n"; else description += "*(this member has a server-specific avatar set)*\n"; if (description != "") eb.Description(description); if (avatar != null) eb.Thumbnail(new Embed.EmbedThumbnail(avatar.TryGetCleanCdnUrl())); if (!member.DisplayName.EmptyOrNull() && member.NamePrivacy.CanAccess(ctx)) eb.Field(new Embed.Field("Display Name", member.DisplayName.Truncate(1024), true)); if (guild != null && guildDisplayName != null) eb.Field(new Embed.Field($"Server Nickname (for {guild.Name})", guildDisplayName.Truncate(1024), true)); if (member.BirthdayFor(ctx) != null) eb.Field(new Embed.Field("Birthdate", member.BirthdayString, true)); if (member.PronounsFor(ctx) is { } pronouns && !string.IsNullOrWhiteSpace(pronouns)) eb.Field(new Embed.Field("Pronouns", pronouns.Truncate(1024), true)); if (member.MessageCountFor(ctx) is { } count && count > 0) eb.Field(new Embed.Field("Message Count", member.MessageCount.ToString(), true)); if (member.HasProxyTags && member.ProxyPrivacy.CanAccess(ctx)) eb.Field(new Embed.Field("Proxy Tags", member.ProxyTagsString("\n").Truncate(1024), true)); // --- For when this gets added to the member object itself or however they get added // if (member.LastMessage != null && member.MetadataPrivacy.CanAccess(ctx)) eb.AddField("Last message:" FormatTimestamp(DiscordUtils.SnowflakeToInstant(m.LastMessage.Value))); // if (member.LastSwitchTime != null && m.MetadataPrivacy.CanAccess(ctx)) eb.AddField("Last switched in:", FormatTimestamp(member.LastSwitchTime.Value)); // if (!member.Color.EmptyOrNull() && member.ColorPrivacy.CanAccess(ctx)) eb.AddField("Color", $"#{member.Color}", true); if (!member.Color.EmptyOrNull()) eb.Field(new Embed.Field("Color", $"#{member.Color}", true)); if (groups.Count > 0) { // More than 5 groups show in "compact" format without ID var content = groups.Count > 5 ? string.Join(", ", groups.Select(g => g.DisplayName ?? g.Name)) : string.Join("\n", groups.Select(g => $"[`{g.DisplayHid(ccfg, isList: true)}`] **{g.DisplayName ?? g.Name}**")); eb.Field(new Embed.Field($"Groups ({groups.Count})", content.Truncate(1000))); } if (member.DescriptionFor(ctx) is { } desc) eb.Field(new Embed.Field("Description", member.Description.NormalizeLineEndSpacing())); return eb.Build(); } public async Task CreateGroupMessageComponents(Context ctx, PKSystem system, PKGroup target) { var pctx = ctx.LookupContextFor(system.Id); var name = target.NameFor(ctx); var systemGuildSettings = ctx.Guild != null ? await _repo.GetSystemGuild(ctx.Guild.Id, system.Id) : null; var systemName = (ctx.Guild != null && systemGuildSettings?.DisplayName != null) ? systemGuildSettings?.DisplayName! : system.NameFor(ctx); var countctx = LookupContext.ByNonOwner; if (ctx.MatchFlag("a", "all")) { if (system.Id == ctx.System?.Id) countctx = LookupContext.ByOwner; else throw Errors.LookupNotAllowed; } var memberCount = await _repo.GetGroupMemberCount(target.Id, countctx == LookupContext.ByOwner ? null : PrivacyLevel.Public); var headerText = ""; if (target.NamePrivacy.CanAccess(pctx) && target.DisplayName != null) headerText += $"\n**Display name:** {target.DisplayName}"; if (ctx.Config != null && ctx.Config.CardShowColorHex && !target.Color.EmptyOrNull()) headerText += $"\n**Color:** #{target.Color}"; if (target.ListPrivacy.CanAccess(pctx)) { headerText += $"\n**Members:** {memberCount}"; if (system.Id == ctx.System?.Id && memberCount == 0) headerText += $" (add one with `{ctx.DefaultPrefix}group {target.Reference(ctx)} add `!)"; else if (memberCount > 0) headerText += $" (see `{ctx.DefaultPrefix}group {target.Reference(ctx)} list`)"; } List descComponents = []; if (target.DescriptionFor(pctx) is { } desc && !string.IsNullOrWhiteSpace(desc)) { descComponents.Add(new() { Type = ComponentType.Separator, }); descComponents.Add(new() { Type = ComponentType.Text, Content = desc.NormalizeLineEndSpacing().Truncate(1024), }); } if (target.BannerPrivacy.CanAccess(pctx) && !string.IsNullOrWhiteSpace(target.BannerImage)) descComponents.Add(new() { Type = ComponentType.MediaGallery, Items = [new() { Media = new() { Url = target.BannerImage } }], }); List header = [ new MessageComponent() { Type = ComponentType.Text, Content = $"### {name}{(systemName != null ? $" ({systemName})" : "")}", }, ]; if (!string.IsNullOrWhiteSpace(headerText)) header.Add(new MessageComponent() { Type = ComponentType.Text, Content = headerText, }); if (target.IconFor(pctx) is { } icon) header = [ new MessageComponent() { Type = ComponentType.Section, Components = [.. header], Accessory = new MessageComponent() { Type = ComponentType.Thumbnail, Media = new() { Url = icon.TryGetCleanCdnUrl() }, }, }, ]; return [ new MessageComponent() { Type = ComponentType.Container, AccentColor = target.Color?.ToDiscordColor(), Components = [ ..header, ..descComponents ], }, new MessageComponent() { Type = ComponentType.Section, Components = [ new MessageComponent() { Type = ComponentType.Text, Content = $"-# System ID: `{system.DisplayHid(ctx.Config)}` \u2219 Group ID: `{target.DisplayHid(ctx.Config)}`{(target.MetadataPrivacy.CanAccess(pctx) ? $"\n-# Created: {target.Created.FormatZoned(ctx.Zone)}" : "")}", }, ], Accessory = new MessageComponent() { Type = ComponentType.Button, Style = ButtonStyle.Link, Label = "View on dashboard", Url = $"{_coreConfig.DashboardBaseUrl}/profile/g/{target.Hid}", }, }, ]; } public async Task CreateGroupEmbed(Context ctx, PKSystem system, PKGroup target) { var pctx = ctx.LookupContextFor(system.Id); var countctx = LookupContext.ByNonOwner; if (ctx.MatchFlag("a", "all")) { if (system.Id == ctx.System?.Id) countctx = LookupContext.ByOwner; else throw Errors.LookupNotAllowed; } var memberCount = await _repo.GetGroupMemberCount(target.Id, countctx == LookupContext.ByOwner ? null : PrivacyLevel.Public); var nameField = target.NameFor(ctx); var systemGuildSettings = ctx.Guild != null ? await _repo.GetSystemGuild(ctx.Guild.Id, system.Id) : null; if (systemGuildSettings != null && systemGuildSettings.DisplayName != null) nameField = $"{nameField} ({systemGuildSettings.DisplayName})"; else if (system.NameFor(ctx) != null) nameField = $"{nameField} ({system.NameFor(ctx)})"; else nameField = $"{nameField}"; var eb = new EmbedBuilder() .Author(new Embed.EmbedAuthor(nameField, IconUrl: target.IconFor(pctx), Url: $"{_coreConfig.DashboardBaseUrl}/profile/g/{target.Hid}")) .Color(target.Color?.ToDiscordColor()); eb.Footer(new Embed.EmbedFooter($"System ID: {system.DisplayHid(ctx.Config)} | Group ID: {target.DisplayHid(ctx.Config)}{(target.MetadataPrivacy.CanAccess(pctx) ? $" | Created on {target.Created.FormatZoned(ctx.Zone)}" : "")}")); if (target.BannerPrivacy.CanAccess(pctx)) eb.Image(new Embed.EmbedImage(target.BannerImage)); if (target.NamePrivacy.CanAccess(pctx) && target.DisplayName != null) eb.Field(new Embed.Field("Display Name", target.DisplayName, true)); if (!target.Color.EmptyOrNull()) eb.Field(new Embed.Field("Color", $"#{target.Color}", true)); if (target.ListPrivacy.CanAccess(pctx)) { if (memberCount == 0 && pctx == LookupContext.ByOwner) // Only suggest the add command if this is actually the owner lol eb.Field(new Embed.Field("Members (0)", $"Add one with `{ctx.DefaultPrefix}group {target.Reference(ctx)} add `!")); else { var name = pctx == LookupContext.ByOwner ? target.Reference(ctx) : target.DisplayHid(ctx.Config); eb.Field(new Embed.Field($"Members ({memberCount})", $"(see `{ctx.DefaultPrefix}group {name} list`)")); } } if (target.DescriptionFor(pctx) is { } desc) eb.Field(new Embed.Field("Description", desc)); if (target.IconFor(pctx) is { } icon) eb.Thumbnail(new Embed.EmbedThumbnail(icon.TryGetCleanCdnUrl())); return eb.Build(); } public async Task CreateFronterEmbed(PKSwitch sw, DateTimeZone zone, LookupContext ctx) { var members = await _db.Execute(c => _repo.GetSwitchMembers(c, sw.Id).ToListAsync().AsTask()); var timeSinceSwitch = SystemClock.Instance.GetCurrentInstant() - sw.Timestamp; var memberStr = "*(no fronter)*"; if (members.Count > 0) { memberStr = ""; foreach (var item in members.Select((value, i) => new { i, value })) { memberStr += item.i == 0 ? "" : ", "; // field limit is 1024, capping after 900 gives us plenty of room // for the remaining count message if (memberStr.Length < 900) memberStr += item.value.NameFor(ctx); else { memberStr += $"*({members.Count - item.i} not shown)*"; break; } } } return new EmbedBuilder() .Color(members.FirstOrDefault()?.Color?.ToDiscordColor()) .Field(new Embed.Field($"Current {"fronter".ToQuantity(members.Count, ShowQuantityAs.None)}", memberStr)) .Field(new Embed.Field("Since", $"{sw.Timestamp.FormatZoned(zone)} ({timeSinceSwitch.FormatDuration()} ago)")) .Build(); } public async Task CreateMessageInfoMessageComponents(FullMessage msg, bool showContent, SystemConfig? ccfg = null) { var channel = await _cache.GetOrFetchChannel(_rest, msg.Message.Guild ?? 0, msg.Message.Channel); var ctx = LookupContext.ByNonOwner; var serverMsg = await _rest.GetMessageOrNull(msg.Message.Channel, msg.Message.Mid); // Need this whole dance to handle cases where: // - the user is deleted (userInfo == null) // - the bot's no longer in the server we're querying (channel == null) // - the member is no longer in the server we're querying (memberInfo == null) // TODO: optimize ordering here a bit with new cache impl; and figure what happens if bot leaves server -> channel still cached -> hits this bit and 401s? GuildMemberPartial memberInfo = null; User userInfo = null; if (channel != null) { GuildMember member = null; try { member = await _rest.GetGuildMember(channel.GuildId!.Value, msg.Message.Sender); } catch (ForbiddenException) { // no permission, couldn't fetch, oh well } if (member != null) // Don't do an extra request if we already have this info from the member lookup userInfo = member.User; memberInfo = member; } if (userInfo == null) userInfo = await _cache.GetOrFetchUser(_rest, msg.Message.Sender); // Calculate string displayed under "Sent by" string userStr; if (showContent && memberInfo != null && memberInfo.Nick != null) userStr = $"**\n Username:** {userInfo.NameAndMention()}\n** Nickname:** {memberInfo.Nick}"; else if (userInfo != null) userStr = userInfo.NameAndMention(); else userStr = $"*(deleted user {msg.Message.Sender})*"; var content = serverMsg?.Content?.NormalizeLineEndSpacing(); if (content == null || !showContent) content = "*(message contents deleted or inaccessible)*"; var systemStr = msg.System == null ? "*(deleted or unknown system)*" : msg.System.NameFor(ctx) != null ? $"{msg.System.NameFor(ctx)} (`{msg.System.DisplayHid(ccfg)}`)" : $"`{msg.System.DisplayHid(ccfg)}`"; var memberStr = msg.Member == null ? "*(deleted member)*" : $"{msg.Member.NameFor(ctx)} (`{msg.Member.DisplayHid(ccfg)}`)"; var roles = memberInfo?.Roles?.ToList(); var rolesContent = ""; if (roles != null && roles.Count > 0 && showContent) { var guild = await _cache.GetGuild(channel.GuildId!.Value); var rolesString = string.Join(", ", (roles .Select(id => { var role = Array.Find(guild.Roles, r => r.Id == id); if (role != null) return role; return new Role { Name = "*(unknown role)*", Position = 0 }; })) .OrderByDescending(role => role.Position) .Select(role => role.Name)); rolesContent = $"**Account Roles ({roles.Count})**\n{rolesString}"; } MessageComponent authorData = new MessageComponent() { Type = ComponentType.Text, Content = $"**System:** {systemStr}\n**Member:** {memberStr}\n**Sent by:** {userStr}\n\n{rolesContent}" }; var avatarURL = msg.Member?.AvatarFor(ctx).TryGetCleanCdnUrl(); MessageComponent header = (avatarURL == "" || avatarURL == null) ? authorData : new MessageComponent() { Type = ComponentType.Section, Components = [authorData], Accessory = new MessageComponent() { Type = ComponentType.Thumbnail, Media = new ComponentMedia() { Url = avatarURL } } }; List body = [ new MessageComponent() { Type = ComponentType.Separator, Spacing = 2 } ]; if (content != "") { body.Add(new MessageComponent() { Type = ComponentType.Text, Content = content }); } if (showContent) { if (serverMsg != null) { var media = new List(); foreach (Message.Attachment attachment in serverMsg?.Attachments) { var url = attachment.Url; if (url != null && url != "") media.Add(new ComponentMediaItem() { Media = new ComponentMedia() { Url = url } }); } if (media.Count > 0) body.Add(new MessageComponent() { Type = ComponentType.MediaGallery, Items = media.ToArray() }); } } MessageComponent footer = new MessageComponent() { Type = ComponentType.Text, Content = $"-# Original Message ID: {msg.Message.OriginalMid} · " }; return [ new MessageComponent() { Type = ComponentType.Container, Components = [ header, ..body ] }, footer ]; } public async Task CreateMessageInfoEmbed(FullMessage msg, bool showContent, SystemConfig? ccfg = null) { var channel = await _cache.GetOrFetchChannel(_rest, msg.Message.Guild ?? 0, msg.Message.Channel); var ctx = LookupContext.ByNonOwner; var serverMsg = await _rest.GetMessageOrNull(msg.Message.Channel, msg.Message.Mid); // Need this whole dance to handle cases where: // - the user is deleted (userInfo == null) // - the bot's no longer in the server we're querying (channel == null) // - the member is no longer in the server we're querying (memberInfo == null) // TODO: optimize ordering here a bit with new cache impl; and figure what happens if bot leaves server -> channel still cached -> hits this bit and 401s? GuildMemberPartial memberInfo = null; User userInfo = null; if (channel != null) { GuildMember member = null; try { member = await _rest.GetGuildMember(channel.GuildId!.Value, msg.Message.Sender); } catch (ForbiddenException) { // no permission, couldn't fetch, oh well } if (member != null) // Don't do an extra request if we already have this info from the member lookup userInfo = member.User; memberInfo = member; } if (userInfo == null) userInfo = await _cache.GetOrFetchUser(_rest, msg.Message.Sender); // Calculate string displayed under "Sent by" string userStr; if (showContent && memberInfo != null && memberInfo.Nick != null) userStr = $"**Username:** {userInfo.NameAndMention()}\n**Nickname:** {memberInfo.Nick}"; else if (userInfo != null) userStr = userInfo.NameAndMention(); else userStr = $"*(deleted user {msg.Message.Sender})*"; var content = serverMsg?.Content?.NormalizeLineEndSpacing(); if (content == null || !showContent) content = "*(message contents deleted or inaccessible)*"; // Put it all together var eb = new EmbedBuilder() .Author(new Embed.EmbedAuthor(msg.Member?.NameFor(ctx) ?? "(deleted member)", IconUrl: msg.Member?.AvatarFor(ctx).TryGetCleanCdnUrl())) .Description(content) .Image(showContent ? new Embed.EmbedImage(serverMsg?.Attachments?.FirstOrDefault()?.Url) : null) .Field(new Embed.Field("System", msg.System == null ? "*(deleted or unknown system)*" : msg.System.NameFor(ctx) != null ? $"{msg.System.NameFor(ctx)} (`{msg.System.DisplayHid(ccfg)}`)" : $"`{msg.System.DisplayHid(ccfg)}`" , true)) .Field(new Embed.Field("Member", msg.Member == null ? "*(deleted member)*" : $"{msg.Member.NameFor(ctx)} (`{msg.Member.DisplayHid(ccfg)}`)" , true)) .Field(new Embed.Field("Sent by", userStr, true)) .Timestamp(DiscordUtils.SnowflakeToInstant(msg.Message.Mid).ToDateTimeOffset().ToString("O")) .Footer(new Embed.EmbedFooter($"Original Message ID: {msg.Message.OriginalMid}")); var roles = memberInfo?.Roles?.ToList(); if (roles != null && roles.Count > 0 && showContent) { var guild = await _cache.GetGuild(channel.GuildId!.Value); var rolesString = string.Join(", ", (roles .Select(id => { var role = Array.Find(guild.Roles, r => r.Id == id); if (role != null) return role; return new Role { Name = "*(unknown role)*", Position = 0 }; })) .OrderByDescending(role => role.Position) .Select(role => role.Name)); eb.Field(new Embed.Field($"Account roles ({roles.Count})", rolesString.Truncate(1024))); } return eb.Build(); } public async Task CreateAuthorMessageComponents(User? user, FullMessage msg) { MessageComponent authorInfo; var author = user != null ? $"{user.Username}#{user.Discriminator}" : $"Deleted user ${msg.Message.Sender}"; var avatarUrl = user?.AvatarUrl(); var authorString = $"{author}\n**ID: **`{msg.Message.Sender.ToString()}`"; if (user != null && avatarUrl != "") { authorInfo = new MessageComponent() { Type = ComponentType.Section, Components = [ new MessageComponent() { Type = ComponentType.Text, Content = authorString } ], Accessory = new MessageComponent() { Type = ComponentType.Thumbnail, Media = new ComponentMedia() { Url = avatarUrl } } }; } else { authorInfo = new MessageComponent() { Type = ComponentType.Text, Content = authorString }; } MessageComponent container = new MessageComponent() { Type = ComponentType.Container, Components = [ authorInfo, ] }; return ( [ new MessageComponent() { Type = ComponentType.Text, Content = user != null ? $"{user.Mention()} ({user.Id})" : $"*(deleted user {msg.Message.Sender})*" }, container ] ); } public async Task CreateCommandMessageInfoMessageComponents(Core.CommandMessage msg, bool showContent) { var content = "*(command message deleted or inaccessible)*"; if (showContent) { var discordMessage = await _rest.GetMessageOrNull(msg.Channel, msg.OriginalMid); if (discordMessage != null) content = discordMessage.Content; } List body = [ new MessageComponent() { Type = ComponentType.Text, Content = $"### Command response message\n**Original message:** https://discord.com/channels/{msg.Guild}/{msg.Channel}/{msg.OriginalMid}\n**Sent By:** <@{msg.Sender}>" }, new MessageComponent() { Type = ComponentType.Separator, }, new MessageComponent() { Type = ComponentType.Text, Content = content }, ]; MessageComponent footer = new MessageComponent() { Type = ComponentType.Text, Content = $"-# Original Message ID: {msg.OriginalMid} · " }; return [ new MessageComponent(){ Type = ComponentType.Container, Components = [ ..body ] }, footer ]; } public async Task CreateCommandMessageInfoEmbed(Core.CommandMessage msg, bool showContent) { var content = "*(command message deleted or inaccessible)*"; if (showContent) { var discordMessage = await _rest.GetMessageOrNull(msg.Channel, msg.OriginalMid); if (discordMessage != null) content = discordMessage.Content; } return new EmbedBuilder() .Title("Command response message") .Description(content) .Field(new("Original message", $"https://discord.com/channels/{msg.Guild}/{msg.Channel}/{msg.OriginalMid}", true)) .Field(new("Sent by", $"<@{msg.Sender}>", true)) .Build(); } public Task CreateFrontPercentEmbed(FrontBreakdown breakdown, PKSystem system, PKGroup group, DateTimeZone tz, LookupContext ctx, string embedTitle, bool ignoreNoFronters, bool showFlat) { var color = system.Color; if (group != null) color = group.Color; var eb = new EmbedBuilder() .Title(embedTitle) .Color(color?.ToDiscordColor()); var footer = $"Since {breakdown.RangeStart.FormatZoned(tz)} ({(breakdown.RangeEnd - breakdown.RangeStart).FormatDuration()} ago)"; Duration period; if (showFlat) { period = Duration.FromTicks(breakdown.MemberSwitchDurations.Values.ToList().Sum(i => i.TotalTicks)); footer += ". Showing flat list (percentages add up to 100%)"; if (!ignoreNoFronters) period += breakdown.NoFronterDuration; else footer += ", ignoring switch-outs"; } else if (ignoreNoFronters) { period = breakdown.RangeEnd - breakdown.RangeStart - breakdown.NoFronterDuration; footer += ". Ignoring switch-outs"; } else { period = breakdown.RangeEnd - breakdown.RangeStart; } eb.Footer(new Embed.EmbedFooter(footer)); var maxEntriesToDisplay = 24; // max 25 fields allowed in embed - reserve 1 for "others" // We convert to a list of pairs so we can add the no-fronter value // Dictionary doesn't allow for null keys so we instead have a pair with a null key ;) var pairs = breakdown.MemberSwitchDurations.ToList(); if (breakdown.NoFronterDuration != Duration.Zero && !ignoreNoFronters) pairs.Add(new KeyValuePair(null, breakdown.NoFronterDuration)); var membersOrdered = pairs.OrderByDescending(pair => pair.Value).Take(maxEntriesToDisplay).ToList(); foreach (var pair in membersOrdered) { var frac = pair.Value / period; eb.Field(new Embed.Field(pair.Key?.NameFor(ctx) ?? "*(no fronter)*", $"{frac * 100:F0}% ({pair.Value.FormatDuration()})")); } if (membersOrdered.Count > maxEntriesToDisplay) eb.Field(new Embed.Field("(others)", membersOrdered.Skip(maxEntriesToDisplay) .Aggregate(Duration.Zero, (prod, next) => prod + next.Value) .FormatDuration(), true)); return Task.FromResult(eb.Build()); } }