Initial commit

1 files changed, 114 insertions, 0 deletions

diff --git a/modules/execute.py b/modules/execute.py
new file mode 100644
index 0000000..6127647
--- /dev/null
+++ b/modules/execute.py
@@ -0,0 +1,114 @@
+import re
+from collections import defaultdict
+from datetime import datetime, timedelta
+from typing import Pattern
+
+from discord import Embed, Color, Message, Guild, TextChannel, Member
+from discord.ext.commands import Bot
+
+from compile_api import execute
+
+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)
+
+PYTHON_3 = ('python3', 2)
+NODEJS = ('nodejs', 2)
+C_LANG = ('c', 3)
+CPP = ('cpp14', 2)
+PHP = ('php', 2)
+PYTHON_2 = ('python2', 1)
+RUBY = ('ruby', 2)
+GO_LANG = ('go', 2)
+SCALA = ('scala', 2)
+BASH = ('bash', 2)
+CSHARP = ('csharp', 2)
+HASKELL = ('haskell', 2)
+BRAINFUCK = ('brainfuck', 0)
+LUA = ('lua', 1)
+DART = ('dart', 2)
+KOTLIN = ('kotlin', 1)
+
+languages = {
+    'kt': KOTLIN,
+    'kotlin': KOTLIN,
+    'dart': DART,
+    'dt': DART,
+    'lua': LUA,
+    'py': PYTHON_3,
+    'python': PYTHON_3,
+    'js': NODEJS,
+    'javascript': NODEJS,
+    'c': C_LANG,
+    'c++': CPP,
+    'cpp': CPP,
+    'py2': PYTHON_2,
+    'go': GO_LANG,
+    'scala': SCALA,
+    'sc': SCALA,
+    'bash': BASH,
+    'hs': HASKELL,
+    'haskell': HASKELL,
+    'brainfuck': BRAINFUCK,
+    'bf': BRAINFUCK,
+}
+
+
+class ExecuteCog(object):
+    def __init__(self, bot: Bot):
+        self.bot: Bot = bot
+        self.last_messaged = defaultdict(lambda: datetime.fromtimestamp(0))
+
+    # noinspection PyMethodMayBeStatic
+    async def on_message(self, message: Message):
+        if message.guild is None:
+            return
+        content: str = message.content
+        guild: Guild = message.guild
+        author: Member = message.author
+        channel: TextChannel = message.channel
+        if guild.me not in message.mentions:
+            return
+        code = ""
+        lang = ""
+        for match in CODE_BLOCK_REGEX.finditer(content):
+            code = match.group('code')
+            lang = match.group('lang')
+            break
+        if lang is "" or code is "":
+            return
+        inp = ""
+        for match in INPUT_BLOCK_REGEX.finditer(content):
+            inp += match.group("text") + '\n'
+        last = self.last_messaged[author.id]
+        delta = datetime.now() - last
+        if delta < timedelta(seconds=30):
+            return await channel.send(
+                embed=Embed(
+                    description=f"You are not allowed to eval code again. Check again in "
+                                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])
+        if response.status_code == 429:
+            return await channel.send(
+                embed=Embed(
+                    color=Color.blurple(),
+                    description="The daily ratelimit for our API is reached. A great alternative is [Ideone]("
+                                "https://ideone.com/) or [Repl.it](https://repl.it/)"))
+        if response.status_code == 401:
+            return await channel.send(
+                embed=Embed(
+                    color=Color.red(),
+                    description="Our API credentials are invalid. Please contact the bot owner"))
+        memory = response.memory
+        output = response.output
+        cpu_time = response.cpu_time
+        await channel.send(
+            embed=Embed(
+                title="Executed your code",
+                description=f"```\n{output}```"
+            ).set_footer(text=f"Executed in {cpu_time}s. Memory: {memory}"))
+
+
+def setup(bot: Bot):
+    bot.add_cog(ExecuteCog(bot))