From 0cfcb42ac40adbb43a5fc842ab28753ecd7b4a74 Mon Sep 17 00:00:00 2001
From: Jeremy Zhang <jkzhang@uw.edu>
Date: Wed, 6 Sep 2017 04:58:46 +0000
Subject: [PATCH] Update and delete messages even if not in discordbot messages
 cache, and raise max_messages deque

---
 discordbot/titanembeds/bot.py | 35 ++++++++++++++++++++++++++++++++++-
 1 file changed, 34 insertions(+), 1 deletion(-)

diff --git a/discordbot/titanembeds/bot.py b/discordbot/titanembeds/bot.py
index 520839e..6b04d20 100644
--- a/discordbot/titanembeds/bot.py
+++ b/discordbot/titanembeds/bot.py
@@ -7,13 +7,14 @@ import aiohttp
 import asyncio
 import sys
 import logging
+import json
 logging.basicConfig(filename='titanbot.log',level=logging.INFO,format='%(asctime)s %(message)s', datefmt='%m/%d/%Y %I:%M:%S %p')
 logging.getLogger('TitanBot')
 logging.getLogger('sqlalchemy')
 
 class Titan(discord.Client):
     def __init__(self):
-        super().__init__()
+        super().__init__(max_messages=20000)
         self.aiosession = aiohttp.ClientSession(loop=self.loop)
         self.http.user_agent += ' TitanEmbeds-Bot'
         self.database = DatabaseInterface(self)
@@ -244,3 +245,35 @@ class Titan(discord.Client):
     async def on_webhooks_update(self, server):
         await self.wait_until_dbonline()
         await self.database.update_guild(server)
+
+    async def on_socket_raw_receive(self, msg):
+        if type(msg) is not str:
+            return
+        msg = json.loads(msg)
+        if msg["op"] != 0:
+            return
+        action = msg["t"]
+        await asyncio.sleep(1)
+        if action == "MESSAGE_UPDATE":
+            if not self.in_messages_cache(msg["d"]["id"]):
+                channel = self.get_channel(msg["d"]["channel_id"])
+                message = await self.get_message(channel, msg["d"]["id"])
+                await self.on_message_edit(None, message)
+        if action == "MESSAGE_DELETE":
+            if not self.in_messages_cache(msg["d"]["id"]):
+                await self.process_raw_message_delete(msg["d"]["id"], msg["d"]["channel_id"])
+        if action == "MESSAGE_DELETE_BULK":
+            for msgid in msg["d"]["ids"]:
+                if not self.in_messages_cache(msgid):
+                    await self.process_raw_message_delete(msgid, msg["d"]["channel_id"])
+    
+    async def process_raw_message_delete(self, msg_id, channel_id):
+        channel = self.get_channel(channel_id)
+        msg = discord.Message(channel=channel, reactions=[], id=msg_id, type=0, timestamp="2017-01-15T02:59:58", content="What fun is there in making sense?") # Procreate a fake message object
+        await self.on_message_delete(msg)
+    
+    def in_messages_cache(self, msg_id):
+        for msg in self.messages:
+            if msg.id == msg_id:
+                return True
+        return False
\ No newline at end of file