PluralKit/PluralKit.Bot/Services/LogChannelService.cs

using System.Threading.Tasks;

using Dapper;

using DSharpPlus;
using DSharpPlus.Entities;

using PluralKit.Core;

using Serilog;

namespace PluralKit.Bot {
    public class LogChannelService {
        private readonly EmbedService _embed;
        private readonly IDatabase _db;
        private readonly ModelRepository _repo;
        private readonly ILogger _logger;

        public LogChannelService(EmbedService embed, ILogger logger, IDatabase db, ModelRepository repo)
        {
            _embed = embed;
            _db = db;
            _repo = repo;
            _logger = logger.ForContext<LogChannelService>();
        }

        public async ValueTask LogMessage(DiscordClient client, MessageContext ctx, ProxyMatch proxy, DiscordMessage trigger, ulong hookMessage)
        {
            if (ctx.SystemId == null || ctx.LogChannel == null || ctx.InLogBlacklist) return;
            
            // Find log channel and check if valid
            var logChannel = await FindLogChannel(client, trigger.Channel.GuildId, ctx.LogChannel.Value);
            if (logChannel == null || logChannel.Type != ChannelType.Text) return;
            
            // Check bot permissions
            if (!logChannel.BotHasAllPermissions(Permissions.SendMessages | Permissions.EmbedLinks))
            {
                _logger.Information(
                    "Does not have permission to proxy log, ignoring (channel: {ChannelId}, guild: {GuildId}, bot permissions: {BotPermissions})", 
                    ctx.LogChannel.Value, trigger.Channel.GuildId, trigger.Channel.BotPermissions());
                return;
            }

            // Send embed!
            await using var conn = await _db.Obtain();
            var embed = _embed.CreateLoggedMessageEmbed(await _repo.GetSystem(conn, ctx.SystemId.Value),
                await _repo.GetMember(conn, proxy.Member.Id), hookMessage, trigger.Id, trigger.Author, proxy.Content,
                trigger.Channel);
            var url = $"https://discord.com/channels/{trigger.Channel.GuildId}/{trigger.ChannelId}/{hookMessage}";
            await logChannel.SendMessageFixedAsync(content: url, embed: embed);
        }

        private async Task<DiscordChannel> FindLogChannel(DiscordClient client, ulong guild, ulong channel)
        {
            // MUST use this client here, otherwise we get strange cache issues where the guild doesn't exist... >.>
            var obj = await client.GetChannel(channel);
            
            if (obj == null)
            {
                // Channel doesn't exist or we don't have permission to access it, let's remove it from the database too
                _logger.Warning("Attempted to fetch missing log channel {LogChannel} for guild {Guild}, removing from database", channel, guild);
                await using var conn = await _db.Obtain();
                await conn.ExecuteAsync("update servers set log_channel = null where id = @Guild",
                    new {Guild = guild});
            }

            return obj;
        }
    }
}