PyDrocsid/CrypticBot

from typing import Iterable
 
import sentry_sdk
from discord import Guild, Intents, Message
from discord.ext.commands import Bot, CommandError, CommandInvokeError, CommandNotFound, Context, UserInputError
 
from PyDrocsid.cog import load_cogs
from PyDrocsid.command import make_error, reply
from PyDrocsid.database import db
from PyDrocsid.environment import TOKEN
from PyDrocsid.events import listener
from PyDrocsid.logger import get_logger
from PyDrocsid.prefix import get_prefix
from PyDrocsid.translations import t
 
from cogs.custom.server_info import CustomServerInfoCog
from cogs.library.administration import PermissionsCog, RolesCog, SettingsCog, SudoCog
from cogs.library.general import CustomCommandsCog, PollsCog, ReactionRoleCog, UtilsCog, VoiceChannelCog
from cogs.library.information import BotInfoCog, HeartbeatCog, InactivityCog, UserInfoCog
from cogs.library.information.help.cog import HelpCog, send_help
from cogs.library.integrations import PythonDocsCog, RedditCog, RunCodeCog
from cogs.library.moderation import (
    AutoRoleCog,
    InvitesCog,
    LoggingCog,
    MessageCog,
    RoleNotificationsCog,
    SpamDetectionCog,
    ThreadsCog,
)
from cogs.library.moderation.mod.cog import ModCog, UserCommandError
from cogs.library.pubsub import send_alert
 
 
logger = get_logger(__name__)
 
t = t.g
 
 
async def fetch_prefix(_, msg: Message) -> Iterable[str]:
    prefix = [await get_prefix(), f"<@!{bot.user.id}> ", f"<@{bot.user.id}> "]
 
    if msg.guild is None:
        prefix.append("")
 
    return prefix
 
 
bot = Bot(command_prefix=fetch_prefix, case_insensitive=True, intents=(Intents.all()))
bot.remove_command("help")
 
 
@listener
async def on_ready():
    logger.info(f"\033[1m\033[36mLogged in as {bot.user}\033[0m")
 
 
@bot.event
async def on_error(*_, **__):
    sentry_sdk.capture_exception()
    raise  # skipcq: PYL-E0704
 
 
@listener
async def on_command_error(ctx: Context, error: CommandError):
    if isinstance(error, CommandInvokeError):
        await reply(ctx, embed=make_error(t.internal_error))
        raise error.original
 
    if isinstance(error, CommandNotFound) and ctx.guild is not None and ctx.prefix == await get_prefix():
        return
    if isinstance(error, UserInputError):
        await send_help(ctx, ctx.command)
    elif isinstance(error, UserCommandError):
        await reply(ctx, embed=make_error(str(error), error.user))
    else:
        await reply(ctx, embed=make_error(str(error)))
 
 
@listener
async def on_permission_error(guild: Guild, error: str):
    await send_alert(guild, error)
 
 
# fmt: off
load_cogs(
    bot,
 
    # Administration
    RolesCog(),
    PermissionsCog(),
    SettingsCog(),
    SudoCog(),
 
    # Moderation
    AutoRoleCog(),
    InvitesCog(),
    LoggingCog(),
    MessageCog(),
    ModCog(),
    RoleNotificationsCog(),
    SpamDetectionCog(),
    ThreadsCog(),
 
    # Information
    BotInfoCog(info_icon="https://github.com/cryptic-game.png"),
    HeartbeatCog(),
    HelpCog(),
    UserInfoCog(),
    CustomServerInfoCog(),
    InactivityCog(),
 
    # Integrations
    PythonDocsCog(),
    RedditCog(),
    RunCodeCog(),
 
    # General
    CustomCommandsCog(),
    PollsCog(team_roles=["head", "head_assistant"]),
    ReactionRoleCog(),
    UtilsCog(),
    VoiceChannelCog(team_roles=["head", "head_assistant"]),
)
# fmt: on
 
 
def run():
    bot.loop.run_until_complete(db.create_tables())
 
    logger.debug("logging in")
    bot.run(TOKEN)