python-app-rocketbot/rocketbot/bot.py

import traceback
from typing import Optional

from discord import Intents
from discord.ext import commands

from config import CONFIG
from rocketbot.cogsetting import CogSetting
from rocketbot.utils import bot_log, dump_stacktrace


class Rocketbot(commands.Bot):
	"""
	Bot subclass
	"""
	def __init__(self, **kwargs):
		# Bot requires command_prefix, even though we're only using slash commands,
		# not commands in message content. Giving an unlikely prefix of private-use
		# Unicode characters. This isn't a security thing, just avoiding showing a
		# "command not found" error if someone happens to start a message with
		# something more common.
		super().__init__(command_prefix='\uED1E\uEA75\uF00D', **kwargs)
		self.__commands_set_up = False
		self.__first_ready = True

	async def on_command_error(self, context: commands.Context, exception: BaseException) -> None:
		bot_log(None, None, f'Command error')
		dump_stacktrace(exception)
		if context.guild is None or \
				context.message.channel is None or \
				context.message.author.bot:
			return
		if not context.message.channel.permissions_for(context.message.author).ban_members:
			# Don't tell non-mods about errors
			return
		if isinstance(exception, (commands.errors.CommandError, )):
			# Reply with the error message for ordinary command errors
			await context.message.reply(
				f'{CONFIG["failure_emoji"]} {exception}',
				mention_author=False)
			return
		# Stack trace everything else

	async def on_error(self, event_method, args=None, kwargs=None):
		bot_log(None, None, 'Event caused error\n' + \
			f'	event method: {event_method}' + \
			f'	event args: {args}' + \
			f'	event kwargs: {kwargs}' + \
			traceback.format_exc())

	async def on_ready(self):
		if self.__first_ready:
			self.__first_ready = False
			if not self.__commands_set_up:
				await self.__set_up_commands()
			bot_log(None, None, 'Bot done initializing')
			bot_log(None, None, f"Type /help in Discord for available commands.")
			print('----------------------------------------------------------')

	async def __set_up_commands(self):
		if self.__commands_set_up:
			return
		self.__commands_set_up = True
		for cog in self.cogs.values():
			from rocketbot.cogs.basecog import BaseCog
			if isinstance(cog, BaseCog):
				bcog: BaseCog = cog
				if len(bcog.settings) > 0:
					CogSetting.set_up_all(bcog, self, bcog.settings)
		try:
			synced_commands = await self.tree.sync()
			for command in synced_commands:
				bot_log(None, None, f'Synced command: /{command.name}')
		except Exception as e:
			dump_stacktrace(e)

# Current active bot instance
rocketbot: Optional[Rocketbot] = None

def __create_bot():
	global rocketbot
	if rocketbot is not None:
		return
	bot_log(None, None, 'Creating bot...')
	intents = Intents.default()
	intents.messages = True
	intents.message_content = True
	intents.members = True
	intents.presences = True
	rocketbot = Rocketbot(intents=intents)
__create_bot()

from rocketbot.cogs.autokickcog import AutoKickCog
from rocketbot.cogs.bangcommandcog import BangCommandCog
from rocketbot.cogs.configcog import ConfigCog
from rocketbot.cogs.crosspostcog import CrossPostCog
from rocketbot.cogs.gamescog import GamesCog
from rocketbot.cogs.generalcog import GeneralCog
from rocketbot.cogs.helpcog import HelpCog
from rocketbot.cogs.joinraidcog import JoinRaidCog
from rocketbot.cogs.logcog import LoggingCog
from rocketbot.cogs.patterncog import PatternCog
from rocketbot.cogs.urlspamcog import URLSpamCog
from rocketbot.cogs.usernamecog import UsernamePatternCog

async def start_bot():
	bot_log(None, None, 'Bot initializing...')

	# Core
	await rocketbot.add_cog(GeneralCog(rocketbot))
	await rocketbot.add_cog(ConfigCog(rocketbot))

	# Optional
	await rocketbot.add_cog(AutoKickCog(rocketbot))
	await rocketbot.add_cog(BangCommandCog(rocketbot))
	await rocketbot.add_cog(CrossPostCog(rocketbot))
	await rocketbot.add_cog(GamesCog(rocketbot))
	await rocketbot.add_cog(HelpCog(rocketbot))
	await rocketbot.add_cog(JoinRaidCog(rocketbot))
	await rocketbot.add_cog(LoggingCog(rocketbot))
	await rocketbot.add_cog(PatternCog(rocketbot))
	await rocketbot.add_cog(URLSpamCog(rocketbot))
	await rocketbot.add_cog(UsernamePatternCog(rocketbot))

	await rocketbot.start(CONFIG['client_token'], reconnect=True)
	print('\nBot aborted')