main.py

import asyncio
import logging
import os
from typing import Any

import discord
import pyfiglet
from disckit import CogEnum, UtilConfig
from disckit.cogs import dis_load_extension
from disckit.utils import MainEmbed
from dotenv import load_dotenv

from core import Bot
from core.config import (
    COG_DIR,
    ERROR_COLOR,
    FOOTER_TEXT,
    LOG_CHANNEL,
    MAIN_COLOR,
    SUCCESS_COLOR,
    SYNC_GUILD_ID,
    BotData,
)
from core.emojis import GREEN_CHECK, RED_CROSS
from core.utils import setup_logging

# Load environment variables from a `.env` file
load_dotenv()

# Set up logging for the application
setup_logging()

_logger: logging.Logger = logging.getLogger(__name__)

# Retrieve the Discord bot token from the environment variables
TOKEN = os.getenv("TOKEN")

# Set up Discord intents for the bot
intents = discord.Intents.all()


async def load_cogs(bot: Bot) -> None:
    """
    Load all cogs from the specified cog directory.

    Parameters
    ----------
    bot : Bot
        The bot instance to which the cogs will be loaded.
    """
    for folder in os.listdir(COG_DIR):
        for cog in os.listdir(COG_DIR + "/" + folder):
            if cog.endswith(".py"):
                cog_path = COG_DIR + "." + folder + "." + cog[:-3]
                await bot.load_extension(cog_path)


async def custom_status(bot: Any, *args: Any) -> tuple[str, ...]:
    """
    Generate a custom status for the bot.

    Parameters
    ----------
    bot : Any
        The bot instance for which the status is being generated.
    *args : Any
        Additional arguments for the status function.

    Returns
    -------
    tuple of str
        A tuple containing status messages to be displayed.
    """
    return (
        f"version {BotData.VERSION}",
        "Another Status Line Here",
        "Use /help for commands",
    )


async def main() -> None:
    """
    The main entry point for the bot application.

    This function initializes the bot, configures settings, loads cogs,
    and starts the bot using the provided token.

    Raises
    ------
    ValueError
        If the Discord bot token is not set in the environment variables.
    """
    bot = Bot(intents=intents)

    # Set up Discord's internal logging
    discord.utils.setup_logging()

    # Display a banner for the bot
    print(pyfiglet.figlet_format("Disckit Example Bot"))  # pyright: ignore[reportUnknownMemberType]

    # Configure utility settings for the bot
    UtilConfig.MAIN_COLOR = MAIN_COLOR
    UtilConfig.SUCCESS_COLOR = SUCCESS_COLOR
    UtilConfig.ERROR_COLOR = ERROR_COLOR

    UtilConfig.SUCCESS_EMOJI = GREEN_CHECK
    UtilConfig.ERROR_EMOJI = RED_CROSS

    UtilConfig.FOOTER_IMAGE = BotData.AVATAR_URL
    UtilConfig.FOOTER_TEXT = FOOTER_TEXT

    UtilConfig.STATUS_FUNC = (custom_status, ())
    UtilConfig.STATUS_TYPE = discord.ActivityType.listening
    UtilConfig.STATUS_COOLDOWN = 600

    UtilConfig.BUG_REPORT_CHANNEL = LOG_CHANNEL
    UtilConfig.OWNER_LIST_URL = (
        "https://images.disutils.com/bot_assets/assets/owners.txt"
    )

    UtilConfig.OWNER_ONLY_HELP_COGS = ("owner commands",)

    UtilConfig.IGNORE_HELP_COGS = (
        "owner commands",
        "help cog",
        "status handler",
        "error handler",
        "owner id handler",
        "user count handler",
    )

    UtilConfig.OVERVIEW_HELP_EMBED = MainEmbed(
        title="Bot's Overview",
        description=(
            "Welcome to the help overview of the bot.\n"
            "Use the select menu from below to see the description of different command groups."
        ),
    )

    UtilConfig.HELP_OWNER_GUILD_ID = SYNC_GUILD_ID

    # Load cogs and extensions
    await load_cogs(bot=bot)
    await dis_load_extension(
        bot,
        CogEnum.ERROR_HANDLER,
        CogEnum.STATUS_HANDLER,
        CogEnum.OWNER_IDS_HANDLER,
        CogEnum.HELP_COG,
    )

    # Ensure TOKEN is not None
    if not TOKEN:
        raise ValueError(
            "Discord bot token is not set in the environment variables."
        )

    # Start the bot using the token
    await bot.start(TOKEN)


if __name__ == "__main__":
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        _logger.info("KeyboardInterrupt detected.")
    except Exception as e:
        _logger.exception(f"An error occurred: {e}")