From 5ae32aeb894f57a3c844ab1ac442cd44dfbc93b6 Mon Sep 17 00:00:00 2001
From: Jeremy Zhang <jkzhang@uw.edu>
Date: Sat, 11 Aug 2018 06:08:31 +0000
Subject: [PATCH] Implement Members list command

---
 discordbot/titanembeds/commands.py       | 44 +++++++++++++++++++++++-
 webapp/titanembeds/blueprints/api/api.py |  8 +++++
 2 files changed, 51 insertions(+), 1 deletion(-)

diff --git a/discordbot/titanembeds/commands.py b/discordbot/titanembeds/commands.py
index acbd968..0aa6f2f 100644
--- a/discordbot/titanembeds/commands.py
+++ b/discordbot/titanembeds/commands.py
@@ -1,4 +1,5 @@
 import aiohttp
+import discord
 
 class Commands():
     def __init__(self, client, config):
@@ -73,4 +74,45 @@ class Commands():
         await message.channel.send("Join the Titan Embeds Discord server! https://discord.gg/pFDDtcN")
         
     async def shard(self, message):
-        await message.channel.send("This instance of Titan Embeds Discord Bot is running on shard **{}**. There are **{}** shards in total.".format(message.guild.shard_id, self.client.shard_count))
\ No newline at end of file
+        await message.channel.send("This instance of Titan Embeds Discord Bot is running on shard **{}**. There are **{}** shards in total.".format(message.guild.shard_id, self.client.shard_count))
+    
+    async def members(self, message):
+        headers = {"Authorization": self.config["titan-web-app-secret"]}
+        payload = {
+            "guild_id": message.guild.id,
+        }
+        users = {"authenticated": [], "unauthenticated": []}
+        url = self.config["titan-web-url"] + "api/bot/members"
+        async with aiohttp.ClientSession() as aioclient:
+            async with aioclient.get(url, params=payload, headers=headers) as resp:
+                if resp.status >= 200 and resp.status < 300:
+                    users = await resp.json()
+        embed_description = ""
+        if users["authenticated"]:
+            embed_description = embed_description + "__(Discord)__\n"
+            count = 1
+            for user in users["authenticated"]:
+                server_user = message.guild.get_member(int(user["id"]))
+                embed_description = embed_description + "**{}.** {}#{}".format(count, server_user.name, server_user.discriminator)
+                if server_user.nick:
+                    embed_description = embed_description + " ({})".format(server_user.nick)
+                embed_description = embed_description + " {}\n".format(server_user.mention)
+                count = count + 1
+        if users["unauthenticated"]:
+            if users["authenticated"]:
+                embed_description = embed_description + "\n"
+            embed_description = embed_description + "__(Guest)__\n"
+            count = 1
+            for user in users["unauthenticated"]:
+                embed_description = embed_description + "**{}.** {}#{}\n".format(count, user["username"], user["discriminator"])
+                count = count + 1
+        if users["authenticated"] or users["unauthenticated"]:
+            embed_description = embed_description + "\n"
+        embed_description = embed_description + "**Total Members Online: __{}__**".format(len(users["authenticated"]) + len(users["unauthenticated"]))
+        embed = discord.Embed(
+            title = "Currently Online Embed Members",
+            url = "https://TitanEmbeds.com/",
+            color = 7964363,
+            description = embed_description
+        )
+        await message.channel.send(embed=embed)
\ No newline at end of file
diff --git a/webapp/titanembeds/blueprints/api/api.py b/webapp/titanembeds/blueprints/api/api.py
index aa93da6..0beb95e 100644
--- a/webapp/titanembeds/blueprints/api/api.py
+++ b/webapp/titanembeds/blueprints/api/api.py
@@ -617,6 +617,14 @@ def bot_revoke():
     db.session.commit()
     return jsonify(success="Successfully kicked **{}#{}**!".format(dbuser.username, dbuser.discriminator))
 
+@api.route("/bot/members")
+def bot_members():
+    if request.headers.get("Authorization", "") != config.get("app-secret", ""):
+        return jsonify(error="Authorization header does not match."), 403
+    guild_id = request.args.get("guild_id")
+    members = get_online_embed_users(guild_id)
+    return jsonify(members)
+
 @api.route("/af/direct_message", methods=["POST"])
 def af_direct_message_post():
     cs = request.form.get('cs', None)