From 3b1dd1b5bd06a8863da7bb33d1d1009b00464127 Mon Sep 17 00:00:00 2001
From: flifloo <flifloo@gmail.com>
Date: Sun, 17 May 2020 23:11:50 +0200
Subject: [PATCH] Add chan permission check on command

---
 extensions/chan.py | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/extensions/chan.py b/extensions/chan.py
index 3ebe8ff..b7f67b4 100644
--- a/extensions/chan.py
+++ b/extensions/chan.py
@@ -7,10 +7,6 @@ from bot_bde.logger import logger
 
 extension_name = "chan"
 logger = logger.getChild(extension_name)
-REACTIONS = []
-for i in range(10):
-    REACTIONS.append(str(i) + "\ufe0f\u20E3")
-REACTIONS.append("\U0001F51F")
 
 
 def check_editable_chan():
@@ -27,6 +23,16 @@ def check_editable_chan():
     return commands.check(predicate)
 
 
+def check_permissions():
+    async def predicate(ctx: commands.Context):
+        if len(ctx.message.channel_mentions) == 1:
+            return ctx.message.channel_mentions[0].permissions_for(ctx.author).manage_channels
+        else:
+            return ctx.author.guild_permissions.manage_channels
+
+    return commands.check(predicate)
+
+
 def chan_permissions(chan: GuildChannel, allow: bool):
     if type(chan) == TextChannel:
         return dict(read_messages=allow, send_messages=allow)
@@ -49,6 +55,7 @@ class Chan(commands.Cog):
 
     @chan.group("create", pass_context=True)
     @commands.guild_only()
+    @check_permissions()
     async def chan_create(self, ctx: commands.Context, name: str):
         chan: TextChannel = await ctx.guild.create_text_channel(name)
         if len(ctx.message.role_mentions) != 0 or len(ctx.message.mentions) != 0:
@@ -61,6 +68,7 @@ class Chan(commands.Cog):
     @chan.group("deny", pass_context=True)
     @commands.guild_only()
     @check_editable_chan()
+    @check_permissions()
     async def chan_deny(self, ctx: commands.Context):
         for r in ctx.message.role_mentions:
             await ctx.message.channel_mentions[0].set_permissions(r,
@@ -74,6 +82,7 @@ class Chan(commands.Cog):
     @chan.group("allow", pass_context=True)
     @commands.guild_only()
     @check_editable_chan()
+    @check_permissions()
     async def allow_deny(self, ctx: commands.Context):
         for r in ctx.message.role_mentions:
             await ctx.message.channel_mentions[0].set_permissions(r,