mirror of
https://github.com/wlinator/luminara.git
synced 2024-10-02 20:43:12 +00:00
112 lines
3.8 KiB
Python
112 lines
3.8 KiB
Python
import sys
|
|
import traceback
|
|
from typing import Any
|
|
|
|
import discord
|
|
from discord import app_commands
|
|
from discord.ext import commands
|
|
from loguru import logger
|
|
|
|
from lib import exceptions
|
|
from lib.client import Luminara
|
|
from lib.const import CONST
|
|
|
|
error_map: dict[type[Exception], str] = {
|
|
commands.BotMissingPermissions: CONST.STRINGS["error_bot_missing_permissions_description"],
|
|
commands.MissingPermissions: CONST.STRINGS["error_missing_permissions_description"],
|
|
commands.NoPrivateMessage: CONST.STRINGS["error_no_private_message_description"],
|
|
commands.NotOwner: CONST.STRINGS["error_not_owner_unknown"],
|
|
commands.PrivateMessageOnly: CONST.STRINGS["error_private_message_only_description"],
|
|
exceptions.BirthdaysDisabled: CONST.STRINGS["error_birthdays_disabled_description"],
|
|
}
|
|
|
|
|
|
async def on_error(event: str, *args: Any, **kwargs: Any) -> None:
|
|
logger.exception(
|
|
f"on_error INFO: errors.event.{event} | '*args': {args} | '**kwargs': {kwargs}",
|
|
)
|
|
logger.exception(f"on_error EXCEPTION: {sys.exc_info()}")
|
|
traceback.print_exc()
|
|
|
|
|
|
async def log_command_error(
|
|
user_name: str,
|
|
command_name: str | None,
|
|
guild_id: int | None,
|
|
error: commands.CommandError | commands.CheckFailure | app_commands.AppCommandError,
|
|
command_type: str,
|
|
) -> None:
|
|
if isinstance(error, commands.NotOwner | exceptions.Blacklisted):
|
|
return
|
|
|
|
log_msg = f"{user_name} executed {command_type}{command_name or 'Unknown'}"
|
|
|
|
log_msg += " in DMs" if guild_id is None else f" | guild: {guild_id}"
|
|
|
|
if CONST.INSTANCE == "dev":
|
|
logger.exception(
|
|
f"{log_msg} | {error.__module__}.{error.__class__.__name__} | {''.join(traceback.format_exception(type(error), error, error.__traceback__))}",
|
|
)
|
|
else:
|
|
logger.error(f"{log_msg} | {error.__module__}.{error.__class__.__name__}")
|
|
|
|
|
|
class ErrorHandler(commands.Cog):
|
|
def __init__(self, bot: Luminara) -> None:
|
|
self.bot = bot
|
|
|
|
async def cog_load(self):
|
|
tree = self.bot.tree
|
|
self._old_tree_error = tree.on_error
|
|
tree.on_error = self.on_app_command_error
|
|
|
|
async def cog_unload(self):
|
|
tree = self.bot.tree
|
|
tree.on_error = self._old_tree_error
|
|
|
|
async def on_app_command_error(
|
|
self,
|
|
interaction: discord.Interaction,
|
|
error: app_commands.AppCommandError,
|
|
) -> None:
|
|
if isinstance(error, commands.CommandNotFound | exceptions.Blacklisted):
|
|
return
|
|
|
|
await log_command_error(
|
|
user_name=interaction.user.name,
|
|
command_name=interaction.command.qualified_name if interaction.command else None,
|
|
guild_id=interaction.guild.id if interaction.guild else None,
|
|
error=error,
|
|
command_type="/",
|
|
)
|
|
|
|
error_msg = error_map.get(type(error), str(error))
|
|
await interaction.response.send_message(content=f"❌ **{interaction.user.name}** {error_msg}", ephemeral=True)
|
|
|
|
@commands.Cog.listener()
|
|
async def on_command_error(
|
|
self,
|
|
ctx: commands.Context[Luminara],
|
|
error: commands.CommandError | commands.CheckFailure,
|
|
) -> None:
|
|
if isinstance(error, commands.CommandNotFound | exceptions.Blacklisted):
|
|
return
|
|
|
|
await log_command_error(
|
|
user_name=ctx.author.name,
|
|
command_name=ctx.command.qualified_name if ctx.command else None,
|
|
guild_id=ctx.guild.id if ctx.guild else None,
|
|
error=error,
|
|
command_type=".",
|
|
)
|
|
|
|
error_msg = error_map.get(type(error), str(error))
|
|
await ctx.send(content=f"❌ **{ctx.author.name}** {error_msg}")
|
|
|
|
@commands.Cog.listener()
|
|
async def on_error(self, event: str, *args: Any, **kwargs: Any) -> None:
|
|
await on_error(event, *args, **kwargs)
|
|
|
|
|
|
async def setup(bot: Luminara) -> None:
|
|
await bot.add_cog(ErrorHandler(bot))
|