tux/.archive/logging/message.py

import discord
from discord.ext import commands

from tux.utils.constants import Constants as CONST
from tux.utils.embeds import EmbedCreator


class GuildLogging(commands.Cog):
    def __init__(self, bot: commands.Bot):
        self.bot = bot
        self.dev_logs_channel_id: int = CONST.LOG_CHANNELS["DEV"]

    async def send_to_dev_log(self, embed: discord.Embed):
        channel = self.bot.get_channel(self.dev_logs_channel_id)
        if isinstance(channel, discord.TextChannel):
            await channel.send(embed=embed)

    @commands.Cog.listener()
    async def on_message(self, message: discord.Message) -> None:
        """
        When a message is sent in a guild

        Parameters
        ----------
        message : discord.Message
            The message that was sent.
        """

        poll_channel = self.bot.get_channel(1228717294788673656)  # TODO: stop hardcoding this

        if message.channel == poll_channel:
            # delete non-poll messages in the poll channel
            if message.poll is None:
                await message.delete()
                embed = EmbedCreator.create_log_embed(
                    title="Non-Poll Deleted",
                    description=f"Message: {message.id}",
                )
                await self.send_to_dev_log(embed)
                return

            # make a thread for the poll
            await message.create_thread(
                name=f"Poll by {message.author.display_name}",
                reason="Poll thread",
            )
            return

        if not message.embeds and not message.attachments and not message.content and not message.stickers:
            # check if the message is not a message
            if message.type != discord.MessageType.default:
                return
            # delete the message and log it
            await message.delete()
            embed = EmbedCreator.create_log_embed(
                title="Poll Deleted",
                description=f"Message: {message.id}",
            )
            await self.send_to_dev_log(embed)


async def setup(bot: commands.Bot) -> None:
    await bot.add_cog(GuildLogging(bot))