Log message edits

Author: kyb3r

bot.py

@@ -418,6 +418,7 @@ async def on_message_edit(self, before, after):
                     if matches and matches[-1] == str(before.id):
                         embed.description = after.content
                         await msg.edit(embed=embed)
+                        await self.modmail_api.edit_message(str(after.id), after.content)
                         break
 
     async def on_command_error(self, ctx, error):

cogs/modmail.py

@@ -2,6 +2,7 @@
 
 import datetime
 from typing import Optional, Union
+import asyncio
 
 import discord
 from discord.ext import commands
@@ -428,8 +429,12 @@ async def edit(self, ctx, message_id: Optional[int]=None, *, new_message):
         if not linked_message_id:
             raise commands.UserInputError
 
-        await thread.edit_message(linked_message_id, new_message)
+        await asyncio.gather(
+            thread.edit_message(linked_message_id, new_message),
+            self.bot.modmail_api.edit_message(linked_message_id, new_message)
+        ) 
         await ctx.message.add_reaction('✅')
+        
 
     @commands.command()
     @trigger_typing

core/clients.py

@@ -151,6 +151,12 @@ def get_log_url(self, recipient, channel, creator):
             }
         })
 
+    async def edit_message(self, message_id, new_content):
+        return await self.request(self.logs + '/edit', method='PATCH', payload={
+            'message_id': str(message_id),
+            'new_content':  new_content
+        })
+
     def append_log(self, message, channel_id='', type='thread_message'):
         channel_id = str(channel_id) or str(message.channel.id)
         payload = {
@@ -254,6 +260,15 @@ async def update_config(self, data):
         valid_keys = self.app.config.valid_keys - self.app.config.protected_keys
         data = {k: v for k, v in data.items() if k in valid_keys}
         return await self.db.config.update_one({'bot_id': self.app.user.id}, {'$set': data})
+    
+    async def edit_message(self, message_id, new_content):
+        await self.logs.update_one(
+            {'messages.message_id': str(message_id)},
+            {'$set': {
+                'messages.$.content': new_content,
+                'messages.$.edited': True
+                }
+            })
 
     async def append_log(self, message, channel_id='', type='thread_message'):
         channel_id = str(channel_id) or str(message.channel.id)