87 lines
2.4 KiB
Python
87 lines
2.4 KiB
Python
import discord
|
|
from discord.ext import commands
|
|
from cogs.cs import CSCog
|
|
from cogs.insult import InsultsCog
|
|
from cogs.roulette import RouletteCog
|
|
from cogs.intern import InternCog
|
|
from setup.logger import LOGGER
|
|
|
|
bot_intents = discord.Intents.default()
|
|
bot_intents.message_content = True
|
|
bot_intents.members = True
|
|
bot_intents.presences = True
|
|
bot_intents.guilds = True
|
|
bot_intents.reactions = True
|
|
bot_intents.voice_states = True
|
|
|
|
bot = commands.Bot(command_prefix='$', intents=bot_intents)
|
|
|
|
COG_REGISTRY = {
|
|
'cs': CSCog,
|
|
'insult': InsultsCog,
|
|
'roulette': RouletteCog,
|
|
'intern': InternCog
|
|
}
|
|
|
|
|
|
def _normalize_cog_names(cogs=None):
|
|
if cogs is None:
|
|
return list(COG_REGISTRY.keys())
|
|
|
|
if isinstance(cogs, str):
|
|
cogs = [cogs]
|
|
|
|
selected_cogs = []
|
|
for cog_name in cogs:
|
|
name = str(cog_name).strip().lower()
|
|
if not name:
|
|
continue
|
|
if name in {'all', '*'}:
|
|
return list(COG_REGISTRY.keys())
|
|
if name in COG_REGISTRY:
|
|
selected_cogs.append(name)
|
|
else:
|
|
LOGGER.warning("Unknown cog '%s'. Available cogs: %s", cog_name, ', '.join(COG_REGISTRY.keys()))
|
|
|
|
return list(dict.fromkeys(selected_cogs))
|
|
|
|
|
|
def register_cogs(cogs=None):
|
|
selected_cogs = _normalize_cog_names(cogs)
|
|
|
|
existing_cog_names = set(COG_REGISTRY.values())
|
|
for existing_name in list(bot.cogs.keys()):
|
|
if existing_name in {cog_class.__name__ for cog_class in existing_cog_names}:
|
|
bot.remove_cog(existing_name)
|
|
|
|
for cog_name in selected_cogs:
|
|
cog_class = COG_REGISTRY[cog_name]
|
|
bot.add_cog(cog_class(bot))
|
|
|
|
return selected_cogs
|
|
|
|
|
|
|
|
def reload_feature(feature: str):
|
|
normalized_feature = str(feature).strip().lower()
|
|
LOGGER.debug(bot.cogs)
|
|
|
|
if normalized_feature in {'all', '*'}:
|
|
register_cogs()
|
|
LOGGER.debug('All cogs reloaded!')
|
|
return
|
|
|
|
if normalized_feature not in COG_REGISTRY:
|
|
LOGGER.warning("Unknown cog '%s'. Available cogs: %s", feature, ', '.join(COG_REGISTRY.keys()))
|
|
return
|
|
|
|
LOGGER.debug(f'Reloading {feature}...')
|
|
target_cog_name = COG_REGISTRY[normalized_feature].__name__
|
|
for existing_name in list(bot.cogs.keys()):
|
|
if existing_name == target_cog_name:
|
|
bot.remove_cog(existing_name)
|
|
break
|
|
|
|
bot.add_cog(COG_REGISTRY[normalized_feature](bot))
|
|
LOGGER.debug(f'{feature} reloaded!')
|