Discord/cogs/logger.py

import logging
from logging import Logger

from discord import Member, Message, TextChannel, MessageType
from discord import utils as ds_utils
from discord.ext import commands
from libbot.utils import config_get

from classes.holo_bot import HoloBot
from classes.holo_user import HoloUser
from modules.database import col_users

logger: Logger = logging.getLogger(__name__)


class Logger(commands.Cog):
    def __init__(self, client: HoloBot):
        self.client: HoloBot = client

    @commands.Cog.listener()
    async def on_message(self, message: Message):
        """Message listener. All actions on messages remain here for now."""
        if (
            (message.author != self.client.user)
            and (message.author.bot is False)
            and (message.author.system is False)
        ):
            await HoloUser.from_user(message.author, cache=self.client.cache)

        if (
            (message.type == MessageType.thread_created)
            and (message.channel is not None)
            and (
                await col_users.count_documents({"custom_channel": message.channel.id})
                > 0
            )
        ):
            try:
                logger.info(
                    "Deleting the thread creation message in a custom channel %s",
                    message.channel.id,
                )
                await message.delete()
            except Exception as exc:
                logger.warning(
                    "Could not delete the thread creation message in a custom channel %s due to %s",
                    message.channel.id,
                    exc,
                )

    @commands.Cog.listener()
    async def on_member_join(self, member: Member) -> None:
        """Member join handler. All actions on member join remain here for now."""
        welcome_chan: TextChannel | None = ds_utils.get(
            self.client.get_guild(await config_get("guild")).channels,
            id=await config_get("welcome", "channels", "text"),
        )
        rules_chan: TextChannel | None = ds_utils.get(
            self.client.get_guild(await config_get("guild")).channels,
            id=await config_get("rules", "channels", "text"),
        )

        if (
            (member != self.client.user)
            and (member.bot is False)
            and (member.system is False)
        ):
            if welcome_chan is not None and rules_chan is not None:
                await welcome_chan.send(
                    content=(await config_get("welcome", "messages")).format(
                        mention=member.mention, rules=rules_chan.mention
                    )
                )
            else:
                logger.warning("Could not find a welcome and/or rules channel by id")

            await HoloUser.from_user(member, cache=self.client.cache)


def setup(client: HoloBot) -> None:
    client.add_cog(Logger(client))