Add oauth whitelist commands

Author: kyb3r

cogs/utility.py

@@ -11,7 +11,7 @@
 from pkg_resources import parse_version
 from textwrap import indent
 
-from discord import Embed, Color, Activity, Role
+from discord import Embed, Color, Activity, Role, Member
 from discord.enums import ActivityType, Status
 from discord.ext import commands
 
@@ -1165,6 +1165,63 @@ def get_level(perm_level):
 
         p_session = PaginatorSession(ctx, *embeds)
         return await p_session.run()
+    
+    @commands.group(invoke_without_command=True)
+    @checks.has_permissions(PermissionLevel.OWNER)
+    async def oauth(self, ctx):
+        """Commands relating to Logviewer oauth2 login authentication."""
+        cmd = self.bot.get_command('help')
+        await ctx.invoke(cmd, command='config')
+    
+    @oauth.command()
+    @checks.has_permissions(PermissionLevel.OWNER)
+    async def whitelist(self, ctx, target: Union[Member, Role]):
+        """Whitelist or un-whitelist a user or role from having access to logs."""
+        whitelisted = self.bot.config['oauth_whitelist']
+
+        if target.id in whitelisted:
+            whitelisted.remove(target.id)
+            removed = True
+        else:
+            whitelisted.append(target.id)
+            removed = False
+        
+        await self.bot.config.update()
+
+        em = Embed(color=self.bot.main_color)
+        em.title = 'Success'
+        em.description = (
+            f"{'Un-w' if removed else 'W'}hitelisted "
+            f"{target.mention} to view logs."
+            )
+
+        await ctx.send(embed=em)
+    
+    @oauth.command()
+    @checks.has_permissions(PermissionLevel.OWNER)
+    async def get(self, ctx):
+        """Shows a list of users and roles that are whitelisted to view logs."""
+        whitelisted = self.bot.config['oauth_whitelist']
+        
+        users = []
+        roles = []
+
+        for id in whitelisted:
+            user = bot.get_user(id)
+            if user:
+                users.append(users)
+            role = self.modmail_guild.get_role(id)
+            if role:
+                roles.append(role)
+        
+        em = Embed(color=self.bot.main_color)
+        em.title = 'Oauth Whitelist'
+        em.add_field(name='Users', value=' '.join(u.mention for u in users) or 'None')
+        em.add_field(name='Roles', value=' '.join(r.mention for r in roles) or 'None')
+
+        await ctx.send(embed=em)
+        
+
 
     @commands.command(hidden=True, name='eval')
     @checks.has_permissions(PermissionLevel.OWNER)

core/config.py

@@ -39,7 +39,7 @@ class ConfigManager(ConfigManagerABC):
 
     internal_keys = {
         # bot presence
-        'activity_message', 'activity_type', 'status',
+        'activity_message', 'activity_type', 'status', 'oauth_whitelist',
 
         # moderation
         'blocked', 'command_permissions', 'level_permissions',
@@ -107,6 +107,7 @@ def populate_cache(self):
             'plugins': [],
             'aliases': {},
             'blocked': {},
+            'oauth_whitelist': [],
             'command_permissions': {},
             'level_permissions': {},
             'notification_squad': {},