Added dynamic snippet configuration.

Author: kyb3r

bot.py

@@ -62,6 +62,7 @@ class ModmailBot(commands.Bot):
 
     def __init__(self):
         super().__init__(command_prefix=self.get_pre)
+        self.version = __version__
         self.start_time = datetime.datetime.utcnow()
         self.threads = ThreadManager(self) 
         self.session = aiohttp.ClientSession(loop=self.loop)
@@ -103,7 +104,11 @@ def run(self):
 
     @property
     def snippets(self):
-        return self.config.get('snippets', {})
+        return {k: v for k, v in self.config.get('snippets', {}).items() if v}
+
+    @property
+    def aliases(self):
+        return {k: v for k, v in self.config.get('aliases', {}).items() if v}     
 
     @property
     def token(self):
@@ -127,12 +132,15 @@ def blocked_users(self):
         if self.guild:
             top_chan = self.main_category.channels[0]
             return [int(i) for i in re.findall(r'\d+', top_chan.topic)]
+    
+    @property
+    def prefix(self):
+        return self.config.get('prefix', '?')
 
     @staticmethod
     async def get_pre(bot, message):
         '''Returns the prefix.'''
-        p = bot.config.get('prefix') or 'm.'
-        return [p, f'<@{bot.user.id}> ', f'<@!{bot.user.id}> ']
+        return [bot.prefix, f'<@{bot.user.id}> ', f'<@!{bot.user.id}> ']
 
     async def on_connect(self):
         print(line)
@@ -184,12 +192,14 @@ async def on_message(self, message):
         if isinstance(message.channel, discord.DMChannel):
             return await self.process_modmail(message)
 
-        prefix = self.config.get('prefix', 'm.')
+        prefix = self.prefix
 
         if message.content.startswith(prefix):
             cmd = message.content[len(prefix):].strip()
             if cmd in self.snippets:
                 message.content = f'{prefix}reply {self.snippets[cmd]}'
+            if cmd in self.aliases:
+                message.content = f'{prefix}{self.aliases[cmd]}'
 
         await self.process_commands(message)
 

cogs/modmail.py

@@ -4,6 +4,7 @@
 import datetime
 import dateutil.parser
 from core.decorators import trigger_typing
+from core.paginator import PaginatorSession
 
 class Modmail:
     '''Commands directly related to Modmail functionality.'''
@@ -32,10 +33,13 @@ async def setup(self, ctx):
 
         await ctx.send('Successfully set up server.')
 
-    @commands.command(name='snippets')
+    @commands.group(name='snippets')
     @commands.has_permissions(manage_messages=True)
-    async def _snippets(self, ctx):
+    async def snippets(self, ctx):
         '''Returns a list of snippets that are currently set.'''
+        if ctx.invoked_subcommand is not None:
+            return 
+
         embeds = []
 
         em = discord.Embed(color=discord.Color.green())
@@ -47,7 +51,8 @@ async def _snippets(self, ctx):
 
         if not self.bot.snippets:
             em.color = discord.Color.red()
-            em.description = 'You dont have any snippets at the moment.'
+            em.description = f'You dont have any snippets at the moment.'
+            em.set_footer(text=f'Do {self.bot.prefix}help snippets for more commands.')
 
         for name, value in self.bot.snippets.items():
             if len(em.fields) == 5:
@@ -58,6 +63,46 @@ async def _snippets(self, ctx):
 
         session = PaginatorSession(ctx, *embeds)
         await session.run()
+    
+    @snippets.command(name='add')
+    async def _add(self, ctx, name: str.lower, *, value):
+        '''Add a snippet to the bot config.'''
+        if 'snippets' not in self.bot.config.cache:
+            self.bot.config['snippets'] = {}
+
+        self.bot.config.snippets[name] = value
+        await self.bot.config.update()
+
+        em = discord.Embed(
+            title='Added snippet',
+            color=discord.Color.green(),
+            description=f'`{name}` points to: {value}'
+        )
+
+        await ctx.send(embed=em)
+
+    @snippets.command(name='del')
+    async def __del(self, ctx, *, name: str.lower):
+        '''Removes a snippet from bot config.'''
+
+        if 'snippets' not in self.bot.config.cache:
+            self.bot.config['snippets'] = {}
+
+        em = discord.Embed(
+            title='Removed snippet',
+            color=discord.Color.green(),
+            description=f'`{name}` no longer exists.'
+        )
+
+        if not self.bot.config.snippets.get(name):
+            em.title = 'Error'
+            em.color = discord.Color.red()
+            em.description = f'Snippet `{name}` does not exist.'
+        else:
+            self.bot.config['snippets'][name] = None
+            await self.bot.config.update()
+
+        await ctx.send(embed=em)
 
     @commands.command(name='close')
     @commands.has_permissions(manage_channels=True)

core/config.py

@@ -42,6 +42,12 @@ async def refresh(self):
     def __getattr__(self, value):
         return self.cache[value]
 
+    def __setitem__(self, key, item):
+        self.cache[key] = item
+
+    def __getitem__(self, key):
+        return self.cache[key]
+
     def get(self, value, default=None):
         return self.cache.get(value) or default