Initial commit

bot/pluralkit/bot.py

@@ -0,0 +1,82 @@
+import logging
+import os
+
+import discord
+
+logging.basicConfig(level=logging.DEBUG)
+logging.getLogger("discord").setLevel(logging.INFO)
+logging.getLogger("websockets").setLevel(logging.INFO)
+
+logger = logging.getLogger("pluralkit.bot")
+logger.setLevel(logging.DEBUG)
+
+client = discord.Client()
+
+@client.event
+async def on_error(evt, *args, **kwargs):
+    logger.exception("Error while handling event {} with arguments {}:".format(evt, args))
+
+@client.event
+async def on_ready():
+    # Print status info
+    logger.info("Connected to Discord.")
+    logger.info("Account: {}#{}".format(client.user.name, client.user.discriminator))
+    logger.info("User ID: {}".format(client.user.id))
+
+@client.event
+async def on_message(message):
+    # Ignore bot messages
+    if message.author.bot:
+        return
+    
+    # Split into args. shlex sucks so we don't bother with quotes
+    args = message.content.split(" ")
+
+    from pluralkit import proxy, utils
+
+    # Find and execute command in map
+    if len(args) > 0 and args[0] in utils.command_map:
+        subcommand_map = utils.command_map[args[0]]
+
+        if len(args) >= 2 and args[1] in subcommand_map:
+            async with client.pool.acquire() as conn:
+                await subcommand_map[args[1]][0](conn, message, args[2:])
+        elif None in subcommand_map:
+            async with client.pool.acquire() as conn:
+                await subcommand_map[None][0](conn, message, args[1:])
+        elif len(args) >= 2:
+            embed = discord.Embed()
+            embed.colour = discord.Colour.dark_red()
+            embed.description = "Subcommand \"{}\" not found.".format(args[1])
+            await client.send_message(message.channel, embed=embed)
+    else:
+        # Try doing proxy parsing
+        async with client.pool.acquire() as conn:
+            await proxy.handle_proxying(conn, message)
+
+@client.event
+async def on_reaction_add(reaction, user):
+    from pluralkit import proxy
+
+    # Pass reactions to proxy system
+    async with client.pool.acquire() as conn:
+        await proxy.handle_reaction(conn, reaction, user)
+
+async def run():
+    from pluralkit import db
+    try:
+        logger.info("Connecting to database...")
+        pool = await db.connect()
+
+        logger.info("Attempting to create tables...")
+        async with pool.acquire() as conn:
+            await db.create_tables(conn)
+
+        logger.info("Connecting to InfluxDB...")
+
+        client.pool = pool
+        logger.info("Connecting to Discord...")
+        await client.start(os.environ["TOKEN"])
+    finally:
+        logger.info("Logging out from Discord...")
+        await client.logout()