python-app-rocketbot/rocketbot/bot.py

import traceback

from discord import Intents
from discord.errors import DiscordException
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, '\u0007Command 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, '\u0007Event 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, "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 sorted(synced_commands, key=lambda c: c.name):
				bot_log(None, None, f'Synced command: /{command.name}')
		except DiscordException as e:
			dump_stacktrace(e)

# Current active bot instance
rocketbot: Rocketbot | None = None

def __create_bot():
	global rocketbot
	if rocketbot is not None:
		return
	bot_log(None, None, 'Creating bot...')
	intents = Intents.default()
	# Set privileged intents. Must agree with configuration in Discord app.
	intents.message_content = CONFIG.get('has_message_content_intent', False)
	intents.members = CONFIG.get('has_members_intent', False)
	intents.presences = False
	rocketbot = Rocketbot(intents=intents)
__create_bot()

async def start_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.kickstartercog import KickstarterCog
	from rocketbot.cogs.logcog import LoggingCog
	from rocketbot.cogs.patterncog import PatternCog
	from rocketbot.cogs.urlspamcog import URLSpamCog
	from rocketbot.cogs.usernamecog import UsernamePatternCog
	from rocketbot.cogs.videopreviewcog import VideoPreviewCog

	bot_log(None, None, 'Bot initializing...')

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

	# Optional
	if AutoKickCog.supports_intents(rocketbot.intents):
		await rocketbot.add_cog(AutoKickCog(rocketbot))
	if BangCommandCog.supports_intents(rocketbot.intents):
		await rocketbot.add_cog(BangCommandCog(rocketbot))
	if CrossPostCog.supports_intents(rocketbot.intents):
		await rocketbot.add_cog(CrossPostCog(rocketbot))
	if GamesCog.supports_intents(rocketbot.intents):
		await rocketbot.add_cog(GamesCog(rocketbot))
	if HelpCog.supports_intents(rocketbot.intents):
		await rocketbot.add_cog(HelpCog(rocketbot))
	if JoinRaidCog.supports_intents(rocketbot.intents):
		await rocketbot.add_cog(JoinRaidCog(rocketbot))
	if LoggingCog.supports_intents(rocketbot.intents):
		await rocketbot.add_cog(LoggingCog(rocketbot))
	if PatternCog.supports_intents(rocketbot.intents):
		await rocketbot.add_cog(PatternCog(rocketbot))
	if URLSpamCog.supports_intents(rocketbot.intents):
		await rocketbot.add_cog(URLSpamCog(rocketbot))
	if UsernamePatternCog.supports_intents(rocketbot.intents):
		await rocketbot.add_cog(UsernamePatternCog(rocketbot))
	if VideoPreviewCog.supports_intents(rocketbot.intents):
		await rocketbot.add_cog(VideoPreviewCog(rocketbot))
	if KickstarterCog.supports_intents(rocketbot.intents):
		await rocketbot.add_cog(KickstarterCog(rocketbot))

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