diff --git a/app/cogs/controlcommands/control.py b/app/cogs/controlcommands/control.py new file mode 100644 index 0000000..74570e0 --- /dev/null +++ b/app/cogs/controlcommands/control.py @@ -0,0 +1,101 @@ +import os +from dotenv import load_dotenv # Import OS variables from Dotenv file. +load_dotenv() # Load Dotenv. Delete this for production +import asyncio # Discord Py Dependency +import discord # Main Lib +from discord.ext import commands # Commands module +from discord_slash import SlashCommand, SlashContext, cog_ext, utils # Slash Command Library +from discord_slash.utils.manage_commands import create_choice, create_option # Slash Command features +from deepdiff import DeepDiff +from pprint import pprint + +from bot import loadCog, unloadCog, cogsDir, checkConfig, parseConfigCheck, yaml_load, configFile + +##### Control Cog +class Control(commands.Cog, name='Cog Control Commands'): + def __init__(self, client): + self.client = client + + #### Check if user is an administrator + async def cog_check(self, ctx:commands.Context): + for role in ctx.author.roles: + if role.permissions.administrator: + return True + return ctx.author.guild_permissions.administrator + + @commands.command( + name='debug', + description='Toggles debug feature for the guild. Enter either `on` or `off`.', + brief='Toggle debug features.' + ) + async def _debug(self, ctx:commands.Context, toggle:str): + if toggle.lower() == 'on': + loadCog(f'./debug/debug.py') + await ctx.reply(f'```Debug commands enabled. Use them carefully.```') + elif toggle.lower() == 'off': + unloadCog(f'./debug/debug.py') + await ctx.reply(f'```Debug commands disabled.```') + else: + raise commands.CommandError(message='Invalid argument.') + # await ctx.reply(f'```Invalid argument.```') + + @commands.command( + name='testconfig', + description='Tests the completeness of the configuration values of the current guild by comparing it to a configuration blueprint.', + brief='Tests config values for current guild.', + aliases=['configtest'] + ) + async def _testconfig(self, ctx:commands.Context): + checkConfig(ctx.guild) + status, output = checkConfig(ctx.guild) + conf = yaml_load(configFile) + if not status: + await ctx.reply(f"```The Bot's configurations are incomplete for the guild {ctx.guild.name}. Some limited functions will still be available, but most features cannot be used until the configurations are complete.\n{parseConfigCheck(output)}\nYou can set these configuration values using the `/config` command.```") + elif status: + await ctx.reply(f"```The Bot's configurations for the guild {ctx.guild.name} are in order. The Bot is ready to interact with the guild.```") + + @commands.command( + name='lockconfig', + description='Administrator command that locks the /config commands, preventing any accidental changes to configurations.', + brief='Toggle locking /config command.', + aliases=['configlock','lock','lockdownconfig'] + ) + async def _lockconfig(self, ctx:commands.Context, toggle:str): + if toggle == 'on': + o = '' + if self.client.get_cog('Configuration Commands') is not None: + unloadCog(f'./{cogsDir}/slashcommands/config.py') + o = ''.join([o,'Configuration Lock turned on. `/config` command has been disabled.']) + else: + o = ''.join([o,'`/config` command has already been disabled.']) + if self.client.get_cog('Manipulate Timeslots') is not None: + unloadCog(f'./{cogsDir}/slashcommands/secondary/manipulate_timeslots.py') + o = ''.join([o,'\nTimeslot configuration sub-commands have been disabled.']) + else: + o = ''.join([o,'\nTimeslot configuration sub-commands have already been disabled.']) + await ctx.reply(f'```{o}```') + await self.client.slash.sync_all_commands() + elif toggle == 'off': + o = '' + if self.client.get_cog('Configuration Commands') is None: + loadCog(f'./{cogsDir}/slashcommands/config.py') + await self.client.slash.sync_all_commands() + o = ''.join([o,'Configuration Lock turned off. `/config` command has been re-enabled.']) + else: + o = ''.join([o,'`/config` command has already been enabled.']) + if self.client.get_cog('Manipulate Timeslots') is None: + if all([len(yaml_load(configFile)[x]['timeslots']) > 0 for x in yaml_load(configFile)]): + loadCog(f'./{cogsDir}/slashcommands/secondary/manipulate_timeslots.py') + await self.client.slash.sync_all_commands() + o = ''.join([o,'\nTimeslot configuration sub-commands have been re-enabled.']) + else: + o = ''.join([o,'\nTimeslot configuration sub-commands are not re-enabled because there are no configured timeslots for the guild.']) + else: + o = ''.join([o,'\nTimeslot configuration sub-commands have already been enabled.']) + await ctx.reply(f'```{o}```') + await self.client.slash.sync_all_commands() + else: + raise commands.CommandError('Invalid argument. `lockconfig` command only accepts the arguments `on` or `off`.') + +def setup(client): + client.add_cog(Control(client)) \ No newline at end of file