bot.py

"""
This is the core bot file. It contains config and database setup functions,
discord and irc login, and a few property and helper functions
"""

import asyncio
import datetime
import glob
import io
import json
import os
import threading
from typing import Self

import botlogging
import discord
import expiringdict
import gino
import ircrelay
import munch
import ui
import yaml
from botlogging import LogContext, LogLevel
from core import auxiliary, custom_errors, databases, extensionconfig, http
from discord import app_commands
from discord.ext import commands

loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)


class TechSupportBot(commands.Bot):
    """Sets up a new TechSupportBot object.
    This does NOT start the bot, the start function must be called for that

    Args:
        intents (discord.Intents): The list of intents that
            the bot needs to request from discord
        allowed_mentions (discord.AllowedMentions): What the bot is, or is not,
            allowed to mention

    Attributes:
        CONFIG_PATH (str): The hard coded path to the yaml config file
        EXTENSIONS_DIR_NAME (str): The hardcoded folder for commands
        EXTENSIONS_DIR (str): The list of all files in the EXTENSIONS_DIR_NAME folder
        FUNCTIONS_DIR_NAME (str):The hardcoded folder for functions
        FUNCTIONS_DIR (str):The list of all files in the FUNCTIONS_DIR_NAME folder
    """

    CONFIG_PATH: str = os.environ.get("CONFIG_YML", "./config.yml")
    EXTENSIONS_DIR_NAME: str = "commands"
    EXTENSIONS_DIR: str = (
        f"{os.path.join(os.path.dirname(__file__))}/{EXTENSIONS_DIR_NAME}"
    )
    FUNCTIONS_DIR_NAME: str = "functions"
    FUNCTIONS_DIR: str = (
        f"{os.path.join(os.path.dirname(__file__))}/{FUNCTIONS_DIR_NAME}"
    )

    def __init__(
        self: Self, intents: discord.Intents, allowed_mentions: discord.AllowedMentions
    ) -> None:
        # Sets a few properires to None to avoid ValueErrors later on
        self.startup_time: datetime = None
        self.owner: discord.User = None
        self.guild_config_lock = None
        self.db = None
        self.file_config = None

        # Sets up some dicts and arrays
        self.guild_configs: dict[str, munch.Munch] = {}
        self.extension_configs = munch.DefaultMunch(None)
        self.extension_states = munch.DefaultMunch(None)
        self.command_rate_limit_bans: expiringdict.ExpiringDict[str, bool] = (
            expiringdict.ExpiringDict(
                max_len=5000,
                max_age_seconds=600,
            )
        )
        self.command_execute_history: dict[str, dict[int, bool]] = {}

        # Loads the file config, which includes things like the token
        self.load_file_config()

        # Call the discord.py init function to create a new commands.Bot object
        super().__init__(
            command_prefix=self.get_prefix,
            intents=intents,
            allowed_mentions=allowed_mentions,
        )

        # Setup the regular or delayed logger, depending on the file config
        if self.file_config.logging.queue_enabled:
            self.logger = botlogging.DelayedLogger(
                discord_bot=self,
                name=self.__class__.__name__,
                send=not self.file_config.logging.block_discord_send,
                wait_time=self.file_config.logging.queue_wait_seconds,
            )
        else:
            self.logger = botlogging.BotLogger(
                discord_bot=self,
                name=self.__class__.__name__,
                send=not self.file_config.logging.block_discord_send,
            )

        # Creates a http calls class and a reference to it to the bot
        self.http_functions = http.HTTPCalls(self)

        # Set the app command on error function to log errors in slash commands
        self.tree.on_error = self.on_app_command_error

        # On interaction will allow us to not run commands if they are disabled
        # And log the calling of commands
        # This will basically replace can_run and on_command for prefix commands
        self.tree.interaction_check = self.interaction_check

    # Entry point

    async def start(self: Self) -> None:
        """Starts the bot, connects to discord, irc, and postgres
        This function should not be used to interact with discord in any way
        Any discord interactions should be done with setup_hook
        """

        if isinstance(self.logger, botlogging.DelayedLogger):
            self.logger.register_queue()
            asyncio.create_task(self.logger.run())

        # Start the IRC bot in an asynchronous task
        irc_config = self.file_config.api.irc
        if irc_config.enable_irc:
            await self.logger.send_log(
                message="Connecting to IRC...", level=LogLevel.DEBUG, console_only=True
            )
            # Make the IRC class in such a way to allow reload without desctruction
            # We need to pass it the running loop so it can interact with discord
            await self.start_irc()

        # this is required for the bot
        await self.logger.send_log(
            message="Connecting to Postgres...", level=LogLevel.DEBUG, console_only=True
        )
        self.db = await self.get_postgres_ref()

        await self.logger.send_log(
            message="Logging into Discord...", level=LogLevel.DEBUG, console_only=True
        )
        self.guild_config_lock = asyncio.Lock()
        await super().start(self.file_config.bot_config.auth_token)

    # Discord.py called functions

    async def setup_hook(self: Self) -> None:
        """This function is automatically called after the bot has been logged into discord
        This creates postgres tables if needed, registers new guild configs if needed,
        Loads extensions, registers the custom help command
        and loads guild configs from the database.

        This function is called only one time, and should never be manually called
        """

        # We have to remove the built in help command
        await self.logger.send_log(
            message="Loading Help commands...", level=LogLevel.DEBUG, console_only=True
        )
        self.remove_command("help")

        # Get all the tables setup and create them all if needed
        await self.logger.send_log(
            message="Syncing Postgres tables...",
            level=LogLevel.DEBUG,
            console_only=True,
        )
        self.models = munch.DefaultMunch(None)
        databases.setup_models(self)
        await self.db.gino.create_all()

        # Load all guild config objects into self.guild_configs object
        all_config = await self.models.Config.query.gino.all()
        for config in all_config:
            self.guild_configs[config.guild_id] = munch.munchify(
                json.loads(config.config)
            )

        # Adds persistent views to the bot
        self.add_view(ui.VotingButtonPersistent())

        # The very last step should be loading extensions
        # Some extensions will require the database or config when loading
        await self.logger.send_log(
            message="Loading extensions...", level=LogLevel.DEBUG, console_only=True
        )
        self.extension_name_list = []
        await self.load_extensions()

    async def on_guild_join(self: Self, guild: discord.Guild) -> None:
        """Configures a new guild upon joining.
        This registers a new guild config, and starts any loop jobs that are configured

        Args:
            guild (discord.Guild): the guild that was joined
        """
        self.register_new_guild_config(str(guild.id))
        for cog in self.cogs.values():
            if getattr(cog, "COG_TYPE", "").lower() == "loop":
                try:
                    await cog.register_new_tasks(guild)
                except Exception as exception:
                    await self.logger.send_log(
                        message="Could not register loop tasks for cog on guild join",
                        level=LogLevel.ERROR,
                        context=LogContext(guild=guild),
                        exception=exception,
                    )

    async def on_ready(self: Self) -> None:
        """Callback for when the bot is finished starting up.
        This function may be called more than once and should not have discord interactions in it
        """
        self.startup_time = datetime.datetime.utcnow()
        await self.logger.send_log(
            message="Bot online", level=LogLevel.INFO, console_only=True
        )
        await self.get_owner()

        # Ensure all guilds have a config
        for guild in self.guilds:
            await self.register_new_guild_config(str(guild.id))

    # DM Logging

    async def log_DM(self: Self, sent_from: str, source: str, content: str) -> None:
        """Logs a DM from any source

        Args:
            sent_from (str): The username of the person who DMed the bot
            source (str): What bot the person DMed
            content (str): The string contents of the message recieved
        """
        owner = await self.get_owner()
        embed = auxiliary.generate_basic_embed(
            f"{source} recieved a PM", f"PM from: {sent_from}\n{content}"
        )
        embed.timestamp = datetime.datetime.utcnow()
        try:
            await owner.send(embed=embed)
        except discord.Forbidden as exception:
            await self.logger.send_log(
                message="Could not DM discord bot owner",
                level=LogLevel.ERROR,
                exception=exception,
            )

    async def on_message(self: Self, message: discord.Message) -> None:
        """Logs DMs and ensure that commands are processed

        Args:
            message (discord.Message): the message object
        """
        owner = await self.get_owner()
        if (
            owner
            and isinstance(message.channel, discord.DMChannel)
            and message.author.id != owner.id
            and not message.author.bot
        ):
            attachment_urls = ", ".join(a.url for a in message.attachments)
            content_string = f'"{message.content}"' if message.content else ""
            attachment_string = f"({attachment_urls})" if attachment_urls else ""
            await self.log_DM(
                message.author,
                "Main Discord Bot",
                f"{content_string} {attachment_string}",
            )

        await self.process_commands(message)

    # Guild config management functions

    async def register_new_guild_config(self: Self, guild_id: str) -> bool:
        """This creates a config for a new guild if needed

        Args:
            guild_id (str): The id of the guild to create config for, in string form

        Returns:
            bool: True if a config was created, False if a config already existed
        """
        async with self.guild_config_lock:
            try:
                config = self.guild_configs[guild_id]
            except KeyError:
                config = None
            if not config:
                await self.create_new_context_config(guild_id)
                return True
            return False

    async def create_new_context_config(self: Self, guild_id: str) -> munch.Munch:
        """Creates a new guild config for a given guild.

        Args:
            guild_id (str): The guild ID the config will be for. Only used for storing the config

        Returns:
            munch.Munch: The new config object ready to use
        """
        extensions_config = munch.DefaultMunch(None)

        for extension_name, extension_config in self.extension_configs.items():
            if extension_config:
                # don't attach to guild config if extension isn't configurable
                extensions_config[extension_name] = extension_config.data
        self.extension_name_list.sort()

        config_ = munch.DefaultMunch(None)

        config_.guild_id = str(guild_id)
        config_.command_prefix = self.file_config.bot_config.default_prefix
        config_.logging_channel = None
        config_.member_events_channel = None
        config_.guild_events_channel = None
        config_.private_channels = []
        config_.enabled_extensions = self.extension_name_list
        config_.nickname_filter = False
        config_.enable_logging = True
        config_.rate_limit = munch.DefaultMunch(None)
        config_.rate_limit.enabled = False
        config_.rate_limit.commands = 4
        config_.rate_limit.time = 10

        config_.extensions = extensions_config

        try:
            await self.logger.send_log(
                message=f"Inserting new config for lookup key: {guild_id}",
                level=LogLevel.DEBUG,
                context=LogContext(guild=self.get_guild(guild_id)),
                console_only=True,
            )
            # Modify the database
            await self.write_new_config(str(guild_id), json.dumps(config_))

            # Modify the local cache
            self.guild_configs[guild_id] = config_

        except Exception as exception:
            # safely finish because the new config is still useful
            await self.logger.send_log(
                message="Could not insert guild config into Postgres",
                level=LogLevel.ERROR,
                context=LogContext(guild=self.get_guild(guild_id)),
                exception=exception,
            )

        return config_

    async def write_new_config(self: Self, guild_id: str, config: str) -> None:
        """Takes a config and guild and updates the config in the database
        This is only needed when a new guild is joined or the config is modifed

        Args:
            guild_id (str): The str ID of the guild the config belongs to
            config (str): The str representation of the json config
        """
        database_config = await self.models.Config.query.where(
            self.models.Config.guild_id == guild_id
        ).gino.first()
        if database_config:
            await database_config.update(
                config=str(config), update_time=datetime.datetime.utcnow()
            ).apply()
        else:
            new_database_config = self.models.Config(
                guild_id=str(guild_id),
                config=str(config),
            )
            await new_database_config.create()

    def add_extension_config(
        self: Self, extension_name: str, config: extensionconfig.ExtensionConfig
    ) -> None:
        """Adds an extensions defined config to the guild config as a whole

        Args:
            extension_name (str): The name of the extension to add config for.
                Will be the key in the config file
            config (extensionconfig.ExtensionConfig): The config class with all
                of the config keys to add

        Raises:
            ValueError: Will be raised if config is not an extensionconfig.ExtensionConfig
        """
        if not isinstance(config, extensionconfig.ExtensionConfig):
            raise ValueError("config must be of type extensionconfig.ExtensionConfig")
        self.extension_configs[extension_name] = config

    async def get_log_channel_from_guild(
        self: Self, guild: discord.Guild, key: str
    ) -> str | None:
        """Gets the log channel ID associated with the given guild.

        This also checks if the channel exists in the correct guild.

        Args:
            guild (discord.Guild): the guild object to reference
            key (str): the key to use when looking up the channel

        Returns:
            str | None: If the log channel exists, this will be the string of the ID
                Otherwise it will be None
        """
        if not guild:
            return None

        config = self.guild_configs[str(guild.id)]
        channel_id = config.get(key)

        if not channel_id:
            return None

        if not guild.get_channel(int(channel_id)):
            return None

        return channel_id

    # File config loading functions

    def load_file_config(self: Self, validate: bool = True) -> None:
        """Loads the config yaml file into a bot object.

        Args:
            validate (bool): True if validations should be ran on the file
        """
        with open(self.CONFIG_PATH, encoding="utf8") as iostream:
            config_ = yaml.safe_load(iostream)

        self.file_config = munch.munchify(config_)

        self.file_config.bot_config.disabled_extensions = (
            self.file_config.bot_config.disabled_extensions or []
        )

        if not validate:
            return

        for subsection in ["required"]:
            self.validate_bot_config_subsection("bot_config", subsection)

    def validate_bot_config_subsection(
        self: Self, section: str, subsection: str
    ) -> None:
        """Loops through a config subsection to check for missing values.

        Args:
            section (str): the section name containing the subsection
            subsection (str): the subsection name

        Raises:
            ValueError: If the subsection validating is missing any keys
        """
        for key, value in self.file_config.get(section, {}).get(subsection, {}).items():
            error_key = None
            if value is None:
                error_key = key
            elif isinstance(value, dict):
                for dict_key, dict_value in value.items():
                    if dict_value is None:
                        error_key = dict_key
            if error_key:
                raise ValueError(
                    f"Config key {error_key} from {section}.{subsection} not supplied"
                )

    # Error handling and logging functions

    async def on_app_command_error(
        self: Self,
        interaction: discord.Interaction,
        error: app_commands.AppCommandError,
    ) -> None:
        """This is called upon any error originating from an app command

        Args:
            interaction (discord.Interaction): The interaction where the error occured at
            error (app_commands.AppCommandError): The error object that occured
        """
        error_message = await self.handle_error(
            exception=error, channel=interaction.channel, guild=interaction.guild
        )

        if not error_message:
            return

        embed = auxiliary.prepare_deny_embed(message=error_message)

        if interaction.response.is_done():
            await interaction.followup.send(embed=embed)
        else:
            await interaction.response.send_message(embed=embed)

    async def handle_error(
        self: Self,
        exception: Exception,
        channel: discord.abc.Messageable,
        guild: discord.Guild,
    ) -> str:
        """Handles the formatting and logging of command and app command errors

        Args:
            exception (Exception): The exception object generated
            channel (discord.abc.Messageable): The channel the command was run in
            guild (discord.Guild): The guild the command was run in

        Returns:
            str: The pretty string format that should be shared with the user
        """
        # Get the custom error response we made for the error
        message_template = custom_errors.COMMAND_ERROR_RESPONSES.get(
            exception.__class__, ""
        )
        # see if we have mapped this error to no response (None)
        # or if we have added it to the global ignore list of errors
        if (
            message_template is None
            or exception.__class__ in custom_errors.IGNORED_ERRORS
        ):
            return
        # otherwise set it a default error message
        if message_template == "":
            message_template = custom_errors.ErrorResponse()

        error_message = message_template.get_message(exception)

        log_channel = await self.get_log_channel_from_guild(
            guild=guild, key="logging_channel"
        )

        # Ensure that error messages aren't too long.
        # This ONLY changes the user facing error, the stack trace isn't impacted
        if len(error_message) > 1000:
            error_message = error_message[:1000]
            error_message += "..."

        # Only log stack trace if you should
        if not getattr(exception, "dont_print_trace", False):
            await self.logger.send_log(
                message=f"Command error: {exception}",
                level=LogLevel.ERROR,
                channel=log_channel,
                context=LogContext(guild=guild, channel=channel),
                exception=exception,
            )

        # Return the string error message and allow the context/interaction to respond properly
        return error_message

    async def on_command_error(
        self: Self, context: commands.Context, exception: Exception
    ) -> None:
        """Catches command errors and sends them to the error logger for processing.

        Args:
            context (commands.Context): the context associated with the exception
            exception (Exception): the exception object associated with the error
        """
        if self.extra_events.get("on_command_error", None):
            return
        if hasattr(context.command, "on_error"):
            return
        if context.cog:
            # pylint: disable=protected-access
            if (
                commands.Cog._get_overridden_method(context.cog.cog_command_error)
                is not None
            ):
                return

        error_message = await self.handle_error(
            exception=exception, channel=context.channel, guild=context.guild
        )
        if not error_message:
            return

        await auxiliary.send_deny_embed(message=error_message, channel=context.channel)

    # Postgres setup function

    async def get_postgres_ref(self: Self) -> gino.GinoEngine:
        """Connects to postgres based on the database login defined the in the file_config
        Adds self.db to be the database reference

        Returns:
            gino.GinoEngine: The database connection as a gino object
        """
        await self.logger.send_log(
            message="Obtaining and binding to Gino instance",
            level=LogLevel.DEBUG,
            console_only=True,
        )

        db_ref = gino.Gino()

        # Pull information from postgres out of the file config
        config_child = self.file_config.database.postgres
        user = config_child.user
        password = config_child.password
        name = config_child.name
        host = config_child.host
        port = config_child.port

        db_url = f"postgres://{user}:{password}@{host}:{port}/{name}"
        url_filtered = f"postgres://{user}:********@{host}:{port}/{name}"

        # don't log the password
        await self.logger.send_log(
            message=f"Generated DB URL: {url_filtered}",
            level=LogLevel.DEBUG,
            console_only=True,
        )

        await db_ref.set_bind(db_url)

        db_ref.Model.__table_args__ = {"extend_existing": True}

        return db_ref

    # Extension loading and management functions

    async def get_potential_extensions(self: Self) -> list[str]:
        """Gets the current list of extensions in the defined directory.
        This ONLY gets commands, not functions

        Returns:
            list[str]: Gets a list of the string names of every python file
                in the commands folder
        """

        self.logger.console.info(f"Searching {self.EXTENSIONS_DIR} for extensions")
        extensions_list = [
            os.path.basename(f)[:-3]
            for f in glob.glob(f"{self.EXTENSIONS_DIR}/*.py")
            if os.path.isfile(f) and not f.endswith("__init__.py")
        ]
        return extensions_list

    async def get_potential_function_extensions(self: Self) -> list[str]:
        """Gets the current list of extensions in the defined directory.
        This ONLY gets functions, not commands

        Returns:
            list[str]: Gets a list of the string names of every python file
                in the functions folder
        """
        self.logger.console.info(f"Searching {self.FUNCTIONS_DIR} for extensions")
        extensions_list = [
            os.path.basename(f)[:-3]
            for f in glob.glob(f"{self.FUNCTIONS_DIR}/*.py")
            if os.path.isfile(f) and not f.endswith("__init__.py")
        ]
        return extensions_list

    async def load_extensions(self: Self, graceful: bool = True) -> None:
        """Loads all extensions currently in the extensions directory.

        Args:
            graceful (bool, optional): True if extensions should gracefully fail to load.
                Defaults to True.

        Raises:
            exception: If graceful is false, this will raise ANY
                exception generated by loading extensions
        """

        self.logger.console.debug("Retrieving commands")
        for extension_name in await self.get_potential_extensions():
            if extension_name in self.file_config.bot_config.disabled_extensions:
                self.logger.console.debug(
                    f"{extension_name} is disabled on startup - ignoring load"
                )
                continue

            try:
                await self.load_extension(
                    f"{self.EXTENSIONS_DIR_NAME}.{extension_name}"
                )
                self.extension_name_list.append(extension_name)
            except Exception as exception:
                self.logger.console.error(
                    f"Failed to load extension {extension_name}: {exception}"
                )
                if not graceful:
                    raise exception

        self.logger.console.debug("Retrieving functions")
        for extension_name in await self.get_potential_function_extensions():
            if extension_name in self.file_config.bot_config.disabled_extensions:
                self.logger.console.debug(
                    f"{extension_name} is disabled on startup - ignoring load"
                )
                continue

            try:
                await self.load_extension(f"{self.FUNCTIONS_DIR_NAME}.{extension_name}")
                self.extension_name_list.append(extension_name)
            except Exception as exception:
                self.logger.console.error(
                    f"Failed to load extension {extension_name}: {exception}"
                )
                if not graceful:
                    raise exception

    def get_command_extension_name(self: Self, command: commands.Command) -> str:
        """Gets the subname of an extension from a command.
        Used only for commands, should never be run for a function

        Args:
            command (commands.Command): the command to reference

        Returns:
            str: The name of the extension name that houses a prefix command.
        """
        if not command.module.startswith(f"{self.EXTENSIONS_DIR_NAME}."):
            return None
        extension_name = command.module.split(".")[1]
        return extension_name

    async def register_file_extension(
        self: Self, extension_name: str, fp: io.BufferedIOBase
    ) -> None:
        """Offers an interface for loading an extension from an external source.

        This saves the external file data to the OS, without any validation.

        Args:
            extension_name (str): the name of the extension to register
            fp (io.BufferedIOBase): the file-like object to save to disk

        Raises:
            NameError: Raised if no extension name is provided
        """
        if not extension_name:
            raise NameError("Invalid extension name")

        try:
            await self.unload_extension(f"{self.EXTENSIONS_DIR_NAME}.{extension_name}")
        except commands.errors.ExtensionNotLoaded:
            pass

        with open(f"{self.EXTENSIONS_DIR}/{extension_name}.py", "wb") as file_handle:
            file_handle.write(fp)

    # Bot properties

    async def is_bot_admin(self: Self, member: discord.Member) -> bool:
        """Processes command context against admin/owner data.
        Command checks are disabled if the context author is the owner.
        They are also ignored if the author is bot admin in the config.

        Args:
            member (discord.Member): the context associated with the command

        Returns:
            bool: True if the member is a bot admin. False if it isn't
        """
        await self.logger.send_log(
            message="Checking context against bot admins",
            level=LogLevel.DEBUG,
            context=LogContext(guild=member.guild),
            console_only=True,
        )
        owner = await self.get_owner()
        if getattr(owner, "id", None) == member.id:
            return True

        if member.id in [int(id) for id in self.file_config.bot_config.admins.ids]:
            return True

        role_is_admin = False
        for role in getattr(member, "roles", []):
            if role.name in self.file_config.bot_config.admins.roles:
                role_is_admin = True
                break
        if role_is_admin:
            return True

        return False

    async def get_owner(self: Self) -> discord.User | None:
        """Gets the owner object from the bot application.

        Returns:
            discord.User | None: The User object of the owner of the application on discords side
        """
        if self.file_config.bot_config.override_owner:
            self.owner = await self.fetch_user(
                int(self.file_config.bot_config.override_owner)
            )
            return self.owner

        if not self.owner:
            try:
                # If this isn't console only, it is a forever recursion
                await self.logger.send_log(
                    message="Looking up bot owner",
                    level=LogLevel.DEBUG,
                    console_only=True,
                )
                app_info = await self.application_info()
                self.owner = app_info.owner
            except discord.errors.HTTPException:
                self.owner = None

        return self.owner

    async def get_prefix(self: Self, message: discord.Message) -> str:
        """Gets the appropriate prefix for a command.
        This is called by discord.py and MUST be async

        Args:
            message (discord.Message): the message to check against

        Returns:
            str: The string of the command prefix by the bot, for the given guild
        """
        guild_config = self.guild_configs[str(message.guild.id)]
        return getattr(
            guild_config, "command_prefix", self.file_config.bot_config.default_prefix
        )

    # Can run command checks

    async def command_run_admin_check(self: Self, member: discord.Member) -> bool:
        """Part of the can_run function set. This is responsible for checking if
        the caller is a bot admin

        Args:
            member (discord.Member): The member who called the command

        Returns:
            bool: True if they are bot admin, false if they aren't
        """
        return await self.is_bot_admin(member)

    def command_run_rate_limit_check(
        self: Self, member: discord.Member, guild: discord.Guild, command_id: int
    ) -> bool:
        """Handle the command rate limiter

        Args:
            member (discord.Member): The member who called the command
            guild (discord.Guild): The guild it was called in
            command_id (int): The ID of the message or interaction

        Returns:
            bool: True if the command should be run, False if under rate limit
        """
        # Assume this is only run if rate limit is enabled
        config = self.guild_configs[str(guild.id)]
        identifier = f"{member.id}-{guild.id}"

        # If this person hasn't run a command in the rate_limit.time
        # We will need to add them to the execute history
        if identifier not in self.command_execute_history:
            self.command_execute_history[identifier] = expiringdict.ExpiringDict(
                max_len=20,
                max_age_seconds=config.rate_limit.time,
            )

        # Ensure that a single command is only ever counted once
        if command_id not in self.command_execute_history[identifier]:
            self.command_execute_history[identifier][command_id] = True

        # Ban the person if they are over the rate limit
        if len(self.command_execute_history[identifier]) > config.rate_limit.commands:
            self.command_rate_limit_bans[identifier] = True

        # If this person is banned, raise an error
        if (
            identifier in self.command_rate_limit_bans
            and not member.guild_permissions.administrator
        ):
            return False

        # Otherwise, return True
        return True

    def command_run_extension_disabled_check(
        self: Self, guild: discord.Guild, extension_name: str
    ) -> bool:
        """Checks if the extension is disabled
        Works for both prefix and slash commands

        Args:
            guild (discord.Guild): The guild the command was run in
            extension_name (str): The name of the extension to check

        Returns:
            bool: False if disabled, True if enabled
        """
        config = self.guild_configs[str(guild.id)]
        if extension_name not in config.enabled_extensions:
            return False
        return True

    # Logging and checking if commands can run

    async def interaction_check(self: Self, interaction: discord.Interaction) -> bool:
        """This is a default function of the command tree that always returns true
        We can use this to log and evaluate if commands should be run

        Args:
            interaction (discord.Interaction): The interaction that started the command

        Raises:
            AppCommandExtensionDisabled: Raised if the guild config hasn't enabled
                the extension belonging to this command
            AppCommandRateLimit: Raised if the command is enabled,
                but the user is under rate limit restrictions

        Returns:
            bool: True if the command should be run, false if it shouldn't be run
        """

        # Since we can't do it anywhere else, log slash command here
        await self.slash_command_log(interaction)

        await self.logger.send_log(
            message="Checking if slash command can run",
            level=LogLevel.DEBUG,
            context=LogContext(guild=interaction.guild, channel=interaction.channel),
            console_only=True,
        )
        config = self.guild_configs[str(interaction.guild.id)]

        # Check 1 - Ensure extension is enabled
        try:
            extension_name = interaction.command.extras["module"]
        except KeyError:
            # Skip extension enabled check if no extras module has been defined
            self.logger.console.warning(
                "No module has been defined, skipping extension enabled check"
            )
            extension_name = None

        if extension_name:
            # If the extension is disabled, raise an error to show it and block execution
            if not self.command_run_extension_disabled_check(
                interaction.guild, extension_name
            ):
                raise custom_errors.AppCommandExtensionDisabled

        # Check 2 - Approve if invoker is bot admin
        result = await self.command_run_admin_check(interaction.user)
        if result:
            return result

        # Check 3 - If rate limiter is enabled, run through the rate limiter
        # If the user is under a rate limit, raise an error to show it and block execution
        if config.rate_limit.get("enabled", False):
            if not self.command_run_rate_limit_check(
                member=interaction.user,
                guild=interaction.guild,
                command_id=interaction.id,
            ):
                raise custom_errors.AppCommandRateLimit

        # Finally, return the default check, which is always True
        return True

    async def slash_command_log(self: Self, interaction: discord.Interaction) -> None:
        """A command to log the call of a slash command

        Args:
            interaction (discord.Interaction): The interaction the slash command generated
        """
        embed = discord.Embed()
        embed.add_field(name="User", value=interaction.user)
        embed.add_field(
            name="Channel", value=getattr(interaction.channel, "name", "DM")
        )
        embed.add_field(name="Server", value=getattr(interaction.guild, "name", "None"))
        embed.add_field(name="Namespace", value=f"{interaction.namespace}")
        embed.set_footer(text=f"Requested by {interaction.user.id}")

        log_channel = await self.get_log_channel_from_guild(
            interaction.guild, key="logging_channel"
        )

        sliced_content = interaction.command.qualified_name[:100]
        message = f"Command detected: `/{sliced_content}`"

        await self.logger.send_log(
            message=message,
            level=LogLevel.INFO,
            context=LogContext(guild=interaction.guild, channel=interaction.channel),
            channel=log_channel,
            embed=embed,
        )

    async def can_run(
        self: Self, ctx: commands.Context, *, call_once: bool = False
    ) -> bool:
        """Wraps the default can_run check to:
        Evaluate bot admin permissions
        Add a rate limiter
        Check if extension is disabled

        Args:
            ctx (commands.Context): The context associated with the command
            call_once (bool, optional): True if the check should be retrieved from the
                call_once attribute. Defaults to False.

        Raises:
            ExtensionDisabled: Raised if the extension holding the command is disabled
            CommandRateLimit: Raised if the user is under rate limit

        Returns:
            bool: True if the user can run the command, False otherwise
        """

        await self.logger.send_log(
            message="Checking if prefix command can run",
            level=LogLevel.DEBUG,
            context=LogContext(guild=ctx.guild, channel=ctx.channel),
            console_only=True,
        )
        config = self.guild_configs[str(ctx.guild.id)]

        # Check 1 - Ensure extension is enabled
        extension_name = self.get_command_extension_name(ctx.command)
        if extension_name:
            # If the extension is disabled, raise an error to show it and block execution
            if not self.command_run_extension_disabled_check(ctx.guild, extension_name):
                raise custom_errors.ExtensionDisabled

        # Check 2 - Approve if invoker is bot admin
        result = await self.command_run_admin_check(ctx.author)
        if result:
            return result

        # Check 3 - If rate limiter is enabled, run through the rate limiter
        if config.rate_limit.get("enabled", False):
            # If the user is under a rate limit, raise an error to show it and block execution
            if not self.command_run_rate_limit_check(
                member=ctx.author, guild=ctx.guild, command_id=ctx.message.id
            ):
                raise custom_errors.CommandRateLimit

        # Finally, return the default check
        return await super().can_run(ctx, call_once=call_once)

    # IRC Stuff

    async def start_irc(self: Self) -> None:
        """Starts the IRC bot in a separate thread."""
        main_loop = asyncio.get_running_loop()
        irc_config = self.file_config.api.irc

        # Create the bot instance
        irc_bot = ircrelay.relay.IRCBot(
            loop=main_loop,
            server=irc_config.server,
            port=irc_config.port,
            channels=irc_config.channels,
            username=irc_config.name,
            password=irc_config.password,
        )

        self.irc = irc_bot

        def run_in_thread() -> None:
            """Run the IRC bot in a separate thread."""
            irc_bot.start_bot()

        # Start the bot in a new thread
        await self.logger.send_log(
            message="Logging into IRC", level=LogLevel.INFO, console_only=True
        )

        bot_thread = threading.Thread(target=run_in_thread)
        bot_thread.start()