added language channel filter

2 files changed, 38 insertions, 1 deletions

diff --git a/filter.py b/filter.py
new file mode 100644
index 0000000..2675b10
--- /dev/null
+++ b/filter.py
@@ -0,0 +1,33 @@
+from discord import Message, Member, TextChannel
+
+allowed_languages = {
+    365161457995743252: ['python3', 'python2'],
+    365161096199536662: ['java'],
+    365162262400598016: ['nodejs'],
+    365162403526082561: ['php'],
+    411245017764462593: ['java'],
+    440476654767177728: ['java'],
+    395547238165643265: ['csharp'],
+    365161847529275393: ['cpp14'],
+    440476086300573696: ['go'],
+    440476142881996810: ['ruby'],
+}
+
+everything = [365162674033524746, 365236060252536833]
+
+
+def verify(message: Message, language: str):
+    if not message.guild:
+        return False
+    author: Member = message.author
+    channel: TextChannel = message.channel
+    if author.guild_permissions.manage_messages:
+        return True
+
+    if channel.id in everything:
+        return True
+
+    if language in allowed_languages.get(channel.id, []):
+        return True
+
+    return False
diff --git a/modules/execute.py b/modules/execute.py
index 4d3f426..1d9102e 100644
--- a/modules/execute.py
+++ b/modules/execute.py
@@ -7,6 +7,7 @@ from discord import Embed, Color, Message, Guild, TextChannel, Member
 from discord.ext.commands import Bot
 
 from compile_api import execute
+from filter import verify
 
 CODE_BLOCK_REGEX: Pattern = re.compile("```(?P<lang>.*)\n(?P<code>[\\s\\S]*?)```")
 INPUT_BLOCK_REGEX: Pattern = re.compile("input[: \t\n]*```(?P<lang>.*)?\n(?P<text>[\\s\\S]*?)```", re.IGNORECASE)
@@ -90,7 +91,10 @@ class ExecuteCog(object):
                                 f"{(timedelta(seconds=30)-delta).seconds}secs"))
         if not author.guild_permissions.manage_messages:
             self.last_messaged[author.id] = datetime.now()
-        response = await execute(code, *languages[lang])
+        language, version = languages[lang]
+        if not verify(message, language):
+            return
+        response = await execute(code, language, version)
         if response.status_code == 429:
             return await channel.send(
                 embed=Embed(